forked from qicosmos/rest_rpc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
codec.h
75 lines (60 loc) · 1.98 KB
/
codec.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#ifndef REST_RPC_CODEC_H_
#define REST_RPC_CODEC_H_
#include <msgpack.hpp>
namespace rest_rpc {
namespace rpc_service {
struct msgpack_codec {
using buffer_type = std::vector<char>;
class buffer_t {
public:
buffer_t() : buffer_t(0) {}
explicit buffer_t(size_t len) : buffer_(len, 0), offset_(0) {}
buffer_t(buffer_t const&) = default;
buffer_t(buffer_t&&) = default;
buffer_t& operator=(buffer_t const&) = default;
buffer_t& operator=(buffer_t&&) = default;
void write(char const* data, size_t length) {
if (buffer_.size() - offset_ < length) buffer_.resize(length + offset_);
std::memcpy(buffer_.data() + offset_, data, length);
offset_ += length;
}
std::vector<char> release() const noexcept { return std::move(buffer_); }
private:
std::vector<char> buffer_;
size_t offset_;
};
template<typename... Args>
static buffer_type pack_args(Args&&... args) {
buffer_t buffer;
auto args_tuple = std::make_tuple(std::forward<Args>(args)...);
msgpack::pack(buffer, args_tuple);
return buffer.release();
}
template<typename Arg, typename... Args,
typename = typename std::enable_if<std::is_enum<Arg>::value>::type>
static std::string pack_args_str(Arg arg, Args&&... args) {
buffer_t buffer;
auto args_tuple = std::make_tuple((int)arg, std::forward<Args>(args)...);
msgpack::pack(buffer, args_tuple);
auto ret = buffer.release();
return std::string(ret.data(), ret.size());
}
template<typename T>
buffer_type pack(T&& t) const {
buffer_t buffer;
msgpack::pack(buffer, std::forward<T>(t));
return buffer.release();
}
template<typename T>
T unpack(char const* data, size_t length) {
try {
msgpack::unpack(&msg_, data, length);
return msg_.get().as<T>();
} catch (...) { throw std::invalid_argument("Args not match!"); }
}
private:
msgpack::unpacked msg_;
};
} // namespace rpc_service
} // namespace rest_rpc
#endif // REST_RPC_CODEC_H_