工作单元模式(Unit of Work)在 .NET 中用于管理多个仓储对数据库的操作,确保所有更改在一个事务中被统一提交或回滚。它的核心作用是协调多个仓储操作,保持数据一致性,特别是在涉及多个实体或聚合根的业务逻辑中。
工作单元模式的作用
在执行一个业务操作时,可能需要更新多个不同的实体,比如订单和库存。如果每个仓储都独立提交更改,就可能出现部分成功、部分失败的情况,导致数据不一致。工作单元通过跟踪所有变更,并提供一个统一的
SaveChanges()方法来解决这个问题。
以 Entity Framework Core 为例,
DbContext本身就实现了工作单元模式: 它跟踪实体的状态变化(新增、修改、删除) 支持事务管理 通过一次
SaveChanges()提交所有更改
与仓储模式结合使用
仓储(Repository)负责封装对特定实体的数据访问逻辑,而工作单元(UnitOfWork)负责协调多个仓储共享同一个数据上下文,从而实现事务一致性。
典型实现方式如下:
public interface IUnitOfWork : IDisposable{
IOrderRepository Orders { get; }
IProductRepository Products { get; }
int Complete(); // 对应 SaveChanges
}
public class UnitOfWork : IUnitOfWork
{
private readonly AppDbContext _context;
public IOrderRepository Orders { get; private set; }
public IProductRepository Products { get; private set; }
public UnitOfWork(AppDbContext context)
{
_context = context;
Orders = new OrderRepository(_context);
Products = new ProductRepository(_context);
}
public int Complete() => _context.SaveChanges();
public void Dispose() => _context.Dispose();
}
在服务层中使用:
public class OrderService{
private readonly IUnitOfWork _unitOfWork;
public OrderService(IUnitOfWork unitOfWork)
{
_unitOfWork = unitOfWork;
}
public void PlaceOrder(Order order)
{
_unitOfWork.Orders.Add(order);
_unitOfWork.Products.DecreaseStock(order.Items);
_unitOfWork.Complete(); // 一次性提交
}
}
依赖注入配置
在
Program.cs或
Startup.cs中注册服务: builder.Services.AddScoped();
builder.Services.AddScoped
builder.Services.AddScoped
这样能保证同一个请求中所有仓储共享同一个
DbContext实例,确保事务一致性。
基本上就这些。工作单元 + 仓储的组合让业务逻辑更清晰,同时保障了数据完整性。.NET 生态中这种模式很常见,尤其在使用 EF Core 的项目里。
