golang zset时间窗口限流-Go服务器开发

在Go语言中，可以使用Redis的ZSet数据类型实现时间窗口限流。具体实现思路是：将每个请求的时间戳作为分数存储在ZSet中，并设置一个过期时间来控制窗口大小。当有新请求到达时，先将当前时间戳作为分数加入ZSet中，然后使用ZREMRANGEBYSCORE命令删除所有分数小于当前时间减去窗口大小的元素，最后通过ZCARD命令获取集合大小与限流阈值进行比较即可。

以下是一段示例代码：

package main

import (
	"fmt"
	"time"

	"github.com/go-redis/redis"
)

func main() {
	client := redis.NewClient(&redis.Options{
		Addr:     "localhost:6379",
		Password: "",
		DB:       0,
	})

	defer client.Close()

	limit := 10          // 时间窗口内允许通过的请求数
	windowSize := 60     // 窗口大小，单位秒
	key := "rate_limit"  // Redis键名

	for {
        currentTime := time.Now().Unix()
        
        err := client.ZAdd(key, &redis.Z{Score: float64(currentTime), Member: currentTime}).Err()
        if err != nil {
            panic(err)
        }
        
        _, _ = client.Expire(key, time.Duration(windowSize)*time.Second).Result()
        
        size, err := client.ZCard(key).Result()
        if err != nil {
            panic(err)
        }
        
        if size > int64(limit) {
            fmt.Println("exceed rate limit")
            
            // TODO 返回错误信息
            
            break
        }
        
        time.Sleep(time.Millisecond * 100)
    }
}

在上述代码中，我们使用go-redis库连接Redis，并通过ZAdd命令将请求的时间戳作为分数存储在ZSet中。然后，使用Expire命令设置过期时间为窗口大小，并通过ZCard命令获取集合大小与限流阈值进行比较。需要注意的是，在实际生产环境中，可能需要考虑多个进程同时访问Redis的情况。为了避免竞争条件和重复执行问题，可以使用Redis的WATCH、MULTI和EXEC命令来保证原子性操作。