ycc-server/app/main/api/internal/middleware/global_sourceinterceptor_middleware.go

package middleware

import (
	"context"
	"net/http"
)

const (
	PlatformKey = "X-Platform"
	PromoteKey  = "X-Promote-Key"
)

func GlobalSourceInterceptor(next http.HandlerFunc) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		// 获取请求头 X-Platform 的值
		platform := r.Header.Get(PlatformKey)
		promoteValue := r.Header.Get(PromoteKey)
		// 将值放入新的 context 中
		ctx := r.Context()
		if platform != "" {
			ctx = context.WithValue(ctx, "platform", platform)
		}
		if promoteValue != "" {
			ctx = context.WithValue(ctx, "promoteKey", promoteValue)
		}
		// 通过 r.WithContext 将更新后的 ctx 传递给后续的处理函数
		r = r.WithContext(ctx)

		// 传递给下一个处理器
		next(w, r)
	}
}