velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

state_manager: make deletion schedule methods sync

This commit is contained in:
Luna 2020-08-02 01:52:20 -03:00
parent 37585760bb
commit 7dfd6d6b7c
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
litecord/gateway/state_manager.py
View File

@ -247,11 +247,11 @@ class StateManager:
state.ws.state = None state.ws.state = None
state.ws = None state.ws = None
async def schedule_deletion(self, state: GatewayState): def schedule_deletion(self, state: GatewayState):
task = app.loop.create_task(self._future_cleanup(state)) task = app.loop.create_task(self._future_cleanup(state))
self.tasks[state.session_id] = task self.tasks[state.session_id] = task
async def unschedule_deletion(self, state: GatewayState): def unschedule_deletion(self, state: GatewayState):
try: try:
task = self.tasks.pop(state.session_id) task = self.tasks.pop(state.session_id)
except KeyError: except KeyError:

2
litecord/gateway/websocket.py
View File

@ -836,7 +836,7 @@ class GatewayWebsocket:
return await self.invalidate_session(False) return await self.invalidate_session(False)
# relink this connection # relink this connection
await self.app.state_manager.unschedule_deletion(state) self.app.state_manager.unschedule_deletion(state)
self.state = state self.state = state
state.ws = self state.ws = self