Question

我有一个asp.net核心项目，它需要能够在运行时支持插件，因此，我需要根据插入的内容生成数据库表。插件分别在不同的项目和他们有自己的DbContext类。在编译期间，仅在运行时才知道要使用的插件。

现在在EF Core我认为会有像“UpdateDatabase”这样的方法，你可以在这里添加表到现有数据库，但我错了。有没有办法实现这个目标？我能够为每个插件生成一个单独的数据库，但这并不是我想到的。我需要一个数据库中的所有表。

以下是“HRContext”插件的代码：

using System;
using System.Collections.Generic;
using System.Text;
using Microsoft.EntityFrameworkCore;
using Plugins.HR.Models.Entities;

namespace Plugins.HR.Contexts
{
    public class HrContext : DbContext
    {
        public HrContext()
        {
        }
        public HrContext(DbContextOptions<HrContext> contextOptions) : base(contextOptions)
        {
        }

        protected override void OnModelCreating(ModelBuilder modelBuilder)
        {
            modelBuilder.HasDefaultSchema("HR");
            base.OnModelCreating(modelBuilder);
        }

        public DbSet<Address> Address { get; set; }
        public DbSet<Attendance> Attendance { get; set; }
        public DbSet<Department> Departments { get; set; }
        public DbSet<Employee> Employees { get; set; }
        public DbSet<JobTitle> JobTitles { get; set; }
    }
}

这是“CoreContext”插件的另一段代码：

using System;
using System.Collections.Generic;
using System.Text;
using Core.Data.Models;
using Microsoft.EntityFrameworkCore;
namespace Core.Data.Contexts
{
    public class CoreContext : DbContext
    {
        public CoreContext()
        {

        }
        public CoreContext(DbContextOptions<CoreContext> contextOptions) : base(contextOptions)
        {

        }

        protected override void OnModelCreating(ModelBuilder modelBuilder)
        {
            modelBuilder.HasDefaultSchema("Core");
            base.OnModelCreating(modelBuilder);

        }
        public DbSet<Test> Tests { get; set; }
    }
}

Startup.cs中的我的ConfigureServices方法：

public void ConfigureServices(IServiceCollection services)
{
    services.AddDbContext<CoreContext>(options => options.UseSqlServer("Data source = localhost; initial catalog = Company.Core; integrated security = true;"))
    .AddDbContext<HrContext>(options => options.UseSqlServer("Data source = localhost; initial catalog = Company.HR; integrated security = true;"));

    // Add framework services.
    services.AddMvc();
}

如果我尝试将连接字符串更改为相同，迟早我会收到一条错误消息，指出一个插件的表不存在。我试过“EnsureCreated”，但这也不行。

Answer 1

我有同样的问题。几天前在GitHub上查看我的解决方案，在这里：EF Core Issue #9238

您需要的是以下内容：

// Using an interface, so that we can swap out the implementation to support PG or MySQL, etc if we wish...
public interface IEntityFrameworkHelper
{
    void EnsureTables<TContext>(TContext context)
        where TContext : DbContext;
}

// Default implementation (SQL Server)
public class SqlEntityFrameworkHelper : IEntityFrameworkHelper
{
    public void EnsureTables<TContext>(TContext context)
        where TContext : DbContext
    {
        string script = context.Database.GenerateCreateScript(); // See issue #2943 for this extension method
        if (!string.IsNullOrEmpty(script))
        {
            try
            {
                var connection = context.Database.GetDbConnection();

                bool isConnectionClosed = connection.State == ConnectionState.Closed;

                if (isConnectionClosed)
                {
                    connection.Open();
                }

                var existingTableNames = new List<string>();
                using (var command = connection.CreateCommand())
                {
                    command.CommandText = "SELECT table_name from INFORMATION_SCHEMA.TABLES WHERE table_type = 'base table'";

                    using (var reader = command.ExecuteReader())
                    {
                        while (reader.Read())
                        {
                            existingTableNames.Add(reader.GetString(0).ToLowerInvariant());
                        }
                    }
                }

                var split = script.Split(new[] { "CREATE TABLE " }, StringSplitOptions.RemoveEmptyEntries);
                foreach (string sql in split)
                {
                    var tableName = sql.Substring(0, sql.IndexOf("(", StringComparison.OrdinalIgnoreCase));
                    tableName = tableName.Split('.').Last();
                    tableName = tableName.Trim().TrimStart('[').TrimEnd(']').ToLowerInvariant();

                    if (existingTableNames.Contains(tableName))
                    {
                        continue;
                    }

                    try
                    {
                        using (var createCommand = connection.CreateCommand())
                        {
                            createCommand.CommandText = "CREATE TABLE " + sql.Substring(0, sql.LastIndexOf(";"));
                            createCommand.ExecuteNonQuery();
                        }
                    }
                    catch (Exception)
                    {
                        // Ignore
                    }
                }

                if (isConnectionClosed)
                {
                    connection.Close();
                }
            }
            catch (Exception)
            {
                // Ignore
            }
        }
    }
}

然后在Startup.Configure()结束时，我解析了IEntityFrameworkHelper个实例，并将其与DbContext的实例一起使用来调用EnsureTables()。

一个问题是我还需要考虑脚本中不是 CREATE TABLE 语句的部分。例如， CREATE INDEX 语句。

我要求他们为我们提供一个干净的解决方案，例如：向CreateTable<TEntity>()添加IRelationalDatabaseCreator方法。虽然没有屏住呼吸......

修改

我忘了发布GenerateCreateScript()的代码。见下文：

using System.Text; using Microsoft.EntityFrameworkCore.Infrastructure; using Microsoft.EntityFrameworkCore.Metadata; using Microsoft.EntityFrameworkCore.Migrations; using Microsoft.EntityFrameworkCore.Storage; public static class DatabaseFacadeExtensions { public static string GenerateCreateScript(this DatabaseFacade database) { var model = database.GetService<IModel>(); var migrationsModelDiffer = database.GetService<IMigrationsModelDiffer>(); var migrationsSqlGenerator = database.GetService<IMigrationsSqlGenerator>(); var sqlGenerationHelper = database.GetService<ISqlGenerationHelper>(); var operations = migrationsModelDiffer.GetDifferences(null, model); var commands = migrationsSqlGenerator.Generate(operations, model); var stringBuilder = new StringBuilder(); foreach (var command in commands) { stringBuilder .Append(command.CommandText) .AppendLine(sqlGenerationHelper.BatchTerminator); } return stringBuilder.ToString(); } }

它基于此处的代码：EF Core Issue #2943

EF Core - 在运行时期间向数据库添加新表

1 个答案: