Support store context when using rpc-multiplex. (#5680)

* Support store context when using `rpc-multiplex`.

* Added test cases.

Author: limingxinleo

src/HttpMessageBuilder.php

@@ -17,14 +17,15 @@
 use Hyperf\HttpMessage\Server\Request;
 use Hyperf\HttpMessage\Stream\SwooleStream;
 use Hyperf\HttpMessage\Uri\Uri;
+use Hyperf\Rpc\Context as RpcContext;
 use Hyperf\RpcMultiplex\Contract\HttpMessageBuilderInterface;
 use Psr\Http\Message\ResponseInterface;
 use Psr\Http\Message\ServerRequestInterface;
 use Psr\Http\Message\UriInterface;
 
 class HttpMessageBuilder implements HttpMessageBuilderInterface
 {
-    public function __construct(protected PackerInterface $packer)
+    public function __construct(protected PackerInterface $packer, protected RpcContext $context)
     {
     }
 
@@ -38,6 +39,8 @@ public function buildRequest(array $data): ServerRequestInterface
 
         $parsedData = $data[Constant::DATA] ?? [];
 
+        $this->context->setData($data[Constant::CONTEXT] ?? []);
+
         $request = new Request('POST', $uri, ['Content-Type' => 'application/json'], new SwooleStream(Json::encode($parsedData)));
 
         return $request->withParsedBody($parsedData);

tests/Cases/HttpMessageBuilderTest.php

@@ -11,6 +11,7 @@
  */
 namespace HyperfTest\RpcMultiplex\Cases;
 
+use Hyperf\Rpc\Context;
 use Hyperf\RpcMultiplex\HttpMessageBuilder;
 use Hyperf\RpcMultiplex\Packer\JsonPacker;
 use Hyperf\Support\Reflection\ClassInvoker;
@@ -24,12 +25,34 @@ class HttpMessageBuilderTest extends AbstractTestCase
 {
     public function testBuildUri()
     {
-        $invoker = new ClassInvoker(new HttpMessageBuilder(new JsonPacker()));
+        $invoker = new ClassInvoker(new HttpMessageBuilder(new JsonPacker(), new Context()));
         /** @var UriInterface $uri */
         $uri = $invoker->buildUri('/', $host = uniqid(), 8806);
         $this->assertSame('http', $uri->getScheme());
         $this->assertSame('/', $uri->getPath());
         $this->assertSame($host, $uri->getHost());
         $this->assertSame(8806, $uri->getPort());
     }
+
+    public function testStoreContext()
+    {
+        $builder = new HttpMessageBuilder(new JsonPacker(), $context = new Context());
+
+        $request = $builder->buildRequest([
+            'path' => '/',
+            'data' => ['id' => 1],
+        ]);
+
+        $this->assertSame(['id' => 1], $request->getParsedBody());
+        $this->assertSame([], $context->getData());
+
+        $request = $builder->buildRequest([
+            'path' => '/',
+            'data' => ['id' => 1],
+            'context' => ['trace_id' => $id = uniqid()],
+        ]);
+
+        $this->assertSame(['id' => 1], $request->getParsedBody());
+        $this->assertSame(['trace_id' => $id], $context->getData());
+    }
 }