Function main

_examples/interceptor/cmd/custom_observability/main.go:55–112

()

)

func main() {
	// Initialize OpenTelemetry providers
	shutdown := initOTel()
	defer shutdown()

	// Start a fake Elasticsearch server
	srv := fake.NewServer(
		fake.WithStatusCode(http.StatusOK),
		fake.WithResponseBody([]byte(`{"cluster_name":"example","version":{"number":"9.2.0"}}`)),
		fake.WithHeaders(func(h http.Header) {
			h.Set("X-Elastic-Product", "Elasticsearch")
			h.Set("Content-Type", "application/json")
		}),
	)
	defer srv.Close()

	// Create metrics instruments
	meter := otel.Meter("elasticsearch-client")
	requestCounter, _ := meter.Int64Counter("elasticsearch.client.requests",
		metric.WithDescription("Number of requests to Elasticsearch"),
		metric.WithUnit("{request}"),
	)
	requestDuration, _ := meter.Float64Histogram("elasticsearch.client.duration",
		metric.WithDescription("Duration of Elasticsearch requests"),
		metric.WithUnit("ms"),
	)

	// Create tracer
	tracer := otel.Tracer("elasticsearch-client")

	// Create an Elasticsearch client with observability interceptors
	es, err := elasticsearch.New(
		elasticsearch.WithAddresses(srv.URL()),
		elasticsearch.WithTransportOptions(elastictransport.WithInterceptors(
			LoggingInterceptor(),
			MetricsInterceptor(requestCounter, requestDuration),
			TracingInterceptor(tracer),
		)),
	)
	if err != nil {
		panic(err)
	}

	// Send some requests to demonstrate the observability
	fmt.Println(">>> Sending requests to demonstrate observability interceptors")
	fmt.Println()

	for i := 1; i <= 3; i++ {
		fmt.Printf("--- Request %d ---\n", i)
		_, _ = es.Info()
		fmt.Println()
		time.Sleep(100 * time.Millisecond) // Small delay between requests
	}

	// Give time for metrics to flush
	fmt.Println(">>> Waiting for metrics to flush...")
	time.Sleep(2 * time.Second)
}