Skip to content

Commit

Permalink
Remove tenant header logs in store gateway (thanos-io#6552)
Browse files Browse the repository at this point in the history
* remove tenant header logs which are too verbose

Signed-off-by: Ben Ye <benye@amazon.com>

* keep the debug level log

Signed-off-by: Ben Ye <benye@amazon.com>

---------

Signed-off-by: Ben Ye <benye@amazon.com>
  • Loading branch information
yeya24 authored and GiedriusS committed Jul 27, 2023
1 parent 61d6188 commit f5af402
Show file tree
Hide file tree
Showing 4 changed files with 16 additions and 26 deletions.
15 changes: 3 additions & 12 deletions pkg/store/bucket.go
Original file line number Diff line number Diff line change
Expand Up @@ -1231,10 +1231,7 @@ func (s *BucketStore) Series(req *storepb.SeriesRequest, srv storepb.Store_Serie
defer s.queryGate.Done()
}

tenant, err := tenancy.GetTenantFromGRPCMetadata(srv.Context())
if err != nil {
level.Warn(s.logger).Log("msg", err)
}
tenant, _ := tenancy.GetTenantFromGRPCMetadata(srv.Context())
level.Debug(s.logger).Log("msg", "Tenant for Series request", "tenant", tenant)

matchers, err := storepb.MatchersToPromMatchers(req.Matchers...)
Expand Down Expand Up @@ -1486,10 +1483,7 @@ func (s *BucketStore) LabelNames(ctx context.Context, req *storepb.LabelNamesReq
return nil, status.Error(codes.InvalidArgument, errors.Wrap(err, "translate request labels matchers").Error())
}

tenant, err := tenancy.GetTenantFromGRPCMetadata(ctx)
if err != nil {
level.Warn(s.logger).Log("msg", err)
}
tenant, _ := tenancy.GetTenantFromGRPCMetadata(ctx)
level.Debug(s.logger).Log("msg", "Tenant for LabelNames request", "tenant", tenant)

resHints := &hintspb.LabelNamesResponseHints{}
Expand Down Expand Up @@ -1680,10 +1674,7 @@ func (s *BucketStore) LabelValues(ctx context.Context, req *storepb.LabelValuesR
return nil, status.Error(codes.InvalidArgument, errors.Wrap(err, "translate request labels matchers").Error())
}

tenant, err := tenancy.GetTenantFromGRPCMetadata(ctx)
if err != nil {
level.Warn(s.logger).Log("msg", err)
}
tenant, _ := tenancy.GetTenantFromGRPCMetadata(ctx)
level.Debug(s.logger).Log("msg", "Tenant for LabelValues request", "tenant", tenant)

resHints := &hintspb.LabelValuesResponseHints{}
Expand Down
13 changes: 6 additions & 7 deletions pkg/store/proxy.go
Original file line number Diff line number Diff line change
Expand Up @@ -307,9 +307,8 @@ func (s *ProxyStore) Series(originalRequest *storepb.SeriesRequest, srv storepb.
// We may arrive here either via the promql engine
// or as a result of a grpc call in layered queries
ctx := srv.Context()
tenant, err := tenancy.GetTenantFromGRPCMetadata(ctx)
if err != nil {
level.Debug(s.logger).Log("msg", "using tenant from context instead of metadata")
tenant, foundTenant := tenancy.GetTenantFromGRPCMetadata(ctx)
if !foundTenant {
if ctx.Value(tenancy.TenantKey) != nil {
tenant = ctx.Value(tenancy.TenantKey).(string)
}
Expand Down Expand Up @@ -467,8 +466,8 @@ func (s *ProxyStore) LabelNames(ctx context.Context, r *storepb.LabelNamesReques

// We may arrive here either via the promql engine
// or as a result of a grpc call in layered queries
tenant, err := tenancy.GetTenantFromGRPCMetadata(gctx)
if err != nil {
tenant, foundTenant := tenancy.GetTenantFromGRPCMetadata(gctx)
if !foundTenant {
level.Debug(s.logger).Log("msg", "using tenant from context instead of metadata")
if gctx.Value(tenancy.TenantKey) != nil {
tenant = gctx.Value(tenancy.TenantKey).(string)
Expand Down Expand Up @@ -546,8 +545,8 @@ func (s *ProxyStore) LabelValues(ctx context.Context, r *storepb.LabelValuesRequ

// We may arrive here either via the promql engine
// or as a result of a grpc call in layered queries
tenant, err := tenancy.GetTenantFromGRPCMetadata(gctx)
if err != nil {
tenant, foundTenant := tenancy.GetTenantFromGRPCMetadata(gctx)
if !foundTenant {
level.Debug(s.logger).Log("msg", "using tenant from context instead of metadata")
if gctx.Value(tenancy.TenantKey) != nil {
tenant = gctx.Value(tenancy.TenantKey).(string)
Expand Down
6 changes: 3 additions & 3 deletions pkg/tenancy/tenancy.go
Original file line number Diff line number Diff line change
Expand Up @@ -102,10 +102,10 @@ func getTenantFromCertificate(r *http.Request, certTenantField string) (string,
return tenant, nil
}

func GetTenantFromGRPCMetadata(ctx context.Context) (string, error) {
func GetTenantFromGRPCMetadata(ctx context.Context) (string, bool) {
md, ok := metadata.FromIncomingContext(ctx)
if !ok || len(md.Get(DefaultTenantHeader)) == 0 {
return DefaultTenant, errors.Errorf("could not get tenant from grpc metadata, using default: %s", DefaultTenantHeader)
return DefaultTenant, false
}
return md.Get(DefaultTenantHeader)[0], nil
return md.Get(DefaultTenantHeader)[0], true
}
8 changes: 4 additions & 4 deletions pkg/tenancy/tenancy_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -92,16 +92,16 @@ func TestTenantFromGRPC(t *testing.T) {
md := metadata.New(map[string]string{tenancy.DefaultTenantHeader: testTenant})
ctx = metadata.NewIncomingContext(ctx, md)

tenant, err := tenancy.GetTenantFromGRPCMetadata(ctx)
testutil.Ok(t, err)
tenant, foundTenant := tenancy.GetTenantFromGRPCMetadata(ctx)
testutil.Equals(t, true, foundTenant)
testutil.Assert(t, tenant == testTenant)
})
t.Run("no-tenant", func(t *testing.T) {

ctx := context.Background()

tenant, err := tenancy.GetTenantFromGRPCMetadata(ctx)
testutil.NotOk(t, err)
tenant, foundTenant := tenancy.GetTenantFromGRPCMetadata(ctx)
testutil.Equals(t, false, foundTenant)
testutil.Assert(t, tenant == tenancy.DefaultTenant)
})
}
Expand Down

0 comments on commit f5af402

Please sign in to comment.