Parcourir la source

When we receive a message the prefetch window is increased, when we accept the
task for processing internally we decrease the prefetch window.

Ask Solem il y a 15 ans
Parent
commit
1f8e955eae
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      celery/worker/__init__.py

+ 1 - 1
celery/worker/__init__.py

@@ -104,11 +104,11 @@ class AMQPListener(object):
 
         eta = message_data.get("eta")
         if eta:
+            self.prefetch_count.increment()
             self.logger.info("Got task from broker: %s[%s] eta:[%s]" % (
                     task.task_name, task.task_id, eta))
             self.hold_queue.put((task, eta, self.prefetch_count.decrement))
         else:
-            self.prefetch_count.decrement()
             self.logger.info("Got task from broker: %s[%s]" % (
                     task.task_name, task.task_id))
             self.bucket_queue.put(task)