ソースを参照

fix :区别 db=4(测试) db=3(正式)

chenjunkai 6 年 前
コミット
475b203fef
2 ファイル変更9 行追加2 行削除
  1. 2 0
      Controller/UserController.py
  2. 7 2
      Object/RedisObject.py

+ 2 - 0
Controller/UserController.py

@@ -295,6 +295,8 @@ class LogoutView(TemplateView):
             return response.json(tko.code)
         MiscellService.add_access_log(request=request, status_code=200)
         Device_User.objects.filter(userID=tko.userID).update(online=False)
+        redisObj = RedisObject(db=3)
+        redisObj.del_data(key=tko.userID)
         return response.json(0)
 
 

+ 7 - 2
Object/RedisObject.py

@@ -14,16 +14,21 @@
 import redis
 
 from Ansjer.config import SERVER_HOST
-
+from Ansjer.config import SERVER_TYPE
 
 # SERVER_HOST = '192.168.136.45'
 
 '''
 db=3  -> 统计在线人数用
 '''
+
+
 class RedisObject:
 
     def __init__(self, db=0):
+        if db == 3:
+            if SERVER_TYPE != 'Ansjer.formal_settings':
+                db = 4
         self.POOL = redis.ConnectionPool(host=SERVER_HOST, port=6379, db=db)
         self.CONN = redis.Redis(connection_pool=self.POOL)
 
@@ -64,4 +69,4 @@ class RedisObject:
 
 redisObj = RedisObject()
 size = redisObj.get_size()
-print(size)
+print(size)