Merge branch 'benchmark-store' into 'main'

feat: add programs to benchmark server chunk storage

See merge request larswirzenius/obnam!23

1 files changed, 28 insertions, 0 deletions

diff --git a/src/bin/benchmark-indexedstore.rs b/src/bin/benchmark-indexedstore.rs
new file mode 100644
index 0000000..a4191ac
--- /dev/null
+++ b/src/bin/benchmark-indexedstore.rs
@@ -0,0 +1,28 @@
+use obnam::benchmark::ChunkGenerator;
+use obnam::indexedstore::IndexedStore;
+use std::path::PathBuf;
+use structopt::StructOpt;
+
+#[derive(Debug, StructOpt)]
+#[structopt(name = "benchmark-store", about = "Benhcmark the store without HTTP")]
+struct Opt {
+    #[structopt(parse(from_os_str))]
+    chunks: PathBuf,
+
+    #[structopt()]
+    num: u32,
+}
+
+fn main() -> anyhow::Result<()> {
+    pretty_env_logger::init();
+
+    let opt = Opt::from_args();
+    let gen = ChunkGenerator::new(opt.num);
+
+    let mut store = IndexedStore::new(&opt.chunks);
+    for (_, _, meta, chunk) in gen {
+        store.save(&meta, &chunk)?;
+    }
+
+    Ok(())
+}