fix: double dispose

atmina · Oct 22, 2024 · a7ec70b · a7ec70b
1 parent 2d85ca2
commit a7ec70b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/Fluss/UnitOfWork/UnitOfWorkFactory.cs b/src/Fluss/UnitOfWork/UnitOfWorkFactory.cs
@@ -23,7 +23,8 @@ public async ValueTask Commit(Func<IWriteUnitOfWork, ValueTask> action)
         await RetryPolicy
             .ExecuteAsync(async () =>
             {
-                using var unitOfWork = serviceProvider.GetRequiredService<UnitOfWork>();
+                using var scope = serviceProvider.CreateScope();
+                var unitOfWork = scope.ServiceProvider.GetRequiredService<UnitOfWork>();
                 await action(unitOfWork);
                 await unitOfWork.CommitInternal();
             });
@@ -36,7 +37,8 @@ public async ValueTask<T> Commit<T>(Func<IWriteUnitOfWork, ValueTask<T>> action)
         return await RetryPolicy
             .ExecuteAsync(async () =>
             {
-                using var unitOfWork = serviceProvider.GetRequiredService<UnitOfWork>();
+                using var scope = serviceProvider.CreateScope();
+                var unitOfWork = scope.ServiceProvider.GetRequiredService<UnitOfWork>();
                 var result = await action(unitOfWork);
                 await unitOfWork.CommitInternal();
                 return result;