Function RunValidationMode

cmd/tempo-vulture/validation.go:19–86

(
	ctx context.Context,
	vultureConfig vultureConfiguration,
	pushEndpoint string,
	logger *zap.Logger,
)

const TempoAccessPolicyToken = "TEMPO_ACCESS_POLICY_TOKEN"

func RunValidationMode(
	ctx context.Context,
	vultureConfig vultureConfiguration,
	pushEndpoint string,
	logger *zap.Logger,
) int {
	accessPolicyToken := os.Getenv(TempoAccessPolicyToken)
	if accessPolicyToken == "" {
		logger.Error("TEMPO_ACCESS_POLICY_TOKEN environment variable is required in validation mode")
		return 1
	}

	// Construct the basic auth token for HTTP headers
	basicAuthToken := constructAuthToken(vultureConfig.tempoOrgID, accessPolicyToken)

	httpClient := httpclient.New(vultureConfig.tempoQueryURL, vultureConfig.tempoOrgID)
	httpClient.SetHeader("Authorization", fmt.Sprintf("Basic %s", basicAuthToken))

	// Create authenticated jaeger client for writing traces
	authJaegerClient, err := utilpkg.NewJaegerToOTLPExporterWithAuth(
		pushEndpoint,
		vultureConfig.tempoOrgID,
		basicAuthToken,
		vultureConfig.tempoPushTLS,
	)
	if err != nil {
		logger.Error("failed to create authenticated OTLP exporter", zap.Error(err))
		os.Exit(1)
	}

	logger.Info("Running in validation mode",
		zap.Int("cycles", validationCycles),
		zap.Duration("timeout", validationTimeout),
		zap.String("org_id", vultureConfig.tempoOrgID),
	)

	validationConfig := ValidationConfig{
		Cycles:                validationCycles,
		TempoOrgID:            vultureConfig.tempoOrgID,
		TempoBasicAuthToken:   basicAuthToken,
		WriteBackoffDuration:  vultureConfig.tempoWriteBackoffDuration,
		SearchBackoffDuration: vultureConfig.tempoSearchBackoffDuration,
	}

	service := NewValidationService(validationConfig, RealClock{}, logger)

	result := service.RunValidation(ctx, authJaegerClient, httpClient, httpClient)

	// Log detailed results before exiting
	logger.Info("Validation completed",
		zap.Int("total_traces", result.TotalTraces),
		zap.Int("validations_passed", result.SuccessCount),
		zap.Int("validations_failed", len(result.Failures)),
		zap.Duration("duration", result.Duration),
	)

	// Optionally log each failure for debugging
	for _, failure := range result.Failures {