add criterion boilerplate

connector/Cargo.toml

@@ -8,6 +8,10 @@ description = "Listen to Solana account updates via geyser or websockets"
 
 [lib]
 
+[[bench]]
+name = "chaindata_bench"
+harness = false
+
 [features]
 default = []
 
@@ -51,3 +55,5 @@ yellowstone-grpc-proto = { workspace = true }
 
 [dev-dependencies]
 clap = { workspace = true }
+# cannot upgrade due to clap_lex requirement MSRV 1.74
+criterion = "0.4.0"

connector/benches/chaindata_bench.rs

@@ -0,0 +1,44 @@
+use criterion::{criterion_group, criterion_main, Criterion};
+use mango_feeds_connector::chain_data::{update_slotvec_logic, AccountData};
+use solana_sdk::account::AccountSharedData;
+use solana_sdk::pubkey::Pubkey;
+use std::hint::black_box;
+
+fn given_v1235(dummy_account_data: AccountSharedData) -> Vec<AccountData> {
+    vec![
+        AccountData {
+            slot: 10,
+            write_version: 10010,
+            account: dummy_account_data.clone(),
+        },
+        AccountData {
+            slot: 20,
+            write_version: 10020,
+            account: dummy_account_data.clone(),
+        },
+        AccountData {
+            slot: 30,
+            write_version: 10030,
+            account: dummy_account_data.clone(),
+        },
+        // no 40
+        AccountData {
+            slot: 50,
+            write_version: 10050,
+            account: dummy_account_data.clone(),
+        },
+    ]
+}
+
+fn criterion_benchmark(c: &mut Criterion) {
+    // overwrite if an entry for the slot already exists, otherwise insert
+    let dummy_account_data = AccountSharedData::new(99999999, 999999, &Pubkey::new_unique());
+    let v = given_v1235(dummy_account_data);
+
+    c.bench_function("update_slotvec_logic", |b| {
+        b.iter(|| update_slotvec_logic(black_box(&v), 40, 10040))
+    });
+}
+
+criterion_group!(benches, criterion_benchmark);
+criterion_main!(benches);

connector/src/chain_data.rs

@@ -1,6 +1,6 @@
 use crate::chain_data::SlotVectorEffect::*;
-use log::trace;
-use smallvec::{SmallVec, smallvec};
+use log::{info, trace};
+use smallvec::{smallvec, SmallVec};
 use solana_sdk::clock::Slot;
 use {
     solana_sdk::account::{AccountSharedData, ReadableAccount},
@@ -184,7 +184,8 @@ impl ChainData {
                 v.insert(smallvec![account]);
             }
             Entry::Occupied(o) => {
-                let v_effect = update_slotvec_logic(o.get().as_slice(), account.slot, account.write_version);
+                let v_effect =
+                    update_slotvec_logic(o.get().as_slice(), account.slot, account.write_version);
 
                 let v = o.into_mut();