mirror of
https://github.com/XTXMarkets/ternfs.git
synced 2026-01-27 05:19:44 -06:00
55 lines
1.7 KiB
C++
55 lines
1.7 KiB
C++
// Copyright 2025 XTX Markets Technologies Limited
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
#pragma once
|
|
|
|
#include <memory>
|
|
#include <shared_mutex>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
#include <rocksdb/db.h>
|
|
#include <rocksdb/options.h>
|
|
#include <rocksdb/utilities/optimistic_transaction_db.h>
|
|
|
|
#include "Env.hpp"
|
|
#include "MsgsGen.hpp"
|
|
|
|
class SharedRocksDB {
|
|
public:
|
|
SharedRocksDB(Logger& logger, std::shared_ptr<XmonAgent>& xmon, const std::string& path, const std::string& statisticsPath);
|
|
~SharedRocksDB();
|
|
void open(rocksdb::Options options, const std::string& path);
|
|
void openTransactionDB(rocksdb::Options options);
|
|
void open(rocksdb::Options options);
|
|
void openForReadOnly(rocksdb::Options options);
|
|
void close();
|
|
|
|
void registerCFDescriptors(const std::vector<rocksdb::ColumnFamilyDescriptor>& _cfDescriptors);
|
|
|
|
rocksdb::ColumnFamilyHandle* getCF(const std::string& name) const;
|
|
void deleteCF(const std::string& name);
|
|
rocksdb::ColumnFamilyHandle* createCF(const rocksdb::ColumnFamilyDescriptor& descriptor);
|
|
|
|
|
|
rocksdb::DB* db() const;
|
|
rocksdb::OptimisticTransactionDB* transactionDB() const;
|
|
void rocksDBMetrics(std::unordered_map<std::string, uint64_t>& stats);
|
|
void dumpRocksDBStatistics();
|
|
|
|
TernError snapshot(const std::string& path);
|
|
|
|
private:
|
|
Env _env;
|
|
bool _transactionDB;
|
|
const std::string _path;
|
|
const std::string _statisticsFilePath;
|
|
std::unique_ptr<rocksdb::DB, void(*)(rocksdb::DB*)> _db;
|
|
std::shared_ptr<rocksdb::Statistics> _dbStatistics;
|
|
std::vector<rocksdb::ColumnFamilyDescriptor> _cfDescriptors;
|
|
mutable std::shared_mutex _stateMutex;
|
|
std::unordered_map<std::string, rocksdb::ColumnFamilyHandle*> _cfs;
|
|
};
|