UnitOfWork：一个支持多数据库，工作单元模式、支持分布式事务以及支持 MySQL 多数据库/表分片的开源项目

针对一些大型项目、高并发的场景的项目，我们往往需要多数据库、分表的方式，才能支撑项目的运行。

今天推荐一个开源项目，让我们轻松实现，多数据库、分表的项目开发。

01 项目简介

UnitOfWork，它是一个基于 Microsoft.EntityFrameworkCore 的插件，目标是简化复杂数据库操作，同时支持以下核心功能：

仓储模式（Repository Pattern）

提供统一的接口，方便对数据库操作进行抽象和封装。
单元工作模式（Unit of Work Pattern）

管理多个仓储的事务，确保一组操作要么全部成功，要么全部失败，从而保证数据一致性。
多数据库支持

支持跨多个数据库的操作，适用于复杂的分布式系统。
分布式事务支持

在多数据库环境中，确保事务的提交或回滚是全局一致的。
MySQL 分库分表支持

动态切换数据库或表，支持 MySQL 的分库分表功能，适用于大规模数据的分布式存储。

02 项目特点

动态切换数据库和表：通过 ChangeDatabase 和 ChangeTable 方法，可以灵活地在运行时切换数据库或表。
简化事务管理：通过 UnitOfWork，可以轻松管理事务，确保数据一致性。
支持多种数据库操作：包括插入、查询、更新、分页等常用功能。
扩展性强：支持自定义仓储和扩展方法，满足复杂业务需求。

03 使用方法

1、项目配置

swift 复制代码

// 注册 DbContext
services
    // 添加 QuickStartContext 数据上下文
    .AddDbContext<QuickStartContext>(opt => 
        // 配置使用内存数据库（适用于测试环境）
        opt.UseInMemoryDatabase()
    )
    // 添加 UnitOfWork 支持
    .AddUnitOfWork<QuickStartContext>()
    // 添加自定义仓储
    // 指定 Blog 实体对应的仓储为 CustomBlogRepository
    .AddCustomRepository<Blog, CustomBlogRepository>();

2、单元模式使用、切换数据库

csharp 复制代码

// 定义一个只读的 IUnitOfWork 字段
private readonly IUnitOfWork _unitOfWork;

// 控制器构造函数，注入 IUnitOfWork
public ValuesController(IUnitOfWork unitOfWork)
{
    _unitOfWork = unitOfWork; // 将注入的 IUnitOfWork 赋值给字段

    // 动态切换数据库，目前仅支持 MySQL
    // 根据当前年份动态切换数据库，例如 "uow_db_2025"
    _unitOfWork.ChangeDatabase($"uow_db_{DateTime.Now.Year}");

    // 获取 User 和 Post 的仓储实例
    var userRepo = _unitOfWork.GetRepository<User>(); // 获取 User 的仓储
    var postRepo = _unitOfWork.GetRepository<Post>(); // 获取 Post 的仓储

    // 根据当前年份和月份动态切换表，例如 "user_202504" 和 "post_202504"
    var ym = DateTime.Now.ToString("yyyyMM");
    userRepo.ChangeTable($"user_{ym}");
    postRepo.ChangeTable($"post_{ym}");

    // 创建一个 User 实体并插入数据库
    var user = new User
    {
        UserName = "rigofunc", // 设置用户名
        Password = "password" // 设置密码
    };
    userRepo.Insert(user); // 插入 User 实体

    // 创建一个 Post 实体并插入数据库
    var post = new Post
    {
        UserId = user.UserId, // 设置关联的用户 ID
        Content = "What a piece of junk!" // 设置帖子内容
    };
    postRepo.Insert(post); // 插入 Post 实体

    // 提交事务，保存所有更改到数据库
    _unitOfWork.SaveChanges();

    // 查询 User 实体并更新密码
    var find = userRepo.Find(user.UserId); // 根据用户 ID 查询 User 实体
    find.Password = "p@ssword"; // 更新密码
    _unitOfWork.SaveChanges(); // 提交更改
}

04 项目地址

github.com/arch/UnitOf...

End -