Merge pull request #391 from splitio/async-redis-split-storage

Added async Redis split storage

Author: chillaq

splitio/storage/adapters/cache_trait.py

@@ -3,7 +3,7 @@
 import threading
 import time
 from functools import update_wrapper
-
+from splitio.optional.loaders import asyncio
 
 DEFAULT_MAX_AGE = 5
 DEFAULT_MAX_SIZE = 100
@@ -84,6 +84,42 @@ def get(self, *args, **kwargs):
             self._rollover()
             return node.value
 
+    async def get_key(self, key):
+        """
+        Fetch an item from the cache, return None if does not exist
+
+        :param key: User supplied key
+        :type key: str/frozenset
+
+        :return: Cached/Fetched object
+        :rtype: object
+        """
+        async with asyncio.Lock():
+            node = self._data.get(key)
+            if node is not None:
+                if self._is_expired(node):
+                    return None
+            if node is None:
+                return None
+            node = self._bubble_up(node)
+            return node.value
+
+    async def add_key(self, key, value):
+        """
+        Add an item from the cache.
+
+        :param key: User supplied key
+        :type key: str/frozenset
+
+        :param value: key value
+        :type value: str
+        """
+        async with asyncio.Lock():
+            node = LocalMemoryCache._Node(key, value, time.time(), None, None)
+            node = self._bubble_up(node)
+            self._data[key] = node
+            self._rollover()
+
     def remove_expired(self):
         """Remove expired elements."""
         with self._lock:
@@ -189,4 +225,4 @@ def _decorator(user_function):
         wrapper = lambda *args, **kwargs: _cache.get(*args, **kwargs)  # pylint: disable=unnecessary-lambda
         return update_wrapper(wrapper, user_function)
 
-    return _decorator
+    return _decorator