初始化v1

api/middleware/rate_limit.go

@@ -0,0 +1,60 @@
+package middleware
+
+import (
+	"fmt"
+	"nl-video-api/utility/response"
+	"time"
+
+	"github.com/gogf/gf/v2/database/gredis"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/net/ghttp"
+)
+
+// RateLimit 限流中间件
+func RateLimit(maxRequests int, window time.Duration) func(r *ghttp.Request) {
+	return func(r *ghttp.Request) {
+		var (
+			ctx    = r.Context()
+			client = g.Redis()
+			key    = fmt.Sprintf("rate_limit:%s", r.GetClientIp())
+		)
+
+		// 获取当前请求次数
+		count, err := client.Get(ctx, key)
+		if err != nil {
+			g.Log().Error(ctx, "Redis获取失败:", err)
+			r.Middleware.Next()
+			return
+		}
+
+		// 检查是否超过限制
+		if count.Int() >= maxRequests {
+			response.Error(r, response.CodeError, "请求过于频繁，请稍后再试")
+			return
+		}
+
+		// 增加计数
+		if count.Int() == 0 {
+			// 第一次请求，设置过期时间
+			seconds := int64(window.Seconds())
+			client.Set(ctx, key, 1, gredis.SetOption{
+				TTLOption: gredis.TTLOption{EX: &seconds},
+			})
+		} else {
+			// 增加计数
+			client.Incr(ctx, key)
+		}
+
+		r.Middleware.Next()
+	}
+}
+
+// APIRateLimit API接口限流（每分钟60次）
+func APIRateLimit(r *ghttp.Request) {
+	RateLimit(60, time.Minute)(r)
+}
+
+// LoginRateLimit 登录接口限流（每分钟5次）
+func LoginRateLimit(r *ghttp.Request) {
+	RateLimit(5, time.Minute)(r)
+}