利用redis实现分布式环境下的限频

版权声明 本站原创文章 由 萌叔 发表
转载请注明 萌叔 | https://vearne.cc

redis 本身有计数器，并且可以做原子的增1操作，特别适合用来做分布式环境下的限频

# coding:utf-8
import time
import threading
from redis import StrictRedis


class Counter(object):
    def __init__(self, redis_url):
        self.redis_client = StrictRedis.from_url(redis_url)

    def increment(self, key):
        t = int(time.time())
        sign = t / 60

        redis_key = key + ':' + str(sign)
        counter = self.redis_client.incr(redis_key)
        # 注：设置key的失效时间没有必要和原子增1操作包含在一个事务中。
        self.redis_client.expire(redis_key, 300)  # 设置key的失效时间300 seconds
        return counter




if __name__ == '__main__':
    redis_url = 'redis://127.0.0.1:6379/0'
    c = Counter(redis_url)
    for i in range(100):
        time.sleep(0.2)
        x = c.increment('hello')
        if x > 50:
            print "over limit"
        print x

这里限频有个前提条件，就是分布式环境中时钟，必须尽量对齐。
在上面的例子中频率限制就是50次/分钟