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.
106 lines
3.7 KiB
C++
106 lines
3.7 KiB
C++
//
|
|
// Created by Ziyang Hu on 2022/7/3.
|
|
//
|
|
|
|
#include <iostream>
|
|
#include "db.h"
|
|
#include "cozorocks/src/bridge/mod.rs.h"
|
|
|
|
shared_ptr<RocksDbBridge> open_db(const DbOpts &opts, RocksDbStatus &status, bool use_cmp, RustComparatorFn cmp_impl) {
|
|
auto options = make_unique<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));
|
|
}
|
|
RustComparator *cmp = nullptr;
|
|
if (use_cmp) {
|
|
cmp = new RustComparator(
|
|
string(opts.comparator_name),
|
|
opts.comparator_different_bytes_can_be_equal,
|
|
cmp_impl);
|
|
options->comparator = cmp;
|
|
}
|
|
|
|
shared_ptr<RocksDbBridge> db_wrapper = shared_ptr<RocksDbBridge>(nullptr);
|
|
if (opts.optimistic) {
|
|
auto db = new OptimisticRocksDb();
|
|
db->options = std::move(options);
|
|
db->db_path = string(opts.db_path);
|
|
db->comparator.reset(cmp);
|
|
|
|
OptimisticTransactionDB *txn_db = nullptr;
|
|
write_status(OptimisticTransactionDB::Open(*db->options, db->db_path, &txn_db), status);
|
|
db->db.reset(txn_db);
|
|
db->destroy_on_exit = opts.destroy_on_exit;
|
|
db_wrapper.reset(db);
|
|
} else {
|
|
auto db = new PessimisticRocksDb();
|
|
db->options = std::move(options);
|
|
db->db_path = string(opts.db_path);
|
|
db->tdb_opts = make_unique<TransactionDBOptions>();
|
|
db->comparator.reset(cmp);
|
|
|
|
TransactionDB *txn_db = nullptr;
|
|
write_status(TransactionDB::Open(*db->options, *db->tdb_opts, db->db_path, &txn_db), status);
|
|
db->db.reset(txn_db);
|
|
db->destroy_on_exit = opts.destroy_on_exit;
|
|
db_wrapper.reset(db);
|
|
}
|
|
|
|
return db_wrapper;
|
|
}
|
|
|
|
PessimisticRocksDb::~PessimisticRocksDb() {
|
|
if (destroy_on_exit) {
|
|
cerr << "destroying database on exit: " << db_path << endl;
|
|
auto status = db->Close();
|
|
if (!status.ok()) {
|
|
cerr << status.ToString() << endl;
|
|
}
|
|
db.reset();
|
|
auto status2 = DestroyDB(db_path, *options);
|
|
if (!status2.ok()) {
|
|
cerr << status.ToString() << endl;
|
|
}
|
|
}
|
|
}
|
|
|
|
OptimisticRocksDb::~OptimisticRocksDb() {
|
|
if (destroy_on_exit) {
|
|
cerr << "destroying database on exit: " << db_path << endl;
|
|
auto status = db->Close();
|
|
if (!status.ok()) {
|
|
cerr << status.ToString() << endl;
|
|
}
|
|
db.reset();
|
|
auto status2 = DestroyDB(db_path, *options);
|
|
if (!status2.ok()) {
|
|
cerr << status.ToString() << endl;
|
|
}
|
|
}
|
|
} |