set rocksdb block cache to 4gb

cberner · Aug 4, 2024 · 5b1d50c · 5b1d50c
1 parent 3ca4467
commit 5b1d50c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/benches/lmdb_benchmark.rs b/benches/lmdb_benchmark.rs
@@ -307,7 +307,15 @@ fn main() {
 
     let rocksdb_results = {
         let tmpfile: TempDir = tempfile::tempdir_in(&tmpdir).unwrap();
-        let db = rocksdb::TransactionDB::open_default(tmpfile.path()).unwrap();
+
+        let mut bb = rocksdb::BlockBasedOptions::default();
+        bb.set_block_cache(&rocksdb::Cache::new_lru_cache(4 * 1_024 * 1_024 * 1_024));
+
+        let mut opts = rocksdb::Options::default();
+        opts.set_block_based_table_factory(&bb);
+        opts.create_if_missing(true);
+
+        let db = rocksdb::TransactionDB::open(&opts, &Default::default(), tmpfile.path()).unwrap();
         let table = RocksdbBenchDatabase::new(&db);
         benchmark(table)
     };