Question

我使用heaps构建了优先级队列。队列包含消息，应按照与优先级相关的顺序发送消息。但是，作为优先级值，我有一段时间应该发送消息，例如我必须将一组消息放入队列：

(10, message1)
(15, message2)
(5, message3)

因此，按照优先级发送消息很容易。但是，如果我在将其放入队列后5秒钟内首先发送messag3，我希望确保下一条消息message1将在放入队列10秒后发送，message3发送后的5秒。有谁知道如何设法做到这一点？

Answer 1

您可以使用纪元作为优先级值，并且每次计时器触发计算何时应该根据当前时间再次触发。以下是实践中的一个简短示例：

import calendar
import time
import heapq
from threading import Timer

def epoch():
    return calendar.timegm(time.gmtime())

start_time = epoch()
heap = []
timer = None

def add_message(seconds, content):
    top = heap[0] if heap else None
    heapq.heappush(heap, (epoch() + seconds, content))
    if timer and top != heap[0]:
        timer.cancel()
        start()

def start():
    global timer
    if heap:
        timer = Timer(heap[0][0] - epoch(), fire)
        timer.start()

def fire():
    _, message = heapq.heappop(heap)
    print '{}: {}'.format(epoch() - start_time, message)
    start()

add_message(10, 'message1')
add_message(15, 'message2')
add_message(5, 'message3')
start()
add_message(1, 'message4')

输出：

1: message4
5: message3
10: message1
15: message2

Python：优先级队列，优先级为时间

1 个答案: