Merge pull request #10 from dotnet-campus/t/lindexi/LulergarwhawhurboLebinawkairyeehu

支持释放队列

Author: lindexi

AsyncWorkerCollection/AsyncQueue.cs

@@ -1,4 +1,5 @@
-using System.Collections.Concurrent;
+using System;
+using System.Collections.Concurrent;
 using System.Collections.Generic;
 using System.Threading;
 using System.Threading.Tasks;
@@ -9,7 +10,7 @@ namespace dotnetCampus.Threading
     /// 提供一个异步的队列。可以使用 await 关键字异步等待出队，当有元素入队的时候，等待就会完成。
     /// </summary>
     /// <typeparam name="T">存入异步队列中的元素类型。</typeparam>
-    public class AsyncQueue<T>
+    public class AsyncQueue<T> : IDisposable
     {
         private readonly SemaphoreSlim _semaphoreSlim;
         private readonly ConcurrentQueue<T> _queue;
@@ -64,7 +65,7 @@ public void EnqueueRange(IEnumerable<T> source)
         /// <returns>可以异步等待的队列返回的元素。</returns>
         public async Task<T> DequeueAsync(CancellationToken cancellationToken = default)
         {
-            while (true)
+            while (!_isDisposed)
             {
                 await _semaphoreSlim.WaitAsync(cancellationToken).ConfigureAwait(false);
 
@@ -73,6 +74,22 @@ public async Task<T> DequeueAsync(CancellationToken cancellationToken = default)
                     return item;
                 }
             }
+
+            return default;
         }
+
+        /// <summary>
+        /// 主要用来释放锁，让 DequeueAsync 方法返回，解决因为锁让此对象内存不释放
+        /// </summary>
+        public void Dispose()
+        {
+            // 当释放的时候，将通过 _queue 的 Clear 清空内容，而通过 _semaphoreSlim 的释放让 DequeueAsync 释放锁
+            // 此时将会在 DequeueAsync 进入 TryDequeue 方法，也许此时依然有开发者在 _queue.Clear() 之后插入元素，但是没关系，我只是需要保证调用 Dispose 之后会让 DequeueAsync 方法返回而已
+            _isDisposed = true;
+            _queue.Clear();
+            _semaphoreSlim.Dispose();
+        }
+
+        private bool _isDisposed;
     }
 }

AsyncWorkerCollection/AsyncTaskQueue_/AsyncTaskQueue.cs

@@ -6,7 +6,7 @@
 namespace dotnetCampus.Threading
 {
     /// <summary>
-    /// 异步任务队列
+    /// 异步任务队列，这是重量级的方案，将会开启一个线程来做
     /// </summary>
     public class AsyncTaskQueue : IDisposable
     {