feat: 增加接口流量配置中间件

router/option.go

@@ -12,6 +12,7 @@ import (
 
 	apiDocDefine "git.zhangdeman.cn/zhangdeman/api-doc/define"
 	"git.zhangdeman.cn/zhangdeman/gin/middleware"
+	"git.zhangdeman.cn/zhangdeman/rate_limit/abstract"
 	"github.com/gin-gonic/gin"
 )
 
@@ -30,6 +31,14 @@ type serverOption struct {
 	disableInitRequest   bool                       // 禁用初始化请求
 	loggerCfg            *middleware.AccessConfig   // 日志配置
 	initContextData      gin.HandlerFunc            // 初始化一些请求数据
+	rateLimitInstance    abstract.IRateLimit        // 服务流控实例
+}
+
+// WithRateLimitInstance 设置流控实例, 配置为 nil, 代表禁用
+func WithRateLimitInstance(i abstract.IRateLimit) SetServerOptionFunc {
+	return func(so *serverOption) {
+		so.rateLimitInstance = i
+	}
 }
 
 // WithDisableInitRequest 禁用自从初始化请求