This commit is contained in:
qicosmos
2025-10-17 21:25:20 +08:00
parent e581820afb
commit 6bd939b177
+15 -9
View File
@@ -1,11 +1,13 @@
#include <rest_rpc.hpp> #include <rest_rpc.hpp>
using namespace rest_rpc; using namespace rest_rpc;
std::string_view echo(std::string_view str) { std::string_view echo(std::string_view str) { return str; }
return str;
asio::awaitable<std::string_view> echo_coro(std::string_view str) {
co_return str;
} }
struct dummy{ struct dummy {
int add(int a, int b) { return a + b; } int add(int a, int b) { return a + b; }
}; };
int add(int a, int b) { return a + b; } int add(int a, int b) { return a + b; }
@@ -24,6 +26,7 @@ person get_person(person p) {
void basict_usage() { void basict_usage() {
rpc_server server("127.0.0.1:9004", 4); rpc_server server("127.0.0.1:9004", 4);
server.register_handler<echo>(); server.register_handler<echo>();
server.register_handler<echo_coro>();
dummy d{}; dummy d{};
server.register_handler<&dummy::add>(&d); server.register_handler<&dummy::add>(&d);
@@ -31,7 +34,7 @@ void basict_usage() {
server.register_handler<get_person>(); server.register_handler<get_person>();
auto ec = server.async_start(); auto ec = server.async_start();
if(ec) { if (ec) {
REST_LOG_ERROR << ec.message(); REST_LOG_ERROR << ec.message();
return; return;
} }
@@ -39,26 +42,29 @@ void basict_usage() {
auto rpc_call = []() -> asio::awaitable<void> { auto rpc_call = []() -> asio::awaitable<void> {
rpc_client client; rpc_client client;
auto ec0 = co_await client.connect("127.0.0.1:9004"); auto ec0 = co_await client.connect("127.0.0.1:9004");
if(ec0) { if (ec0) {
REST_LOG_ERROR << ec0.message(); REST_LOG_ERROR << ec0.message();
co_return; co_return;
} }
auto r = co_await client.call<echo>("test"); auto r = co_await client.call<echo>("test");
if(r.ec==rpc_errc::ok) { if (r.ec == rpc_errc::ok) {
REST_LOG_INFO << "call result: " << r.value; REST_LOG_INFO << "call result: " << r.value;
assert(r.value == "test"); assert(r.value == "test");
} }
r = co_await client.call<echo_coro>("test1");
assert(r.value == "test1");
auto r1 = co_await client.call<&dummy::add>(1, 2); auto r1 = co_await client.call<&dummy::add>(1, 2);
if(r1.ec==rpc_errc::ok) { if (r1.ec == rpc_errc::ok) {
REST_LOG_INFO << "call result: " << r1.value; REST_LOG_INFO << "call result: " << r1.value;
assert(r1.value == 3); assert(r1.value == 3);
} }
person p{1, "tom", 20}; person p{1, "tom", 20};
auto r2 = co_await client.call<get_person>(p); auto r2 = co_await client.call<get_person>(p);
if(r2.ec==rpc_errc::ok) { if (r2.ec == rpc_errc::ok) {
REST_LOG_INFO << "call result: " << r2.value.name; REST_LOG_INFO << "call result: " << r2.value.name;
assert(r2.value.name == "jack"); assert(r2.value.name == "jack");
} }
@@ -78,7 +84,7 @@ void publish() {
std::string str; std::string str;
while (true) { while (true) {
std::cin >> str; std::cin >> str;
if(str == "quit") { if (str == "quit") {
break; break;
} }