如何重写EF Core命令发送机制，将SQL命令转发至其他应用？

当然可以，EF Core提供了**命令拦截器（DbCommandInterceptor）**这一核心扩展点，让你能拦截所有即将发送到数据库的SQL命令，替换为自定义的转发逻辑。以下是具体实现方案：

1. 实现自定义命令拦截器 #

继承DbCommandInterceptor，重写命令执行前的方法，在其中捕获SQL命令和参数，转发到你的多租户管理应用，并阻止EF Core自行执行命令。

public class TenantSqlForwardingInterceptor : DbCommandInterceptor
{
    private readonly ITenantManagementClient _tenantClient;

    public TenantSqlForwardingInterceptor(ITenantManagementClient tenantClient)
    {
        _tenantClient = tenantClient;
    }

    // 拦截同步非查询命令（比如INSERT/UPDATE/DELETE）
    public override InterceptionResult<int> NonQueryExecuting(
        DbCommand command,
        CommandEventData eventData,
        InterceptionResult<int> result)
    {
        // 提取SQL命令和参数
        var sql = command.CommandText;
        var parameters = ExtractParameters(command.Parameters);

        // 转发到多租户管理应用
        _tenantClient.ForwardSqlCommand(sql, parameters);

        // 阻止EF Core继续执行原命令，返回自定义结果（根据业务需求调整）
        return InterceptionResult<int>.SuppressWithResult(0);
    }

    // 拦截异步非查询命令
    public override async ValueTask<InterceptionResult<int>> NonQueryExecutingAsync(
        DbCommand command,
        CommandEventData eventData,
        InterceptionResult<int> result,
        CancellationToken cancellationToken = default)
    {
        var sql = command.CommandText;
        var parameters = ExtractParameters(command.Parameters);

        await _tenantClient.ForwardSqlCommandAsync(sql, parameters, cancellationToken);

        return InterceptionResult<int>.SuppressWithResult(0);
    }

    // 拦截同步查询命令（返回数据的SELECT）
    public override InterceptionResult<DbDataReader> ReaderExecuting(
        DbCommand command,
        CommandEventData eventData,
        InterceptionResult<DbDataReader> result)
    {
        var sql = command.CommandText;
        var parameters = ExtractParameters(command.Parameters);

        // 调用多租户应用执行SQL并获取结果
        var queryResult = _tenantClient.ExecuteQuery(sql, parameters);

        // 将结果转换为DbDataReader返回给EF Core
        var dataReader = queryResult.CreateDataReader();
        return InterceptionResult<DbDataReader>.SuppressWithResult(dataReader);
    }

    // 拦截异步查询命令
    public override async ValueTask<InterceptionResult<DbDataReader>> ReaderExecutingAsync(
        DbCommand command,
        CommandEventData eventData,
        InterceptionResult<DbDataReader> result,
        CancellationToken cancellationToken = default)
    {
        var sql = command.CommandText;
        var parameters = ExtractParameters(command.Parameters);

        var queryResult = await _tenantClient.ExecuteQueryAsync(sql, parameters, cancellationToken);
        var dataReader = queryResult.CreateDataReader();

        return InterceptionResult<DbDataReader>.SuppressWithResult(dataReader);
    }

    // 辅助方法：提取命令参数为字典
    private Dictionary<string, object> ExtractParameters(DbParameterCollection parameters)
    {
        var paramDict = new Dictionary<string, object>();
        foreach (DbParameter param in parameters)
        {
            paramDict.Add(param.ParameterName, param.Value ?? DBNull.Value);
        }
        return paramDict;
    }
}

2. 注册拦截器到EF Core上下文 #

将自定义拦截器注入到EF Core的上下文配置中，确保所有命令都会被拦截：

方式一：在DbContext的OnConfiguring中注册 #

public class MyDbContext : DbContext
{
    private readonly ITenantManagementClient _tenantClient;

    public MyDbContext(ITenantManagementClient tenantClient)
    {
        _tenantClient = tenantClient;
    }

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        // 填写占位符连接字符串（无需实际可用，因为不会用它执行命令）
        optionsBuilder.UseSqlServer("Server=.;Database=Dummy;Trusted_Connection=True;")
                      .AddInterceptors(new TenantSqlForwardingInterceptor(_tenantClient));
    }
}

方式二：在依赖注入容器中注册 #

services.AddScoped<ITenantManagementClient, TenantManagementClient>();
services.AddScoped<TenantSqlForwardingInterceptor>();

services.AddDbContext<MyDbContext>(options =>
{
    options.UseSqlServer("Server=.;Database=Dummy;Trusted_Connection=True;")
           .AddInterceptors(sp => sp.GetRequiredService<TenantSqlForwardingInterceptor>());
});

关键注意事项 #

• 覆盖所有执行路径：必须同时重写同步和异步方法，EF Core会根据你的代码调用方式（比如SaveChanges vs SaveChangesAsync）使用对应的拦截方法。
• 结果处理：如果需要EF Core接收查询结果，必须将多租户应用返回的数据转换为DbDataReader（或兼容的实现），否则EF会抛出异常。
• 事务处理：如果你的业务涉及事务，需要额外重写TransactionStarting、TransactionCommitting等事务相关的拦截方法，将事务操作同步到多租户管理应用。
• 连接字符串占位符：EF Core初始化上下文时会尝试验证连接，因此需要提供一个格式合法的占位符连接字符串，无需实际数据库存在。

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