如何实现无需指定具体DbContext的通用MediatR命令保存管道

核心思路 #

让命令处理程序通过接口暴露自身依赖的DbContext，而非在管道行为中硬编码具体上下文类型，同时保持IPipelineBehavior的泛型兼容性，确保DI注册正常。

步骤1：定义接口规范 #

先定义两个关键接口：一个标记命令支持自动保存配置，另一个让处理程序暴露DbContext实例：

// 标记命令可配置自动保存
public interface ISaveableCommand
{
    bool DisableAutoSave { get; }
}

// 标记处理程序持有DbContext
public interface IDbContextHandler
{
    DbContext DbContext { get; }
}

步骤2：改造抽象处理程序基类 #

让你的DbCommandHandler基类实现IDbContextHandler，并注入通用DbContext：

public abstract class DbCommandHandler<TCommand, TResponse> 
    : IRequestHandler<TCommand, TResponse>, IDbContextHandler
    where TCommand : IRequest<TResponse>, ISaveableCommand
{
    protected readonly DbContext _dbContext;

    protected DbCommandHandler(DbContext dbContext)
    {
        _dbContext = dbContext;
    }

    // 实现接口暴露DbContext
    public DbContext DbContext => _dbContext;

    public abstract Task<TResponse> Handle(TCommand request, CancellationToken cancellationToken);
}

步骤3：修改管道行为类 #

更新SaveCommandBehavior，通过IDbContextHandler接口从处理程序获取DbContext，不再依赖具体类型：

public class SaveCommandBehavior<TRequest, TResponse> : IPipelineBehavior<TRequest, TResponse>
    where TRequest : IRequest<TResponse>, ISaveableCommand
{
    public async Task<TResponse> Handle(TRequest request, RequestHandlerDelegate<TResponse> next, CancellationToken cancellationToken)
    {
        // 先执行命令处理逻辑
        var response = await next();

        // 检查是否需要跳过自动保存
        if (request.DisableAutoSave)
        {
            return response;
        }

        // 从处理程序实例中获取DbContext
        if (next.Target is IDbContextHandler handler)
        {
            await handler.DbContext.SaveChangesAsync(cancellationToken);
        }

        return response;
    }
}

步骤4：DI注册配置 #

正常注册MediatR、管道行为和你的DbContext即可，无需修改行为的泛型参数：

// 注册MediatR及处理程序
services.AddMediatR(cfg => cfg.RegisterServicesFromAssembly(typeof(Program).Assembly));
// 注册管道行为
services.AddScoped(typeof(IPipelineBehavior<,>), typeof(SaveCommandBehavior<,>));
// 注册具体DbContext（这里依然注册你的MyDbContext，但行为不再绑定它）
services.AddDbContext<MyDbContext>(options =>
{
    // 你的数据库配置逻辑
});

示例代码 #

命令类

public class CreateUserCommand : IRequest<Guid>, ISaveableCommand
{
    public string Username { get; set; }
    // 默认开启自动保存，可按需关闭
    public bool DisableAutoSave { get; set; } = false;
}

处理程序类

public class CreateUserCommandHandler : DbCommandHandler<CreateUserCommand, Guid>
{
    // 注入具体的MyDbContext，基类会转为通用DbContext
    public CreateUserCommandHandler(MyDbContext dbContext) : base(dbContext)
    {
    }

    public override async Task<Guid> Handle(CreateUserCommand request, CancellationToken cancellationToken)
    {
        var user = new User { Username = request.Username };
        _dbContext.Users.Add(user);
        // 无需手动调用SaveChanges，管道行为会自动处理
        return user.Id;
    }
}

原理说明 #

• 管道行为通过next.Target获取到处理程序实例，再通过IDbContextHandler接口拿到DbContext，彻底解除了对具体上下文类型的绑定。
• 保持了IPipelineBehavior<TRequest, TResponse>的标准泛型约束，DI注册不会出现冲突。
• 命令通过ISaveableCommand的DisableAutoSave属性灵活控制是否执行自动保存。

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