Use ConnectionStringHelper when creating any repository derived from …

…EFRepositoryBase
VirtoCommerce · Aug 4, 2017 · 37fda4d · 37fda4d
1 parent 1c6425a
commit 37fda4d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/VirtoCommerce.Platform.Core/Common/ConnectionStringHelper.cs b/VirtoCommerce.Platform.Core/Common/ConnectionStringHelper.cs
@@ -5,13 +5,21 @@ namespace VirtoCommerce.Platform.Core.Common
 {
     public class ConnectionStringHelper
     {
-        public static string GetConnectionString(string name)
+        public static string GetConnectionString(string nameOrConnectionString)
         {
-            var environmentConnectionString = Environment.GetEnvironmentVariable($"VIRTO_CONN_STR_{name}");
+            var result = nameOrConnectionString;
 
-            return !string.IsNullOrEmpty(environmentConnectionString)
-                ? environmentConnectionString
-                : ConfigurationManager.ConnectionStrings[name]?.ConnectionString;
+            if (nameOrConnectionString.IndexOf('=') < 0)
+            {
+                result = Environment.GetEnvironmentVariable($"VIRTO_CONN_STR_{nameOrConnectionString}");
+
+                if (string.IsNullOrEmpty(result))
+                {
+                    result = ConfigurationManager.ConnectionStrings[nameOrConnectionString]?.ConnectionString;
+                }
+            }
+
+            return result;
         }
     }
 }
diff --git a/VirtoCommerce.Platform.Data/Infrastructure/EFRepositoryBase.cs b/VirtoCommerce.Platform.Data/Infrastructure/EFRepositoryBase.cs
@@ -38,7 +38,7 @@ protected EFRepositoryBase()
         /// <param name="unitOfWork">The unit of work.</param>
         /// <param name="interceptors">The interceptors.</param>
         protected EFRepositoryBase(string nameOrConnectionString, IUnitOfWork unitOfWork = null, IInterceptor[] interceptors = null)
-            : base(nameOrConnectionString)
+            : base(ConnectionStringHelper.GetConnectionString(nameOrConnectionString))
         {
             _unitOfWork = unitOfWork;
             _interceptors = interceptors;