Fixed a freeze bug.

tosh-coding · Jul 15, 2024 · 59d4d51 · 59d4d51
1 parent 390874d
commit 59d4d51
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/src/AsyncFiberWorks/Fibers/AsyncFiber.cs b/src/AsyncFiberWorks/Fibers/AsyncFiber.cs
@@ -53,13 +53,20 @@ bool TryDequeue(out Func<Task> result)
         /// </summary>
         async void Run()
         {
-            while (TryDequeue(out var func))
+            while (true)
             {
-                await _executor.Execute(func).ConfigureAwait(false);
-            }
-            lock (_lockObj)
-            {
-                _running = false;
+                while (TryDequeue(out var func))
+                {
+                    await _executor.Execute(func).ConfigureAwait(false);
+                }
+                lock (_lockObj)
+                {
+                    if (_queue.Count <= 0)
+                    {
+                        _running = false;
+                        return;
+                    }
+                }
             }
         }