seal PollingContext

typelevel · Oct 27, 2024 · c125e73 · c125e73
1 parent 26ef827
commit c125e73
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/core/jvm-native/src/main/scala/cats/effect/unsafe/PollingSystem.scala b/core/jvm-native/src/main/scala/cats/effect/unsafe/PollingSystem.scala
@@ -104,7 +104,7 @@ abstract class PollingSystem {
 
 }
 
-trait PollingContext[P] {
+sealed trait PollingContext[P] {
 
   /**
    * Register a callback to obtain a thread-local `Poller`
@@ -119,6 +119,8 @@ trait PollingContext[P] {
   def ownPoller(poller: P): Boolean
 }
 
+private[unsafe] trait UnsealedPollingContext[P] extends PollingContext[P]
+
 object PollingSystem {
 
   /**

diff --git a/core/jvm/src/main/scala/cats/effect/unsafe/WorkStealingThreadPool.scala b/core/jvm/src/main/scala/cats/effect/unsafe/WorkStealingThreadPool.scala
@@ -72,7 +72,7 @@ private[effect] final class WorkStealingThreadPool[P <: AnyRef](
     reportFailure0: Throwable => Unit
 ) extends ExecutionContextExecutor
     with Scheduler
-    with PollingContext[P] {
+    with UnsealedPollingContext[P] {
 
   import TracingConstants._
   import WorkStealingThreadPoolConstants._

diff --git a/core/native/src/main/scala/cats/effect/unsafe/IORuntimeCompanionPlatform.scala b/core/native/src/main/scala/cats/effect/unsafe/IORuntimeCompanionPlatform.scala
@@ -31,7 +31,7 @@ private[unsafe] abstract class IORuntimeCompanionPlatform { this: IORuntime.type
     val loop = new EventLoopExecutorScheduler[system.Poller](64, system)
     val poller = loop.poller
     val api = system.makeApi(
-      new PollingContext[system.Poller] {
+      new UnsealedPollingContext[system.Poller] {
         def accessPoller(cb: system.Poller => Unit) = cb(poller)
         def ownPoller(poller: system.Poller) = true
       }