Allow passing fiber factory to tracing fiber factory

revoltphp · May 1, 2024 · f897d2f · f897d2f
1 parent 7cc40ca
commit f897d2f
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/EventLoop/TracingFiberFactory.php b/src/EventLoop/TracingFiberFactory.php
@@ -18,10 +18,11 @@ final class TracingFiberFactory implements FiberFactory, Countable, IteratorAggr
     /**
      * @var \WeakMap<\Fiber, null>
      */
-    private \WeakMap $map;
+    private readonly \WeakMap $map;
 
-    public function __construct()
-    {
+    public function __construct(
+        private readonly FiberFactory $fiberFactory = new DefaultFiberFactory()
+    ) {
         /** @var \WeakMap<\Fiber, null> */
         $this->map = new \WeakMap();
     }
@@ -35,7 +36,7 @@ public function __construct()
      */
     public function create(callable $callback): \Fiber
     {
-        $f = new \Fiber($callback);
+        $f = $this->fiberFactory->create($callback);
         $this->map[$f] = null;
         return $f;
     }