我有一个Python脚本,它打开一个websocket到Twitter API,然后等待。当事件通过amq传递给脚本时,我需要打开一个新的websocket连接并立即在新连接注册后立即关闭旧连接。
它看起来像这样:
stream = TwitterStream()
stream.start()
for message in broker.listen():
if message:
new_stream = TwitterStream()
# need to close the old connection as soon as the
# new one connects here somehow
stream = new_stream()
我正试图弄清楚我如何建立一个"回调",以便通知我的脚本何时建立新的连接。TwitterStream类有一个"is_running"布尔变量,我可以引用,所以我想也许是这样的:
while not new_stream.is_running:
time.sleep(1)
但是看起来有点乱。有人知道更好的方法吗?
繁忙循环不是正确的方法,因为它明显浪费CPU。相反,有一些线程结构可以让您通信此类事件。例如:http://docs.python.org/library/threading.html#event-objects
下面是一个线程事件的例子:
import threading
from time import sleep
evt = threading.Event()
result = None
def background_task():
global result
print("start")
result = "Started"
sleep(5)
print("stop")
result = "Finished"
evt.set()
t = threading.Thread(target=background_task)
t.start()
# optional timeout
timeout=3
evt.wait(timeout=timeout)
print(result)