瀏覽代碼

Fixes worker crash after using --purge

Ask Solem 9 年之前
父節點
當前提交
ded6e062cb
共有 1 個文件被更改,包括 5 次插入4 次删除
  1. 5 4
      celery/apps/worker.py

+ 5 - 4
celery/apps/worker.py

@@ -162,10 +162,11 @@ class Worker(WorkController):
         )
 
     def purge_messages(self):
-        count = self.app.control.purge()
-        if count:  # pragma: no cover
-            print('purge: Erased {0} {1} from the queue.\n'.format(
-                count, pluralize(count, 'message')))
+        with self.app.connection_for_write() as connection:
+            count = self.app.control.purge(connection=connection)
+            if count:  # pragma: no cover
+                print('purge: Erased {0} {1} from the queue.\n'.format(
+                    count, pluralize(count, 'message')))
 
     def tasklist(self, include_builtins=True, sep='\n', int_='celery.'):
         return sep.join(