线程本地pyzmq ioloop

时间:2015-06-11 11:04:35

标签: python multithreading tornado pyzmq

我想知道是否有办法在pyzmq中拥有一个线程本地ioloop。代码中的注释表示(https://github.com/zeromq/pyzmq/blob/master/zmq/eventloop/minitornado/ioloop.py第172行):

  

通常,您应该使用IOLoop.current作为默认值           构造异步对象,并使用IOLoop.instance           当你的意思是从一个不同的主线程沟通           之一。

根据这个我假设使用.current我应该能够创建线程本地ioloop,我可以独立于主线程ioloop启动/停止。

奇怪的是,这不是我观察到的行为(我在本地线程中使用.current,但在此循环中执行.stop也会停止主线程循环)。而且,这个简单的测试代码:

import threading
from zmq.eventloop import ioloop

loop = ioloop.IOLoop.current()
print 'main id(loop):', id(loop)

class w(threading.Thread):
    def __init__(self, loop=None):
        super(w, self).__init__()

        self.loop = loop or ioloop.IOLoop.current()

        print 'w.__init__ id(loop):', id(self.loop)

    def run(self):
        print 'w.__run__ id(loop):', id(self.loop)
        print 'current:', id(ioloop.IOLoop.current())


w(loop).start()

为两个循环打印相同的ID:

main id(loop): 33576144
w.__init__ id(loop): 33576144
w.__run__ id(loop): 33576144
current: 33576144

我很高兴明白我错过了什么。

亚历

1 个答案:

答案 0 :(得分:2)

由于历史原因,

IOLoop.current()有一些怪癖。它将引用线程局部事件循环(如果存在),但它不会创建一个,而是回退到单例(主线程)循环。

解决方案是在w.__init__()self.loop = ioloop.IOLoop())中创建一个新循环,并使其在w.run() self.loop.make_current()中保持最新。loop.start()将调用{{ 1}}为你)。