wasilak/go-hello-world

package gorilla

import (
    "context"
    "log/slog"
    "net/http"

    "github.com/arl/statsviz"
    "github.com/gorilla/mux"
    "github.com/prometheus/client_golang/prometheus/promhttp"
    "github.com/wasilak/go-hello-world/utils"
    "go.opentelemetry.io/contrib/instrumentation/github.com/gorilla/mux/otelmux"
    "go.opentelemetry.io/otel/trace"
)

var tracer trace.Tracer

func Init(ctx context.Context, listenAddr *string, otelEnabled, statsvizEnabled *bool, tr trace.Tracer) {
    slog.DebugContext(ctx, "Features supported", "loggergo", true, "statsviz", true, "tracing", true)
    tracer = tr
    router := mux.NewRouter()

    router.Use(prometheusMiddleware)
    router.Path("/metrics").Handler(promhttp.Handler())

    router.HandleFunc("/", chain(rootHandler, logging()))
    router.HandleFunc("/health", chain(healthHandler, logging()))

    if *statsvizEnabled {
        // Create statsviz server and register the handlers on the router.
        srv, _ := statsviz.NewServer()

        slog.DebugContext(ctx, "Statsviz enabled", "address", "/debug/statsviz/")
        router.Methods("GET").Path("/debug/statsviz/ws").Name("GET /debug/statsviz/ws").HandlerFunc(srv.Ws())
        router.Methods("GET").PathPrefix("/debug/statsviz/").Name("GET /debug/statsviz/").Handler(srv.Index())
    }

    if *otelEnabled {
        router.Use(otelmux.Middleware(utils.GetAppName()))
    }

    slog.DebugContext(ctx, "Starting server", "address", *listenAddr)
    http.ListenAndServe(*listenAddr, router)
}