【EF Core】为 DatabaseFacade 扩展“创建”与“删除”数据表功能

坏级尹 · 8 小时前

对于玩 EF 的大伙伴来说，对 DatabaseFacade 类的 EnsureCreated 和 EnsureDeleted 方法应该很熟悉。这对方法可在运行阶段创建或删除数据库。创建数据库时，会连同数据表一起创建；至于说删除数据库时嘛……库都没了，哪还有表呢。
不过，有些时候，不，不是有些时候，很多时候我们其实只想删除数据表。比如要初始化应用程序；或者数据库已存在的情况下，咱们只考虑判断数据表是否存在，不存在的话就创建表。
最简单的方法是直接发送 SQL 语句——如 DROP TABLE、DROP DATABASE 等。这个不在本文的讨论范围内。老周这次讨论的是运用 EF Core 自身的功能去实现。
咱们来热热身——先学点理论知识，有了这些知识，后面实战起来会容易很多。老掉牙的话：EF Core 也是基于服务容器。即支持依赖注入，这个东西好用着呢，也方便定制和扩展。微软在 .NET 上基本贯彻这个路线了。
咱们都知道，为了方便我们用面向对象的方式操作数据库，EF Core 内部实现了将相关操作以及 LINQ 转译为 SQL 的功能。尽管不能够完全覆盖 SQL 所有功能，但常用的一个不落。
下面介绍几个重量级角色：
NO.1：IMigrationsSqlGenerator 接口。

这个接口的功能很关键，看名字就知道，用来生成 SQL 语句的。这个接口只有一个 Generate 方法，它的签名比较长：

public IReadOnlyList<MigrationCommand> Generate(IReadOnlyList<MigrationOperation> operations, IModel? model = default, MigrationsSqlGenerationOptions options = MigrationsSqlGenerationOptions.Default);

复制代码

第三个参数一般保留默认，第一个参数是一系列 MigrationOperation 对象；第二个参数是模型对象，这个可以从 DbContext.Model 属性返回。MigrationOperation 是一个抽象类，表示一个数据库迁移操作（对的，dotnet ef migration 命令调用的正是迁移 API 来生成 SQL，再将其发送到数据库执行）。它的每一个派生类都代表一种操作。比如，DropColumnOperation 表示从数据表中删除一列，AddUniqueConstraintOperation 用于向数据表添加唯一约束。
注意，这些 Operation 类型其实是“参数封装”器，各个属性用于收集此操作所需要的信息，各个数据库提供者都应该实现具体的操作。比如，在 SQL Server 提供者中，用于创建数据库的 SqlServerCreateDatabaseOperation 类，Name 属性用于设置数据库的名字；FileName 属性设置数据库文件的路径，比如连接字符串中使用了 AttachDbFile 字段。
现在说回 IMigrationsSqlGenerator 接口，它有个基础实现类—— MigrationsSqlGenerator。数据库提供者通常不必直接实现接口，而是从该类派生。例如官方默认实现的 SqliteMigrationsSqlGenerator 类和 SqlServerMigrationsSqlGenerator 类。Generate 方法返回一个 MigrationCommand 列表，这个列表传递给 IMigrationCommandExecutor 接口的 ExecuteNonQuery 或 ExecuteNonQueryAsync 方法就会执行 SQL 命令了。
NO.2：IMigrationCommandExecutor 接口

这个流程很明确，XXXSqlGenerator 负责生成 SQL，XXXCommandExecutor 负责执行命令。这个逻辑相信大伙伴们都能理解。这还不够，咱们还不知道怎么连接数据库呢，执行 SQL 前肯定得连接数据库的哟。于是，有请第三位。
NO.3：IRelationalConnection 接口

这个也不必过多介绍，就是用来连接数据库的。默认的基础类是 RelationalConnection。不用意外，这货一定是抽象类的。毕竟，不同数据库的连接方式是不同的，所以，这个得让数据库提供者们自己去实现。比如面向 SQL Server 的 SqlServerConnection 类，面向 SQLite 的 SqliteRelationalConnection 类。
当然了，IRelationalConnection 对象只是套个壳添加必要的逻辑，真正完成连接数据库任务的是 DbConnection 类（通过 DbConnection 属性引用）。如连接 Sqlite 数据库的 SqliteConnection 类。
NO.4：IDatabaseCreator 接口

这个接口从名字上也能猜到，它是用来创建或删除数据库的。这个接口包含了咱们非常熟悉的 EnsureCreated 和 EnsureDeleted 方法。通常，咱们在用的时候不会从服务容器中获取此接口，而是它的派生接口—— IRelationalDatabaseCreator。重点来了，这。

再明显不过了，HasTables 返回 bool 值，表示数据表是否已存在；CreateTables 就是创建数据表。不过，DatabaseFacade 类没有这些方法，它只能创建、删除数据库。所以，咱们要做的，就是给它加扩展方法，把创建、删除表的方法封装出来。
于是，咱们在实战部分，要分两个阶段。
=====================================================================================
理论课上完了，下面咱们动手实践。由于 IRelationalDatabaseCreator 已经实现了判断数据表存在性和创建数据表的功能，那咱们直接用它。先要定义一个静态类，比如，叫 MyDatabaseFacadeExtension。

public static class MyDatabaseFacadeExtension
{
……
}

复制代码

为了尽量少写重复代码，可以先搞个私有的扩展方法，用来获取 IRelationalDatabaseCreator。

private static IRelationalDatabaseCreator GetDBCreator(this DatabaseFacade db)
{
return db.GetService<IRelationalDatabaseCreator>();
}

复制代码

这个很简单，DatabaseFacade 对象就有一个 GetService 方法，可直接从服务容器中取对象。
接着实现数据表的存在性判断—— HasTables。

/// <summary>
/// 判断表是否存在
/// </summary>
public static bool HasTables(this DatabaseFacade db)
{
IRelationalDatabaseCreator creator = db.GetDBCreator();
return creator.HasTables();
}
public static Task<bool> HasTablesAsync(this DatabaseFacade db)
{
IRelationalDatabaseCreator creator = db.GetDBCreator();
return creator.HasTablesAsync();
}

复制代码

随后就是创建表的封装，也很简单。

/// <summary>
/// 创建表
/// </summary>
public static void CreateTables(this DatabaseFacade db)
{
IRelationalDatabaseCreator creator = db.GetDBCreator();
creator.CreateTables();
}
public static async Task CreateTablesAsync(this DatabaseFacade db)
{
IRelationalDatabaseCreator creator = db.GetDBCreator();
await creator.CreateTablesAsync();
}

复制代码

上面的都好弄，最难的来了！IRelationalDatabaseCreator 只实现了创建表，可没有删除表啊。所以，只好自己动手了。
回想一下前文的理论热身，咱们是不是需要三个服务接口？
1、IRelationalConnection：负责连接数据库；
2、IMigrationsSqlGenerator：负责生成 SQL 命令；
3、IMigrationCommandExecutor：负责执行命令。
由于在调用服务方法时咱们需要 IModel，而它一般可从 DbContext 对象的 Model 属性获取。所以咱们还要想办法从 DatabaseFacade 对象中获取当前 DbContext 对象的引用（其实是从 DbContext 派生的类实例）。仔细观察，可发现 DatabaseFacade 类显示实现了 IDatabaseFacadeDependenciesAccessor 接口，而这个接口有个 Context 属性，正好能获取到 DbContext 实例。
好了，现在，所有难题都解决了，可以开干了。

/// <summary>
/// 删除表
/// </summary>
public static void RemoveTables(this DatabaseFacade db)
{
// 获取服务
IRelationalConnection conn = db.GetService<IRelationalConnection>();
IMigrationsSqlGenerator generator = db.GetService<IMigrationsSqlGenerator>();
IMigrationCommandExecutor executor = db.GetService<IMigrationCommandExecutor>
();
// 获取 DbContext 实例
DbContext context = ((IDatabaseFacadeDependenciesAccessor)db).Context;
// 获取 Model 实例
IModel model = context.Model;
List<MigrationOperation> operations = new();
// 看看要删除哪些表
foreach (var entity in model.GetEntityTypes())
{
DropTableOperation drpopr = new()
{
// 被删除表的架构名
Schema = entity.GetSchema(),
// 被删除表的表名
Name = entity.GetTableName()!
};
operations.Add(drpopr);
}
// 构建命令
var commands = generator.Generate(operations, model);
// 执行命令
executor.ExecuteNonQuery(commands, conn);
}

复制代码

删除数据表，对应的迁移操作是 DropTable，所以实例化一个 DropTableOperation 对象，并设置要删除的表名（可能还有架构名）。接着生成 SQL 命令，最后执行它。完事。
为了兼容，还可以实现异步版本。

public static async Task RemoveTablesAsync(this DatabaseFacade db)
{
// 获取DbContext实例
DbContext context = ((IDatabaseFacadeDependenciesAccessor)db).Context;
// 获取服务
IMigrationCommandExecutor executor = context.
GetService<IMigrationCommandExecutor>();
IMigrationsSqlGenerator generator = context.
GetService<IMigrationsSqlGenerator>();
IRelationalConnection conn = context.GetService<IRelationalConnection>();
// 构建operation列表
List<MigrationOperation> operations = new();
// 删除所有表
foreach (var ent in context.Model.GetEntityTypes())
{
operations.Add(new DropTableOperation()
{
Schema = ent.GetSchema(),
Name = ent.GetTableName()!
});
}
// 生成命令
var cmds = generator.Generate(operations, context.Model);
// 执行命令
await executor.ExecuteNonQueryAsync(cmds, conn);
}

复制代码

这里注意，Model 可能不只一个实体类，即不只一个表，所以要 foreach 逐个访问 IModel.GetEntityTypes 方法返回的实体类型集合。把实体对应的表都删除。
可以测试一下。假设用 SQL Server ，事先建一个数据库，里面没有表。

public class Student
{
public int ID { get; set; }
public string Name { get; set; } = string.Empty;
public string? Major { get; set; }
public int Age { get; set; }
public string? Email { get; set; }
}
public class MyDbContext : DbContext
{
protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
{
optionsBuilder.UseSqlServer("server=(localdb)\\MSSQLLOCALDB; database=Demo");
}
protected override void OnModelCreating(ModelBuilder modelBuilder)
{
modelBuilder.Entity<Student>(et =>
{
et.HasKey(x => x.ID).HasName("PK_Student");
et.Property(x => x.ID).HasColumnName("sid")
.UseIdentityColumn(100, 1);
});
}
}

复制代码

模型配置时，让主键例的名字为 sid。
运行程序时，检查有没有数据表，没有就创建。

var dc = new MyDbContext();
if (!dc.Database.HasTables())
{
dc.Database.CreateTables();
}

复制代码

若需要删除数据表，可以这样写：

dc.Database.RemoveTables();

复制代码

怎么样，这样封装后是不是就和官方 API 那样方便？

来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！

账号		自动登录	找回密码
密码			立即注册