wehub-resource-sync
87 lines
2.5 KiB
C++
87 lines
2.5 KiB
C++
// Copyright 2020-2021 The Ray Authors.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#pragma once
|
|
|
|
#include <ray/api/msgpack_adaptor.h>
|
|
#include <ray/api/ray_exception.h>
|
|
#include <ray/api/xlang_function.h>
|
|
|
|
#include <msgpack.hpp>
|
|
|
|
namespace ray {
|
|
namespace internal {
|
|
|
|
class Serializer {
|
|
public:
|
|
template <typename T>
|
|
static msgpack::sbuffer Serialize(const T &t) {
|
|
msgpack::sbuffer buffer;
|
|
msgpack::pack(buffer, t);
|
|
return buffer;
|
|
}
|
|
|
|
static msgpack::sbuffer Serialize(const char *data, size_t size) {
|
|
msgpack::sbuffer buffer;
|
|
msgpack::packer<msgpack::sbuffer> packer(&buffer);
|
|
packer.pack_bin(size);
|
|
packer.pack_bin_body(data, size);
|
|
return buffer;
|
|
}
|
|
|
|
template <typename T>
|
|
static T Deserialize(const char *data, size_t size) {
|
|
msgpack::unpacked unpacked;
|
|
msgpack::unpack(unpacked, data, size);
|
|
return unpacked.get().as<T>();
|
|
}
|
|
|
|
template <typename T>
|
|
static T Deserialize(const char *data, size_t size, size_t offset) {
|
|
return Deserialize<T>(data + offset, size - offset);
|
|
}
|
|
|
|
template <typename T>
|
|
static T Deserialize(const char *data, size_t size, size_t *off) {
|
|
msgpack::unpacked unpacked = msgpack::unpack(data, size, *off);
|
|
return unpacked.get().as<T>();
|
|
}
|
|
|
|
template <typename T>
|
|
static std::pair<bool, T> DeserializeWhenNil(const char *data, size_t size) {
|
|
T val;
|
|
size_t off = 0;
|
|
msgpack::unpacked unpacked = msgpack::unpack(data, size, off);
|
|
if (!unpacked.get().convert_if_not_nil(val)) {
|
|
return {false, {}};
|
|
}
|
|
|
|
return {true, val};
|
|
}
|
|
|
|
static bool HasError(char *data, size_t size) {
|
|
msgpack::unpacked unpacked = msgpack::unpack(data, size);
|
|
return unpacked.get().is_nil() && size > 1;
|
|
}
|
|
|
|
static bool IsXLang(char *data, size_t size) {
|
|
msgpack::unpacked unpacked = msgpack::unpack(data, size);
|
|
return unpacked.get().type == msgpack::type::POSITIVE_INTEGER &&
|
|
size >= XLANG_HEADER_LEN;
|
|
}
|
|
};
|
|
|
|
} // namespace internal
|
|
} // namespace ray
|