Guard againstmultiple wstps

typelevel · Dec 28, 2022 · 6581dc4 · 6581dc4
1 parent 6c4a9d1
commit 6581dc4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/core/jvm/src/main/scala/cats/effect/unsafe/WorkStealingThreadPool.scala b/core/jvm/src/main/scala/cats/effect/unsafe/WorkStealingThreadPool.scala
@@ -97,8 +97,10 @@ private[effect] final class WorkStealingThreadPool(
 
           if (thread.isInstanceOf[WorkerThread]) {
             val worker = thread.asInstanceOf[WorkerThread]
-            if (worker.isOwnedBy(pool)) ioa
-            else IO.cede *> ioa
+            if (worker.isOwnedBy(pool)) // we're good
+              ioa
+            else // possibly a blocking worker thread, possibly on another wstp
+              IO.cede *> ioa.evalOn(pool)
           } else ioa.evalOn(pool)
         }
       }