Initial commit

core/service/redis.py

@@ -0,0 +1,34 @@
+import json
+
+import aioredis
+from aioredis import Redis
+from aioredis.client import ExpiryT
+
+from core.config import DatabaseConfig
+
+
+def redis_db(db: DatabaseConfig) -> Redis:
+    # Redis client bound to pool of connections (auto-reconnecting).
+    redis = aioredis.from_url(
+        db.url,
+        encoding="utf-8",
+        decode_responses=True,
+        db=int(db.db) or 0
+    )
+    return redis
+
+
+async def redis_set(key: str, value: any, db: DatabaseConfig, ex=ExpiryT):
+    redis_c = redis_db(db)
+    await redis_c.set(key, json.dumps(value), ex=ex)
+    await redis_c.close()
+    return True
+
+
+async def redis_get(key: str, db: DatabaseConfig):
+    redis_c = redis_db(db)
+    _return = await redis_c.get(key)
+    await redis_c.close()
+    if not _return or isinstance(_return, (int, float, complex)):
+        return _return
+    return json.loads(_return)