promote

app/main/api/internal/middleware/global/reqHeaderCtxMiddleware.go

@@ -0,0 +1,34 @@
+package middleware
+
+import (
+	"context"
+	"net/http"
+)
+
+type contextKey string
+
+const (
+	brandKey    contextKey = "brand"
+	platformKey contextKey = "platform"
+	promoteKey  contextKey = "promoteKey"
+)
+
+func ReqHeaderCtxMiddleware(next http.HandlerFunc) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		brand := r.Header.Get("X-Brand")
+		platform := r.Header.Get("X-Platform")
+		promoteValue := r.Header.Get("X-Promote-Key")
+		ctx := r.Context()
+		if brand != "" {
+			ctx = context.WithValue(ctx, brandKey, brand)
+		}
+		if platform != "" {
+			ctx = context.WithValue(ctx, platformKey, platform)
+		}
+		if promoteValue != "" {
+			ctx = context.WithValue(ctx, promoteKey, promoteValue)
+		}
+		r = r.WithContext(ctx)
+		next(w, r)
+	}
+}