47 lines
1.3 KiB
C#
47 lines
1.3 KiB
C#
using FreeSql;
|
||
using F9WebAiTest.Application.Interfaces.UOW;
|
||
|
||
namespace F9WebAiTest.Infrastructure.FreeSqls.Persistence;
|
||
|
||
/// <summary>
|
||
/// 工作单元实现：基于 FreeSql 的 UnitOfWorkManager，事务内 Orm 通过 AsyncLocal 暴露给仓储。
|
||
/// 仓储通过 UnitOfWork.CurrentOrm 获取事务感知的 Orm。
|
||
/// </summary>
|
||
public class UnitOfWork_F9 : IUnitOfWork_F9
|
||
{
|
||
private readonly IFreeSql _fsql;
|
||
private readonly UnitOfWorkManager _uowManager;
|
||
private static readonly AsyncLocal<IFreeSql> _currentOrm = new();
|
||
|
||
/// <summary>当前事务内的 Orm（非事务上下文返回 null）</summary>
|
||
public static IFreeSql CurrentOrm => _currentOrm.Value;
|
||
|
||
public UnitOfWork_F9(IFreeSql fsql)
|
||
{
|
||
_fsql = fsql;
|
||
_uowManager = new UnitOfWorkManager(fsql);
|
||
}
|
||
|
||
public async Task<bool> ExecuteWithUow(Func<Task<bool>> func)
|
||
{
|
||
using var uow = _uowManager.Begin();
|
||
_currentOrm.Value = uow.Orm;
|
||
try
|
||
{
|
||
var result = await func();
|
||
if (result) uow.Commit();
|
||
else uow.Rollback();
|
||
return result;
|
||
}
|
||
catch
|
||
{
|
||
uow.Rollback();
|
||
throw;
|
||
}
|
||
finally
|
||
{
|
||
_currentOrm.Value = null;
|
||
}
|
||
}
|
||
}
|