dongzhaorui 3 years ago
parent
commit
cb66476ebc
1 changed files with 4 additions and 4 deletions
  1. 4 4
      find_source/crawler/schedule.py

+ 4 - 4
find_source/crawler/schedule.py

@@ -4,14 +4,14 @@ from settings import REDIS_KEY
 
 class Scheduler:
 
-    def __init__(self, queue=RedisQueue()):
-        self.mrq = queue
+    def __init__(self, queue=None):
+        self.mrq = (queue or RedisQueue())
 
     def get_task(self):
         return self.mrq.pop_task([REDIS_KEY])
 
     def insert_task(self, task):
-        self.mrq.push_task(REDIS_KEY, [task])
+        self.mrq.push_task(REDIS_KEY, [task], level=1)
 
     def insert_tasks(self, tasks):
-        self.mrq.push_task(REDIS_KEY, tasks)
+        self.mrq.push_task(REDIS_KEY, tasks, level=1)