78 lines
2.5 KiB
C
78 lines
2.5 KiB
C
|
#pragma once
|
||
|
|
#ifndef MESSMER_BLOCKSTORE_IMPLEMENTATIONS_COMPRESSING_COMPRESSINGBLOCKSTORE_H_
|
||
|
|
#define MESSMER_BLOCKSTORE_IMPLEMENTATIONS_COMPRESSING_COMPRESSINGBLOCKSTORE_H_
|
||
|
|
|
||
|
|
#include "../../interface/BlockStore.h"
|
||
|
|
#include "CompressedBlock.h"
|
||
|
|
|
||
|
|
namespace blockstore {
|
||
|
|
namespace compressing {
|
||
|
|
|
||
|
|
template<class Compressor>
|
||
|
|
class CompressingBlockStore final: public BlockStore {
|
||
|
|
public:
|
||
|
|
CompressingBlockStore(cpputils::unique_ref<BlockStore> baseBlockStore);
|
||
|
|
~CompressingBlockStore();
|
||
|
|
|
||
|
|
Key createKey() override;
|
||
|
|
boost::optional<cpputils::unique_ref<Block>> tryCreate(const Key &key, cpputils::Data data) override;
|
||
|
|
boost::optional<cpputils::unique_ref<Block>> load(const Key &key) override;
|
||
|
|
void remove(cpputils::unique_ref<Block> block) override;
|
||
|
|
uint64_t numBlocks() const override;
|
||
|
|
|
||
|
|
private:
|
||
|
|
cpputils::unique_ref<BlockStore> _baseBlockStore;
|
||
|
|
|
||
|
|
DISALLOW_COPY_AND_ASSIGN(CompressingBlockStore);
|
||
|
|
};
|
||
|
|
|
||
|
|
template<class Compressor>
|
||
|
|
CompressingBlockStore<Compressor>::CompressingBlockStore(cpputils::unique_ref<BlockStore> baseBlockStore)
|
||
|
|
: _baseBlockStore(std::move(baseBlockStore)) {
|
||
|
|
}
|
||
|
|
|
||
|
|
template<class Compressor>
|
||
|
|
CompressingBlockStore<Compressor>::~CompressingBlockStore() {
|
||
|
|
}
|
||
|
|
|
||
|
|
template<class Compressor>
|
||
|
|
Key CompressingBlockStore<Compressor>::createKey() {
|
||
|
|
return _baseBlockStore->createKey();
|
||
|
|
}
|
||
|
|
|
||
|
|
template<class Compressor>
|
||
|
|
boost::optional<cpputils::unique_ref<Block>> CompressingBlockStore<Compressor>::tryCreate(const Key &key, cpputils::Data data) {
|
||
|
|
auto result = CompressedBlock<Compressor>::TryCreateNew(_baseBlockStore.get(), key, std::move(data));
|
||
|
|
if (result == boost::none) {
|
||
|
|
return boost::none;
|
||
|
|
}
|
||
|
|
return cpputils::unique_ref<Block>(std::move(*result));
|
||
|
|
}
|
||
|
|
|
||
|
|
template<class Compressor>
|
||
|
|
boost::optional<cpputils::unique_ref<Block>> CompressingBlockStore<Compressor>::load(const Key &key) {
|
||
|
|
auto loaded = _baseBlockStore->load(key);
|
||
|
|
if (loaded == boost::none) {
|
||
|
|
return boost::none;
|
||
|
|
}
|
||
|
|
return boost::optional<cpputils::unique_ref<Block>>(CompressedBlock<Compressor>::Decompress(std::move(*loaded)));
|
||
|
|
}
|
||
|
|
|
||
|
|
template<class Compressor>
|
||
|
|
void CompressingBlockStore<Compressor>::remove(cpputils::unique_ref<Block> block) {
|
||
|
|
auto _block = cpputils::dynamic_pointer_move<CompressedBlock<Compressor>>(block);
|
||
|
|
ASSERT(_block != boost::none, "Wrong block type");
|
||
|
|
auto baseBlock = (*_block)->releaseBaseBlock();
|
||
|
|
return _baseBlockStore->remove(std::move(baseBlock));
|
||
|
|
}
|
||
|
|
|
||
|
|
template<class Compressor>
|
||
|
|
uint64_t CompressingBlockStore<Compressor>::numBlocks() const {
|
||
|
|
return _baseBlockStore->numBlocks();
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|