#pragma once
#ifndef MESSMER_BLOBSTORE_IMPLEMENTATIONS_ONBLOCKS_BLOBONBLOCKS_H_
#define MESSMER_BLOBSTORE_IMPLEMENTATIONS_ONBLOCKS_BLOBONBLOCKS_H_

#include "../../interface/Blob.h"

#include <memory>
#include <boost/optional.hpp>

namespace blobstore {
namespace onblocks {
namespace datanodestore {
class DataLeafNode;
}
namespace parallelaccessdatatreestore {
class DataTreeRef;
}

class BlobOnBlocks final: public Blob {
public:
  BlobOnBlocks(cpputils::unique_ref<parallelaccessdatatreestore::DataTreeRef> datatree);
  ~BlobOnBlocks();

  const blockstore::Key &key() const override;

  uint64_t size() const override;
  void resize(uint64_t numBytes) override;

  cpputils::Data readAll() const override;
  void read(void *target, uint64_t offset, uint64_t size) const override;
  uint64_t tryRead(void *target, uint64_t offset, uint64_t size) const override;
  void write(const void *source, uint64_t offset, uint64_t size) override;

  void flush() override;

  cpputils::unique_ref<parallelaccessdatatreestore::DataTreeRef> releaseTree();

private:

  uint64_t _size() const;

  void _read(void *target, uint64_t offset, uint64_t count) const;
  uint64_t _tryRead(void *target, uint64_t offset, uint64_t size) const;
  void _traverseLeaves(uint64_t offsetBytes, uint64_t sizeBytes, std::function<void (uint64_t, datanodestore::DataLeafNode *, uint32_t, uint32_t)>) const;

  cpputils::unique_ref<parallelaccessdatatreestore::DataTreeRef> _datatree;
  mutable boost::optional<uint64_t> _sizeCache;

  DISALLOW_COPY_AND_ASSIGN(BlobOnBlocks);
};

}
}

#endif
