Add benchmark script

.gitignore

@@ -1,6 +1,6 @@
 /.idea/
 /.vagrant
-/flamegraph.svg
+/benchmark
 /index.lmdb
 /index.redb
 /ord.log

bin/benchmark

@@ -0,0 +1,21 @@
+#!/usr/bin/env bash
+
+set -euxo pipefail
+
+rm -rf benchmark
+mkdir benchmark
+
+for HEIGHT_LIMIT in 100 250 500 1000 2000; do
+  sudo \
+    CARGO_PROFILE_RELEASE_DEBUG=true \
+    RUST_LOG=info \
+    cargo flamegraph \
+    --deterministic \
+    --bin ord \
+    --output benchmark/$HEIGHT_LIMIT.svg \
+    -- \
+    --chain signet \
+    --data-dir benchmark \
+    --height-limit $HEIGHT_LIMIT \
+    index
+done

justfile

@@ -96,3 +96,6 @@ graph:
 
 flamegraph:
   CARGO_PROFILE_RELEASE_DEBUG=true sudo cargo flamegraph -- index
+
+benchmark:
+  ./bin/benchmark

src/index.rs

@@ -17,6 +17,7 @@ pub(crate) struct Index {
   client: Client,
   database: Database,
   database_path: PathBuf,
+  height_limit: Option<Height>,
 }
 
 pub(crate) enum List {
@@ -61,6 +62,7 @@ impl Index {
       client,
       database,
       database_path,
+      height_limit: options.height_limit,
     })
   }
 
@@ -121,6 +123,12 @@ impl Index {
 
   pub(crate) fn index_ranges(&self) -> Result {
     loop {
+      if let Some(height_limit) = self.height_limit {
+        if self.height()? >= height_limit {
+          break;
+        }
+      }
+
       let mut wtx = self.database.begin_write()?;
 
       let done = self.index_block(&mut wtx)?;

src/options.rs

@@ -17,6 +17,8 @@ pub(crate) struct Options {
   data_dir: Option<PathBuf>,
   #[clap(long)]
   bitcoin_data_dir: Option<PathBuf>,
+  #[clap(long)]
+  pub(crate) height_limit: Option<Height>,
 }
 
 #[derive(ValueEnum, Copy, Clone, Debug)]

tests/index.rs

@@ -19,3 +19,19 @@ fn custom_index_size() {
     1 << 20
   );
 }
+
+#[test]
+fn height_limit() {
+  Test::new()
+    .command("find 5000000000")
+    .expected_stdout("150ba822b458a19615e70a604d8dd9d3482fc165fa4e9cc150d74e11916ce8ae:0:0\n")
+    .blocks(1)
+    .run();
+
+  Test::new()
+    .command("--height-limit 0 find 5000000000")
+    .expected_stderr("error: Ordinal has not been mined as of index height\n")
+    .expected_status(1)
+    .blocks(1)
+    .run();
+}