(rocketmq) branch develop updated: Get namespace from header (#7782)

[zhouxzhan]

This is an automated email from the ASF dual-hosted git repository.

zhouxzhan pushed a commit to branch develop
in repository https://gitbox.apache.org/repos/asf/rocketmq.git


The following commit(s) were added to refs/heads/develop by this push:
     new 04eec34978 Get namespace from header (#7782)
04eec34978 is described below

commit 04eec3497895a1fa8a9c8fbe520406c62a05a5b0
Author: Zhouxiang Zhan <zhouxiang....@alibaba-inc.com>
AuthorDate: Wed Jan 24 20:18:59 2024 +0800

    Get namespace from header (#7782)
---
 .../java/org/apache/rocketmq/proxy/common/ContextVariable.java   | 2 +-
 .../main/java/org/apache/rocketmq/proxy/common/ProxyContext.java | 9 +++++++++
 .../apache/rocketmq/proxy/grpc/v2/GrpcMessagingApplication.java  | 3 ++-
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git 
a/proxy/src/main/java/org/apache/rocketmq/proxy/common/ContextVariable.java 
b/proxy/src/main/java/org/apache/rocketmq/proxy/common/ContextVariable.java
index 0760826de7..93b4eacd8a 100644
--- a/proxy/src/main/java/org/apache/rocketmq/proxy/common/ContextVariable.java
+++ b/proxy/src/main/java/org/apache/rocketmq/proxy/common/ContextVariable.java
@@ -27,5 +27,5 @@ public class ContextVariable {
     public static final String REMAINING_MS = "remaining-ms";
     public static final String ACTION = "action";
     public static final String PROTOCOL_TYPE = "protocol-type";
-
+    public static final String NAMESPACE = "namespace";
 }
diff --git 
a/proxy/src/main/java/org/apache/rocketmq/proxy/common/ProxyContext.java 
b/proxy/src/main/java/org/apache/rocketmq/proxy/common/ProxyContext.java
index 77a6791f04..e6fc989fcc 100644
--- a/proxy/src/main/java/org/apache/rocketmq/proxy/common/ProxyContext.java
+++ b/proxy/src/main/java/org/apache/rocketmq/proxy/common/ProxyContext.java
@@ -131,4 +131,13 @@ public class ProxyContext {
         return this.getVal(ContextVariable.PROTOCOL_TYPE);
     }
 
+    public ProxyContext setNamespace(String namespace) {
+        this.withVal(ContextVariable.NAMESPACE, namespace);
+        return this;
+    }
+
+    public String getNamespace() {
+        return this.getVal(ContextVariable.NAMESPACE);
+    }
+
 }
diff --git 
a/proxy/src/main/java/org/apache/rocketmq/proxy/grpc/v2/GrpcMessagingApplication.java
 
b/proxy/src/main/java/org/apache/rocketmq/proxy/grpc/v2/GrpcMessagingApplication.java
index 2cb395ad60..a344b0590c 100644
--- 
a/proxy/src/main/java/org/apache/rocketmq/proxy/grpc/v2/GrpcMessagingApplication.java
+++ 
b/proxy/src/main/java/org/apache/rocketmq/proxy/grpc/v2/GrpcMessagingApplication.java
@@ -175,7 +175,8 @@ public class GrpcMessagingApplication extends 
MessagingServiceGrpc.MessagingServ
             .setProtocolType(ChannelProtocolType.GRPC_V2.getName())
             .setLanguage(getDefaultStringMetadataInfo(headers, 
InterceptorConstants.LANGUAGE))
             .setClientVersion(getDefaultStringMetadataInfo(headers, 
InterceptorConstants.CLIENT_VERSION))
-            .setAction(getDefaultStringMetadataInfo(headers, 
InterceptorConstants.SIMPLE_RPC_NAME));
+            .setAction(getDefaultStringMetadataInfo(headers, 
InterceptorConstants.SIMPLE_RPC_NAME))
+            .setNamespace(getDefaultStringMetadataInfo(headers, 
InterceptorConstants.NAMESPACE_ID));
         if (ctx.getDeadline() != null) {
             
context.setRemainingMs(ctx.getDeadline().timeRemaining(TimeUnit.MILLISECONDS));
         }