Go app framework

pkg/srv/http.go

@@ -0,0 +1,131 @@
+package srv
+
+import (
+	"context"
+	"net"
+	"net/http"
+	"time"
+
+	"github.com/prometheus/client_golang/prometheus/promhttp"
+	"github.com/rs/zerolog"
+	"go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp"
+	"go.opentelemetry.io/otel/metric"
+	"go.opentelemetry.io/otel/trace"
+
+	"gitea.libretechconsulting.com/rmcguire/go-app/pkg/config"
+	"gitea.libretechconsulting.com/rmcguire/go-app/pkg/otel"
+)
+
+var (
+	httpMeter    metric.Meter
+	httpTracer   trace.Tracer
+	readTimeout  = 10 * time.Second
+	writeTimeout = 10 * time.Second
+	idleTimeout  = 15 * time.Second
+)
+
+type HTTPFunc struct {
+	Path        string
+	HandlerFunc http.HandlerFunc
+}
+
+func prepHTTPServer(ctx context.Context, handleFuncs []HTTPFunc, hcFuncs ...HealthCheckFunc) *http.Server {
+	var (
+		cfg = config.MustFromCtx(ctx)
+		l   = zerolog.Ctx(ctx)
+		mux = &http.ServeMux{}
+	)
+
+	// NOTE: Wraps handle func with otelhttp handler and
+	// inserts route tag
+	otelHandleFunc := func(pattern string, handlerFunc func(http.ResponseWriter, *http.Request)) {
+		handler := otelhttp.WithRouteTag(pattern, http.HandlerFunc(handlerFunc))
+		mux.Handle(pattern, handler) // Associate pattern with handler
+	}
+
+	healthChecks := handleHealthCheckFunc(ctx, hcFuncs...)
+	otelHandleFunc("/health", healthChecks)
+	otelHandleFunc("/", healthChecks)
+
+	for _, f := range handleFuncs {
+		otelHandleFunc(f.Path, f.HandlerFunc)
+	}
+
+	// Prometheus metrics endpoint
+	if cfg.OTEL.PrometheusEnabled {
+		mux.Handle(cfg.OTEL.PrometheusPath, promhttp.Handler())
+		l.Info().Str("prometheusPath", cfg.OTEL.PrometheusPath).
+			Msg("mounted prometheus metrics endpoint")
+	}
+
+	// Add OTEL, skip health-check spans
+	// NOTE: Add any other span exclusions here
+	handler := otelhttp.NewHandler(mux, "/",
+		otelhttp.WithFilter(func(r *http.Request) bool {
+			switch r.URL.Path {
+			case "/health":
+				return false
+			case cfg.OTEL.PrometheusPath:
+				return false
+			default:
+				return true
+			}
+		}))
+
+	return &http.Server{
+		Addr:         cfg.HTTP.Listen,
+		ReadTimeout:  readTimeout,
+		WriteTimeout: writeTimeout,
+		IdleTimeout:  idleTimeout,
+		Handler:      handler,
+		BaseContext: func(_ net.Listener) context.Context {
+			return ctx
+		},
+	}
+}
+
+// Returns a shutdown func and a done channel if the
+// server aborts abnormally. Panics on error.
+func MustInitHTTPServer(ctx context.Context, funcs []HTTPFunc, hcFuncs ...HealthCheckFunc) (
+	func(context.Context) error, <-chan interface{},
+) {
+	shutdownFunc, doneChan, err := InitHTTPServer(ctx, funcs, hcFuncs...)
+	if err != nil {
+		panic(err)
+	}
+	return shutdownFunc, doneChan
+}
+
+// Returns a shutdown func and a done channel if the
+// server aborts abnormally. Returns error on failure to start
+func InitHTTPServer(ctx context.Context, funcs []HTTPFunc, hcFuncs ...HealthCheckFunc) (
+	func(context.Context) error, <-chan interface{}, error,
+) {
+	l := zerolog.Ctx(ctx)
+	doneChan := make(chan interface{})
+
+	var server *http.Server
+
+	httpMeter = otel.GetMeter(ctx, "http")
+	httpTracer = otel.GetTracer(ctx, "http")
+
+	server = prepHTTPServer(ctx, funcs, hcFuncs...)
+
+	go func() {
+		l.Debug().Msg("HTTP Server Started")
+		err := server.ListenAndServe()
+		if err != nil && err != http.ErrServerClosed {
+			l.Err(err).Msg("HTTP server error")
+		} else {
+			l.Info().Msg("HTTP server shut down")
+		}
+		doneChan <- nil
+	}()
+
+	// Shut down http server with a deadline
+	return func(shutdownCtx context.Context) error {
+		l.Debug().Msg("stopping http server")
+		server.Shutdown(shutdownCtx)
+		return nil
+	}, doneChan, nil
+}

pkg/srv/http_health.go

@@ -0,0 +1,67 @@
+package srv
+
+import (
+	"context"
+	"errors"
+	"math/rand"
+	"net/http"
+	"sync"
+	"time"
+
+	"github.com/rs/zerolog"
+)
+
+type HealthCheckFunc func(context.Context) error
+
+func handleHealthCheckFunc(ctx context.Context, hcFuncs ...HealthCheckFunc) func(w http.ResponseWriter, r *http.Request) {
+	// Return http handle func
+	return func(w http.ResponseWriter, r *http.Request) {
+		var (
+			healthChecksFailed bool
+			errs               error
+			hcWg               sync.WaitGroup
+		)
+
+		if len(hcFuncs) < 1 {
+			zerolog.Ctx(ctx).Warn().Msg("no health checks given responding with dummy 200")
+			hcFuncs = append(hcFuncs, dummyHealthCheck)
+		}
+
+		// Run all health check funcs concurrently
+		// log all errors
+		hcWg.Add(len(hcFuncs))
+		for _, hc := range hcFuncs {
+			go func() {
+				defer hcWg.Done()
+				errs = errors.Join(errs, hc(ctx))
+			}()
+		}
+		hcWg.Wait()
+
+		if errs != nil {
+			healthChecksFailed = true
+		}
+
+		if healthChecksFailed {
+			w.WriteHeader(http.StatusInternalServerError)
+		}
+
+		if errs != nil {
+			w.Write([]byte(errs.Error()))
+		} else {
+			w.Write([]byte("ok"))
+		}
+	}
+}
+
+func dummyHealthCheck(ctx context.Context) error {
+	workFor := rand.Intn(750)
+	ticker := time.NewTicker(time.Duration(time.Duration(workFor) * time.Millisecond))
+
+	select {
+	case <-ticker.C:
+		return nil
+	case <-ctx.Done():
+		return ctx.Err()
+	}
+}