82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
package middleware
|
||
|
||
import (
|
||
"context"
|
||
"net/http"
|
||
|
||
"git.toowon.com/jimmy/go-common/tools"
|
||
)
|
||
|
||
// TimezoneKey context中存储时区的key
|
||
type timezoneKey struct{}
|
||
|
||
// TimezoneHeaderName 时区请求头名称
|
||
const TimezoneHeaderName = "X-Timezone"
|
||
|
||
// DefaultTimezone 默认时区
|
||
const DefaultTimezone = tools.AsiaShanghai
|
||
|
||
// GetTimezoneFromContext 从context中获取时区
|
||
func GetTimezoneFromContext(ctx context.Context) string {
|
||
if tz, ok := ctx.Value(timezoneKey{}).(string); ok && tz != "" {
|
||
return tz
|
||
}
|
||
return DefaultTimezone
|
||
}
|
||
|
||
// Timezone 时区处理中间件
|
||
// 从请求头 X-Timezone 读取时区信息，如果未传递则使用默认时区 AsiaShanghai
|
||
// 时区信息会存储到context中，可以通过 GetTimezoneFromContext 获取
|
||
func Timezone(next http.Handler) http.Handler {
|
||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
// 从请求头获取时区
|
||
timezone := r.Header.Get(TimezoneHeaderName)
|
||
|
||
// 如果未传递时区信息，使用默认时区
|
||
if timezone == "" {
|
||
timezone = DefaultTimezone
|
||
}
|
||
|
||
// 验证时区是否有效
|
||
if _, err := tools.GetLocation(timezone); err != nil {
|
||
// 如果时区无效，使用默认时区
|
||
timezone = DefaultTimezone
|
||
}
|
||
|
||
// 将时区存储到context中
|
||
ctx := context.WithValue(r.Context(), timezoneKey{}, timezone)
|
||
next.ServeHTTP(w, r.WithContext(ctx))
|
||
})
|
||
}
|
||
|
||
// TimezoneWithDefault 时区处理中间件（可自定义默认时区）
|
||
// defaultTimezone: 默认时区，如果未指定则使用 AsiaShanghai
|
||
func TimezoneWithDefault(defaultTimezone string) func(http.Handler) http.Handler {
|
||
// 验证默认时区是否有效
|
||
if _, err := tools.GetLocation(defaultTimezone); err != nil {
|
||
defaultTimezone = DefaultTimezone
|
||
}
|
||
|
||
return func(next http.Handler) http.Handler {
|
||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
// 从请求头获取时区
|
||
timezone := r.Header.Get(TimezoneHeaderName)
|
||
|
||
// 如果未传递时区信息，使用指定的默认时区
|
||
if timezone == "" {
|
||
timezone = defaultTimezone
|
||
}
|
||
|
||
// 验证时区是否有效
|
||
if _, err := tools.GetLocation(timezone); err != nil {
|
||
// 如果时区无效，使用默认时区
|
||
timezone = defaultTimezone
|
||
}
|
||
|
||
// 将时区存储到context中
|
||
ctx := context.WithValue(r.Context(), timezoneKey{}, timezone)
|
||
next.ServeHTTP(w, r.WithContext(ctx))
|
||
})
|
||
}
|
||
}
|