如何在Autofac中使用泛型UnitOfWork<TContext>解决多DBContext调用问题

我来帮你搞定这个泛型UnitOfWork的DbContext解析问题，核心是让Autofac能根据泛型参数TContext动态提供对应的DbContextOptions<TContext>，并正确实例化DbContext注入到UnitOfWork中。下面是具体的实现步骤：

1. 调整DbContext的Autofac注册方式 #

首先，我们需要让Autofac能针对不同的TContext生成对应的DbContextOptions<TContext>。你可以用泛型注册的方式来实现这一点，这样不需要为每个DbContext单独写重复的注册代码：

var containerBuilder = new ContainerBuilder();

// 注册泛型DbContextOptions<>，根据上下文类型动态生成配置
containerBuilder.RegisterGeneric(typeof(DbContextOptions<>))
                .AsSelf()
                .InstancePerLifetimeScope()
                .WithParameter(
                    // 匹配DbContextOptions<T>类型的参数
                    (parameterInfo, context) => 
                        parameterInfo.ParameterType.IsGenericType && 
                        parameterInfo.ParameterType.GetGenericTypeDefinition() == typeof(DbContextOptions<>),
                    // 根据上下文类型构建对应的选项
                    (parameterInfo, context) => 
                    {
                        var contextType = parameterInfo.ParameterType.GetGenericArguments()[0];
                        var optionsBuilder = new DbContextOptionsBuilder();
                        
                        // 针对不同的DbContext配置连接字符串或其他选项
                        if (contextType == typeof(DBContext1))
                        {
                            optionsBuilder.UseSqlServer("你的DBContext1连接字符串");
                        }
                        // 可以添加更多DbContext的配置分支，比如DBContext2、DBContext3等
                        // else if (contextType == typeof(DBContext2)) { ... }
                        
                        return optionsBuilder.Options;
                    });

// 注册具体的DbContext（比如你的DBContext1）
containerBuilder.RegisterType<DBContext1>()
                .AsSelf()
                .InstancePerLifetimeScope();

// 如果有多个DbContext，继续添加类似的注册语句
// containerBuilder.RegisterType<DBContext2>().AsSelf().InstancePerLifetimeScope();

如果你的DbContext数量不多，也可以直接为每个DbContext单独注册并指定选项，这种方式更直观：

// 直接注册DBContext1并传入对应的DbContextOptions
containerBuilder.RegisterType<DBContext1>()
                .AsSelf()
                .InstancePerLifetimeScope()
                .WithParameter(
                    typeof(DbContextOptions<DBContext1>),
                    new DbContextOptionsBuilder<DBContext1>()
                        .UseSqlServer("你的DBContext1连接字符串")
                        .Options);

2. 实现泛型UnitOfWork #

你的泛型UnitOfWork只需要在构造函数中接收TContext类型的参数即可，Autofac会自动帮你解析对应的DbContext实例：

public class UnitOfWork<TContext> : IUnitOfWork<TContext> where TContext : DbContext
{
    private readonly TContext _dbContext;

    // 构造函数直接注入TContext，Autofac会自动处理其依赖的DbContextOptions<TContext>
    public UnitOfWork(TContext dbContext)
    {
        _dbContext = dbContext ?? throw new ArgumentNullException(nameof(dbContext));
    }

    // 示例：实现UnitOfWork的核心方法
    public async Task<int> SaveChangesAsync(CancellationToken cancellationToken = default)
    {
        return await _dbContext.SaveChangesAsync(cancellationToken);
    }

    // 其他UnitOfWork方法，比如事务管理等
}

// 对应的接口（可选，推荐使用接口便于依赖注入和解耦）
public interface IUnitOfWork<TContext> where TContext : DbContext
{
    Task<int> SaveChangesAsync(CancellationToken cancellationToken = default);
}

3. 注册泛型UnitOfWork #

最后，把泛型UnitOfWork注册到Autofac中：

containerBuilder.RegisterGeneric(typeof(UnitOfWork<>))
                .As(typeof(IUnitOfWork<>))
                .InstancePerLifetimeScope();

为什么这样可行？ #

当你需要使用IUnitOfWork<DBContext1>时，Autofac会：

1. 尝试解析DBContext1实例
2. 发现DBContext1的构造函数需要DbContextOptions<DBContext1>
3. 根据我们注册的泛型DbContextOptions<>工厂方法（或直接注册的参数）生成对应的选项
4. 实例化DBContext1并注入到UnitOfWork<DBContext1>中

这样就实现了运行时根据传入的TContext动态创建DbContext实例的需求。

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