From bb5ee88d4faf904edc1dcdbb57a290b81966897c Mon Sep 17 00:00:00 2001 From: Claudiu Dragalina-Paraipan Date: Tue, 3 Sep 2024 10:39:11 +0300 Subject: [PATCH] added stackIdExtractor for cloud mode Co-Authored-By: Gabriel MABILLE --- pkg/storage/unified/resource/client_wrapper.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/pkg/storage/unified/resource/client_wrapper.go b/pkg/storage/unified/resource/client_wrapper.go index 6bdf13768a1..788b384900d 100644 --- a/pkg/storage/unified/resource/client_wrapper.go +++ b/pkg/storage/unified/resource/client_wrapper.go @@ -86,7 +86,7 @@ func NewResourceStoreClientCloud(conn *grpc.ClientConn, cfg *setting.Cfg) (Resou opts := []authnlib.GrpcClientInterceptorOption{ authnlib.WithIDTokenExtractorOption(idTokenExtractor), - authnlib.WithMetadataExtractorOption(orgIdExtractor), + authnlib.WithMetadataExtractorOption(stackIdExtractor(cfg.StackID)), } if cfg.Env == setting.Dev { @@ -137,3 +137,9 @@ func orgIdExtractor(ctx context.Context) (key string, values []string, err error return authzlib.DefaultStackIDMetadataKey, []string{fmt.Sprintf("%d", requester.GetOrgID())}, nil } + +func stackIdExtractor(stackID string) func(ctx context.Context) (key string, values []string, err error) { + return func(ctx context.Context) (key string, values []string, err error) { + return authzlib.DefaultStackIDMetadataKey, []string{stackID}, nil + } +}