Added support for JSONRPC requests coming from the server side

dark_clearing_improvements
Esmeralda Gallardo 2022-10-19 09:59:29 -03:00
parent cc1694760c
commit db0b59abaa
No known key found for this signature in database
GPG Key ID: AD67FBB417CAC71B
2 changed files with 26 additions and 11 deletions

View File

@ -95,9 +95,14 @@ class JSONRPCResult(Struct):
error: Optional[dict] = None
usIn: int
usOut: int
usDiff: int
usDiff: int
testnet: bool
class JSONRPCChannel(Struct):
jsonrpc: str = '2.0'
method: str
params: dict
class KLinesResult(Struct):
close: list[float]

View File

@ -174,7 +174,9 @@ class JSONRPCResult(Struct):
async def open_jsonrpc_session(
url: str,
start_id: int = 0,
dtype: type = JSONRPCResult
response_type: type = JSONRPCResult,
request_type: Optional[type] = None,
request_hook: Optional[Callable] = None
) -> Callable[[str, dict], dict]:
async with (
@ -221,18 +223,26 @@ async def open_jsonrpc_session(
field, then sets the event to wakeup original sender tasks.
'''
async for msg in ws:
msg = dtype(**msg)
try:
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))
n.start_soon(recv_task)