This commit is contained in:
Dobromir Popov
2024-11-11 22:38:05 +02:00
parent efd7d8df5a
commit 5851af8f80
8 changed files with 115 additions and 199 deletions

View File

@ -84,7 +84,18 @@ class SolanaWS:
self.on_message = on_message
self.websocket = None
self.last_msg_responded = False
async def save_log(log):
try:
os.makedirs('./logs', exist_ok=True)
timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S_%f")
filename = f"./logs/ws_response_{timestamp}.json"
with open(filename, 'w') as f:
json.dump(log, f, indent=2)
except Exception as e:
logging.error(f"Error saving RPC log: {e}")
async def connect(self):
while True:
try:
@ -116,7 +127,7 @@ class SolanaWS:
response = await self.websocket.recv()
response_data = json.loads(response)
self.last_msg_responded = True
await self.save_log(response_data)
if 'result' in response_data:
return response_data['result']
elif 'error' in response_data:
@ -158,7 +169,7 @@ class SolanaWS:
async def receive_messages(self, one = False):
while True:
try:
response = await self.websocket.recv()
response_data = json.loads(response)
self.last_msg_responded = True