Reference same coroutine id in same request

Author: m3m0r7

src/Coroutine/Context.php

@@ -7,6 +7,8 @@
 
 class Context
 {
+    protected const MAX_RECURSE_CONTEXT_ID = 50;
+
     /**
      * The app containers in different coroutine environment.
      *
@@ -90,10 +92,20 @@ public static function clear()
     }
 
     /**
-     * Get current coroutine id.
+     * Get current requesting coroutine id.
      */
-    public static function getCoroutineId()
+    public static function getCoroutineId(): int
     {
-        return Coroutine::getuid();
+        $currentId = Coroutine::getuid();
+        if ($currentId === -1) {
+            return -1;
+        }
+
+        $counter = 0;
+        while (($topCoroutineId = Coroutine::getPcid($currentId)) !== -1 && $counter <= self::MAX_RECURSE_CONTEXT_ID) {
+            $currentId = $topCoroutineId;
+            $counter++;
+        }
+        return $currentId;
     }
 }

tests/Coroutine/ContextTest.php

@@ -52,6 +52,31 @@ public function testGetDataKeys()
         $this->assertSame(['foo', 'sea'], Context::getDataKeys());
     }
 
+    public function testGetDataKeyInCoroutine()
+    {
+        $data1 = null;
+        $data2 = null;
+        $coroutineId1 = null;
+        $coroutineId2 = null;
+        \Swoole\Coroutine\run(function () use (&$data1, &$data2, &$coroutineId1, &$coroutineId2) {
+            Context::setData('foo', 'bar');
+
+            $data1 = Context::getData('foo');
+            $data2 = 'baz';
+            $coroutineId1 = Context::getCoroutineId();
+            $coroutineId2 = -1;
+
+            go(function () use (&$data2, &$coroutineId2) {
+                $data2 = Context::getData('foo');
+                $coroutineId2 = Context::getCoroutineId();
+            });
+        });
+
+        $this->assertSame('bar', $data1);
+        $this->assertSame($data1, $data2);
+        $this->assertSame($coroutineId1, $coroutineId2);
+    }
+
     public function testClear()
     {
         Context::setApp(m::mock(Container::class));