解决这个问题的方法是不让UnitOfWork 负责通过容器注入创建每个Repository，而是让每个Repository 负责确保UnitOfWork 知道它的存在在实例化时。
这将确保
- 您的
UnitOfWork 不需要为每个新的Repository 更改
- 您没有使用服务定位器（许多人认为是anti-pattern）
最好用一些代码来演示——我使用SimpleInjector，所以示例是基于这个：
从Repository抽象开始：
public interface IRepository
{
void Submit();
}
public interface IRepository<T> :IRepository where T : class { }
public abstract class GenericRepository<T> : IRepository<T> where T : class { }
还有UnitOfWork
public interface IUnitOfWork
{
void Register(IRepository repository);
void Commit();
}
每个Repository必须向UnitOfWork注册自己，这可以通过更改抽象父类GenericRepository来确保完成：
public abstract class GenericRepository<T> : IRepository<T> where T : class
{
public GenericRepository(IUnitOfWork unitOfWork)
{
unitOfWork.Register(this);
}
}
每个真实的Repository 都继承自GenericRepository：
public class Department { }
public class Student { }
public class DepartmentRepository : GenericRepository<Department>
{
public DepartmentRepository(IUnitOfWork unitOfWork): base(unitOfWork) { }
}
public class StudentRepository : GenericRepository<Student>
{
public StudentRepository(IUnitOfWork unitOfWork) : base(unitOfWork) { }
}
添加UnitOfWork 的物理实现，一切就绪：
public class UnitOfWork : IUnitOfWork
{
private readonly Dictionary<string, IRepository> _repositories;
public UnitOfWork()
{
_repositories = new Dictionary<string, IRepository>();
}
public void Register(IRepository repository)
{
_repositories.Add(repository.GetType().Name, repository);
}
public void Commit()
{
_repositories.ToList().ForEach(x => x.Value.Submit());
}
}
可以将容器注册设置为自动获取所有已定义的 IRepository 实例并在生命周期范围内注册它们，以确保它们在事务的生命周期内都存在：
public static class BootStrapper
{
public static void Configure(Container container)
{
var lifetimeScope = new LifetimeScopeLifestyle();
container.Register<IUnitOfWork, UnitOfWork>(lifetimeScope);
container.RegisterManyForOpenGeneric(
typeof(IRepository<>),
lifetimeScope,
typeof(IRepository<>).Assembly);
}
}
通过这些抽象和围绕 DI 构建的架构，您有一个 UnitOfWork，它知道在任何服务调用中已实例化的所有 Repository，并且您已在编译时验证您的所有存储库都已定义。你的代码是open for extension but closed for modification。
要测试所有这些 - 添加这些类
public class SomeActivity
{
public SomeActivity(IRepository<Department> departments) { }
}
public class MainActivity
{
private readonly IUnitOfWork _unitOfWork;
public MainActivity(IUnitOfWork unitOfWork, SomeActivity activity)
{
_unitOfWork = unitOfWork;
}
public void test()
{
_unitOfWork.Commit();
}
}
将这些行添加到BootStrapper.Configure()
//register the test classes
container.Register<SomeActivity>();
container.Register<MainActivity>();
在代码行中设置断点：
_repositories.ToList().ForEach(x => x.Value.Submit());
最后，运行这个控制台测试代码：
class Program
{
static void Main(string[] args)
{
Container container = new Container();
BootStrapper.Configure(container);
container.Verify();
using (container.BeginLifetimeScope())
{
MainActivity entryPoint = container.GetInstance<MainActivity>();
entryPoint.test();
}
}
}
您会发现代码在断点处停止，并且您有一个 IRepository 的活动实例准备就绪并等待 Submit() 对数据库进行任何更改。
你可以装饰你的 UnitOfWork 来处理事务等。我会听从强大的 .NetJunkie 并推荐你阅读这两篇文章 here 和 here。