Explorar o código

Worker: Events: Remember changes from remote control commands

Ask Solem %!s(int64=11) %!d(string=hai) anos
pai
achega
8d76a7b968
Modificáronse 1 ficheiros con 5 adicións e 1 borrados
  1. 5 1
      celery/worker/consumer.py

+ 5 - 1
celery/worker/consumer.py

@@ -487,7 +487,7 @@ class Events(bootsteps.StartStopStep):
         c.event_dispatcher = None
 
     def start(self, c):
-        # Flush events sent while connection was down.
+        # flush events sent while connection was down.
         prev = c.event_dispatcher
         dis = c.event_dispatcher = c.app.events.Dispatcher(
             c.connect(), hostname=c.hostname,
@@ -499,6 +499,10 @@ class Events(bootsteps.StartStopStep):
 
     def stop(self, c):
         if c.event_dispatcher:
+            # remember changes from remote control commands:
+            self.groups = c.event_dispatcher.groups
+
+            # close custom connection
             if c.event_dispatcher.connection:
                 ignore_errors(c, c.event_dispatcher.connection.close)
             ignore_errors(c, c.event_dispatcher.close)