RocksDatabase.h

Go to the documentation of this file.

 
#pragma once
#include "RocksPruningFilter.h"
#include "catapult/utils/FileSize.h"
#include "catapult/types.h"
#include <memory>
#include <string>
#include <vector>
 
namespace rocksdb {
    class ColumnFamilyHandle;
    class DB;
    class PinnableSlice;
    class Slice;
    class WriteBatch;
}
 
namespace catapult { namespace cache {
 
    class RdbDataIterator {
    private:
        enum class StorageStrategy { Allocate, Do_Not_Allocate };
 
        explicit RdbDataIterator(StorageStrategy storageStrategy);
 
    public:
        RdbDataIterator();
 
        ~RdbDataIterator();
 
    public:
        RdbDataIterator(RdbDataIterator&&);
 
        RdbDataIterator& operator=(RdbDataIterator&&);
 
    public:
        static RdbDataIterator End();
 
    public:
        bool operator==(const RdbDataIterator& rhs) const;
 
        bool operator!=(const RdbDataIterator& rhs) const;
 
    public:
        rocksdb::PinnableSlice& storage() const;
 
        void setFound(bool found);
 
        RawBuffer buffer() const;
 
    private:
        struct Impl;
        std::shared_ptr<Impl> m_pImpl;
        bool m_isFound;
    };
 
    struct RocksDatabaseSettings {
    public:
        RocksDatabaseSettings();
 
        RocksDatabaseSettings(
                const std::string& databaseDirectory,
                const std::vector<std::string>& columnFamilyNames,
                utils::FileSize maxDatabaseWriteBatchSize,
                FilterPruningMode pruningMode);
 
    public:
        const std::string DatabaseDirectory;
 
        const std::vector<std::string> ColumnFamilyNames;
 
        const utils::FileSize MaxDatabaseWriteBatchSize;
 
        const FilterPruningMode PruningMode;
    };
 
    class RocksDatabase {
    public:
        RocksDatabase();
 
        explicit RocksDatabase(const RocksDatabaseSettings& settings);
 
        ~RocksDatabase();
 
    public:
        const std::vector<std::string>& columnFamilyNames() const;
 
        bool canPrune() const;
 
    public:
        void get(size_t columnId, const rocksdb::Slice& key, RdbDataIterator& result);
 
        void put(size_t columnId, const rocksdb::Slice& key, const std::string& value);
 
        void del(size_t columnId, const rocksdb::Slice& key);
 
        size_t prune(size_t columnId, uint64_t boundary);
 
        void flush();
 
    private:
        void saveIfBatchFull();
 
    private:
        const RocksDatabaseSettings m_settings;
        RocksPruningFilter m_pruningFilter;
        std::unique_ptr<rocksdb::WriteBatch> m_pWriteBatch;
 
        std::unique_ptr<rocksdb::DB> m_pDb;
        std::vector<rocksdb::ColumnFamilyHandle*> m_handles;
    };
}}