add token (#46)

Co-authored-by: wangminglang@kanzhun.com <wangminglang@kanzhun.com>
polarismesh · Aug 10, 2023 · 75c2b2c · 75c2b2c
1 parent 017665b
commit 75c2b2c
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/grpc-java-polaris/src/main/java/com/tencent/polaris/grpc/server/PolarisGrpcServer.java b/grpc-java-polaris/src/main/java/com/tencent/polaris/grpc/server/PolarisGrpcServer.java
@@ -188,6 +188,7 @@ private void registerOne(String serviceName) {
         request.setNamespace(builder.getNamespace());
         request.setService(serviceName);
         request.setHost(host);
+        request.setToken(builder.getToken());
         request.setVersion(builder.getVersion());
         request.setProtocol("grpc");
         request.setWeight(builder.getWeight());

diff --git a/...-java-polaris/src/main/java/com/tencent/polaris/grpc/server/PolarisGrpcServerBuilder.java b/...-java-polaris/src/main/java/com/tencent/polaris/grpc/server/PolarisGrpcServerBuilder.java
@@ -60,6 +60,8 @@ public final class PolarisGrpcServerBuilder extends ServerBuilder<PolarisGrpcSer
 
     private String host;
 
+    private String token;
+
     private DelayRegister delayRegister;
 
     private RegisterHook registerHook;
@@ -168,6 +170,17 @@ public PolarisGrpcServerBuilder host(String host) {
         return this;
     }
 
+    /**
+     * Set the token.
+     *
+     * @param token token
+     * @return PolarisGrpcServerBuilder
+     */
+    public PolarisGrpcServerBuilder token(String token) {
+        this.token = token;
+        return this;
+    }
+
     @Override
     public PolarisGrpcServerBuilder directExecutor() {
         return executor(MoreExecutors.directExecutor());
@@ -305,6 +318,10 @@ String getHost() {
         return host;
     }
 
+    String getToken() {
+        return token;
+    }
+
     SDKContext getContext() {
         return context;
     }