Browse Source

Optimization for #2358

Ask Solem 10 years ago
parent
commit
f8b2efaa7e
1 changed files with 2 additions and 1 deletions
  1. 2 1
      celery/worker/strategy.py

+ 2 - 1
celery/worker/strategy.py

@@ -61,6 +61,7 @@ def default(task, app, consumer,
     call_at = consumer.timer.call_at
     apply_eta_task = consumer.apply_eta_task
     rate_limits_enabled = not consumer.disable_rate_limits
+    get_bucket = consumer.task_buckets.__getitem__
     handle = consumer.on_task_request
     limit_task = consumer._limit_task
     body_can_be_buffer = consumer.pool.body_can_be_buffer
@@ -115,7 +116,7 @@ def default(task, app, consumer,
                 call_at(eta, apply_eta_task, (req, ), priority=6)
         else:
             if rate_limits_enabled:
-                bucket = consumer.task_buckets[task.name]
+                bucket = get_bucket(task.name)
                 if bucket:
                     return limit_task(req, bucket, 1)
             task_reserved(req)