Added scaling support via redisv1.1.0-rc1

1 files changed, 20 insertions, 0 deletions

diff --git a/ttun_server/redis.py b/ttun_server/redis.py
new file mode 100644
index 0000000..344c107
--- /dev/null
+++ b/ttun_server/redis.py
@@ -0,0 +1,20 @@
+import os
+
+from aioredis import ConnectionPool, Redis
+
+
+class RedisConnectionPool():
+    instance: 'RedisConnectionPool' = None
+
+    def __init__(self):
+       self.pool = ConnectionPool.from_url(os.environ.get('REDIS_URL'))
+
+    def __del__(self):
+        self.pool.disconnect()
+
+    @classmethod
+    def get_connection(cls) -> Redis:
+        if cls.instance is None:
+            cls.instance = RedisConnectionPool()
+
+        return Redis(connection_pool=cls.instance.pool)