浏览代码

Mark dispatch of broadcast message as blocking section

Ask Solem 11 年之前
父节点
当前提交
a0963263d9
共有 1 个文件被更改,包括 3 次插入1 次删除
  1. 3 1
      celery/worker/pidbox.py

+ 3 - 1
celery/worker/pidbox.py

@@ -3,6 +3,7 @@ from __future__ import absolute_import
 import socket
 import threading
 
+from kombu.async import maybe_block
 from kombu.common import ignore_errors
 from kombu.utils.encoding import safe_str
 
@@ -34,7 +35,8 @@ class Pidbox(object):
         self._forward_clock()  # just increase clock as clients usually don't
                                # have a valid clock to adjust with.
         try:
-            self.node.handle_message(body, message)
+            with maybe_block():
+                self.node.handle_message(body, message)
         except KeyError as exc:
             error('No such control command: %s', exc)
         except Exception as exc: