.NetCore基于SqlSugar的工作单元—UnitOfWork - ToB企服应用市场:ToB评测及商务社交产业平台

/// <summary>
/// 工作单元依赖接口
/// </summary>
public interface IUnitOfWork
{
/// <summary>
/// 开启工作单元处理
/// </summary>
/// <param name="context"></param>
/// <param name="unitOfWork"></param>
void BeginTransaction(ActionExecutingContext context);
/// <summary>
/// 提交工作单元处理
/// </summary>
/// <param name="resultContext"></param>
/// <param name="unitOfWork"></param>
void CommitTransaction(ActionExecutedContext resultContext);
/// <summary>
/// 回滚工作单元处理
/// </summary>
/// <param name="resultContext"></param>
/// <param name="unitOfWork"></param>
void RollbackTransaction(ActionExecutedContext resultContext);
/// <summary>
/// 执行完毕（无论成功失败）
/// </summary>
/// <param name="context"></param>
/// <param name="resultContext"></param>
void OnCompleted(ActionExecutingContext context, ActionExecutedContext resultContext);
}

复制代码

public class SqlSugarUnitOfWork : IUnitOfWork
{
private readonly ISqlSugarClient _sqlSugarClient;
public SqlSugarUnitOfWork(ISqlSugarClient sqlSugarClient)
{
this._sqlSugarClient = sqlSugarClient;
}
public void BeginTransaction(ActionExecutingContext context)
{
_sqlSugarClient.AsTenant().BeginTran();
}
public void CommitTransaction(ActionExecutedContext resultContext)
{
_sqlSugarClient.AsTenant().CommitTran();
}
public void OnCompleted(ActionExecutingContext context, ActionExecutedContext resultContext)
{
_sqlSugarClient.Dispose();
}
public void RollbackTransaction(ActionExecutedContext resultContext)
{
_sqlSugarClient.AsTenant().RollbackTran();
}
}

复制代码

services.AddTransient<IUnitOfWork, SqlSugarUnitOfWork>(); // 注册工作单元到容器

复制代码

/// <summary>
/// 工作单元配置特性
/// </summary>
[AttributeUsage(AttributeTargets.Method)]
public class UnitOfWorkAttribute : Attribute {}

复制代码

/// <summary>
/// 工作单元Action过滤器
/// </summary>
public class UnitOfWorkFilter : IAsyncActionFilter, IOrderedFilter
{
private readonly ILogger<UnitOfWorkFilter> _logger;
public UnitOfWorkFilter(ILogger<UnitOfWorkFilter> logger)
{
this._logger = logger;
}
/// <summary>
/// 过滤器排序
/// </summary>
internal const int FilterOrder = 999;
/// <summary>
/// 排序属性
/// </summary>
public int Order => FilterOrder;
/// <summary>
/// 拦截请求
/// </summary>
/// <param name="context">动作方法上下文</param>
/// <param name="next">中间件委托</param>
public async Task OnActionExecutionAsync(ActionExecutingContext context, ActionExecutionDelegate next)
{
// 获取动作方法描述器
var actionDescriptor = context.ActionDescriptor as ControllerActionDescriptor;
var method = actionDescriptor.MethodInfo;
// 获取请求上下文
var httpContext = context.HttpContext;
// 如果没有定义工作单元过滤器，则跳过
if (!method.IsDefined(typeof(UnitOfWorkAttribute), true))
{
// 调用方法
_ = await next();
return;
}
// 打印工作单元开始消息
_logger.LogInformation($@"{nameof(UnitOfWorkFilter)} Beginning");
// 解析工作单元服务
var _unitOfWork = httpContext.RequestServices.GetRequiredService<IUnitOfWork>();
// 调用开启事务方法
_unitOfWork.BeginTransaction(context);
// 获取执行 Action 结果
var resultContext = await next();
if (resultContext == null || resultContext.Exception == null)
{
// 调用提交事务方法
_unitOfWork.CommitTransaction(resultContext);
}
else
{
// 调用回滚事务方法
_unitOfWork.RollbackTransaction(resultContext);
}
// 调用执行完毕方法
_unitOfWork.OnCompleted(context, resultContext);
// 打印工作单元结束消息
_logger.LogInformation($@"{nameof(UnitOfWorkFilter)} Ending");
}
}

复制代码

services.AddControllers(options =>
{
// 添加工作单元过滤器
options.Filters.Add<UnitOfWorkFilter>();
});

复制代码

[Route("test")]
public class TestController : ControllerBase
{
private readonly IBaseRepository<Student> _studentRepository;
private readonly IBaseRepository<Teacher> _teacherRepository;
public TestController(IBaseRepository<Student> studentRepository,
IBaseRepository<Teacher> teacherRepository)
{
this._studentRepository = studentRepository;
this._teacherRepository = teacherRepository;
}
[HttpGet, UnitOfWork]
public async Task<IActionResult> AddTestAsync()
{
await _studentRepository.InsertAsync(new Student
{
Name = "Hello",
Age = 22
});
await _teacherRepository.InsertAsync(new Teacher
{
Name = "World",
Age = 35,
Subject = 1
});
return Ok("Ok");
}
}

复制代码

https://github.com/young-qiang/UnitOfWork.Demo.git

复制代码