bool Layout::write_block_meta(BlockMeta* meta, BlockWriter& writer)
{
if (!writer.writeUInt64(meta->offset)) return false;
if (!writer.writeUInt32(meta->skeleton_size)) return false;
if (!writer.writeUInt32(meta->total_size)) return false;
if (!writer.writeUInt16(meta->crc)) return false;
return true;
}
bool LeafNode::write_buckets_info(BlockWriter& writer)
{
if (!writer.writeUInt32(buckets_info_.size())) return false;
for (size_t i = 0; i < buckets_info_.size(); i++ ) {
if (!writer.writeSlice(buckets_info_[i].key)) return false;
if (!writer.writeUInt32(buckets_info_[i].offset)) return false;
if (!writer.writeUInt32(buckets_info_[i].length)) return false;
if (!writer.writeUInt32(buckets_info_[i].uncompressed_length)) return false;
}
return true;
}
bool LeafNode::write_bucket(BlockWriter& writer, RecordBucket *bucket)
{
if (!writer.writeUInt32(bucket->size())) return false;
for (size_t j = 0; j < bucket->size(); j++) {
if (!(*bucket)[j].write_to(writer)) return false;
}
return true;
}
bool MsgBuf::write_to(BlockWriter& writer)
{
if (!writer.writeUInt32(container_.size())) return false;
for (ContainerType::iterator it = container_.begin();
it != container_.end(); it ++ ) {
if (!it->write_to(writer)) return false;
}
return true;
}
bool Layout::write_index(BlockWriter& writer)
{
ScopedMutex block_index_lock(&block_index_mtx_);
if (!writer.writeUInt32(block_index_.size())) return false;
for(map<bid_t, BlockMeta*>::iterator it = block_index_.begin();
it != block_index_.end(); it++ ) {
if (!writer.writeUInt64(it->first)) return false;
if (!write_block_meta(it->second, writer)) return false;
}
return true;
}
bool InnerNode::write_to(BlockWriter& writer, size_t& skeleton_size)
{
// get length of skeleton and reserve space for skeleton
size_t skeleton_offset = writer.pos();
size_t skeleton_length = 1 + 4 + 8 + 4 + 4 + 4;
for (size_t i = 0; i < pivots_.size(); i++) {
skeleton_length += pivot_size(pivots_[i].key);
}
if (!writer.skip(skeleton_length)) return false;
// prepare buffer if compression is enabled
Slice buffer;
if (tree_->compressor_) {
// get buffer length to serialize msgbuf
size_t buffer_length = first_msgbuf_->size();
for (size_t i = 0; i < pivots_.size(); i++) {
if (pivots_[i].msgbuf->size() > buffer_length)
buffer_length = pivots_[i].msgbuf->size();
}
buffer = Slice::alloc(buffer_length);
}
// write the first msgbuf
first_msgbuf_offset_ = writer.pos();
if (!write_msgbuf(writer, first_msgbuf_, buffer)) return false;
first_msgbuf_length_ = writer.pos() - first_msgbuf_offset_;
first_msgbuf_uncompressed_length_ = first_msgbuf_->size();
// write rest msgbufs
for (size_t i = 0; i < pivots_.size(); i++) {
pivots_[i].offset = writer.pos();
if (!write_msgbuf(writer, pivots_[i].msgbuf, buffer)) return false;
pivots_[i].length = writer.pos() - pivots_[i].offset;
pivots_[i].uncompressed_length = pivots_[i].msgbuf->size();
}
if (buffer.size()) {
buffer.destroy();
}
size_t last_offset = writer.pos();
// seek to the head and write index
writer.seek(skeleton_offset);
if (!writer.writeBool(bottom_)) return false;
if (!writer.writeUInt32(pivots_.size())) return false;
if (!writer.writeUInt64(first_child_)) return false;
if (!writer.writeUInt32(first_msgbuf_offset_)) return false;
if (!writer.writeUInt32(first_msgbuf_length_)) return false;
if (!writer.writeUInt32(first_msgbuf_uncompressed_length_)) return false;
for (size_t i = 0; i < pivots_.size(); i++) {
if (!writer.writeSlice(pivots_[i].key)) return false;
if (!writer.writeUInt64(pivots_[i].child)) return false;
if (!writer.writeUInt32(pivots_[i].offset)) return false;
if (!writer.writeUInt32(pivots_[i].length)) return false;
if (!writer.writeUInt32(pivots_[i].uncompressed_length)) return false;
}
writer.seek(last_offset);
skeleton_size = skeleton_length;
return true;
}