feat: 保留兼容逻辑

polarismesh · Mar 28, 2024 · 48db22e · 48db22e
1 parent 798c0df
commit 48db22e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/apiserver/grpcserver/discover/v1/client_access.go b/apiserver/grpcserver/discover/v1/client_access.go
@@ -144,6 +144,11 @@ func (g *DiscoverServer) Discover(server apiservice.PolarisGRPC_DiscoverServer)
 			})
 		}()
 
+		// 兼容。如果请求中带了token，优先使用该token
+		if in.GetService().GetToken().GetValue() != "" {
+			ctx = context.WithValue(ctx, utils.ContextAuthTokenKey, in.GetService().GetToken().GetValue())
+		}
+
 		switch in.Type {
 		case apiservice.DiscoverRequest_INSTANCE:
 			action = metrics.ActionDiscoverInstance