enhance: remain token after authorize (#39453)

Signed-off-by: aoiasd <[email protected]>
milvus-io · Jan 20, 2025 · a41de38 · a41de38
1 parent 140c5a0
commit a41de38
Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/internal/proxy/authentication_interceptor.go b/internal/proxy/authentication_interceptor.go
@@ -106,6 +106,7 @@ func AuthenticationInterceptor(ctx context.Context) (context.Context, error) {
 				metrics.UserRPCCounter.WithLabelValues(user).Inc()
 				userToken := fmt.Sprintf("%s%s%s", user, util.CredentialSeperator, util.PasswordHolder)
 				md[strings.ToLower(util.HeaderAuthorize)] = []string{crypto.Base64Encode(userToken)}
+				md[util.HeaderToken] = []string{rawToken}
 				ctx = metadata.NewIncomingContext(ctx, md)
 			} else {
 				// username+password authentication

diff --git a/pkg/util/constant.go b/pkg/util/constant.go
@@ -46,6 +46,7 @@ const (
 	FieldIndexPrefix   = "field-index"
 
 	HeaderAuthorize = "authorization"
+	HeaderToken     = "token"
 	// HeaderSourceID identify requests from Milvus members and client requests
 	HeaderSourceID = "sourceId"
 	// MemberCredID id for Milvus members (data/index/query node/coord component)