milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

enhance: [2.5] remain token after authorize (#39455) 

relate: https://github.com/milvus-io/milvus/issues/39285

Signed-off-by: aoiasd <zhicheng.yue@zilliz.com>
pull/39439/merge
aoiasd 2025-01-20 20:27:04 +08:00 committed by GitHub
parent f62eca09a0
commit 15d60c60c2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
internal/proxy/authentication_interceptor.go
View File

@ -106,6 +106,7 @@ func AuthenticationInterceptor(ctx context.Context) (context.Context, error) {
metrics.UserRPCCounter.WithLabelValues(user).Inc() metrics.UserRPCCounter.WithLabelValues(user).Inc()
userToken := fmt.Sprintf("%s%s%s", user, util.CredentialSeperator, util.PasswordHolder) userToken := fmt.Sprintf("%s%s%s", user, util.CredentialSeperator, util.PasswordHolder)
md[strings.ToLower(util.HeaderAuthorize)] = []string{crypto.Base64Encode(userToken)} md[strings.ToLower(util.HeaderAuthorize)] = []string{crypto.Base64Encode(userToken)}
md[util.HeaderToken] = []string{rawToken}
ctx = metadata.NewIncomingContext(ctx, md) ctx = metadata.NewIncomingContext(ctx, md)
} else { } else {
// username+password authentication // username+password authentication

1
pkg/util/constant.go
View File

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