chore(grpc): set metadata if not exist

midwayjs · Nov 12, 2024 · 4354f91 · 4354f91
1 parent 863b31e
commit 4354f91
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/packages/grpc/src/comsumer/clients.ts b/packages/grpc/src/comsumer/clients.ts
@@ -115,31 +115,39 @@ export class GRPCClients extends Map {
     let genericFunctionName;
     switch (genericFunctionSelector) {
       case 0:
-        options.metadata.set('rpc.method.type', 'unary');
+        if (! options.metadata.get('rpc.method.type')) {
+          options.metadata.set('rpc.method.type', 'unary');
+        }
         genericFunctionName = new ClientUnaryRequest(
           client,
           originalFunction,
           options
         );
         break;
       case 1:
-        options.metadata.set('rpc.method.type', 'server'); // server streaming
+        if (! options.metadata.get('rpc.method.type')) {
+          options.metadata.set('rpc.method.type', 'server'); // server streaming
+        }
         genericFunctionName = new ClientReadableRequest(
           client,
           originalFunction,
           options
         );
         break;
       case 2:
-        options.metadata.set('rpc.method.type', 'client'); // client streaming
+        if (! options.metadata.get('rpc.method.type')) {
+          options.metadata.set('rpc.method.type', 'client'); // client streaming
+        }
         genericFunctionName = new ClientWritableRequest(
           client,
           originalFunction,
           options
         );
         break;
       case 3:
-        options.metadata.set('rpc.method.type', 'bidi'); // bidirectional streaming
+        if (! options.metadata.get('rpc.method.type')) {
+          options.metadata.set('rpc.method.type', 'bidi'); // bidirectional streaming
+        }
         genericFunctionName = new ClientDuplexStreamRequest(
           client,
           originalFunction,