Procházet zdrojové kódy

kvbackends: Use save_group directly

Ask Solem před 12 roky
rodič
revize
e37689e2e8
2 změnil soubory, kde provedl 4 přidání a 3 odebrání
  1. 3 2
      celery/backends/base.py
  2. 1 1
      celery/backends/cache.py

+ 3 - 2
celery/backends/base.py

@@ -215,8 +215,9 @@ class BaseBackend(object):
 
     def save_group(self, group_id, result):
         """Store the result and status of a task."""
+
         raise NotImplementedError(
-            'save_group is not supported by this backend.')
+            'save_group is not supported by %s.' % (type(self).__name__, ))
 
     def restore_group(self, group_id, cache=True):
         """Get the result of a group."""
@@ -476,7 +477,7 @@ class KeyValueStoreBackend(BaseDictBackend):
 
     def on_chord_apply(self, group_id, body, result=None, **kwargs):
         if self.implements_incr:
-            self.app.GroupResult(group_id, result).save()
+            self.save_group(group_id, self.app.GroupResult(group_id, result))
         else:
             self.fallback_chord_unlock(group_id, body, result, **kwargs)
 

+ 1 - 1
celery/backends/cache.py

@@ -115,7 +115,7 @@ class CacheBackend(KeyValueStoreBackend):
 
     def on_chord_apply(self, group_id, body, result=None, **kwargs):
         self.client.set(self.get_key_for_chord(group_id), '0', time=86400)
-        self.app.GroupResult(group_id, result).save()
+        self.save_group(group_id, self.app.GroupResult(group_id, result))
 
     def incr(self, key):
         return self.client.incr(key)