UOW+Repository+Autofac加载两个不同的DbContext

本文关键字：DbContext 两个 加载 UOW+Repository+Autofac | 更新日期: 2023-09-27 18:19:35

我今天遇到一个问题，无法解决，我搜索了很多，无法找到解决方案，如果可以的话，请帮助我。

我正在实现一个MVC应用程序，它使用EF+Repository Pattern+Unit Of Work和Autofac作为依赖项注入器。

我能够使用一个DbContext类，但我面临的情况是，我需要使用另一个DbContextInstance（使用另一用户凭据访问另一个数据库）

让我更好地解释：我有来自数据库A的EntityA（并且有一个DatabaseA_Context类）。所以我需要一个EntityB，它来自数据库B（有自己的DatabaseB_Context类）。

当我向AutoFac注册它们时，只有最后配置的依赖项被注入到GenericRepository实现中。

我已经发现一些文章说，Autofac会用最后一个值覆盖注册。

我已经找到了另一篇文章，它表明如果我在UnitOfWork构造函数上传递一个IEnumerable，我可以看到它的所有注册类型，但我想要一个特定的类型。

我够清楚吗？

我下面的代码：

我的控制器：

public class MyController : Controller
{
    private readonly IBaseBLL<EntityA> _aBLL;
    private readonly IBaseBLL<EntityB> _bBll;
    public MyController(IBaseBLL<EntityA> aBLL, IBaseBLL<EntityB> bBLL)
    {
        _aBLL = aBLL;
        _bBLL = bBLL;
    }
}

我的业务层

public interface IBaseBLL<T> where T : class
{
    T Select(Expression<Func<T, bool>> predicate);
    T AddT entity);
    void Update(T entity);
    T Delete(T entity);
}
public class BaseBLL<T> : IBaseBLL<T> where T : class
{
    private readonly IUnitOfWork _uow;
    public BaseBLL(IUnitOfWork uow)
    {
        _uow = uow;
    }
    //implementation goes here...
}

我的UOW实施

public interface IUnitOfWork : IDisposable
{
    int SaveChanges();
    IGenericRepository<T> Repository<T>() where T : class;
}
public class UnitOfWork : IUnitOfWork
{
    private readonly DbContext _dbContext;
    private bool disposed = false;
    private Dictionary<Type, object> repositories;

    public UnitOfWork(DbContext dbContext)
    {
        _dbContext = dbContext;
        repositories = new Dictionary<Type, object>();
    }
    public IGenericReposity<T> Repository<T>() where T : class
    {
        if (repositories.Keys.Contains(typeof(T)))
            return repositories[typeof(T)] as IGenericReposity<T>;
        IGenericReposity<T> repository = new GenericRepository<T>(_dbContext);
        repositories.Add(typeof(T), repository );
        return repository ;
    }
    public int SaveChanges()
    {
        return _dbContext.SaveChanges();
    }
    protected virtual void Dispose(bool disposing)
    {
        if (!this.disposed)
            if (disposing)
                _dbContext.Dispose();
        this.disposed = true;
    }
    public void Dispose()
    {
        Dispose(true);
        GC.SuppressFinalize(this);
    }
}

我的存储库实现

public class GenericRepository<T> : IGenericRepositoryT> where T : class
{
    protected readonly DbContext _dbContext;
    protected IDbSet<T> _dbSet;
    public GenericRepository(DbContext dbContext)
    {
        _dbContext = dbContext;
        _dbSet = _dbContext.Set<T>();
    }
    //implementation goes here...
}

我的AutoFac注册（在Global.asax文件中）

var builder = new ContainerBuilder();
builder.RegisterType(typeof(DatabaseA_Context)).As(typeof(DbContext)).InstancePerLifetimeScope();
builder.RegisterType(typeof(DatabaseB_Context)).As(typeof(DbContext)).InstancePerLifetimeScope();
builder.RegisterType(typeof(UnitOfWork)).As(typeof(IUnitOfWork)).InstancePerRequest();

请帮助

UOW+Repository+Autofac加载两个不同的DbContext

您应该使用命名和键控服务

builder.RegisterType<DatabaseA_Context>()
       .Named<DbContext>("databaseA")
       .InstancePerLifetimeScope();
builder.RegisterType<DatabaseB_Context>()
       .Named<DbContext>("databaseB")
       .InstancePerLifetimeScope();

然后，您可以在注册时为组件指定所需的DbContext

builder.RegisterType<MyService>()
       .As<IService>()
       .WithParameter((pi, c) => pi.Name == "dbContext", 
                      (pi, c) => c.ResolveNamed<DbContext>("databaseA"))

或者通过使用IIndex<,>

public class MyService : IService
{
    public MyService(IIndex<String, DbContext> dbContexts)
    {
        var databaseA = dbContexts["databaseA"];
    }
}

Autofac还支持使用WithKeyAttribute 指定命名注册

public class MyService : IService
{
    public MyService([WithKey("DatabaseA")DbContext dbContext)
    {
    }
}

有关如何设置WithKeyAttribute的更多信息，请参阅元数据文档。

使用此解决方案，将不会注册DbContext。如果你想要一个默认的DbContext，你可以注册一个这样的：

builder.Register(c => c.ResolveNamed<DbContext>("databaseA"))
       .As<DbContext>()
       .InstancePerLifetimeScope();

您还可以使用一个模块，该模块将根据参数的名称选择正确的注册：

public class MyService : IService
{
    public MyService(DbContext dbContextA, DbContext dbContextB)
    {
    }
}

为此，您需要注册此Autofac模块

public class DbContextModule : Module
{
    protected override void AttachToComponentRegistration(
        IComponentRegistry componentRegistry, IComponentRegistration registration)
    {
        registration.Preparing += Registration_Preparing;
    }
    private void Registration_Preparing(Object sender, PreparingEventArgs e)
    {
        Parameter parameter = new ResolvedParameter(
                                (pi, c) => pi.ParameterType == typeof(DbContext),
                                (pi, c) =>
                                {
                                    if (pi.Name.Equals("dbContextA", StringComparison.OrdinalIgnoreCase))
                                    {
                                        return c.ResolveNamed<DbContext>("databaseA");
                                    }
                                    else if (pi.Name.Equals("dbContextB", StringComparison.OrdinalIgnoreCase))
                                    {
                                        return c.ResolveNamed<DbContext>("databaseB");
                                    }
                                    else
                                    {
                                        throw new NotSupportedException($"DbContext not found for '{pi.Name}' parameter name");
                                    }
                                });
        e.Parameters = e.Parameters.Concat(new Parameter[] { parameter });
    }
}

和

builder.RegisterModule<DbContextModule>()
相关文章：