ip rate limiter

config/config.go

@@ -281,8 +281,8 @@ type IPRateLimiter struct {
 }
 
 type RateLimiterStore struct {
-	Type string `yaml:"redis" json:"redis"`
-	Dsn  string `yaml:"dsn" json:"dsn"`
+	Type  string `yaml:"type" json:"type"`
+	Redis Redis  `yaml:"redis" json:"redis"`
 }
 
 func (x IPRateLimiter) Get() (time.Duration, int) {

gin/xgin/xgin.go

@@ -93,7 +93,14 @@ func New() *gin.Engine {
 		}
 		if app.IPRateLimiter.Type == "sync" {
 			if app.IPRateLimiter.Store.Type == "redis" {
-				redisCli, err := goredis.Init(app.IPRateLimiter.Store.Dsn, goredis.WithLogger(logger.Get()))
+				redisConf := app.IPRateLimiter.Store.Redis
+				opts := []goredis.Option{
+					goredis.WithLogger(logger.Get()),
+					goredis.WithDialTimeout(time.Duration(redisConf.DialTimeout) * time.Second),
+					goredis.WithReadTimeout(time.Duration(redisConf.ReadTimeout) * time.Second),
+					goredis.WithWriteTimeout(time.Duration(redisConf.WriteTimeout) * time.Second),
+				}
+				redisCli, err := goredis.Init(redisConf.Dsn, opts...)
 				if err != nil {
 					logger.Fatal("init redis error: " + err.Error())
 				}