From f15f180a07e2a3375b7f6fd6de235ed5d7b4154a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?M=C3=A5rten=20Blankfors?= <marten@blankfors.se>
Date: Sat, 9 Nov 2024 15:19:29 +0700
Subject: [PATCH] feat: simple bench target for measuring transaction batch
 lookup times

---
 Cargo.lock                                |   1 +
 benches/Cargo.toml                        |   5 +
 benches/benches/end_to_end_query_times.rs | 124 ++++++++++++++++++++++
 3 files changed, 130 insertions(+)
 create mode 100644 benches/benches/end_to_end_query_times.rs

diff --git a/Cargo.lock b/Cargo.lock
index 67a97f72c3e..33c27003ca7 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -3263,6 +3263,7 @@ dependencies = [
  "ethnum",
  "fuel-core",
  "fuel-core-chain-config",
+ "fuel-core-client",
  "fuel-core-database",
  "fuel-core-services",
  "fuel-core-storage",
diff --git a/benches/Cargo.toml b/benches/Cargo.toml
index e4ee7db927b..d2546fd5876 100644
--- a/benches/Cargo.toml
+++ b/benches/Cargo.toml
@@ -23,6 +23,7 @@ fuel-core = { path = "../crates/fuel-core", default-features = false, features =
   "rocksdb-production",
 ] }
 fuel-core-chain-config = { workspace = true }
+fuel-core-client = { path = "./../crates/client" }
 fuel-core-database = { path = "./../crates/database" }
 fuel-core-services = { path = "./../crates/services" }
 fuel-core-storage = { path = "./../crates/storage", features = ["smt"] }
@@ -76,3 +77,7 @@ name = "transaction_throughput"
 [[bench]]
 harness = false
 name = "db_lookup_times"
+
+[[bench]]
+harness = false
+name = "end_to_end_query_times"
diff --git a/benches/benches/end_to_end_query_times.rs b/benches/benches/end_to_end_query_times.rs
new file mode 100644
index 00000000000..7b0e2a16760
--- /dev/null
+++ b/benches/benches/end_to_end_query_times.rs
@@ -0,0 +1,124 @@
+use fuel_core::{
+    combined_database::CombinedDatabase,
+    fuel_core_graphql_api::database::ReadDatabase,
+};
+
+use fuel_core_storage::{
+    tables::{
+        FuelBlocks,
+        Transactions,
+    },
+    transactional::WriteTransaction,
+    StorageAsMut,
+};
+use fuel_core_types::{
+    blockchain::block::CompressedBlock,
+    fuel_tx::{
+        TxId,
+        UniqueIdentifier,
+    },
+    fuel_types::{
+        BlockHeight,
+        ChainId,
+    },
+};
+use rand::{
+    rngs::StdRng,
+    SeedableRng,
+};
+
+#[tokio::main]
+async fn main() -> anyhow::Result<()> {
+    println!("Setting up bench harness.");
+    let mut harness = Harness::new(StdRng::seed_from_u64(2322)).await?;
+
+    println!("Populating storage with transactions.");
+    let tx_ids = harness.populate_database().await?;
+
+    println!("Querying transactions from storage.");
+    harness.query_database_many_times(&tx_ids).await?;
+
+    Ok(())
+}
+
+struct Harness<Rng> {
+    rng: Rng,
+    params: Parameters,
+    db: CombinedDatabase,
+}
+
+impl<Rng: rand::RngCore + rand::CryptoRng + Send> Harness<Rng> {
+    async fn new(rng: Rng) -> anyhow::Result<Self> {
+        let params = Parameters::hard_coded();
+        let db = CombinedDatabase::default();
+
+        Ok(Self { rng, params, db })
+    }
+
+    async fn populate_database(&mut self) -> anyhow::Result<Vec<TxId>> {
+        let mut tx_ids = Vec::new();
+
+        for block_height in 0..self.params.num_blocks {
+            let block_height = BlockHeight::from(block_height as u32);
+            let mut compressed_block = CompressedBlock::default();
+            compressed_block.header_mut().set_block_height(block_height);
+
+            let mut transaction = self.db.on_chain_mut().write_transaction();
+
+            transaction
+                .storage::<FuelBlocks>()
+                .insert(&block_height, &compressed_block)
+                .unwrap();
+
+            for tx in (0..self.params.tx_count_per_block).map(|i| {
+                let script_gas_limit = 26; // Cost of OP_RET * 2
+                test_helpers::make_tx(&mut self.rng, i, script_gas_limit)
+            }) {
+                let tx_id = tx.id(&ChainId::default());
+                transaction
+                    .storage::<Transactions>()
+                    .insert(&tx_id, &tx)
+                    .unwrap();
+
+                tx_ids.push(tx_id);
+            }
+
+            transaction.commit().unwrap();
+        }
+
+        Ok(tx_ids)
+    }
+
+    async fn query_database_many_times(&mut self, tx_ids: &[TxId]) -> anyhow::Result<()> {
+        let read_database = ReadDatabase::new(
+            0,
+            BlockHeight::new(0),
+            self.db.on_chain().clone(),
+            self.db.off_chain().clone(),
+        );
+
+        let read_view = read_database.view()?;
+
+        for _ in 0..self.params.num_queries {
+            let _ = read_view.transactions(tx_ids.to_vec()).await;
+        }
+
+        Ok(())
+    }
+}
+
+struct Parameters {
+    num_queries: usize,
+    num_blocks: usize,
+    tx_count_per_block: u64,
+}
+
+impl Parameters {
+    fn hard_coded() -> Self {
+        Self {
+            num_queries: 100,
+            num_blocks: 100,
+            tx_count_per_block: 1000,
+        }
+    }
+}