- update DbContextOptionsBuilderExtensions

The-Poolz · Nov 14, 2023 · ed81801 · ed81801
1 parent b53058d
commit ed81801
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 6 deletions.
diff --git a/ConfiguredSqlConnection/Extensions/DbContextOptionsBuilderExtensions.cs b/ConfiguredSqlConnection/Extensions/DbContextOptionsBuilderExtensions.cs
@@ -1,36 +1,47 @@
 using SecretsManager;
 using EnvironmentManager;
 using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Migrations.Internal;
 
 namespace ConfiguredSqlConnection.Extensions;
 
 public static class DbContextOptionsBuilderExtensions
 {
     private static readonly EnvManager envManager = new();
 
-    public static DbContextOptionsBuilder ConfigureFromActionConnection(this DbContextOptionsBuilder optionsBuilder)
+    public static DbContextOptionsBuilder ConfigureFromActionConnection(this DbContextOptionsBuilder optionsBuilder, string? migrationsAssembly = null)
     {
         if (optionsBuilder.IsConfigured) return optionsBuilder;
-
         var connectionString = envManager.GetEnvironmentValue<string>("CONFIGUREDSQLCONNECTION_ACTION_CONNECTION");
-        if (!string.IsNullOrEmpty(connectionString))
+        if (string.IsNullOrEmpty(connectionString)) return optionsBuilder;
+
+        if (string.IsNullOrWhiteSpace(migrationsAssembly))
         {
             optionsBuilder.UseSqlServer(connectionString);
         }
+        else
+        {
+            optionsBuilder.UseSqlServer(connectionString, options => options.MigrationsAssembly(migrationsAssembly));
+        }
 
         return optionsBuilder;
     }
 
-    public static DbContextOptionsBuilder ConfigureFromSecretConnection(this DbContextOptionsBuilder optionsBuilder)
+    public static DbContextOptionsBuilder ConfigureFromSecretConnection(this DbContextOptionsBuilder optionsBuilder, string? migrationsAssembly = null)
     {
         if (optionsBuilder.IsConfigured) return optionsBuilder;
-
         var secretValue = envManager.GetEnvironmentValue<string>("CONFIGUREDSQLCONNECTION_SECRET_NAME_OF_CONNECTION", true);
         var connectionString = new SecretManager().GetSecretValue(secretValue, "connectionString");
-        if (!string.IsNullOrEmpty(connectionString))
+        if (string.IsNullOrEmpty(connectionString)) return optionsBuilder;
+
+        if (string.IsNullOrWhiteSpace(migrationsAssembly))
         {
             optionsBuilder.UseSqlServer(connectionString);
         }
+        else
+        {
+            optionsBuilder.UseSqlServer(connectionString, options => options.MigrationsAssembly(migrationsAssembly));
+        }
 
         return optionsBuilder;
     }

diff --git a/README.md b/README.md
@@ -95,6 +95,11 @@ optionsBuilder
 optionsBuilder
     .ConfigureFromSecretConnection();
 
+// Try to configure action connection also try to configure db connection via SecretManager.
+optionsBuilder
+    .ConfigureFromActionConnection()
+    .ConfigureFromSecretConnection();
+
 // Try to configure action connection also try to configure db connection via SecretManager.
 optionsBuilder
     .ConfigureFromActionConnection()