[GO]go redis实现滑动窗口限流-redis版

2020-12-16 15:33:24 浏览数 (1)

上一篇是单机当前进程的滑动窗口限流 , 这一个是使用go redis list结构实现的滑动窗口限流 , 原理都一样 , 但是支持分布式

原理可以参考上一篇介绍

代码语言:javascript复制
func LimitFreqs(queueName string, count uint, timeWindow int64) bool {
    currTime := time.Now().Unix()
    length := uint(ListLen(queueName))
    if length < count {
        ListPush(queueName, currTime)
        return true
    }
    //队列满了,取出最早访问的时间
    earlyTime, _ := strconv.ParseInt(ListIndex(queueName, int64(length)-1), 10, 64)
    //说明最早期的时间还在时间窗口内,还没过期,所以不允许通过
    if currTime-earlyTime <= timeWindow {
        return false
    } else {
        //说明最早期的访问应该过期了,去掉最早期的
        ListPop(queueName)
        ListPush(queueName, currTime)
    }
    return true
}

0 人点赞