Просмотр исходного кода

feat(策略): 添加TransactionApi并增加客户端连接检查

在策略初始化时添加TransactionApi客户端,并增加对SignerClient连接状态的检查,确保客户端连接正常后再继续执行策略
skyfffire 1 неделя назад
Родитель
Сommit
914f788d72
1 измененных файлов с 7 добавлено и 0 удалено
  1. 7 0
      src/record/strategy.py

+ 7 - 0
src/record/strategy.py

@@ -43,6 +43,7 @@ class TradingStrategy:
         self.account_index = 281474976643718
         self.api_client = lighter.ApiClient()
         self.account_api = lighter.AccountApi(self.api_client)
+        self.transaction_api = lighter.TransactionApi(self.api_client)
         self.signer_client = lighter.SignerClient(  
             url='https://mainnet.zklighter.elliot.ai',  
             private_key='0xf3625c4662ab0b338e405f61b7555e90aeda8fa28dd607588c9e275dc6f326ddcbd9341e18ca2950',  
@@ -50,6 +51,12 @@ class TradingStrategy:
             api_key_index=0
         )
         
+        # Check client connection
+        err = self.signer_client.check_client()
+        if err is not None:
+            logger.error(f"SignerClient CheckClient error: {trim_exception(err)}")
+            return
+        
         logger.info("策略初始化完成,当前状态: WAITING_INIT")
     
     async def do_strategy(self, market_data):