内存计数器实现限频

以分钟为单位计数,超过相应数值则超过频率限制,很简单,不多解释

import time
import threading


class Counter(object):
    def __init__(self):
        self.curr = 0
        self.internal_dd = {}
        self.lock = threading.Lock()

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

        # lock
        self.lock.acquire()
        try:
            if sign != self.curr:
                self.curr = sign
                self.internal_dd = {}

            if key not in self.internal_dd:
                self.internal_dd[key] = 0

            self.internal_dd[key] += 1

            # unlock
            return self.internal_dd.get(key, 0)
        finally:
            self.lock.release()


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

你可能感兴趣的:(限频)