支持等待完成

Author: lindexi

AsyncWorkerCollection/DoubleBuffer_/DoubleBufferTask.cs

@@ -3,12 +3,17 @@
 using System.Collections.Generic;
 using System.Threading.Tasks;
 
+#if NETFRAMEWORK
+using ValueTask = System.Threading.Tasks.Task;
+#endif
+
 namespace dotnetCampus.Threading
 {
     /// <summary>
     /// 双缓存任务
     /// </summary>
-    public class DoubleBufferTask<T, TU> where T : class, ICollection<TU>
+    public class DoubleBufferTask<T, TU> : IAsyncDisposable
+        where T : class, ICollection<TU>
     {
         /// <summary>
         /// 创建双缓存任务，执行任务的方法放在 <paramref name="doTask"/> 方法
@@ -92,5 +97,12 @@ public Task WaitAllTaskFinish()
         private readonly Func<T, Task> _doTask;
 
         private DoubleBuffer<T, TU> DoubleBuffer { get; }
+
+        /// <inheritdoc />
+        public async ValueTask DisposeAsync()
+        {
+            Finish();
+            await WaitAllTaskFinish();
+        }
     }
 }

AsyncWorkerCollection/IAsyncDisposable.cs

@@ -4,10 +4,15 @@
 using System.Text;
 using System.Threading.Tasks;
 
+#if NETFRAMEWORK
+using ValueTask = System.Threading.Tasks.Task;
+#endif
+
 namespace dotnetCampus.Threading
 {
     // 这个接口在 .NET Framework 4.5 没有
     interface IAsyncDisposable
     {
+        ValueTask DisposeAsync();
     }
 }