-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathclient.cpp
64 lines (57 loc) · 2.37 KB
/
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
#include <gflags/gflags.h>
#include <butil/logging.h>
#include <brpc/channel.h>
#include <brpc/stream.h>
#include "echo.pb.h"
DEFINE_bool(send_attachment, true, "Carry attachment along with requests");
DEFINE_string(connection_type, "", "Connection type. Available values: single, pooled, short");
DEFINE_string(server, "0.0.0.0:8001", "IP Address of server");
DEFINE_int32(timeout_ms, 100, "RPC timeout in milliseconds");
DEFINE_int32(max_retry, 3, "Max retries(not including the first RPC)");
int main(int argc, char* argv[]) {
// Parse gflags. We recommend you to use gflags as well.
google::ParseCommandLineFlags(&argc, &argv, true);
// A Channel represents a communication line to a Server. Notice that
// Channel is thread-safe and can be shared by all threads in your program.
brpc::Channel channel;
// Initialize the channel, NULL means using default options.
brpc::ChannelOptions options;
options.protocol = brpc::PROTOCOL_BAIDU_STD;
options.connection_type = FLAGS_connection_type;
options.timeout_ms = FLAGS_timeout_ms/*milliseconds*/;
options.max_retry = FLAGS_max_retry;
if (channel.Init(FLAGS_server.c_str(), NULL) != 0) {
LOG(ERROR) << "Fail to initialize channel";
return -1;
}
// Normally, you should not call a Channel directly, but instead construct
// a stub Service wrapping it. stub can be shared by all threads as well.
example::EchoService_Stub stub(&channel);
brpc::Controller cntl;
brpc::StreamId stream;
if (brpc::StreamCreate(&stream, cntl, NULL) != 0) {
LOG(ERROR) << "Fail to create stream";
return -1;
}
LOG(INFO) << "Created Stream=" << stream;
example::EchoRequest request;
example::EchoResponse response;
request.set_message("I'm a RPC to connect stream");
stub.Echo(&cntl, &request, &response, NULL);
if (cntl.Failed()) {
LOG(ERROR) << "Fail to connect stream, " << cntl.ErrorText();
return -1;
}
while (!brpc::IsAskedToQuit()) {
butil::IOBuf msg1;
msg1.append("abcdefghijklmnopqrstuvwxyz");
CHECK_EQ(0, brpc::StreamWrite(stream, msg1));
butil::IOBuf msg2;
msg2.append("0123456789");
CHECK_EQ(0, brpc::StreamWrite(stream, msg2));
sleep(1);
}
CHECK_EQ(0, brpc::StreamClose(stream));
LOG(INFO) << "EchoClient is going to quit";
return 0;
}