功能变化: 优化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 self.channel_name
) )
await self.accept() await self.accept()
# 发送连接成功
#await self.send_json(set_message('system', 'SYSTEM', '连接成功'))
# 主动推送消息 # 主动推送消息
unread_count = await _get_message_unread(self.user_id) unread_count = await _get_message_unread(self.user_id)
await self.send_json( if unread_count == 0:
set_message('system', 'SYSTEM', "请查看您的未读消息~", # 发送连接成功
unread=unread_count)) await self.send_json(set_message('system', 'SYSTEM', '连接成功'))
else:
await self.send_json(
set_message('system', 'SYSTEM', "请查看您的未读消息~",
unread=unread_count))
except InvalidSignatureError: except InvalidSignatureError:
await self.disconnect(None) await self.disconnect(None)