support runtime.ServeMux opts for grpc-gateway

pkg/ambient/ambient.go

@@ -318,6 +318,18 @@ func (aw *AmbientWeather) enrichStation(update *weather.WeatherUpdate) {
 	}
 }
 
+func (aw *AmbientWeather) GetLogger() *zerolog.Logger {
+	aw.RLock()
+	defer aw.RUnlock()
+	return aw.l
+}
+
+func (aw *AmbientWeather) GetCtx() context.Context {
+	aw.RLock()
+	defer aw.RUnlock()
+	return aw.appCtx
+}
+
 func (aw *AmbientWeather) GetRecorder() *recorder.WeatherRecorder {
 	aw.RLock()
 	defer aw.RUnlock()

pkg/ambient/app.go

@@ -0,0 +1,33 @@
+package ambient
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
+)
+
+func (aw *AmbientWeather) GetGatewayOpts() []runtime.ServeMuxOption {
+	return []runtime.ServeMuxOption{
+		runtime.WithMiddlewares(func(hf runtime.HandlerFunc) runtime.HandlerFunc {
+			return func(w http.ResponseWriter, r *http.Request, pathParams map[string]string) {
+				aw.GetLogger().Trace().
+					Any("pathParams", pathParams).
+					Msg("inbound request called")
+			}
+		}),
+		runtime.WithRoutingErrorHandler(
+			func(
+				ctx context.Context, mux *runtime.ServeMux, mshl runtime.Marshaler,
+				w http.ResponseWriter, r *http.Request, code int,
+			) {
+				aw.GetLogger().Trace().
+					Int("code", code).
+					Str("pattern", r.Pattern).
+					Msg("requesting grpc-gateway route")
+
+				// Pass back to default
+				runtime.DefaultRoutingErrorHandler(ctx, mux, mshl, w, r, code)
+			}),
+	}
+}

pkg/weather/recorder/recorder.go

@@ -29,6 +29,10 @@ type Opts struct {
 	KeepLast  int
 }
 
+func (r *WeatherRecorder) Ping(ctx context.Context) error {
+	return r.recorder.Ping(ctx)
+}
+
 func MustNewWeatherRecorder(opts *Opts) *WeatherRecorder {
 	if opts.KeepLast < 1 {
 		opts.KeepLast = 1

pkg/weather/recorder/recorders/memory/memory.go

@@ -25,6 +25,11 @@ type MemoryRecorder struct {
 	*sync.RWMutex
 }
 
+// No meaningful health check to do here
+func (r *MemoryRecorder) Ping(ctx context.Context) error {
+	return nil
+}
+
 func (r *MemoryRecorder) Init(ctx context.Context, opts *recorders.RecorderOpts) {
 	if opts.RetainLast < 1 {
 		opts.RetainLast = DEF_RETAIN_LAST

pkg/weather/recorder/recorders/noop/noop.go

@@ -20,4 +20,6 @@ func (n *NoopRecorder) Get(context.Context, *pb.GetWeatherRequest) ([]*weather.W
 
 func (n *NoopRecorder) Count(context.Context) int { return 0 }
 
+func (n *NoopRecorder) Ping(context.Context) error { return nil }
+
 func (r *NoopRecorder) Name() string { return "no-op recorder" }

pkg/weather/recorder/recorders/recorders.go

@@ -19,5 +19,6 @@ type Recorder interface {
 	Set(context.Context, *weather.WeatherUpdate) error
 	Get(context.Context, *pb.GetWeatherRequest) ([]*weather.WeatherUpdate, error)
 	Count(context.Context) int // Best Effort
+	Ping(context.Context) error
 	Name() string
 }

pkg/weather/recorder/recorders/redis/redis.go

@@ -36,6 +36,11 @@ type RedisRecorder struct {
 	*sync.RWMutex
 }
 
+func (r *RedisRecorder) Ping(ctx context.Context) error {
+	_, err := r.redis.Ping(ctx).Result()
+	return err
+}
+
 func (r *RedisRecorder) Init(ctx context.Context, opts *recorders.RecorderOpts) {
 	if opts.AppConfig.RecorderConfig.RedisConfig == nil {
 		panic("refusing to init redis recorder with no redisConfig")