42 lines
906 B
Go
42 lines
906 B
Go
package middleware
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
)
|
|
|
|
const (
|
|
BrandKey = "X-Brand"
|
|
PlatformKey = "X-Platform"
|
|
)
|
|
|
|
type SourceInterceptorMiddleware struct {
|
|
}
|
|
|
|
func NewSourceInterceptorMiddleware() *SourceInterceptorMiddleware {
|
|
return &SourceInterceptorMiddleware{}
|
|
}
|
|
|
|
func (m *SourceInterceptorMiddleware) Handle(next http.HandlerFunc) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
// 获取请求头 X-Brand 和 X-Platform 的值
|
|
brand := r.Header.Get(BrandKey)
|
|
platform := r.Header.Get(PlatformKey)
|
|
|
|
// 将值放入新的 context 中
|
|
ctx := r.Context()
|
|
if brand != "" {
|
|
ctx = context.WithValue(ctx, "brand", brand)
|
|
}
|
|
if platform != "" {
|
|
ctx = context.WithValue(ctx, "platform", platform)
|
|
}
|
|
|
|
// 通过 r.WithContext 将更新后的 ctx 传递给后续的处理函数
|
|
r = r.WithContext(ctx)
|
|
|
|
// 传递给下一个处理器
|
|
next(w, r)
|
|
}
|
|
}
|