fix: 修复数据库连接没有关闭的问题

apps/authentication/utils.py

@@ -59,6 +59,10 @@ def check_different_city_login(user, request):
     else:
         city = get_ip_city(ip) or DEFAULT_CITY
 
-    last_user_login = UserLoginLog.objects.filter(username=user.username, status=True).first()
-    if last_user_login and last_user_login.city != city:
-        DifferentCityLoginMessage(user, ip, city).publish_async()
+    city_white = ['LAN', ]
+    if city not in city_white:
+        last_user_login = UserLoginLog.objects.exclude(city__in=city_white) \
+            .filter(username=user.username, status=True).first()
+
+        if last_user_login and last_user_login.city != city:
+            DifferentCityLoginMessage(user, ip, city).publish_async()

apps/notifications/ws.py

@@ -2,6 +2,7 @@ import threading
 import json
 from redis.exceptions import ConnectionError
 from channels.generic.websocket import JsonWebsocketConsumer
+from django.db import close_old_connections
 
 from common.utils import get_logger
 from .site_msg import SiteMessageUtil
@@ -65,8 +66,11 @@ class SiteMsgWebsocket(JsonWebsocketConsumer):
                     logger.debug('Decode json error: ', e)
         except ConnectionError:
             logger.debug('Redis chan closed')
+        finally:
+            close_old_connections()
 
     def disconnect(self, close_code):
         if self.chan is not None:
             self.chan.close()
         self.close()
+        close_old_connections()