You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

65 lines
1.8 KiB
C++

#pragma once
#include <ipc/IpcMessage.h>
#include <functional>
#include <string>
#include <tuple>
#include <unordered_map>
class RpcInvoker {
public:
template<typename Obj, typename Ret, typename... Args>
void registerMethod(Obj* instance,
const std::string& name,
Ret (Obj::*method)(Args...)) {
handlers[name] =
[instance, method](const IpcMessage& req) -> IpcMessage {
IpcMessage msg = req;
// пропустить имя метода
(void)msg.template get<std::string>();
// читать аргументы и вызвать метод
return callMethod<Ret, Obj, Args...>(instance, method, msg);
};
}
IpcMessage dispatch(const IpcMessage& request) {
IpcMessage tmp = request;
const std::string method = tmp.get<std::string>();
auto it = handlers.find(method);
if (it == handlers.end()) {
return IpcMessage{};
}
return it->second(request);
}
private:
template<typename Ret, typename Obj, typename... Args>
static IpcMessage callMethod(Obj* obj,
Ret (Obj::*method)(Args...),
IpcMessage& msg) {
auto tuple = readArgs<Args...>(msg);
Ret result =
std::apply(method, std::tuple_cat(std::make_tuple(obj), tuple));
IpcMessage out;
out.add(result);
return out;
}
template<typename... Args>
static std::tuple<Args...> readArgs(IpcMessage& msg) {
return std::tuple<Args...>{msg.template get<Args>()...};
}
std::unordered_map<std::string,
std::function<IpcMessage(const IpcMessage&)>>
handlers;
};