You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
153 lines
5.1 KiB
C++
153 lines
5.1 KiB
C++
// Copyright 2022, The Cozo Project Authors.
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0.
|
|
// If a copy of the MPL was not distributed with this file,
|
|
// You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
|
|
#include <iostream>
|
|
#include <memory>
|
|
#include "db.h"
|
|
#include "cozorocks/src/bridge/mod.rs.h"
|
|
#include "rocksdb/utilities/options_util.h"
|
|
|
|
Options default_db_options() {
|
|
Options options = Options();
|
|
options.bottommost_compression = kZSTD;
|
|
options.compression = kLZ4Compression;
|
|
options.level_compaction_dynamic_level_bytes = true;
|
|
options.max_background_jobs = 6;
|
|
options.bytes_per_sync = 1048576;
|
|
options.compaction_pri = kMinOverlappingRatio;
|
|
BlockBasedTableOptions table_options;
|
|
table_options.block_size = 16 * 1024;
|
|
table_options.cache_index_and_filter_blocks = true;
|
|
table_options.pin_l0_filter_and_index_blocks_in_cache = true;
|
|
table_options.format_version = 5;
|
|
|
|
auto table_factory = NewBlockBasedTableFactory(table_options);
|
|
options.table_factory.reset(table_factory);
|
|
|
|
return options;
|
|
}
|
|
|
|
ColumnFamilyOptions default_cf_options() {
|
|
ColumnFamilyOptions options = ColumnFamilyOptions();
|
|
options.bottommost_compression = kZSTD;
|
|
options.compression = kLZ4Compression;
|
|
options.level_compaction_dynamic_level_bytes = true;
|
|
options.compaction_pri = kMinOverlappingRatio;
|
|
|
|
// auto cache = NewLRUCache(128 << 20);
|
|
|
|
BlockBasedTableOptions table_options;
|
|
// table_options.block_cache = cache;
|
|
|
|
table_options.block_size = 16 * 1024;
|
|
table_options.cache_index_and_filter_blocks = true;
|
|
table_options.pin_l0_filter_and_index_blocks_in_cache = true;
|
|
table_options.format_version = 5;
|
|
|
|
auto table_factory = NewBlockBasedTableFactory(table_options);
|
|
options.table_factory.reset(table_factory);
|
|
|
|
return options;
|
|
}
|
|
|
|
shared_ptr <RocksDbBridge> open_db(const DbOpts &opts, RocksDbStatus &status) {
|
|
auto options = default_db_options();
|
|
|
|
shared_ptr<Cache> cache = nullptr;
|
|
|
|
if (opts.block_cache_size > 0) {
|
|
cache = NewLRUCache(1 * 1024 * 1024 * 1024);
|
|
}
|
|
|
|
if (!opts.options_path.empty()) {
|
|
DBOptions loaded_db_opt;
|
|
std::vector<ColumnFamilyDescriptor> loaded_cf_descs;
|
|
ConfigOptions config_options;
|
|
string options_path = string(opts.options_path);
|
|
Status s = LoadOptionsFromFile(config_options, options_path, &loaded_db_opt,
|
|
&loaded_cf_descs);
|
|
if (!s.ok()) {
|
|
write_status(s, status);
|
|
return nullptr;
|
|
}
|
|
|
|
if (cache != nullptr) {
|
|
for (size_t i = 0; i < loaded_cf_descs.size(); ++i) {
|
|
auto* loaded_bbt_opt =
|
|
loaded_cf_descs[0]
|
|
.options.table_factory->GetOptions<BlockBasedTableOptions>();
|
|
loaded_bbt_opt->block_cache = cache;
|
|
}
|
|
}
|
|
|
|
options = Options(loaded_db_opt, loaded_cf_descs[0].options);
|
|
}
|
|
|
|
if (opts.prepare_for_bulk_load) {
|
|
options.PrepareForBulkLoad();
|
|
}
|
|
if (opts.increase_parallelism > 0) {
|
|
options.IncreaseParallelism(opts.increase_parallelism);
|
|
}
|
|
if (opts.optimize_level_style_compaction) {
|
|
options.OptimizeLevelStyleCompaction();
|
|
}
|
|
options.create_if_missing = opts.create_if_missing;
|
|
options.paranoid_checks = opts.paranoid_checks;
|
|
if (opts.enable_blob_files) {
|
|
options.enable_blob_files = true;
|
|
|
|
options.min_blob_size = opts.min_blob_size;
|
|
|
|
options.blob_file_size = opts.blob_file_size;
|
|
|
|
options.enable_blob_garbage_collection = opts.enable_blob_garbage_collection;
|
|
}
|
|
if (opts.use_bloom_filter) {
|
|
BlockBasedTableOptions table_options;
|
|
table_options.filter_policy.reset(NewBloomFilterPolicy(opts.bloom_filter_bits_per_key, false));
|
|
table_options.whole_key_filtering = opts.bloom_filter_whole_key_filtering;
|
|
options.table_factory.reset(NewBlockBasedTableFactory(table_options));
|
|
}
|
|
if (opts.use_capped_prefix_extractor) {
|
|
options.prefix_extractor.reset(NewCappedPrefixTransform(opts.capped_prefix_extractor_len));
|
|
}
|
|
if (opts.use_fixed_prefix_extractor) {
|
|
options.prefix_extractor.reset(NewFixedPrefixTransform(opts.fixed_prefix_extractor_len));
|
|
}
|
|
options.create_missing_column_families = true;
|
|
|
|
shared_ptr <RocksDbBridge> db = make_shared<RocksDbBridge>();
|
|
|
|
db->db_path = string(opts.db_path);
|
|
|
|
TransactionDB *txn_db = nullptr;
|
|
write_status(
|
|
TransactionDB::Open(options, TransactionDBOptions(), db->db_path, &txn_db),
|
|
status);
|
|
db->db.reset(txn_db);
|
|
db->destroy_on_exit = opts.destroy_on_exit;
|
|
|
|
|
|
return db;
|
|
}
|
|
|
|
RocksDbBridge::~RocksDbBridge() {
|
|
if (destroy_on_exit && (db != nullptr)) {
|
|
cerr << "destroying database on exit: " << db_path << endl;
|
|
auto status = db->Close();
|
|
if (!status.ok()) {
|
|
cerr << status.ToString() << endl;
|
|
}
|
|
db.reset();
|
|
Options options{};
|
|
auto status2 = DestroyDB(db_path, options);
|
|
if (!status2.ok()) {
|
|
cerr << status2.ToString() << endl;
|
|
}
|
|
}
|
|
}
|