From f4a23b6a684c5d4180545f58918bfc3749957c76 Mon Sep 17 00:00:00 2001 From: gussf <25396922+gussf@users.noreply.github.com> Date: Wed, 5 Jun 2024 11:24:04 -0300 Subject: [PATCH] fix: rate limiter config --- config/default.yaml | 5 +++-- config/docker_test.yaml | 2 +- config/test.yaml | 5 +++-- extensions/rate_limiter.go | 4 +++- 4 files changed, 10 insertions(+), 6 deletions(-) diff --git a/config/default.yaml b/config/default.yaml index 1793173..744cc97 100644 --- a/config/default.yaml +++ b/config/default.yaml @@ -96,5 +96,6 @@ feedbackListeners: rateLimiter: limit: 100 redis: - host: "localhost:6379" - pwd: "" \ No newline at end of file + host: "localhost" + port: "6379" + password: "" \ No newline at end of file diff --git a/config/docker_test.yaml b/config/docker_test.yaml index c14f428..2242c1f 100644 --- a/config/docker_test.yaml +++ b/config/docker_test.yaml @@ -97,4 +97,4 @@ rateLimiter: limit: 100 redis: host: "redis:6379" - pwd: "" + password: "" diff --git a/config/test.yaml b/config/test.yaml index c631e99..50c76bb 100644 --- a/config/test.yaml +++ b/config/test.yaml @@ -94,5 +94,6 @@ feedbackListeners: rateLimiter: limit: 100 redis: - host: "localhost:6379" - pwd: "" \ No newline at end of file + host: "localhost" + port: "6379" + password: "" \ No newline at end of file diff --git a/extensions/rate_limiter.go b/extensions/rate_limiter.go index 0c983c7..e77ad16 100644 --- a/extensions/rate_limiter.go +++ b/extensions/rate_limiter.go @@ -19,10 +19,12 @@ type rateLimiter struct { } func NewRateLimiter(config *viper.Viper, logger *logrus.Logger) rateLimiter { - addr := config.GetString("rateLimiter.redis.host") + host := config.GetString("rateLimiter.redis.host") + port := config.GetString("rateLimiter.redis.port") pwd := config.GetString("rateLimiter.redis.password") limit := config.GetInt("rateLimiter.limit") + addr := fmt.Sprintf("%s:%s", host, port) rdb := redis.NewClient(&redis.Options{ Addr: addr, Password: pwd,