功能变化: 优化websocket

pull/84/head
猿小天 2023-01-03 13:30:20 +08:00
parent b57d500e0c
commit 4712530100
1 changed files with 7 additions and 5 deletions

View File

@ -69,13 +69,15 @@ class DvadminWebSocket(AsyncJsonWebsocketConsumer):
self.channel_name
)
await self.accept()
# 发送连接成功
#await self.send_json(set_message('system', 'SYSTEM', '连接成功'))
# 主动推送消息
unread_count = await _get_message_unread(self.user_id)
await self.send_json(
set_message('system', 'SYSTEM', "请查看您的未读消息~",
unread=unread_count))
if unread_count == 0:
# 发送连接成功
await self.send_json(set_message('system', 'SYSTEM', '连接成功'))
else:
await self.send_json(
set_message('system', 'SYSTEM', "请查看您的未读消息~",
unread=unread_count))
except InvalidSignatureError:
await self.disconnect(None)