Replaced try/except block in recv_task() by match msg, and added new changes to description comment

dark_clearing_improvements
Esmeralda Gallardo 2022-10-19 15:38:35 -03:00
parent db0b59abaa
commit 3f5dec82ed
No known key found for this signature in database
GPG Key ID: AD67FBB417CAC71B
1 changed files with 20 additions and 15 deletions

View File

@ -221,28 +221,33 @@ async def open_jsonrpc_session(
'''
receives every ws message and stores it in its corresponding result
field, then sets the event to wakeup original sender tasks.
also, recieves responses to requests originated from the server side.
'''
async for msg in ws:
try:
msg = response_type(**msg)
match msg:
case {
'result': _
}:
msg = response_type(**msg)
if msg.id not in rpc_results:
log.warning(f'Wasn\'t expecting ws msg: {json.dumps(msg, indent=4)}')
if msg.id not in rpc_results:
log.warning(f'Wasn\'t expecting ws msg: {json.dumps(msg, indent=4)}')
res = rpc_results.setdefault(
msg.id,
{'result': None, 'event': trio.Event()}
)
res = rpc_results.setdefault(
msg.id,
{'result': None, 'event': trio.Event()}
)
res['result'] = msg
res['event'].set()
res['result'] = msg
res['event'].set()
except TypeError:
if request_type == None:
raise
await request_hook(request_type(**msg))
case {
'method': _,
'params': _
}:
if request_hook:
await request_hook(request_type(**msg))
n.start_soon(recv_task)