mod util;
use std::{fs, io::Read as _};
use rocksdb::{BlockBasedOptions, DataBlockIndexType, Options, ReadOptions, DB};
use util::DBPath;
#[test]
fn test_set_num_levels() {
let n = DBPath::new("_rust_rocksdb_test_set_num_levels");
{
let mut opts = Options::default();
opts.create_if_missing(true);
opts.set_num_levels(2);
let _db = DB::open(&opts, &n).unwrap();
}
}
#[test]
fn test_increase_parallelism() {
let n = DBPath::new("_rust_rocksdb_test_increase_parallelism");
{
let mut opts = Options::default();
opts.create_if_missing(true);
opts.increase_parallelism(4);
let _db = DB::open(&opts, &n).unwrap();
}
}
#[test]
fn test_set_level_compaction_dynamic_level_bytes() {
let n = DBPath::new("_rust_rocksdb_test_set_level_compaction_dynamic_level_bytes");
{
let mut opts = Options::default();
opts.create_if_missing(true);
opts.set_level_compaction_dynamic_level_bytes(true);
let _db = DB::open(&opts, &n).unwrap();
}
}
#[test]
fn test_block_based_options() {
let path = "_rust_rocksdb_test_block_based_options";
let n = DBPath::new(path);
{
let mut opts = Options::default();
opts.create_if_missing(true);
let mut block_opts = BlockBasedOptions::default();
block_opts.set_cache_index_and_filter_blocks(true);
block_opts.set_pin_l0_filter_and_index_blocks_in_cache(true);
block_opts.set_format_version(4);
block_opts.set_index_block_restart_interval(16);
opts.set_block_based_table_factory(&block_opts);
let _db = DB::open(&opts, &n).unwrap();
let mut rocksdb_log = fs::File::open(format!("{}/LOG", (&n).as_ref().to_str().unwrap()))
.expect("rocksdb creates a LOG file");
let mut settings = String::new();
rocksdb_log.read_to_string(&mut settings).unwrap();
assert!(settings.contains("cache_index_and_filter_blocks: 1"));
assert!(settings.contains("pin_l0_filter_and_index_blocks_in_cache: 1"));
assert!(settings.contains("format_version: 4"));
assert!(settings.contains("index_block_restart_interval: 16"));
}
}
#[test]
fn test_read_options() {
let mut read_opts = ReadOptions::default();
read_opts.set_verify_checksums(false);
}
#[test]
fn test_set_data_block_index_type() {
let path = "_rust_rocksdb_test_set_data_block_index_type";
let n = DBPath::new(path);
{
let mut opts = Options::default();
opts.create_if_missing(true);
let block_opts = BlockBasedOptions::default();
opts.set_block_based_table_factory(&block_opts);
let _db = DB::open(&opts, &n).expect("open a db works");
let mut rocksdb_log = fs::File::open(format!("{}/LOG", (&n).as_ref().to_str().unwrap()))
.expect("rocksdb creates a LOG file");
let mut settings = String::new();
rocksdb_log
.read_to_string(&mut settings)
.expect("can read the LOG file");
assert!(settings.contains("data_block_index_type: 0"));
assert!(settings.contains("data_block_hash_table_util_ratio: 0.750000"));
}
{
let mut opts = Options::default();
opts.create_if_missing(false);
let mut block_opts = BlockBasedOptions::default();
block_opts.set_data_block_index_type(DataBlockIndexType::BinaryAndHash);
block_opts.set_data_block_hash_ratio(0.35);
opts.set_block_based_table_factory(&block_opts);
let _db = DB::open(&opts, &n).expect("open a db works");
let mut rocksdb_log = fs::File::open(format!("{}/LOG", (&n).as_ref().to_str().unwrap()))
.expect("rocksdb creates a LOG file");
let mut settings = String::new();
rocksdb_log
.read_to_string(&mut settings)
.expect("can read the LOG file");
assert!(settings.contains("data_block_index_type: 1"));
assert!(settings.contains("data_block_hash_table_util_ratio: 0.350000"));
}
}