Time set by server

This commit is contained in:
BarsTiger
2023-07-26 16:54:46 +03:00
parent 333cc8ddf2
commit 80242a809e
2 changed files with 5 additions and 3 deletions

View File

@@ -1,6 +1,7 @@
from attrs import define
from fastapi import WebSocket
from dragonion_core.proto.web.webmessage import (
set_time,
webmessages_union,
webmessage_error_message_literal,
WebErrorMessage
@@ -20,7 +21,7 @@ class Connection(object):
:param obj: Should be some type of WebMessage
:return:
"""
await self.ws.send_text(obj.to_json())
await self.ws.send_text(set_time(obj).to_json())
async def send_error(
self,

View File

@@ -8,6 +8,7 @@ from json.decoder import JSONDecodeError
from dragonion_core.proto.web.webmessage import (
webmessages_union,
set_time,
WebMessageMessage,
WebBroadcastableMessage,
WebNotificationMessage,
@@ -37,9 +38,9 @@ class Room(object):
await ws.receive_text()
)
except JSONDecodeError:
await ws.send_text(WebErrorMessage(
await ws.send_text(set_time(WebErrorMessage(
'invalid_webmessage'
).to_json())
)).to_json())
await ws.close(reason='invalid_webmessage')
return