first commit

backend/F9WebAiTest.Infrastructure/FreeSqls/Persistence/UnitOfWork.cs

@@ -0,0 +1,46 @@
+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;
+        }
+    }
+}