ASP.NET Core 2 + 获取数据库上下文实例

Question

我正在尝试获取 DbContext 的实例（因此我可以在启动时做一些额外的工作），尝试在 Configure 方法中获取实例时出现以下错误：

System.InvalidOperationException: '无法从根提供者解析范围服务'MyApp.Data.MyDbContext'。'

public void ConfigureServices(IServiceCollection services)
{
 services.AddDbContext<MyDbContext>(
                options => options.UseSqlServer(Configuration.GetConnectionString("MyDbContext")));
}

public void Configure(IApplicationBuilder app, IHostingEnvironment env)
{

    var dbContext = app.ApplicationServices.GetService(typeof(MyDbContext)) as MyDbContext;
}

我可以通过控制器等访问 DbContext 的实例

Answer 1

Paul Hiles 的评论是正确的，但该方法在 .NET Core 1.0 中效果更好。

在 ASP.NET Core 2.0 中，在 Startup.cs 中运行任何数据库设置通常不是一个好主意。这是因为如果您从 CLI 或 Visual Studio 运行任何迁移，它将运行所有 Startup.cs 并尝试运行将失败的配置。当然，如果您不使用 Entity-Framework，那么这不是问题，但它仍然不是 2.0 中推荐的方式。现在建议在Program.cs 中进行。

例如，您可以创建IWebHost 的扩展方法，它将运行您需要的任何设置。

public static IWebHost MigrateDatabase(this IWebHost webHost)
{
    var serviceScopeFactory = (IServiceScopeFactory)webHost.Services.GetService(typeof(IServiceScopeFactory));

    using (var scope = serviceScopeFactory.CreateScope())
    {
        var services = scope.ServiceProvider;
        var dbContext = services.GetRequiredService<YourDbContext>();

        dbContext.Database.Migrate();
    }

    return webHost;
}

然后在Program.cs 中，您可以在运行之前调用该方法。

public static void Main(string[] args)
{
    BuildWebHost(args)
        .MigrateDatabase()
        .Run();
}

Answer 2

Core 2.1 以后的更新

只是为了补充@Travis Boatman 的出色答案，首选的Main 方法语法具有changed slightly from Core 2.1 onwards，默认Main 方法现在具有CreateWebHostBuilder 而不是BuildWebHost。

调用扩展方法的修改代码如下所示。

注意：这里的顺序很重要，Build 方法返回一个WebHost，这是扩展方法正在扩展的，所以你需要在Build() 之后和Run() 之前调用 migrate 方法）：

public static void Main(string[] args)
{
    CreateWebHostBuilder(args)
        .Build()
        .MigrateDatabase()
        .Run();
}

迁移多个 DbContext

我们的项目中有不止一个DbContext，所以我把扩展方法改成了可以接受任何类型DbContext的泛型方法：

public static IWebHost MigrateDatabase<T>(this IWebHost webHost) where T:DbContext
{
    var serviceScopeFactory = (IServiceScopeFactory)webHost
        .Services.GetService(typeof(IServiceScopeFactory));

    using (var scope = serviceScopeFactory.CreateScope())
    {
        var services = scope.ServiceProvider;

        var dbContext = services.GetRequiredService<T>();
        dbContext.Database.Migrate();
    }

    return webHost;
}

然后您可以链接调用以迁移不同的上下文：

CreateWebHostBuilder(args)
    .Build()
    .MigrateDatabase<ApiAuthDbContext>()
    .MigrateDatabase<MainDbContext>()
    .MigrateDatabase<SomeOtherDbContext>()
    .Run();