adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

split initial data into separate requests

Browse Source
This commit is contained in:
Timothy Pogue
2022-09-01 23:52:13 -06:00
parent dccde88c83
commit eb4cd6ba82
4 changed files with 35 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
freqtrade/rpc/api_server/api_ws.py
View File

@@ -33,7 +33,7 @@ async def _process_consumer_request(
return
if not isinstance(data, list):
logger.error(f"Improper request from channel: {channel} - {request}")
logger.error(f"Improper subscribe request from channel: {channel} - {request}")
return
# If all topics passed are a valid RPCMessageType, set subscriptions on channel
@@ -42,19 +42,19 @@ async def _process_consumer_request(
logger.debug(f"{channel} subscribed to topics: {data}")
channel.set_subscriptions(data)
elif type == RPCRequestType.INITIAL_DATA:
# Acquire the data
initial_data = rpc._ws_initial_data()
elif type == RPCRequestType.WHITELIST:
# They requested the whitelist
whitelist = rpc._ws_request_whitelist()
# We now loop over it sending it in pieces
whitelist_data, analyzed_df = initial_data.get('whitelist'), initial_data.get('analyzed_df')
await channel.send({"type": RPCMessageType.WHITELIST, "data": whitelist})
if whitelist_data:
await channel.send({"type": RPCMessageType.WHITELIST, "data": whitelist_data})
elif type == RPCRequestType.ANALYZED_DF:
# They requested the full historical analyzed dataframes
analyzed_df = rpc._ws_request_analyzed_df()
if analyzed_df:
for pair, message in analyzed_df.items():
await channel.send({"type": RPCMessageType.ANALYZED_DF, "data": message})
# For every dataframe, send as a separate message
for _, message in analyzed_df.items():
await channel.send({"type": RPCMessageType.ANALYZED_DF, "data": message})
@router.websocket("/message/ws")