result work not block

binux · binux · commit 77c2d5fa3100 · 2014-11-11T13:14:46.000+08:00
diff --git a/result/result_worker.py b/result/result_worker.py
@@ -28,18 +28,20 @@ def on_result(self, task, result):
         assert 'url' in task, 'need url in task'
         return self.resultdb.save(project=task['project'], taskid=task['taskid'], url=task['url'], result=result)
 
+    def quit(self):
+        self._quit = True
+
     def run(self):
         while not self._quit:
             try:
-                task, result = self.inqueue.get()
+                task, result = self.inqueue.get(timeout=1)
                 if 'taskid' in task and 'project' in task and 'url' in task:
                     logger.info('result %s:%s %s -> %.30r' % (
                         task['project'], task['taskid'], task['url'], result))
                 else:
                     logger.warning('result UNKNOW -> %.30r' % result)
                 ret = self.on_result(task, result)
             except Queue.Empty as e:
-                time.sleep(1)
                 continue
             except KeyboardInterrupt:
                 break
