-
Notifications
You must be signed in to change notification settings - Fork 0
/
ping_client.cpp
108 lines (80 loc) · 2.28 KB
/
ping_client.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
//
// Copyright (c) 2023-present DeepGrace (complex dot invoke at gmail dot com)
//
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
// Official repository: https://github.com/deepgrace/srpc
//
#define BOOST_ASIO_HAS_IO_URING
#define BOOST_ASIO_DISABLE_EPOLL
#include <thread>
#include <iostream>
#include <srpc.hpp>
#include <ping.pb.h>
namespace net = boost::asio;
namespace gp = google::protobuf;
struct task
{
srpc::controller controller;
pb::request request;
pb::response response;
srpc::Closure* done;
};
class client
{
public:
client(snp::asio_context& ctx, const std::string& host, const std::string& port) : ctx(ctx), work(ctx.get_io_context()), host(host), port(port)
{
channel = new srpc::channel(ctx);
service = new pb::service::Stub(channel, pb::service::STUB_OWNS_CHANNEL);
}
void ping()
{
auto t = std::make_shared<task>();
auto& controller = t->controller;
controller.host(host);
controller.port(port);
controller.timeout(80);
t->request.set_command("ping");
t->done = gp::NewCallback(this, &client::done, t);
service->execute(&t->controller, &t->request, &t->response, t->done);
}
void done(std::shared_ptr<task> t)
{
auto& controller = t->controller;
if (controller.Failed())
std::cerr << "ErrorCode: " << controller.ErrorCode() << " ErrorText: " << controller.ErrorText() << std::endl;
std::cout << t->response.DebugString();
}
~client()
{
delete service;
}
private:
snp::asio_context& ctx;
net::io_context::work work;
srpc::channel* channel;
pb::service* service;
std::string host;
std::string port;
};
int main(int argc, char* argv[])
{
if (argc != 3)
{
std::cout << "Usage: " << argv[0] << " <host> <port>" << std::endl;
return 1;
}
std::string host(argv[1]);
std::string port(argv[2]);
snp::asio_context ctx;
client c(ctx, host, port);
std::thread t([&]{ ctx.run(); });
constexpr size_t size = 100;
char buff[size];
while (fgets(buff, size, stdin))
c.ping();
t.join();
return 0;
}