#include "CachedBlock.h"
#include "CachingBlockStore.h"

using std::unique_ptr;
using cpputils::unique_ref;

namespace blockstore {
namespace caching {

CachedBlock::CachedBlock(unique_ref<Block> baseBlock, CachingBlockStore *blockStore)
    :Block(baseBlock->key()),
     _blockStore(blockStore),
     _baseBlock(std::move(baseBlock)) {
}

CachedBlock::~CachedBlock() {
  if (_baseBlock.get() != nullptr) {
    _blockStore->release(std::move(_baseBlock));
  }
}

const void *CachedBlock::data() const {
  return _baseBlock->data();
}

void CachedBlock::write(const void *source, uint64_t offset, uint64_t size) {
  return _baseBlock->write(source, offset, size);
}

void CachedBlock::flush() {
  return _baseBlock->flush();
}

size_t CachedBlock::size() const {
  return _baseBlock->size();
}

unique_ref<Block> CachedBlock::releaseBlock() {
  return std::move(_baseBlock);
}

}
}