如何让EF Core的Script-Migration在脚本开头添加USE语句？

除了构建后操作或自定义MSBuild任务，有两种更直接的实现方式：

方法一：自定义MigrationSqlGenerator（EF Core扩展点） #

这是最贴合EF Core迁移流程的方案，通过扩展EF Core的SQL生成逻辑，让每次生成的脚本自动包含开头的USE语句。

步骤1：创建自定义SQL生成器 #

继承SqlServerMigrationSqlGenerator（针对SQL Server，若使用其他数据库需对应调整），重写Generate方法插入前置语句：

using Microsoft.EntityFrameworkCore.Migrations;
using Microsoft.EntityFrameworkCore.Migrations.Sql;

public class CustomSqlServerMigrationSqlGenerator : SqlServerMigrationSqlGenerator
{
    private readonly string _targetDbName;

    public CustomSqlServerMigrationSqlGenerator(
        MigrationsSqlGeneratorDependencies dependencies,
        IMigrationsAnnotationProvider annotationsProvider,
        string targetDbName)
        : base(dependencies, annotationsProvider)
    {
        _targetDbName = targetDbName;
    }

    public override IReadOnlyList<MigrationCommand> Generate(
        IReadOnlyList<MigrationOperation> operations,
        IModel model = null)
    {
        // 先调用默认逻辑生成迁移命令
        var migrationCommands = base.Generate(operations, model).ToList();
        
        // 构造USE和GO命令插入到最前面
        var useDbCommand = new MigrationCommand(
            $"USE [{_targetDbName}]\nGO",
            annotations: null);
        
        migrationCommands.Insert(0, useDbCommand);
        
        return migrationCommands;
    }
}

步骤2：在DbContext中注册自定义生成器 #

在OnConfiguring方法里替换默认的迁移SQL生成服务：

protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
{
    optionsBuilder.UseSqlServer("你的数据库连接字符串")
        .ReplaceService<IMigrationsSqlGenerator, CustomSqlServerMigrationSqlGenerator>(
            (serviceProvider, dependencies) => new CustomSqlServerMigrationSqlGenerator(
                dependencies,
                serviceProvider.GetRequiredService<IMigrationsAnnotationProvider>(),
                "你的目标数据库名称"));
}

完成后，每次运行Script-Migration命令，生成的脚本开头都会自动带上USE [database-name]/GO语句，完全集成在EF Core的迁移流程中。

方法二：通过命令行管道直接拼接脚本 #

无需修改代码，利用命令行的文本处理能力，将Script-Migration的输出与前置语句拼接后保存：

PowerShell示例（Package Manager Console或终端） #

# 定义数据库名称和前置语句
$targetDb = "你的数据库名称"
$preScript = "USE [$targetDb]`nGO`n"

# 生成迁移脚本并拼接后保存
$migrationScript = Script-Migration
$fullScript = $preScript + $migrationScript
$fullScript | Out-File .\GeneratedMigrations.sql

.NET CLI终端示例 #

$targetDb = "你的数据库名称"
$preScript = "USE [$targetDb]`n`nGO`n`n"
# 直接通过管道拼接输出
$preScript + (dotnet ef migrations script --output -) | Out-File .\GeneratedMigrations.sql

这种方式灵活，适合临时调整或不想修改项目代码的场景，不属于构建后任务或自定义MSBuild任务范畴。

内容的提问来源于stack exchange，提问作者Mike