mirror of
https://github.com/hl-archive-node/nanoreth.git
synced 2025-12-06 10:59:55 +00:00
5c55c57c5e
* feat(db): Add mdbx-rs apache licenced code 55e234 * feat(db): replace mdbx with reth-mdbx, metadata changes * chore(db): bump mdbx-sys to 0.12.1
27 lines
683 B
Rust
27 lines
683 B
Rust
use reth_libmdbx::{Environment, NoWriteMap, WriteFlags};
|
|
use tempfile::{tempdir, TempDir};
|
|
|
|
pub fn get_key(n: u32) -> String {
|
|
format!("key{}", n)
|
|
}
|
|
|
|
pub fn get_data(n: u32) -> String {
|
|
format!("data{}", n)
|
|
}
|
|
|
|
pub fn setup_bench_db(num_rows: u32) -> (TempDir, Environment<NoWriteMap>) {
|
|
let dir = tempdir().unwrap();
|
|
let env = Environment::new().open(dir.path()).unwrap();
|
|
|
|
{
|
|
let txn = env.begin_rw_txn().unwrap();
|
|
let db = txn.open_db(None).unwrap();
|
|
for i in 0..num_rows {
|
|
txn.put(&db, &get_key(i), &get_data(i), WriteFlags::empty())
|
|
.unwrap();
|
|
}
|
|
txn.commit().unwrap();
|
|
}
|
|
(dir, env)
|
|
}
|