nex/college-bot-v1
1
0
Fork 0
mirror of https://github.com/nexy7574/LCC-bot.git synced 2024-09-19 18:16:34 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix websocket

Browse source
This commit is contained in:
Nexus 2023-06-29 12:59:15 +01:00
parent a8c0508eac
commit b54c1ffc19
Signed by: nex
GPG key ID: 0FA334385D0B689F
2 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
cogs/events.py
View file

@ -385,6 +385,7 @@ class Events(commands.Cog):
} }
# dumped = json.dumps(payload, separators=(",", ":")) # dumped = json.dumps(payload, separators=(",", ":"))
self.bot.bridge_queue.put_nowait(payload) self.bot.bridge_queue.put_nowait(payload)
print("Added %s to queue" % payload)
if message.channel.name == "pinboard": if message.channel.name == "pinboard":
if message.type == discord.MessageType.pins_add: if message.type == discord.MessageType.pins_add:

8
web/server.py
View file

@ -1,3 +1,4 @@
import asyncio
import ipaddress import ipaddress
import sys import sys
@ -318,7 +319,12 @@ async def bridge_recv(ws: WebSocket, secret: str = Header(None)):
await ws.accept() await ws.accept()
while True: while True:
data = await app.state.bot.bridge_queue.get() try:
data = app.state.bot.bridge_queue.get_nowait()
except asyncio.QueueEmpty:
await asyncio.sleep(0.5)
continue
try: try:
await ws.send_json(data) await ws.send_json(data)
except WebSocketDisconnect: except WebSocketDisconnect: