update

ZjzMisaka · Nov 11, 2023 · 0bff54c · 0bff54c
1 parent 3dd7508
commit 0bff54c
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/PowerThreadPool/PowerThreadPool.cs b/PowerThreadPool/PowerThreadPool.cs
@@ -767,20 +767,20 @@ internal void SetWork(WorkBase work)
         /// <returns></returns>
         private Worker GetWorker()
         {
-
-                Worker worker = null;
-                while (idleWorkerQueue.TryDequeue(out string firstWorkerID))
+
+            Worker worker = null;
+            while (idleWorkerQueue.TryDequeue(out string firstWorkerID))
+            {
+                if (idleWorkerDic.TryRemove(firstWorkerID, out worker))
                 {
-                    if (idleWorkerDic.TryRemove(firstWorkerID, out worker))
+                    if (Interlocked.Increment(ref worker.gettedLock) == -99)
                     {
-                        if (Interlocked.Increment(ref worker.gettedLock) == -99)
-                        {
-                            Interlocked.Exchange(ref worker.gettedLock, -100);
-                            continue;
-                        }
-                        return worker;
+                        Interlocked.Exchange(ref worker.gettedLock, -100);
+                        continue;
                     }
+                    return worker;
                 }
+            }
 
             lock (this)
             {
@@ -817,9 +817,9 @@ private Worker GetWorker()
                         }
                     }
                 }
-
-                return worker;
             }
+
+            return worker;
         }
 
         /// <summary>