浏览代码

set - WeakSet is not allowed, so real requests now stored in state.requests. Closes #3256

Ask Solem 8 年之前
父节点
当前提交
df624a5deb
共有 1 个文件被更改,包括 3 次插入3 次删除
  1. 3 3
      celery/worker/state.py

+ 3 - 3
celery/worker/state.py

@@ -43,10 +43,10 @@ REVOKES_MAX = 50000
 REVOKE_EXPIRES = 10800
 
 #: Mapping of reserved task_id->Request.
-requests = weakref.WeakValueDictionary()
+requests = {}
 
 #: set of all reserved :class:`~celery.worker.request.Request`'s.
-reserved_requests = set()
+reserved_requests = weakref.WeakSet()
 
 #: set of currently active :class:`~celery.worker.request.Request`'s.
 active_requests = weakref.WeakSet()
@@ -65,9 +65,9 @@ should_terminate = None
 
 
 def reset_state():
+    requests.clear()
     reserved_requests.clear()
     active_requests.clear()
-    requests.clear()
     total_count.clear()
     all_total_count[:] = [0]
     revoked.clear()