You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
2.2 KiB
100 lines
2.2 KiB
#ifndef VTABLE_FORMAT_H
|
|
#define VTABLE_FORMAT_H
|
|
|
|
#include "leveldb/slice.h"
|
|
#include "leveldb/status.h"
|
|
#include "table/format.h"
|
|
|
|
namespace leveldb {
|
|
|
|
const uint64_t kRecordHeaderSize = 4;
|
|
|
|
struct VTableRecord {
|
|
Slice key;
|
|
Slice value;
|
|
|
|
void Encode(std::string* target) const;
|
|
Status Decode(Slice* input);
|
|
|
|
size_t size() const { return key.size() + value.size(); }
|
|
|
|
friend bool operator==(const VTableRecord& a, const VTableRecord& b) {
|
|
return a.key == b.key && a.value == b.value;
|
|
}
|
|
};
|
|
|
|
class RecordEncoder {
|
|
public:
|
|
// TODO: Support compression while encoding a record
|
|
RecordEncoder() = default;
|
|
|
|
// Encode a vTable record
|
|
void Encode(const VTableRecord& record);
|
|
|
|
// Get the size of encoded record
|
|
size_t GetEncodedSize() const { return sizeof(header_) + record_.size(); }
|
|
|
|
// Get the header
|
|
Slice GetHeader() const { return {header_, sizeof(header_)}; }
|
|
|
|
// Get the encoded record
|
|
Slice GetRecord() const {return record_; }
|
|
private:
|
|
char header_[kRecordHeaderSize];
|
|
Slice record_;
|
|
|
|
std::string record_buff_;
|
|
};
|
|
|
|
class RecordDecoder {
|
|
public:
|
|
Status DecodeHeader(Slice* input);
|
|
|
|
Status DecodeRecord(Slice* input, VTableRecord* record) const;
|
|
|
|
size_t GetDecodedSize() const { return record_size_; }
|
|
|
|
private:
|
|
uint32_t record_size_{0};
|
|
};
|
|
|
|
struct VTableHandle {
|
|
uint64_t offset{0};
|
|
uint64_t size{0};
|
|
|
|
void Encode(std::string* target) const;
|
|
Status Decode(Slice* input);
|
|
|
|
friend bool operator==(const VTableHandle& a, const VTableHandle& b) {
|
|
return a.offset == b.offset && a.size == b.size;
|
|
}
|
|
};
|
|
|
|
struct VTableIndex {
|
|
enum Type : unsigned char {
|
|
kVTableIndex = 1,
|
|
};
|
|
|
|
uint64_t file_number{0};
|
|
VTableHandle vtable_handle;
|
|
|
|
void Encode(std::string* target) const;
|
|
Status Decode(Slice* input);
|
|
|
|
friend bool operator==(const VTableIndex& a, const VTableIndex& b) {
|
|
return a.file_number == b.file_number && a.vtable_handle == b.vtable_handle;
|
|
}
|
|
};
|
|
|
|
template <typename T>
|
|
Status DecodeSrcIntoObj(const Slice& src, T* target) {
|
|
Slice input = src;
|
|
Status s = target->Decode(&input);
|
|
if (s.ok() && !input.empty()) {
|
|
s = Status::Corruption(Slice());
|
|
}
|
|
return s;
|
|
}
|
|
|
|
} // namespace leveldb
|
|
#endif //VTABLE_FORMAT_H
|