Function checkClusterFromIncomingContext

middleware/grpc_cluster.go:101–163

(
	ctx context.Context, method string, expectedClusters []string, softValidationEnabled bool,
	invalidClusterRequests *prometheus.CounterVec, logger log.Logger,
)

}

func checkClusterFromIncomingContext(
	ctx context.Context, method string, expectedClusters []string, softValidationEnabled bool,
	invalidClusterRequests *prometheus.CounterVec, logger log.Logger,
) error {
	reqCluster, err := clusterutil.GetClusterFromIncomingContext(ctx)
	if err == nil && clusterutil.IsClusterAllowed(reqCluster, expectedClusters) {
		return nil
	}

	logger = log.With(
		logger,
		"method", method,
		"cluster_validation_labels", fmt.Sprintf("%v", expectedClusters),
		"soft_validation", softValidationEnabled,
	)
	if tenantID, err := user.ExtractOrgID(ctx); err == nil {
		logger = log.With(logger, "tenant", tenantID)
	}
	if p, ok := peer.FromContext(ctx); ok {
		logger = log.With(logger, "client_address", p.Addr.String())
	}
	if traceID, ok := tracing.ExtractSampledTraceID(ctx); ok {
		logger = log.With(logger, "trace_id", traceID)
	}

	if err == nil {
		// No error, but request's and server's cluster validation labels didn't match.
		var wrongClusterErr error
		if !softValidationEnabled {
			wrongClusterErr = fmt.Errorf("rejected request with wrong cluster validation label %q - it should be one of %v", reqCluster, expectedClusters)
		}

		// Use first expected cluster for metrics compatibility
		expectedClusterForMetrics := expectedClusters[0]
		invalidClusterRequests.WithLabelValues("grpc", method, expectedClusterForMetrics, reqCluster).Inc()
		level.Warn(logger).Log("msg", "request with wrong cluster validation label", "request_cluster_validation_label", reqCluster)
		return wrongClusterErr
	}

	if errors.Is(err, clusterutil.ErrNoClusterValidationLabel) {
		var emptyClusterErr error
		if !softValidationEnabled {
			emptyClusterErr = fmt.Errorf("rejected request with empty cluster validation label - it should be one of %v", expectedClusters)
		}

		// Use first expected cluster for metrics compatibility
		expectedClusterForMetrics := expectedClusters[0]
		invalidClusterRequests.WithLabelValues("grpc", method, expectedClusterForMetrics, "").Inc()
		level.Warn(logger).Log("msg", "request with no cluster validation label")
		return emptyClusterErr
	}

	var rejectedRequestErr error
	if !softValidationEnabled {
		rejectedRequestErr = fmt.Errorf("rejected request: %w", err)
	}

	// Use first expected cluster for metrics compatibility