限速器

gin/middleware/sync_ratelimiter/sync_ratelimiter.go

 package sync_ratelimiter
 
 import (
+	"context"
 	"fmt"
 	"github.com/RussellLuo/slidingwindow"
-	"github.com/go-redis/redis"
+	"github.com/redis/go-redis/v9"
 	"strconv"
 	"sync"
 	"time"
@@ -20,17 +21,17 @@ func (d *RedisDatastore) fullKey(key string, start int64) string {
 
 func (d *RedisDatastore) Add(key string, start, value int64) (int64, error) {
 	k := d.fullKey(key, start)
-	c, err := d.client.IncrBy(k, value).Result()
+	c, err := d.client.IncrBy(context.Background(), k, value).Result()
 	if err != nil {
 		return 0, err
 	}
-	_, _ = d.client.Expire(k, d.ttl).Result()
+	_, _ = d.client.Expire(context.Background(), k, d.ttl).Result()
 	return c, err
 }
 
 func (d *RedisDatastore) Get(key string, start int64) (int64, error) {
 	k := d.fullKey(key, start)
-	value, err := d.client.Get(k).Result()
+	value, err := d.client.Get(context.Background(), k).Result()
 	if err != nil {
 		if err == redis.Nil {
 			err = nil

gin/middleware/sync_ratelimiter/sync_ratelimiter_test.go

@@ -14,9 +14,10 @@ func TestRateLimiter(t *testing.T) {
 	rl := NewIPRateLimiter(
 		WithWindowSize(1*time.Second),
 		WithMaxRequests(4),
-		WithRedisAddr("localhost:6379"),
+		//WithRedisAddr("localhost:6379"),
+		WithRedisAddr("r-bp1fm63ps25tp3k0vypd.redis.rds.aliyuncs.com:6379"),
 		WithRedisDB(0),
-		WithRedisPassword(""),
+		WithRedisPassword("$3ktkf6z8m"),
 	)
 	ip := "192.168.1.1"
 	for i := 0; i < 100; i++ {