feat: add config to allow selecting to skip or not the pre-flight (#34)

Author: GabrielePicco

vrf-oracle/src/args.rs

@@ -20,4 +20,8 @@ pub struct Args {
 
     #[arg(long, env = "VRF_ORACLE_HTTP_PORT")]
     pub http_port: Option<u16>,
+
+    // Whether to skip preflight when submitting transactions
+    #[arg(long, env = "VRF_ORACLE_SKIP_PREFLIGHT", default_value_t = true)]
+    pub skip_preflight: bool,
 }

vrf-oracle/src/main.rs

@@ -75,6 +75,7 @@ async fn main() -> Result<()> {
         args.websocket_url,
         args.laserstream_endpoint,
         args.laserstream_api_key,
+        args.skip_preflight,
     ));
 
     // Start minimal HTTP server exposing /stats

vrf-oracle/src/oracle/client.rs

@@ -48,6 +48,8 @@ pub struct OracleClient {
     pub response_counts: Arc<RwLock<HashMap<String, u64>>>,
     // In-flight requests per queue: request_id -> enqueue slot
     pub inflight_requests: Arc<RwLock<InflightRequestsMap>>,
+    // Whether to skip preflight when sending transactions
+    pub skip_preflight: bool,
 }
 
 #[async_trait]
@@ -63,6 +65,7 @@ impl OracleClient {
         websocket_url: String,
         laserstream_endpoint: Option<String>,
         laserstream_api_key: Option<String>,
+        skip_preflight: bool,
     ) -> Self {
         let (oracle_vrf_sk, oracle_vrf_pk) = generate_vrf_keypair(&keypair);
         Self {
@@ -77,6 +80,7 @@ impl OracleClient {
             avg_response_slots: Arc::new(RwLock::new(HashMap::new())),
             response_counts: Arc::new(RwLock::new(HashMap::new())),
             inflight_requests: Arc::new(RwLock::new(HashMap::new())),
+            skip_preflight,
         }
     }
 

vrf-oracle/src/oracle/processor.rs

@@ -192,7 +192,7 @@ pub async fn process_oracle_queue(
                                         Err(_) => {
                                             warn!("Failed to parse signature");
                                             attempts += 1;
-                                            sleep(Duration::from_millis(10 * attempts)).await;
+                                            sleep(Duration::from_millis(20 * attempts)).await;
                                             continue;
                                         }
                                     };
@@ -202,6 +202,7 @@ pub async fn process_oracle_queue(
                                     Err(_) => {
                                         warn!("Failed to confirm transaction");
                                         attempts += 1;
+                                        sleep(Duration::from_millis(20 * attempts)).await;
                                         continue;
                                     }
                                 };
@@ -211,12 +212,14 @@ pub async fn process_oracle_queue(
                                     break;
                                 } else {
                                     warn!("Transaction failed");
+                                    sleep(Duration::from_millis(20 * attempts)).await;
                                     attempts += 1;
                                 }
                             }
                             Err(e) => {
                                 error!("Failed to send transaction: {e:?}");
                                 attempts += 1;
+                                sleep(Duration::from_millis(20 * attempts)).await;
                                 blockhash_cache.refresh_blockhash().await;
                             }
                         }
@@ -301,7 +304,7 @@ impl ProcessableItem {
             .send_transaction_with_config(
                 &tx,
                 RpcSendTransactionConfig {
-                    skip_preflight: true,
+                    skip_preflight: oracle_client.skip_preflight,
                     preflight_commitment: Some(
                         solana_sdk::commitment_config::CommitmentLevel::Processed,
                     ),