Merge pull request #249 from eval-exec/jemalloc

Use tikv-jemallocator on non msvc platform and set rocksdb options

Author: quake

Cargo.lock

@@ -33,6 +33,9 @@ rocksdb = { package = "ckb-rocksdb", version = "=0.21.1", features = [
 env_logger = "0.11"
 anyhow = "1.0.56"
 
+[target.'cfg(not(target_env = "msvc"))'.dependencies]
+tikv-jemallocator = "0.6"
+
 [dev-dependencies]
 rand = "0.8"
 serde_json = "1.0"

light-client-bin/Cargo.toml

@@ -9,6 +9,10 @@ mod tests;
 use cli::AppConfig;
 use env_logger::{Builder, Env, Target};
 
+#[cfg(not(target_env = "msvc"))]
+#[global_allocator]
+static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
+
 fn main() -> anyhow::Result<()> {
     let mut builder = Builder::from_env(Env::default());
     builder.target(Target::Stdout);

light-client-bin/src/main.rs

@@ -21,7 +21,7 @@ use ckb_types::{
 use rocksdb::{
     ops::{Delete, GetPinned},
     prelude::{Get, Iterate, Open, Put, WriteOps},
-    DBPinnableSlice, Direction, IteratorMode, Snapshot, WriteBatch, DB,
+    DBPinnableSlice, Direction, IteratorMode, Options, Snapshot, WriteBatch, DB,
 };
 use std::{
     collections::{HashMap, HashSet},
@@ -52,7 +52,12 @@ pub struct Storage {
 
 impl Storage {
     pub fn new<P: AsRef<Path>>(path: P) -> Self {
-        let db = Arc::new(DB::open_default(path).expect("Failed to open rocksdb"));
+        let mut opts = Options::default();
+        opts.create_if_missing(true);
+        opts.set_max_total_wal_size(128 * 1024 * 1024);
+        opts.set_write_buffer_size(128 * 1024 * 1024);
+        opts.set_max_write_buffer_number(2);
+        let db = Arc::new(DB::open(&opts, path).expect("Failed to open rocksdb"));
         Self { db }
     }