EntityFrameworkCore 模型自动更新（下）

🚀 优质资源分享 🚀

话题

上一篇我们讨论到获取将要执行的迁移操作，到这一步为止，针对所有数据库都通用，在此之后需要生成SQL脚本对于不同数据库将有不同差异，我们一起来瞅一瞅

SQLite脚本生成差异

在上一篇拿到的迁移操作类即MigrationOperation为执行所有其他操作类的父类，比如添加列操作（AddColumnOperation），修改列操作（AlterColumnOperation）、创建表操作（CreateTableOperation）等等，我们知道SQLite不支持修改列，所以我们需要去除列修改操作，代码如下：

// Sqlite不支持修改操作，所以需过滤修改迁移操作
var operations = migrationOperations.Except(migrationOperations.Where(o => o is AlterColumnOperation)).ToList();

if (!operations.Any())
{
 return;
}
1
2
3
4
5
6
7

然后获取生成SQL脚本接口，拿到执行操作命令类里面的脚本文本即可

var migrationsSqlGenerator = context.GetService();

var commandList = migrationsSqlGenerator.Generate(operations);

if (!commandList.Any())
{
 return;
}

var sqlScript = string.Concat(commandList.Select(c => c.CommandText));

if (string.IsNullOrEmpty(sqlScript))
{
 return;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

MySQL脚本生成差异

因为我们可能会修改主键，此时Pomelo.EntityFrameworkCore.MySql使用的方式则是创建一个临时存储过程，先删除主键，然后则执行完相关脚本后，最后重建主键，然后删除临时存储过程，临时存储过程如下：

#region Custom Sql
 #region BeforeDropPrimaryKey

 private const string BeforeDropPrimaryKeyMigrationBegin = @"DROP PROCEDURE IF EXISTS `POMELO\_BEFORE\_DROP\_PRIMARY\_KEY`;
CREATE PROCEDURE `POMELO\_BEFORE\_DROP\_PRIMARY\_KEY`(IN `SCHEMA\_NAME\_ARGUMENT` VARCHAR(255), IN `TABLE\_NAME\_ARGUMENT` VARCHAR(255))
BEGIN
    DECLARE HAS\_AUTO\_INCREMENT\_ID TINYINT(1);
 DECLARE PRIMARY\_KEY\_COLUMN\_NAME VARCHAR(255);
 DECLARE PRIMARY\_KEY\_TYPE VARCHAR(255);
 DECLARE SQL\_EXP VARCHAR(1000);
 SELECT COUNT(*)
 INTO HAS\_AUTO\_INCREMENT\_ID
 FROM `information\_schema`.`COLUMNS`
 WHERE `TABLE\_SCHEMA`
1
2
3
4
5
6
7
8
9
10
11
12
13