Ver Fonte

Pool now compatible with thread-based transports again

Ask Solem há 11 anos atrás
pai
commit
4700c49194
1 ficheiros alterados com 7 adições e 1 exclusões
  1. 7 1
      celery/concurrency/processes.py

+ 7 - 1
celery/concurrency/processes.py

@@ -987,11 +987,17 @@ class TaskPool(BasePool):
         self.restart = P.restart
         self.maybe_handle_result = P._result_handler.handle_event
         self.handle_result_event = P.handle_result_event
-        self.register_with_event_loop = P.register_with_event_loop
 
     def did_start_ok(self):
         return self._pool.did_start_ok()
 
+    def register_with_event_loop(self, loop):
+        try:
+            reg = self._pool.register_with_event_loop
+        except AttributeError:
+            return
+        return reg(loop)
+
     def on_stop(self):
         """Gracefully stop the pool."""
         if self._pool is not None and self._pool._state in (RUN, CLOSE):