Updating instrumented example

Author: lightsta1ker

examples/dice/instrumented/main.go

@@ -7,12 +7,14 @@ package main
 
 import (
 	"context"
+	"encoding/json"
 	"errors"
 	"log"
 	"net"
 	"net/http"
 	"os"
 	"os/signal"
+	"strconv"
 	"time"
 
 	"go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp"
@@ -40,8 +42,12 @@ func run() error {
 	}()
 
 	// Start HTTP server.
+	port := os.Getenv("APPLICATION_PORT")
+	if port == "" {
+		port = "8080"
+	}
 	srv := &http.Server{
-		Addr:         ":8080",
+		Addr:         ":" + port,
 		BaseContext:  func(net.Listener) context.Context { return ctx },
 		ReadTimeout:  time.Second,
 		WriteTimeout: 10 * time.Second,
@@ -80,10 +86,55 @@ func newHTTPHandler() http.Handler {
 	}
 
 	// Register handlers.
-	handleFunc("/rolldice/", rolldice)
-	handleFunc("/rolldice/{player}", rolldice)
+	handleFunc("/rolldice", handleRollDice)
 
 	// Add HTTP instrumentation for the whole server.
 	handler := otelhttp.NewHandler(mux, "/")
 	return handler
 }
+
+func handleRollDice(w http.ResponseWriter, r *http.Request) {
+	// Parse query parameters
+	rollsParam := r.URL.Query().Get("rolls")
+	player := r.URL.Query().Get("player")
+
+	// Default rolls = 1 if not defined
+	if rollsParam == "" {
+		rollsParam = "1"
+	}
+
+	// Check if rolls is a number
+	rolls, err := strconv.Atoi(rollsParam)
+	if err != nil {
+		w.WriteHeader(http.StatusBadRequest)
+		msg := "Parameter rolls must be a positive integer"
+		_ = json.NewEncoder(w).Encode(map[string]string{
+			"status":  "error",
+			"message": msg,
+		})
+		log.Printf("WARN: %s", msg)
+		return
+	}
+
+	results, err := rollDice(r.Context(), rolls)
+	if err != nil {
+		// Signals invalid input (<=0)
+		w.WriteHeader(http.StatusInternalServerError)
+		log.Printf("ERROR: %v", err)
+		return
+	}
+
+	if player == "" {
+		log.Printf("DEBUG: anonymous player rolled %v", results)
+	} else {
+		log.Printf("DEBUG: player=%s rolled %v", player, results)
+	}
+	log.Printf("INFO: %s %s -> 200 OK", r.Method, r.URL.String())
+
+	w.Header().Set("Content-Type", "application/json")
+	if len(results) == 1 {
+		json.NewEncoder(w).Encode(results[0])
+	} else {
+		json.NewEncoder(w).Encode(results)
+	}
+}

examples/dice/instrumented/otel.go

@@ -16,6 +16,7 @@ import (
 	"go.opentelemetry.io/otel/propagation"
 	"go.opentelemetry.io/otel/sdk/log"
 	"go.opentelemetry.io/otel/sdk/metric"
+	"go.opentelemetry.io/otel/sdk/resource"
 	"go.opentelemetry.io/otel/sdk/trace"
 )
 
@@ -42,12 +43,19 @@ func setupOTelSDK(ctx context.Context) (func(context.Context) error, error) {
 		err = errors.Join(inErr, shutdown(ctx))
 	}
 
+	res, _ := resource.New(ctx,
+		resource.WithFromEnv(), // reads OTEL_SERVICE_NAME & OTEL_RESOURCE_ATTRIBUTES
+		resource.WithProcess(),
+		resource.WithHost(),
+		resource.WithTelemetrySDK(),
+	)
+
 	// Set up propagator.
 	prop := newPropagator()
 	otel.SetTextMapPropagator(prop)
 
 	// Set up trace provider.
-	tracerProvider, err := newtracerProvider()
+	tracerProvider, err := newtracerProvider(res)
 	if err != nil {
 		handleErr(err)
 		return shutdown, err
@@ -56,7 +64,7 @@ func setupOTelSDK(ctx context.Context) (func(context.Context) error, error) {
 	otel.SetTracerProvider(tracerProvider)
 
 	// Set up meter provider.
-	meterProvider, err := newMeterProvider()
+	meterProvider, err := newMeterProvider(res)
 	if err != nil {
 		handleErr(err)
 		return shutdown, err
@@ -65,7 +73,7 @@ func setupOTelSDK(ctx context.Context) (func(context.Context) error, error) {
 	otel.SetMeterProvider(meterProvider)
 
 	// Set up logger provider.
-	loggerProvider, err := newLoggerProvider()
+	loggerProvider, err := newLoggerProvider(res)
 	if err != nil {
 		handleErr(err)
 		return shutdown, err
@@ -83,43 +91,41 @@ func newPropagator() propagation.TextMapPropagator {
 	)
 }
 
-func newtracerProvider() (*trace.TracerProvider, error) {
-	traceExporter, err := stdouttrace.New(
-		stdouttrace.WithPrettyPrint())
+func newtracerProvider(res *resource.Resource) (*trace.TracerProvider, error) {
+	traceExporter, err := stdouttrace.New(stdouttrace.WithPrettyPrint())
 	if err != nil {
 		return nil, err
 	}
 
 	tracerProvider := trace.NewTracerProvider(
-		trace.WithBatcher(traceExporter,
-			// Default is 5s. Set to 1s for demonstrative purposes.
-			trace.WithBatchTimeout(time.Second)),
+		trace.WithBatcher(traceExporter, trace.WithBatchTimeout(time.Second)), // Default is 5s. Set to 1s for demonstrative purposes.
+		trace.WithResource(res),
 	)
 	return tracerProvider, nil
 }
 
-func newMeterProvider() (*metric.MeterProvider, error) {
+func newMeterProvider(res *resource.Resource) (*metric.MeterProvider, error) {
 	metricExporter, err := stdoutmetric.New()
 	if err != nil {
 		return nil, err
 	}
 
 	meterProvider := metric.NewMeterProvider(
-		metric.WithReader(metric.NewPeriodicReader(metricExporter,
-			// Default is 1m. Set to 3s for demonstrative purposes.
-			metric.WithInterval(3*time.Second))),
+		metric.WithReader(metric.NewPeriodicReader(metricExporter, metric.WithInterval(3*time.Second))), // Default is 1m. Set to 3s for demonstrative purposes.
+		metric.WithResource(res),
 	)
 	return meterProvider, nil
 }
 
-func newLoggerProvider() (*log.LoggerProvider, error) {
+func newLoggerProvider(res *resource.Resource) (*log.LoggerProvider, error) {
 	logExporter, err := stdoutlog.New()
 	if err != nil {
 		return nil, err
 	}
 
 	loggerProvider := log.NewLoggerProvider(
 		log.WithProcessor(log.NewBatchProcessor(logExporter)),
+		log.WithResource(res),
 	)
 	return loggerProvider, nil
 }

examples/dice/instrumented/rolldice.go

@@ -4,10 +4,9 @@
 package main
 
 import (
-	"io"
+	"context"
+	"errors"
 	"math/rand"
-	"net/http"
-	"strconv"
 
 	"go.opentelemetry.io/otel"
 	"go.opentelemetry.io/otel/attribute"
@@ -19,42 +18,95 @@ import (
 const name = "go.opentelemetry.io/contrib/examples/dice"
 
 var (
-	tracer  = otel.Tracer(name)
-	meter   = otel.Meter(name)
-	logger  = otelslog.NewLogger(name)
-	rollCnt metric.Int64Counter
+	tracer         = otel.Tracer(name)
+	meter          = otel.Meter(name)
+	logger         = otelslog.NewLogger(name)
+	rollCnt        metric.Int64Counter
+	outcomeHist    metric.Int64Histogram
+	lastRollsGauge metric.Int64ObservableGauge
+	lastRolls      int64
 )
 
 func init() {
 	var err error
 	rollCnt, err = meter.Int64Counter("dice.rolls",
-		metric.WithDescription("The number of rolls by roll value"),
+		metric.WithDescription("The number of rolls"),
 		metric.WithUnit("{roll}"))
 	if err != nil {
 		panic(err)
 	}
+
+	outcomeHist, err = meter.Int64Histogram(
+		"dice.outcome",
+		metric.WithDescription("Distribution of dice outcomes (1-6)"),
+		metric.WithUnit("{count}"),
+	)
+	if err != nil {
+		panic(err)
+	}
+
+	lastRollsGauge, err = meter.Int64ObservableGauge(
+		"dice.last.rolls",
+		metric.WithDescription("The last rolls value observed"),
+	)
+	if err != nil {
+		panic(err)
+	}
+
+	// Register the gauge callback
+	_, err = meter.RegisterCallback(
+		func(ctx context.Context, o metric.Observer) error {
+			o.ObserveInt64(lastRollsGauge, lastRolls)
+			return nil
+		},
+		lastRollsGauge,
+	)
+	if err != nil {
+		panic(err)
+	}
 }
 
-func rolldice(w http.ResponseWriter, r *http.Request) {
-	ctx, span := tracer.Start(r.Context(), "roll")
+// rollDice is the outer function which Does the error handling.
+func rollDice(ctx context.Context, rolls int) ([]int, error) {
+	ctx, span := tracer.Start(ctx, "rollDice")
 	defer span.End()
 
-	roll := 1 + rand.Intn(6) //nolint:gosec // G404: Use of weak random number generator (math/rand instead of crypto/rand) is ignored as this is not security-sensitive.
+	if rolls <= 0 {
+		err := errors.New("rolls must be positive")
+		span.RecordError(err)
+		logger.ErrorContext(ctx, "error", "error", err)
+		return nil, err
+	}
+
+	if rolls == 1 {
+		val := rollOnce(ctx)
+		outcomeHist.Record(ctx, int64(val))
+		lastRolls = int64(rolls)
+		return []int{val}, nil
+	}
 
-	var msg string
-	if player := r.PathValue("player"); player != "" {
-		msg = player + " is rolling the dice"
-	} else {
-		msg = "Anonymous player is rolling the dice"
+	results := make([]int, rolls)
+	for i := 0; i < rolls; i++ {
+		results[i] = rollOnce(ctx)
+		outcomeHist.Record(ctx, int64(results[i]))
 	}
-	logger.InfoContext(ctx, msg, "result", roll)
+
+	rollsAttr := attribute.Int("rolls", rolls)
+	span.SetAttributes(rollsAttr)
+	rollCnt.Add(ctx, 1, metric.WithAttributes(rollsAttr))
+	lastRolls = int64(rolls)
+	return results, nil
+}
+
+// rollOnce is the inner function — returns a random number 1–6.
+func rollOnce(ctx context.Context) int {
+	ctx, span := tracer.Start(ctx, "rollOnce")
+	defer span.End()
+
+	roll := 1 + rand.Intn(6) //nolint:gosec // G404: Use of weak random number generator (math/rand instead of crypto/rand) is ignored as this is not security-sensitive.
 
 	rollValueAttr := attribute.Int("roll.value", roll)
 	span.SetAttributes(rollValueAttr)
-	rollCnt.Add(ctx, 1, metric.WithAttributes(rollValueAttr))
 
-	resp := strconv.Itoa(roll) + "\n"
-	if _, err := io.WriteString(w, resp); err != nil {
-		logger.ErrorContext(ctx, "Write failed", "error", err)
-	}
+	return roll
 }

examples/dice/uninstrumented/main.go

@@ -96,7 +96,7 @@ func handleRollDice(w http.ResponseWriter, r *http.Request) {
 
 	results, err := rolldice(rolls)
 	if err != nil {
-		// Library signals invalid input (<=0)
+		// Signals invalid input (<=0)
 		w.WriteHeader(http.StatusInternalServerError)
 		log.Printf("ERROR: %v", err)
 		return

examples/dice/uninstrumented/rolldice.go

@@ -8,6 +8,7 @@ import (
 	"math/rand"
 )
 
+// rolldice is the outer function which does the error handling.
 func rolldice(rolls int) ([]int, error) {
 	if rolls <= 0 {
 		return nil, errors.New("rolls must be positive")