golang利用redis实现限速器

最新推荐文章于 2024-08-20 20:04:06 发布

liyunlong41

最新推荐文章于 2024-08-20 20:04:06 发布

阅读量2.4k

点赞数 1

分类专栏： redis Golang学习文章标签： golang redis 限速器限流

本文链接：https://blog.csdn.net/liyunlong41/article/details/89854808

版权

本文介绍了如何使用Golang结合Redis的String数据结构和过期功能创建一个简单的限速器。通过用户ID作为key，访问次数作为value进行存储和 incr 操作。然而，该方法存在不精确和非原子性问题。对于不精确性，由于Redis的限速器在某些情况下可能导致超过限制的访问次数。为了解决非原子性问题，可以使用Lua脚本确保incr和expire操作的原子性，或者通过检查key的ttl并在必要时补偿过期时间。

摘要由CSDN通过智能技术生成

利用redis的String数据结构可以实现计数器的功能，利用incr或者incrby来实现计数器的增加或者减少。

同时redis对可以对key实现过期功能，这样我们就可以利用计数器和过期功能来实现限速器。

例如很多时候我们需要对单个用户进行限速，每分钟或每秒钟限制limit次，可以使用user_id作为key，访问次数作为value，到redis进行存储，当用户访问一次，我们就incr一下；同时我们设置key的过期时间为1秒或1分钟，当用户访问超出限制后，我们禁止用户的访问，当key过期后，我们重新设置key，这样用户在下一分钟或者下一秒就又有了limit次访问机会。

具体实现如下：

const (
	timeout = 60 //60 seconds
)

func RateLimitIncr(key string) (err error) {
	con := pool.Get()
	defer con.Close()
	count, err := redis.Int64(con.Do("incr", key))
	if err != nil {
		return
	}
	if count == 1 {
		_, err = con.Do("expire", timeout)
	}
	return
}

func GetRateLimitCount(key string) (count int64, err error) {
	con := pool.Get()
	defer con.Close()
	count, err = redis.Int64(con.Do("get", key))
	if err != nil {
		if err == redis.ErrN