如何用Redis实现一个指定时间的限速器

栏目: 数据库 · 发布时间: 5年前

内容简介:使用Redis的在redis的官方文档中也有详细的示例举个例子:

使用 RedisIncr 可以很容易的实现一个限速器

在redis的官方文档中也有详细的示例

FUNCTION LIMIT_API_CALL(ip)
ts = CURRENT_UNIX_TIME()
keyname = ip+":"+ts
current = GET(keyname)

IF current != NULL AND current > 10 THEN
    ERROR "too many requests per second"
END

IF current == NULL THEN
    MULTI
        INCR(keyname, 1)
        EXPIRE(keyname, 1)
    EXEC
ELSE
    INCR(keyname, 1)
END

PERFORM_API_CALL()

举个例子:

现有一个服务,1分钟内只能接受一个用户不超过10次的请求,这时我们可以将用户的ip地址设置为key,用户每次时程序去redis中获取该key的值,如果大于等于10则返回错误,否则给key对应的value+1即可,如果value为0,那么再将该key设置1分钟的过期时间。

但是现在有一个需求,我们可以在一个指定的时间内给用户推送一条消息,但是要求用户每分钟内只能接受一条消息,每小时内接受的消息不超过5条,一天内接受的消息不超过10条。

比如,我现在向这个接口提交了一条数据,要求在2019-11-11 11:11:11时向一个用户发送一条数据,那么当我再提交一条数据,要求在2019-11-11 11:11:12是向同样的用户发送一条数据,那么接口就会返回错误。

此时,仅使用 Incr 是无法满足该需求的。

解决方法:

使用 set 或者 zset 将用户的 ip+发送日期 作为key,发送时间转换为当天的秒数作为value,插入到 set 或者 zset 中,每次向用户提交信息时,可以获取到 set 中的所有发送时间,然后再一一比对,如果不满足条件就返回错误。

以下是golang的实现

func RateLimit(ctx context.Context, ip string, sendTime time.Time) error {
    sendTime = sendTime.UTC()
    if sendTime.Before(time.Now()) {
        return nil
    }
    zeroStr := sendTime.Format("2006-01-02")
    key := ip + zeroStr
    zero, _ := time.Parse("2006-01-02", zeroStr)
    // 获取发送时间距当天时间的秒数
    second := int(sendTime.Sub(zero).Seconds())
    ress, err := cache.SMembers(ctx, key)
    if err != nil {
        return err
    }
    // 处理返回参数,将[]string转换为[]int
    var sends []int
    for _, v := range ress {
        if vv, err := strconv.Atoi(v); err == nil {
            sends = append(sends, vv)
        }
    }
    if len(sends) >= 10 {
        return errors.ErrMsg1DayLimit
    }

    var expire bool

    if len(sends) == 0 {
        expire = true
    }

    var hourCount int

    for _, s := range sends {
        if math.Abs(float64(second-s)) < 60 {
            return errors.ErrMsg1MinuteLimit
        }
        if math.Abs(float64(second-s)) < 3600 {
            hourCount++
        }
    }
    if hourCount > 5 {
        return errors.ErrMsg1HourLimit
    }
    // 异步更新
    go func() {
        // 开启事务
        t := cache.Pipeline()
        defer t.Close()
        t.SAdd(context.Background(), key, []byte(strconv.Itoa(second)))
        if expire {
            ttl := sendTime.Sub(time.Now()).Seconds()
            if ttl <= 0 {
                t.Rollback()
                return
            }
            t.Expire( key, int64(ttl))
        }
        t.Commit()
    }()
    return nil
}

以上就可以实现一个指定时间的限速器。


以上所述就是小编给大家介绍的《如何用Redis实现一个指定时间的限速器》,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对 码农网 的支持!

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

JAVA核心技术(卷1)

JAVA核心技术(卷1)

Cay S. Horstmann、Gary Cornell / 杜永萍、邝劲筠、叶乃文 / 机械工业出版社 / 2008-6 / 98.00元

《JAVA核心技术(卷1):基础知识(原书第8版)》是《Java核心技术》的最新版,《Java核心技术》出版以来一直畅销不衰,深受读者青睐,每个新版本都尽可能快地跟上Java开发工具箱发展的步伐,而且每一版都重新改写了的部分内容,以便适应Java的最新特性。本版也不例外,它反遇了Java SE6的新特性。全书共14章,包括Java基本的程序结构、对象与类、继承、接口与内部类、图形程序设计、事件处理......一起来看看 《JAVA核心技术(卷1)》 这本书的介绍吧!

HTML 压缩/解压工具
HTML 压缩/解压工具

在线压缩/解压 HTML 代码

HTML 编码/解码
HTML 编码/解码

HTML 编码/解码

html转js在线工具
html转js在线工具

html转js在线工具