Fixed asp.net core 2.0 startup error about issue #44 of MySql and Pos…

…tgreSql (#51) Fixed asp.net core 2.0 startup error about issue #44 of MySql and PostgreSql
dotnetcore · Sep 29, 2017 · 3562973 · 3562973
1 parent d0d8b40
commit 3562973
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/src/DotNetCore.CAP.MySql/CAP.MySqlCapOptionsExtension.cs b/src/DotNetCore.CAP.MySql/CAP.MySqlCapOptionsExtension.cs
@@ -31,9 +31,13 @@ public void AddServices(IServiceCollection services)
             {
                 services.AddSingleton(x =>
                 {
-                    var dbContext = (DbContext)x.GetService(mysqlOptions.DbContextType);
-                    mysqlOptions.ConnectionString = dbContext.Database.GetDbConnection().ConnectionString;
-                    return mysqlOptions;
+                    using (var scope = x.CreateScope())
+                    {
+                        var provider = scope.ServiceProvider;
+                        var dbContext = (DbContext)provider.GetService(mysqlOptions.DbContextType);
+                        mysqlOptions.ConnectionString = dbContext.Database.GetDbConnection().ConnectionString;
+                        return mysqlOptions;
+                    }
                 });
             }
             else

diff --git a/src/DotNetCore.CAP.PostgreSql/CAP.PostgreSqlCapOptionsExtension.cs b/src/DotNetCore.CAP.PostgreSql/CAP.PostgreSqlCapOptionsExtension.cs
@@ -31,9 +31,13 @@ public void AddServices(IServiceCollection services)
             {
                 services.AddSingleton(x =>
                 {
-                    var dbContext = (DbContext)x.GetService(postgreSqlOptions.DbContextType);
-                    postgreSqlOptions.ConnectionString = dbContext.Database.GetDbConnection().ConnectionString;
-                    return postgreSqlOptions;
+                    using (var scope = x.CreateScope())
+                    {
+                        var provider = scope.ServiceProvider;
+                        var dbContext = (DbContext)provider.GetService(postgreSqlOptions.DbContextType);
+                        postgreSqlOptions.ConnectionString = dbContext.Database.GetDbConnection().ConnectionString;
+                        return postgreSqlOptions;
+                    }
                 });
             }
             else