github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

fix: 完善消息队列 redis 连接未关闭

v2.11
xinwen 2022-02-10 14:40:22 +08:00 committed by Jiangjie.Bai
parent 5cfd5afc7c
commit bc7dfc3dc7
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/notifications/ws.py
View File

@ -4,7 +4,6 @@ import json
from channels.generic.websocket import JsonWebsocketConsumer from channels.generic.websocket import JsonWebsocketConsumer
from common.utils import get_logger from common.utils import get_logger
from .models import SiteMessage
from .site_msg import SiteMessageUtil from .site_msg import SiteMessageUtil
from .signals_handler import new_site_msg_chan from .signals_handler import new_site_msg_chan
@ -14,6 +13,7 @@ logger = get_logger(__name__)
class SiteMsgWebsocket(JsonWebsocketConsumer): class SiteMsgWebsocket(JsonWebsocketConsumer):
disconnected = False disconnected = False
refresh_every_seconds = 10 refresh_every_seconds = 10
subscribe = None
def connect(self): def connect(self):
user = self.scope["user"] user = self.scope["user"]
@ -68,5 +68,6 @@ class SiteMsgWebsocket(JsonWebsocketConsumer):
def disconnect(self, close_code): def disconnect(self, close_code):
self.disconnected = True self.disconnected = True
self.subscribe.close()
self.close() self.close()
if self.subscribe:
self.subscribe.close()