|
@@ -70,6 +70,8 @@ class TaskBucket(object):
|
|
|
|
|
|
def put(self, job):
|
|
def put(self, job):
|
|
"""Put a task into the appropiate bucket."""
|
|
"""Put a task into the appropiate bucket."""
|
|
|
|
+ if job.task_name not in self.buckets:
|
|
|
|
+ self.add_bucket_for_type(job.task_name)
|
|
self.buckets[job.task_name].put_nowait(job)
|
|
self.buckets[job.task_name].put_nowait(job)
|
|
put_nowait = put
|
|
put_nowait = put
|
|
|
|
|