|
|
@@ -10,8 +10,8 @@ class Scheduler:
|
|
|
def get_task(self):
|
|
|
return self.mrq.pop_task([REDIS_KEY])
|
|
|
|
|
|
- def insert_task(self, task):
|
|
|
- self.mrq.push_task(REDIS_KEY, [task], level=1)
|
|
|
+ def insert_task(self, task, level=1):
|
|
|
+ self.mrq.push_task(REDIS_KEY, [task], level=level)
|
|
|
|
|
|
- def insert_tasks(self, tasks):
|
|
|
- self.mrq.push_task(REDIS_KEY, tasks, level=1)
|
|
|
+ def insert_tasks(self, tasks, level=1):
|
|
|
+ self.mrq.push_task(REDIS_KEY, tasks, level=level)
|
