fix: 工作单元 支持 取消令牌

Author: jianxuanbing

framework/src/Bing.Datas.EntityFramework/Core/UnitOfWorkBase.cs

@@ -365,11 +365,12 @@ public int Commit()
     /// <summary>
     /// 异步提交，返回影响的行数
     /// </summary>
-    public async Task<int> CommitAsync()
+    /// <param name="cancellationToken">取消令牌</param>
+    public async Task<int> CommitAsync(CancellationToken cancellationToken = default)
     {
         try
         {
-            return await SaveChangesAsync();
+            return await SaveChangesAsync(cancellationToken);
         }
         catch (DbUpdateConcurrencyException ex)
         {

framework/src/Bing.FreeSQL/Bing/Uow/UnitOfWorkBase.cs

@@ -190,11 +190,12 @@ public int Commit()
     /// <summary>
     /// 提交，返回影响的行数
     /// </summary>
-    public async Task<int> CommitAsync()
+    /// <param name="cancellationToken">取消令牌</param>
+    public async Task<int> CommitAsync(CancellationToken cancellationToken = default)
     {
         try
         {
-            return await SaveChangesAsync();
+            return await SaveChangesAsync(cancellationToken);
         }
         catch (Exception ex)
         {

framework/src/Bing.Uow/Bing/Uow/IUnitOfWork.cs

@@ -16,5 +16,6 @@ public interface IUnitOfWork : IDisposable
     /// <summary>
     /// 提交，返回影响的行数
     /// </summary>
-    Task<int> CommitAsync();
+    /// <param name="cancellationToken">取消令牌</param>
+    Task<int> CommitAsync(CancellationToken cancellationToken = default);
 }

framework/src/Bing.Uow/Bing/Uow/IUnitOfWorkManager.cs

@@ -15,7 +15,8 @@ public interface IUnitOfWorkManager : IScopedDependency
     /// <summary>
     /// 提交
     /// </summary>
-    Task CommitAsync();
+    /// <param name="cancellationToken">取消令牌</param>
+    Task CommitAsync(CancellationToken cancellationToken = default);
 
     /// <summary>
     /// 注册工作单元

framework/src/Bing.Uow/Bing/Uow/UnitOfWorkManager.cs

@@ -27,10 +27,11 @@ public void Commit()
     /// <summary>
     /// 提交
     /// </summary>
-    public async Task CommitAsync()
+    /// <param name="cancellationToken">取消令牌</param>
+    public async Task CommitAsync(CancellationToken cancellationToken = default)
     {
         foreach (var unitOfWork in _unitOfWorks)
-            await unitOfWork.CommitAsync();
+            await unitOfWork.CommitAsync(cancellationToken);
     }
 
     /// <summary>

tests/Bing.Tests.Samples/Bing/Tests/Samples/CrudServiceSample.cs

@@ -1,4 +1,5 @@
-using System.Threading.Tasks;
+using System.Threading;
+using System.Threading.Tasks;
 using Bing.Application.Services;
 using Bing.Data.Queries;
 using Bing.Extensions;
@@ -34,7 +35,8 @@ public void Dispose()
     /// <summary>
     /// 提交，返回影响的行数
     /// </summary>
-    public Task<int> CommitAsync() => Task.FromResult(1);
+    /// <param name="cancellationToken">取消令牌</param>
+    public Task<int> CommitAsync(CancellationToken cancellationToken = default) => Task.FromResult(1);
 }
 
 /// <summary>
@@ -60,4 +62,4 @@ public CrudServiceSample(IUnitOfWork unitOfWork, IRepositorySample repository) :
     /// </summary>
     /// <param name="parameter">查询参数</param>
     protected override IQueryBase<EntitySample> CreateQuery(QueryParameterSample parameter) => new Query<EntitySample>(parameter);
-}
+}