初始版本修改

api/api_v1/authz/authz.py

@@ -10,7 +10,6 @@ from db import get_database
 from db.ckdb import CKDrive, get_ck_db
 from db.redisdb import RedisDrive, get_redis_pool
 from models.behavior_analysis import BehaviorAnalysis
-from utils import casbin_enforcer
 
 router = APIRouter()
 

api/api_v1/endpoints/project.py

@@ -9,7 +9,6 @@ from db import get_database
 from db.ckdb import CKDrive, get_ck_db
 from schemas.project import ProjectCreate
 # from utils import casbin_enforcer
-from utils import casbin_enforcer
 
 router = APIRouter()
 

db/redisdb.py

@@ -0,0 +1,32 @@
+from aredis import StrictRedis
+import asyncio
+
+
+class RedisDrive:
+    client: StrictRedis = None
+
+    # async def get_keys(self, *keys, prefix='') -> list:
+    #     res = []
+    #     for key in keys:
+    #         key = prefix + key
+    #         data = await self.client.get(key, encoding='utf8')
+    #         res.append(data)
+    #     return res
+
+    async def smembers_keys(self, *keys, prefix='') -> dict:
+        tasks = []
+        for key in keys:
+            task = asyncio.create_task(self.client.smembers(prefix + key))
+            tasks.append(task)
+        data = await asyncio.gather(*tasks)
+        return {k: v for k, v in zip(keys, data)}
+
+    def __getattr__(self, item):
+        return getattr(self.client, item)
+
+
+rdb = RedisDrive()
+
+
+def get_redis_pool() -> RedisDrive:
+    return rdb

db/redisdb_utils.py

@@ -0,0 +1,12 @@
+from aredis import StrictRedis
+
+from core.config import settings
+from .redisdb import RedisDrive
+
+
+async def connect_to_redis():
+    RedisDrive.client = StrictRedis(**settings.REDIS_CONF)
+
+
+async def close_redis_connection():
+    pass

utils/__init__.py

@@ -1,4 +1,3 @@
-from .adapter import *
 
 from . import casbin
 from .func import *