Zengtudor/ccls
1
0
Fork 0
mirror of https://github.com/MaskRay/ccls.git synced 2024-11-29 11:01:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

WIP serializers/msgpack.h

Browse Source
This commit is contained in:
Fangrui Song 2018-01-06 20:10:08 -08:00
parent aa139a4b65
commit 299e687d37
2 changed files with 35 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/serializer.cc
View File

@ -244,8 +244,17 @@ std::unique_ptr<IndexFile> Deserialize(SerializeFormat format,
} }
case SerializeFormat::MessagePack: { case SerializeFormat::MessagePack: {
msgpack::object_handle oh = msgpack::unpack(serialized.data(), serialized.size()); try {
(void)oh; msgpack::object_handle oh =
msgpack::unpack(serialized.data(), serialized.size());
file = MakeUnique<IndexFile>(path);
MessagePackReader reader(oh.get());
Reflect(reader, *file);
if (file->version != expected_version)
return nullptr;
} catch (msgpack::insufficient_bytes&) {
return nullptr;
}
break; break;
} }
} }

64
src/serializers/msgpack.h
View File

@ -5,48 +5,27 @@
#include <msgpack.hpp> #include <msgpack.hpp>
class MessagePackReader : public Reader { class MessagePackReader : public Reader {
msgpack::unpacker* m_; msgpack::object o_;
msgpack::object_handle oh_; size_t idx_ = 0;
std::vector<msgpack::object> children_;
void next() { m_->next(oh_); }
public: public:
MessagePackReader(msgpack::unpacker* m) : m_(m) { next(); } MessagePackReader(msgpack::object o) : o_(o) {}
SerializeFormat Format() const override { return SerializeFormat::MessagePack; } SerializeFormat Format() const override { return SerializeFormat::MessagePack; }
bool IsNull() override { return oh_.get().is_nil(); } bool IsNull() override { return o_.is_nil(); }
bool IsArray() override { return oh_.get().type == msgpack::type::ARRAY; } bool IsArray() override { return o_.type == msgpack::type::ARRAY; }
bool IsInt() override { bool IsInt() override {
return oh_.get().type == msgpack::type::POSITIVE_INTEGER || return o_.type == msgpack::type::POSITIVE_INTEGER ||
oh_.get().type == msgpack::type::NEGATIVE_INTEGER; o_.type == msgpack::type::NEGATIVE_INTEGER;
} }
bool IsString() override { return oh_.get().type == msgpack::type::STR; } bool IsString() override { return o_.type == msgpack::type::STR; }
bool GetBool() override { bool GetBool() override { return o_.as<bool>(); }
auto ret = oh_.get().as<bool>(); int GetInt() override { return o_.as<int>(); }
next(); int64_t GetInt64() override { return o_.as<int64_t>(); }
return ret; uint64_t GetUint64() override { return o_.as<uint64_t>(); }
} const char* GetCString() override { return o_.as<char*>(); }
int GetInt() override {
auto ret = oh_.get().as<int>();
next();
return ret;
}
int64_t GetInt64() override {
auto ret = oh_.get().as<int64_t>();
next();
return ret;
}
uint64_t GetUint64() override {
auto ret = oh_.get().as<uint64_t>();
next();
return ret;
}
const char* GetCString() override {
auto ret = oh_.get().as<char*>();
next();
return ret;
}
bool HasMember(const char* x) override { return true; } bool HasMember(const char* x) override { return true; }
std::unique_ptr<Reader> operator[](const char* x) override { std::unique_ptr<Reader> operator[](const char* x) override {
@ -54,11 +33,17 @@ class MessagePackReader : public Reader {
} }
void IterArray(std::function<void(Reader&)> fn) override { void IterArray(std::function<void(Reader&)> fn) override {
for (auto& entry : o_.as<std::vector<msgpack::object>>()) {
MessagePackReader sub(entry);
fn(sub);
}
} }
void DoMember(const char* name, std::function<void(Reader&)> fn) override { void DoMember(const char* name, std::function<void(Reader&)> fn) override {
const char* key = GetCString(); if (idx_ == 0)
fn(*this); children_ = o_.as<std::vector<msgpack::object>>();
MessagePackReader sub(children_[idx_++]);
fn(sub);
} }
}; };
@ -79,8 +64,7 @@ class MessagePackWriter : public Writer {
void String(const char* x, size_t len) override { m_->pack(std::string(x, len)); } void String(const char* x, size_t len) override { m_->pack(std::string(x, len)); }
void StartArray(size_t n) override { m_->pack_array(uint32_t(n)); } void StartArray(size_t n) override { m_->pack_array(uint32_t(n)); }
void EndArray() override {} void EndArray() override {}
// TODO pack_array void StartObject(size_t n) override { m_->pack_array(uint32_t(n)); }
void StartObject(size_t n) override { m_->pack_map(uint32_t(n)); }
void EndObject() override {} void EndObject() override {}
void Key(const char* name) override { m_->pack(name); } void Key(const char* name) override {}
}; };