saberbud/rest_rpc
1
0
Fork 0
mirror of https://github.com/qicosmos/rest_rpc.git synced 2026-08-29 08:34:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

add unsafe response

Browse Source
This commit is contained in:
qicosmos
2025-10-04 16:54:07 +08:00
parent 500303ad66
commit d5b6edc907
3 changed files with 49 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/rest_rpc/rest_rpc_protocol.hpp
+1 -1
View File
@@ -6,7 +6,7 @@
#include <arpa/inet.h>
#endif
#ifdef __APPLE__
#if defined(__APPLE__) || defined(_WIN32)
#else
inline uint64_t htonll(uint64_t value) {
return ((uint64_t)htonl(value & 0xFFFFFFFF) << 32) | htonl(value >> 32);
include/rest_rpc/rpc_connection.hpp
+22 -5
View File
@@ -27,10 +27,15 @@ public:
auto get_executor();
template <auto func, typename... Args>
asio::awaitable<std::error_code> response(Args &&...args);
asio::awaitable<std::error_code> response_s(Args &&...args);
template <auto func, typename... Args>
std::error_code sync_response(Args &&...args);
std::error_code sync_response_s(Args &&...args);
template <typename... Args>
asio::awaitable<std::error_code> response(Args &&...args);
template <typename... Args> std::error_code sync_response(Args &&...args);
private:
rpc_context() = default;
@@ -177,12 +182,24 @@ private:
// zero or one arguments
template <auto func, typename... Args>
asio::awaitable<std::error_code> rpc_context::response(Args &&...args) {
asio::awaitable<std::error_code> rpc_context::response_s(Args &&...args) {
using args_tuple =
typename util::function_traits<decltype(func)>::return_type;
static_assert(
std::is_constructible_v<args_tuple, Args...>,
"rpc function return type and response arguments are not match");
return response(std::forward<Args>(args)...);
}
template <auto func, typename... Args>
std::error_code rpc_context::sync_response_s(Args &&...args) {
return sync_wait(conn_->get_executor(),
response_s<func>(std::forward<Args>(args)...));
}
template <typename... Args>
asio::awaitable<std::error_code> rpc_context::response(Args &&...args) {
if (has_response_) {
co_return make_error_code(rpc_errc::has_response);
}
@@ -193,10 +210,10 @@ asio::awaitable<std::error_code> rpc_context::response(Args &&...args) {
co_return co_await conn_->response(result);
}
template <auto func, typename... Args>
template <typename... Args>
std::error_code rpc_context::sync_response(Args &&...args) {
return sync_wait(conn_->get_executor(),
response<func>(std::forward<Args>(args)...));
response(std::forward<Args>(args)...));
}
auto rpc_context::get_executor() { return conn_->get_executor(); }