"""
延迟队列
"""
import threading
import time


class BatchQueue:
    def __init__(self):
        self.buf = []

    def add_one(self, obj):
        self.buf.append(obj)

    def get_all(self):
        _buf = self.buf.copy()
        self.buf = []
        return _buf


class DelayQueue:

    def __init__(self, interval, process_func):
        self.queue = BatchQueue()
        self.interval = interval
        self.process_func = process_func
        # 启动定时器
        threading.Timer(self.interval, self.process).start()

    def add(self, obj):
        self.queue.add_one(obj)

    def process(self):
        buf = self.queue.get_all()
        if len(buf) > 0:
            self.process_func(buf)
        # 定时器需要再激活才可使用
        threading.Timer(self.interval, self.process).start()


if __name__ == '__main__':

    def process(buf):
        print(buf)


    def thread_fun(dq):
        for i in range(1000):
            dq.add(i)
            time.sleep(0.01)


    dq = DelayQueue(5, process)
    threading.Thread(target=thread_fun, args=(dq,)).start()
    threading.Thread(target=thread_fun, args=(dq,)).start()
    threading.Thread(target=thread_fun, args=(dq,)).start()
    input()