Update pkg/store/bucket.go

Co-authored-by: Saswata Mukherjee <saswataminsta@yahoo.com>
Signed-off-by: Thibault Mange <22740367+thibaultmg@users.noreply.github.com>

pkg/store/bucket.go

@@ -1574,13 +1574,11 @@ func (s *BucketStore) LabelNames(ctx context.Context, req *storepb.LabelNamesReq
 	s.mtx.RUnlock()
 
 	if err := g.Wait(); err != nil {
-		if statusErr, ok := status.FromError(err); ok {
-			if statusErr.Code() == codes.ResourceExhausted {
-				return nil, status.Error(codes.ResourceExhausted, err.Error())
-			}
+		code := codes.Internal
+		if s, ok := status.FromError(errors.Cause(err)); ok {
+			code = s.Code()
 		}
-
-		return nil, status.Error(codes.Internal, err.Error())
+		return nil, status.Error(code, err.Error())
 	}
 
 	anyHints, err := types.MarshalAny(resHints)