Replaced try/except block in recv_task() by match msg, and added new changes to description comment
parent
db0b59abaa
commit
3f5dec82ed
|
@ -221,9 +221,13 @@ 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:
|
||||
match msg:
|
||||
case {
|
||||
'result': _
|
||||
}:
|
||||
msg = response_type(**msg)
|
||||
|
||||
if msg.id not in rpc_results:
|
||||
|
@ -237,14 +241,15 @@ async def open_jsonrpc_session(
|
|||
res['result'] = msg
|
||||
res['event'].set()
|
||||
|
||||
except TypeError:
|
||||
if request_type == None:
|
||||
raise
|
||||
case {
|
||||
'method': _,
|
||||
'params': _
|
||||
}:
|
||||
|
||||
if request_hook:
|
||||
await request_hook(request_type(**msg))
|
||||
|
||||
|
||||
|
||||
|
||||
n.start_soon(recv_task)
|
||||
yield json_rpc
|
||||
n.cancel_scope.cancel()
|
||||
|
|
Loading…
Reference in New Issue