初始版本,工具基础类

middleware/timezone.go

@@ -0,0 +1,82 @@
+package middleware
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/go-common/datetime"
+)
+
+// TimezoneKey context中存储时区的key
+type timezoneKey struct{}
+
+// TimezoneHeaderName 时区请求头名称
+const TimezoneHeaderName = "X-Timezone"
+
+// DefaultTimezone 默认时区
+const DefaultTimezone = datetime.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 := datetime.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 := datetime.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 := datetime.GetLocation(timezone); err != nil {
+				// 如果时区无效，使用默认时区
+				timezone = defaultTimezone
+			}
+
+			// 将时区存储到context中
+			ctx := context.WithValue(r.Context(), timezoneKey{}, timezone)
+			next.ServeHTTP(w, r.WithContext(ctx))
+		})
+	}
+}
+