Преглед изворни кода

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 пре 16 година
родитељ
комит
1f8e955eae
1 измењених фајлова са 1 додато и 1 уклоњено
  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)