feat(swqos): add Stellium support and update minimum tip requirements

src/swqos/mod.rs

@@ -10,6 +10,7 @@ pub mod node1;
 pub mod flashblock;
 pub mod blockrazor;
 pub mod astralane;
+pub mod stellium;
 
 use std::sync::Arc;
 
@@ -20,29 +21,31 @@ use tokio::sync::RwLock;
 use anyhow::Result;
 
 use crate::{
-    common::SolanaRpcClient, 
+    common::SolanaRpcClient,
     constants::swqos::{
-        SWQOS_ENDPOINTS_BLOX, 
-        SWQOS_ENDPOINTS_JITO, 
-        SWQOS_ENDPOINTS_NEXTBLOCK, 
-        SWQOS_ENDPOINTS_TEMPORAL, 
-        SWQOS_ENDPOINTS_ZERO_SLOT, 
-        SWQOS_ENDPOINTS_NODE1, 
+        SWQOS_ENDPOINTS_BLOX,
+        SWQOS_ENDPOINTS_JITO,
+        SWQOS_ENDPOINTS_NEXTBLOCK,
+        SWQOS_ENDPOINTS_TEMPORAL,
+        SWQOS_ENDPOINTS_ZERO_SLOT,
+        SWQOS_ENDPOINTS_NODE1,
         SWQOS_ENDPOINTS_FLASHBLOCK,
         SWQOS_ENDPOINTS_BLOCKRAZOR,
-        SWQOS_ENDPOINTS_ASTRALANE
-    }, 
+        SWQOS_ENDPOINTS_ASTRALANE,
+        SWQOS_ENDPOINTS_STELLIUM
+    },
     swqos::{
-        bloxroute::BloxrouteClient, 
-        jito::JitoClient, 
-        nextblock::NextBlockClient, 
-        solana_rpc::SolRpcClient, 
-        temporal::TemporalClient, 
-        zeroslot::ZeroSlotClient, 
-        node1::Node1Client, 
+        bloxroute::BloxrouteClient,
+        jito::JitoClient,
+        nextblock::NextBlockClient,
+        solana_rpc::SolRpcClient,
+        temporal::TemporalClient,
+        zeroslot::ZeroSlotClient,
+        node1::Node1Client,
         flashblock::FlashBlockClient,
         blockrazor::BlockRazorClient,
-        astralane::AstralaneClient
+        astralane::AstralaneClient,
+        stellium::StelliumClient
     }
 };
 
@@ -81,6 +84,7 @@ pub enum SwqosType {
     FlashBlock,
     BlockRazor,
     Astralane,
+    Stellium,
     Default,
 }
 
@@ -96,6 +100,7 @@ impl SwqosType {
             Self::FlashBlock,
             Self::BlockRazor,
             Self::Astralane,
+            Self::Stellium,
             Self::Default,
         ]
     }
@@ -144,6 +149,8 @@ pub enum SwqosConfig {
     BlockRazor(String, SwqosRegion, Option<String>),
     /// Astralane(api_token, region, custom_url)
     Astralane(String, SwqosRegion, Option<String>),
+    /// Stellium(api_token, region, custom_url)
+    Stellium(String, SwqosRegion, Option<String>),
 }
 
 impl SwqosConfig {
@@ -151,7 +158,7 @@ impl SwqosConfig {
         if let Some(custom_url) = url {
             return custom_url;
         }
-        
+
         match swqos_type {
             SwqosType::Jito => SWQOS_ENDPOINTS_JITO[region as usize].to_string(),
             SwqosType::NextBlock => SWQOS_ENDPOINTS_NEXTBLOCK[region as usize].to_string(),
@@ -162,6 +169,7 @@ impl SwqosConfig {
             SwqosType::FlashBlock => SWQOS_ENDPOINTS_FLASHBLOCK[region as usize].to_string(),
             SwqosType::BlockRazor => SWQOS_ENDPOINTS_BLOCKRAZOR[region as usize].to_string(),
             SwqosType::Astralane => SWQOS_ENDPOINTS_ASTRALANE[region as usize].to_string(),
+            SwqosType::Stellium => SWQOS_ENDPOINTS_STELLIUM[region as usize].to_string(),
             SwqosType::Default => "".to_string(),
         }
     }
@@ -249,11 +257,20 @@ impl SwqosConfig {
                 );
                 Arc::new(astralane_client)
             },
+            SwqosConfig::Stellium(auth_token, region, url) => {
+                let endpoint = SwqosConfig::get_endpoint(SwqosType::Stellium, region, url);
+                let stellium_client = StelliumClient::new(
+                    rpc_url.clone(),
+                    endpoint.to_string(),
+                    auth_token
+                );
+                Arc::new(stellium_client)
+            },
             SwqosConfig::Default(endpoint) => {
                 let rpc = SolanaRpcClient::new_with_commitment(
                     endpoint,
                     commitment
-                );   
+                );
                 let rpc_client = SolRpcClient::new(Arc::new(rpc));
                 Arc::new(rpc_client)
             }

src/swqos/stellium.rs

@@ -0,0 +1,190 @@
+use crate::swqos::common::{poll_transaction_confirmation, serialize_transaction_and_encode};
+use rand::seq::IndexedRandom;
+use reqwest::Client;
+use serde_json::json;
+use std::{sync::Arc, time::Instant};
+use std::sync::atomic::{AtomicBool, Ordering};
+
+use std::time::Duration;
+use solana_transaction_status::UiTransactionEncoding;
+
+use anyhow::Result;
+use solana_sdk::transaction::VersionedTransaction;
+use crate::swqos::{SwqosType, TradeType};
+use crate::swqos::SwqosClientTrait;
+
+use crate::{common::SolanaRpcClient, constants::swqos::STELLIUM_TIP_ACCOUNTS};
+
+
+#[derive(Clone)]
+pub struct StelliumClient {
+    pub endpoint: String,
+    pub auth_token: String,
+    pub rpc_client: Arc<SolanaRpcClient>,
+    pub http_client: Client,
+    keep_alive_running: Arc<AtomicBool>,
+}
+
+#[async_trait::async_trait]
+impl SwqosClientTrait for StelliumClient {
+    async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction) -> Result<()> {
+        self.send_transaction(trade_type, transaction).await
+    }
+
+    async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>) -> Result<()> {
+        self.send_transactions(trade_type, transactions).await
+    }
+
+    fn get_tip_account(&self) -> Result<String> {
+        let tip_account = *STELLIUM_TIP_ACCOUNTS.choose(&mut rand::rng()).or_else(|| STELLIUM_TIP_ACCOUNTS.first()).unwrap();
+        Ok(tip_account.to_string())
+    }
+
+    fn get_swqos_type(&self) -> SwqosType {
+        SwqosType::Stellium
+    }
+}
+
+impl StelliumClient {
+    pub fn new(rpc_url: String, endpoint: String, auth_token: String) -> Self {
+        let rpc_client = SolanaRpcClient::new(rpc_url);
+        let http_client = Client::builder()
+            // Optimized connection pool settings for high performance
+            .pool_idle_timeout(Duration::from_secs(120))
+            .pool_max_idle_per_host(256)  // Increased from 64 to 256
+            .tcp_keepalive(Some(Duration::from_secs(60)))  // Reduced from 1200 to 60
+            .tcp_nodelay(true)  // Disable Nagle's algorithm for lower latency
+            .http2_keep_alive_interval(Duration::from_secs(10))
+            .http2_keep_alive_timeout(Duration::from_secs(5))
+            .http2_adaptive_window(true)  // Enable adaptive flow control
+            .timeout(Duration::from_millis(3000))  // Reduced from 10s to 3s
+            .connect_timeout(Duration::from_millis(2000))  // Reduced from 5s to 2s
+            .build()
+            .unwrap();
+
+        let keep_alive_running = Arc::new(AtomicBool::new(true));
+
+        let client = Self {
+            rpc_client: Arc::new(rpc_client),
+            endpoint: endpoint.clone(),
+            auth_token: auth_token.clone(),
+            http_client: http_client.clone(),
+            keep_alive_running: keep_alive_running.clone(),
+        };
+
+        // Start ping task
+        let client_clone = client.clone();
+        tokio::spawn(async move {
+            client_clone.start_ping_task().await;
+        });
+
+        client
+    }
+
+    /// Start periodic ping task to keep connections active
+    async fn start_ping_task(&self) {
+        let endpoint = self.endpoint.clone();
+        let auth_token = self.auth_token.clone();
+        let http_client = self.http_client.clone();
+        let stop_ping = self.keep_alive_running.clone();
+
+        tokio::spawn(async move {
+            let mut interval = tokio::time::interval(Duration::from_secs(60)); // Ping every 60 seconds
+
+            loop {
+                interval.tick().await;
+
+                if stop_ping.load(Ordering::Relaxed) {
+                    break;
+                }
+
+                // Send ping request
+                let url = format!("{}/{}", endpoint, auth_token);
+                match http_client.get(&url).send().await {
+                    Ok(response) => {
+                        if !response.status().is_success() {
+                            eprintln!(" [Stellium] Ping failed with status: {}", response.status());
+                        }
+                    }
+                    Err(e) => {
+                        eprintln!(" [Stellium] Ping request error: {:?}", e);
+                    }
+                }
+            }
+        });
+    }
+
+    pub async fn send_transaction(&self, trade_type: TradeType, transaction: &VersionedTransaction) -> Result<()> {
+        let start_time = Instant::now();
+        let (content, signature) = serialize_transaction_and_encode(transaction, UiTransactionEncoding::Base64).await?;
+
+        // Stellium uses standard Solana sendTransaction format
+        let request_body = serde_json::to_string(&json!({
+            "jsonrpc": "2.0",
+            "id": 1,
+            "method": "sendTransaction",
+            "params": [
+                content,
+                { "encoding": "base64" }
+            ]
+        }))?;
+
+        // Build the URL with the API key
+        let url = format!("{}/{}", self.endpoint, self.auth_token);
+
+        // Send request to Stellium
+        let response_text = self.http_client.post(&url)
+            .body(request_body)
+            .header("Content-Type", "application/json")
+            .header("Connection", "keep-alive")
+            .header("Keep-Alive", "timeout=30, max=1000")
+            .send()
+            .await?
+            .text()
+            .await?;
+
+        // Parse response
+        if let Ok(response_json) = serde_json::from_str::<serde_json::Value>(&response_text) {
+            if response_json.get("result").is_some() {
+                println!(" [Stellium] {} submitted: {:?}", trade_type, start_time.elapsed());
+            } else if let Some(_error) = response_json.get("error") {
+                eprintln!(" [Stellium] {} submission failed: {:?}", trade_type, _error);
+            }
+        } else {
+            eprintln!(" [Stellium] {} submission failed: {:?}", trade_type, response_text);
+        }
+
+        let start_time: Instant = Instant::now();
+        match poll_transaction_confirmation(&self.rpc_client, signature).await {
+            Ok(_) => (),
+            Err(e) => {
+                println!(" signature: {:?}", signature);
+                println!(" [Stellium] {} confirmation failed: {:?}", trade_type, start_time.elapsed());
+                return Err(e);
+            },
+        }
+        println!(" signature: {:?}", signature);
+        println!(" [Stellium] {} confirmed: {:?}", trade_type, start_time.elapsed());
+
+        Ok(())
+    }
+
+    pub async fn send_transactions(&self, trade_type: TradeType, transactions: &Vec<VersionedTransaction>) -> Result<()> {
+        for transaction in transactions {
+            self.send_transaction(trade_type, transaction).await?;
+        }
+        Ok(())
+    }
+
+    /// Stop the ping task
+    pub fn stop_ping_task(&self) {
+        self.keep_alive_running.store(false, Ordering::Relaxed);
+    }
+}
+
+impl Drop for StelliumClient {
+    fn drop(&mut self) {
+        // Stop ping task when client is dropped
+        self.keep_alive_running.store(false, Ordering::Relaxed);
+    }
+}