3

我正在使用 python 绑定实现一个带有 zeroMQ 的管道模式。

任务被分发给工作人员，这些工作人员通过无限循环来监听新任务，如下所示：

    while True:
        socks = dict(self.poller.poll())
        if self.receiver in socks and socks[self.receiver] == zmq.POLLIN:
            msg = self.receiver.recv_unicode(encoding='utf-8')
            self.process(msg)
        if self.hear in socks and socks[self.hear] == zmq.POLLIN:
            msg = self.hear.recv()
            print self.pid,":",  msg
            sys.exit(0)

当他们从接收节点收到消息时退出，确认已收到所有预期的结果。

但是，工人可能会错过这样的消息而无法完成。当工人无法知道时（除了通过已经提到的消息，没有进一步的任务要处理），让工人始终完成的最佳方法是什么。

这是我为检查工人状态而编写的测试代码：

#-*- coding:utf-8 -*-
"""
Test module containing tests for all modules of pypln 

"""
import unittest
from servers.ventilator import Ventilator
from subprocess import Popen, PIPE
import time
class testWorkerModules(unittest.TestCase):
    def setUp(self):
        self.nw = 4
        #spawn 4 workers
        self.ws = [Popen(['python', 'workers/dummy_worker.py'], stdout=None) for i in range(self.nw)]
        #spawn a sink
        self.sink = Popen(['python', 'sinks/dummy_sink.py'], stdout=None)
        #start a ventilator
        self.V = Ventilator()
        # wait for workers and sinks to connect
        time.sleep(1)

    def test_send_unicode(self):
        '''
        Pushing unicode strings through workers to sinks.
        '''

        self.V.push_load([u'são joão' for i in xrange(80)])
        time.sleep(1)
        #[p.wait() for p in self.ws]#wait for the workers to terminate
        wsr = [p.poll() for p in self.ws]
        while None in wsr:
            print wsr, [p.pid for p in self.ws if p.poll() == None] #these are the unfinished workers
            time.sleep(0.5)
            wsr = [p.poll() for p in self.ws]
        self.sink.wait()
        self.sink = self.sink.returncode
        self.assertEqual([0]*self.nw, wsr)
        self.assertEqual(0, self.sink)

if __name__ == '__main__':
    unittest.main()
4

1 回答 1

1

所有的消息传递最终都以心跳结束。如果您（作为工作人员或接收器或其他任何人）发现您需要使用的组件已死，您基本上可以尝试连接其他地方或自杀。因此，如果您作为工人发现水槽已不复存在，请退出。这也意味着即使接收器仍然存在但连接断开，您也可以退出。但我不确定你能做更多，也许更合理地设置所有超时......

于 2013-01-15T01:30:18.217 回答