feat: add transaction confirmation wait option and parallel execution optimization

- Add wait_transaction_confirmed parameter to all trading methods
- Support async transaction sending without waiting for confirmation for better performance
- Optimize parallel execution logic to return on first successful transaction
- Fix direct priority_fee modification issue
- Improve error handling and resource management
- Update all example code to support new parameter

src/trading/core/executor.rs

@@ -76,7 +76,12 @@ impl TradeExecutor for GenericTradeExecutor {
         timer.stage("rpc提交确认");
 
         // 发送交易
-        rpc.send_and_confirm_transaction(&transaction).await?;
+        if params.wait_transaction_confirmed {
+            rpc.send_and_confirm_transaction(&transaction).await?;
+        } else {
+            // 异步发送交易
+            rpc.send_transaction(&transaction).await?;
+        }
         timer.finish();
 
         Ok(())
@@ -104,6 +109,7 @@ impl TradeExecutor for GenericTradeExecutor {
             lookup_table_key: params.lookup_table_key,
             recent_blockhash: params.recent_blockhash,
             data_size_limit: params.data_size_limit,
+            wait_transaction_confirmed: params.wait_transaction_confirmed,
             protocol_params: params.protocol_params.clone(),
         };
 
@@ -134,6 +140,7 @@ impl TradeExecutor for GenericTradeExecutor {
             middleware_manager,
             self.protocol_name.to_string(),
             true,
+            params.wait_transaction_confirmed,
         )
         .await?;
 
@@ -179,7 +186,11 @@ impl TradeExecutor for GenericTradeExecutor {
         timer.stage("卖出交易签名");
 
         // 发送交易
-        rpc.send_and_confirm_transaction(&transaction).await?;
+        if params.wait_transaction_confirmed {
+            rpc.send_and_confirm_transaction(&transaction).await?;
+        } else {
+            rpc.send_transaction(&transaction).await?;
+        }
         timer.finish();
 
         Ok(())
@@ -203,6 +214,7 @@ impl TradeExecutor for GenericTradeExecutor {
             priority_fee: params.priority_fee.clone(),
             lookup_table_key: params.lookup_table_key,
             recent_blockhash: params.recent_blockhash,
+            wait_transaction_confirmed: params.wait_transaction_confirmed,
             protocol_params: params.protocol_params.clone(),
         };
 
@@ -233,6 +245,7 @@ impl TradeExecutor for GenericTradeExecutor {
             middleware_manager,
             self.protocol_name.to_string(),
             false,
+            params.wait_transaction_confirmed,
         )
         .await?;
 

src/trading/core/parallel.rs

@@ -2,6 +2,7 @@ use anyhow::{anyhow, Result};
 use solana_hash::Hash;
 use solana_sdk::{instruction::Instruction, pubkey::Pubkey, signature::Keypair};
 use std::{str::FromStr, sync::Arc};
+use tokio::sync::mpsc;
 use tokio::task::JoinHandle;
 
 use crate::{
@@ -30,6 +31,7 @@ pub async fn parallel_execute_with_tips(
     middleware_manager: Option<Arc<MiddlewareManager>>,
     protocol_name: String,
     is_buy: bool,
+    wait_transaction_confirmed: bool,
 ) -> Result<()> {
     let cores = core_affinity::get_core_ids().unwrap();
     let mut handles: Vec<JoinHandle<Result<()>>> = vec![];
@@ -97,7 +99,7 @@ pub async fn parallel_execute_with_tips(
             } else {
                 let tip_account = swqos_client.get_tip_account()?;
                 let tip_account = Arc::new(Pubkey::from_str(&tip_account).map_err(|e| anyhow!(e))?);
-                priority_fee.buy_tip_fee = priority_fee.buy_tip_fees[i];
+                priority_fee.buy_tip_fee = priority_fee.buy_tip_fees[i % priority_fee.buy_tip_fees.len()];
 
                 build_tip_transaction_with_priority_fee(
                     payer,
@@ -125,22 +127,36 @@ pub async fn parallel_execute_with_tips(
         handles.push(handle);
     }
 
-    // 等待所有任务完成
-    let mut errors = Vec::new();
+    // 任意一个成功即返回
+    let (tx, mut rx) = mpsc::channel(swqos_clients.len());
+
+    // 启动监听任务
     for handle in handles {
-        match handle.await {
-            Ok(Ok(_)) => (),
+        let tx = tx.clone();
+        tokio::spawn(async move {
+            let result = handle.await;
+            let _ = tx.send(result).await;
+        });
+    }
+    drop(tx); // 关闭发送端
+
+    // 等待第一个成功的结果
+    let mut errors = Vec::new();
+
+    if !wait_transaction_confirmed {
+        return Ok(());
+    }
+
+    while let Some(result) = rx.recv().await {
+        match result {
+            Ok(Ok(_)) => {
+                return Ok(());
+            }
             Ok(Err(e)) => errors.push(format!("Task error: {}", e)),
             Err(e) => errors.push(format!("Join error: {}", e)),
         }
     }
 
-    if !errors.is_empty() {
-        for error in &errors {
-            println!("{}", error);
-        }
-        return Err(anyhow!("Some tasks failed: {:?}", errors));
-    }
-
-    Ok(())
+    // 如果没有成功的，返回错误
+    return Err(anyhow!("所有交易都失败了: {:?}", errors));
 }

src/trading/core/params.rs

@@ -36,6 +36,7 @@ pub struct BuyParams {
     pub lookup_table_key: Option<Pubkey>,
     pub recent_blockhash: Hash,
     pub data_size_limit: u32,
+    pub wait_transaction_confirmed: bool,
     pub protocol_params: Box<dyn ProtocolParams>,
 }
 
@@ -54,6 +55,7 @@ pub struct BuyWithTipParams {
     pub lookup_table_key: Option<Pubkey>,
     pub recent_blockhash: Hash,
     pub data_size_limit: u32,
+    pub wait_transaction_confirmed: bool,
     pub protocol_params: Box<dyn ProtocolParams>,
 }
 
@@ -70,6 +72,7 @@ pub struct SellParams {
     pub priority_fee: PriorityFee,
     pub lookup_table_key: Option<Pubkey>,
     pub recent_blockhash: Hash,
+    pub wait_transaction_confirmed: bool,
     pub protocol_params: Box<dyn ProtocolParams>,
 }
 
@@ -87,6 +90,7 @@ pub struct SellWithTipParams {
     pub priority_fee: PriorityFee,
     pub lookup_table_key: Option<Pubkey>,
     pub recent_blockhash: Hash,
+    pub wait_transaction_confirmed: bool,
     pub protocol_params: Box<dyn ProtocolParams>,
 }
 
@@ -456,6 +460,7 @@ impl BuyParams {
             lookup_table_key: self.lookup_table_key,
             recent_blockhash: self.recent_blockhash,
             data_size_limit: self.data_size_limit,
+            wait_transaction_confirmed: self.wait_transaction_confirmed,
             protocol_params: self.protocol_params,
         }
     }
@@ -476,6 +481,7 @@ impl SellParams {
             priority_fee: self.priority_fee,
             lookup_table_key: self.lookup_table_key,
             recent_blockhash: self.recent_blockhash,
+            wait_transaction_confirmed: self.wait_transaction_confirmed,
             protocol_params: self.protocol_params,
         }
     }