mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 08:55:51 +08:00
Barak Michener
85 lines
2.7 KiB
C++
85 lines
2.7 KiB
C++
|
|
#pragma once
|
|
|
|
#include <ray/api/serializer.h>
|
|
|
|
#include <msgpack.hpp>
|
|
|
|
namespace ray {
|
|
namespace api {
|
|
|
|
class Arguments {
|
|
public:
|
|
static void WrapArgs(msgpack::packer<msgpack::sbuffer> &packer);
|
|
|
|
template <typename Arg1Type>
|
|
static void WrapArgs(msgpack::packer<msgpack::sbuffer> &packer, Arg1Type &arg1);
|
|
|
|
template <typename Arg1Type, typename... OtherArgTypes>
|
|
static void WrapArgs(msgpack::packer<msgpack::sbuffer> &packer, Arg1Type &arg1,
|
|
OtherArgTypes &... args);
|
|
|
|
static void UnwrapArgs(msgpack::unpacker &unpacker);
|
|
|
|
template <typename Arg1Type>
|
|
static void UnwrapArgs(msgpack::unpacker &unpacker, std::shared_ptr<Arg1Type> *arg1);
|
|
|
|
template <typename Arg1Type, typename... OtherArgTypes>
|
|
static void UnwrapArgs(msgpack::unpacker &unpacker, std::shared_ptr<Arg1Type> *arg1,
|
|
std::shared_ptr<OtherArgTypes> *... args);
|
|
};
|
|
|
|
// --------- inline implementation ------------
|
|
#include <typeinfo>
|
|
|
|
inline void Arguments::WrapArgs(msgpack::packer<msgpack::sbuffer> &packer) {}
|
|
|
|
template <typename Arg1Type>
|
|
inline void Arguments::WrapArgs(msgpack::packer<msgpack::sbuffer> &packer,
|
|
Arg1Type &arg1) {
|
|
/// Notice ObjectRefClassPrefix should be modified by ObjectRef class name or namespace.
|
|
static const std::string ObjectRefClassPrefix = "N3ray3api9ObjectRef";
|
|
std::string type_name = typeid(arg1).name();
|
|
if (type_name.rfind(ObjectRefClassPrefix, 0) == 0) {
|
|
/// Pass by reference.
|
|
Serializer::Serialize(packer, true);
|
|
} else {
|
|
/// Pass by value.
|
|
Serializer::Serialize(packer, false);
|
|
}
|
|
Serializer::Serialize(packer, arg1);
|
|
}
|
|
|
|
template <typename Arg1Type, typename... OtherArgTypes>
|
|
inline void Arguments::WrapArgs(msgpack::packer<msgpack::sbuffer> &packer, Arg1Type &arg1,
|
|
OtherArgTypes &... args) {
|
|
WrapArgs(packer, arg1);
|
|
WrapArgs(packer, args...);
|
|
}
|
|
|
|
inline void Arguments::UnwrapArgs(msgpack::unpacker &unpacker) {}
|
|
|
|
template <typename Arg1Type>
|
|
inline void Arguments::UnwrapArgs(msgpack::unpacker &unpacker,
|
|
std::shared_ptr<Arg1Type> *arg1) {
|
|
bool is_object_ref;
|
|
Serializer::Deserialize(unpacker, &is_object_ref);
|
|
if (is_object_ref) {
|
|
ObjectRef<Arg1Type> object_ref;
|
|
Serializer::Deserialize(unpacker, &object_ref);
|
|
*arg1 = object_ref.Get();
|
|
} else {
|
|
Serializer::Deserialize(unpacker, arg1);
|
|
}
|
|
}
|
|
|
|
template <typename Arg1Type, typename... OtherArgTypes>
|
|
inline void Arguments::UnwrapArgs(msgpack::unpacker &unpacker,
|
|
std::shared_ptr<Arg1Type> *arg1,
|
|
std::shared_ptr<OtherArgTypes> *... args) {
|
|
UnwrapArgs(unpacker, arg1);
|
|
UnwrapArgs(unpacker, args...);
|
|
}
|
|
|
|
} // namespace api
|
|
} // namespace ray
|