Commit cd6d1f92 by James Cammarata

Fix pickling errors with cache plugins (v2)

Fixes #10945
parent 198476e3
...@@ -26,6 +26,9 @@ from six import add_metaclass ...@@ -26,6 +26,9 @@ from six import add_metaclass
@add_metaclass(ABCMeta) @add_metaclass(ABCMeta)
class BaseCacheModule: class BaseCacheModule:
def __init__(self):
self.__getstate__ = self.copy
@abstractmethod @abstractmethod
def get(self, key): def get(self, key):
pass pass
...@@ -53,3 +56,4 @@ class BaseCacheModule: ...@@ -53,3 +56,4 @@ class BaseCacheModule:
@abstractmethod @abstractmethod
def copy(self): def copy(self):
pass pass
...@@ -113,6 +113,8 @@ class CacheModuleKeys(collections.MutableSet): ...@@ -113,6 +113,8 @@ class CacheModuleKeys(collections.MutableSet):
self._cache = cache self._cache = cache
self._keyset = dict(*args, **kwargs) self._keyset = dict(*args, **kwargs)
super(CacheModule, self).__init__()
def __contains__(self, key): def __contains__(self, key):
return key in self._keyset return key in self._keyset
......
...@@ -24,6 +24,8 @@ class CacheModule(BaseCacheModule): ...@@ -24,6 +24,8 @@ class CacheModule(BaseCacheModule):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
self._cache = {} self._cache = {}
super(CacheModule, self).__init__()
def get(self, key): def get(self, key):
return self._cache.get(key) return self._cache.get(key)
......
...@@ -51,6 +51,8 @@ class CacheModule(BaseCacheModule): ...@@ -51,6 +51,8 @@ class CacheModule(BaseCacheModule):
self._cache = StrictRedis(*connection) self._cache = StrictRedis(*connection)
self._keys_set = 'ansible_cache_keys' self._keys_set = 'ansible_cache_keys'
super(CacheModule, self).__init__()
def _make_key(self, key): def _make_key(self, key):
return self._prefix + key return self._prefix + key
...@@ -100,3 +102,4 @@ class CacheModule(BaseCacheModule): ...@@ -100,3 +102,4 @@ class CacheModule(BaseCacheModule):
for key in self.keys(): for key in self.keys():
ret[key] = self.get(key) ret[key] = self.get(key)
return ret return ret
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment