118 lines
2.5 KiB
C++
118 lines
2.5 KiB
C++
|
#include "examples/protobuf/resolver/resolver.pb.h"
|
||
|
|
|
||
|
|
#include "muduo/base/Logging.h"
|
||
|
|
#include "muduo/net/EventLoop.h"
|
||
|
|
#include "muduo/net/InetAddress.h"
|
||
|
|
#include "muduo/net/TcpClient.h"
|
||
|
|
#include "muduo/net/TcpConnection.h"
|
||
|
|
#include "muduo/net/protorpc/RpcChannel.h"
|
||
|
|
|
||
|
|
#include <arpa/inet.h> // inet_ntop
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
|
||
|
|
using namespace muduo;
|
||
|
|
using namespace muduo::net;
|
||
|
|
|
||
|
|
class RpcClient : noncopyable
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
RpcClient(EventLoop* loop, const InetAddress& serverAddr)
|
||
|
|
: loop_(loop),
|
||
|
|
client_(loop, serverAddr, "RpcClient"),
|
||
|
|
channel_(new RpcChannel),
|
||
|
|
got_(0),
|
||
|
|
total_(0),
|
||
|
|
stub_(get_pointer(channel_))
|
||
|
|
{
|
||
|
|
client_.setConnectionCallback(
|
||
|
|
std::bind(&RpcClient::onConnection, this, _1));
|
||
|
|
client_.setMessageCallback(
|
||
|
|
std::bind(&RpcChannel::onMessage, get_pointer(channel_), _1, _2, _3));
|
||
|
|
// client_.enableRetry();
|
||
|
|
}
|
||
|
|
|
||
|
|
void connect()
|
||
|
|
{
|
||
|
|
client_.connect();
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
void onConnection(const TcpConnectionPtr& conn)
|
||
|
|
{
|
||
|
|
if (conn->connected())
|
||
|
|
{
|
||
|
|
//channel_.reset(new RpcChannel(conn));
|
||
|
|
channel_->setConnection(conn);
|
||
|
|
total_ = 4;
|
||
|
|
resolve("www.example.com");
|
||
|
|
resolve("www.chenshuo.com");
|
||
|
|
resolve("www.google.com");
|
||
|
|
resolve("acme.chenshuo.org");
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
loop_->quit();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void resolve(const std::string& host)
|
||
|
|
{
|
||
|
|
resolver::ResolveRequest request;
|
||
|
|
request.set_address(host);
|
||
|
|
resolver::ResolveResponse* response = new resolver::ResolveResponse;
|
||
|
|
|
||
|
|
stub_.Resolve(NULL, &request, response,
|
||
|
|
NewCallback(this, &RpcClient::resolved, response, host));
|
||
|
|
}
|
||
|
|
|
||
|
|
void resolved(resolver::ResolveResponse* resp, std::string host) // pass by value for NewCallback above
|
||
|
|
{
|
||
|
|
if (resp->resolved())
|
||
|
|
{
|
||
|
|
char buf[32];
|
||
|
|
uint32_t ip = resp->ip(0);
|
||
|
|
inet_ntop(AF_INET, &ip, buf, sizeof buf);
|
||
|
|
|
||
|
|
LOG_INFO << "resolved " << host << " : " << buf << "\n"
|
||
|
|
<< resp->DebugString();
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
LOG_INFO << "resolved " << host << " failed";
|
||
|
|
}
|
||
|
|
|
||
|
|
if (++got_ >= total_)
|
||
|
|
{
|
||
|
|
client_.disconnect();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
EventLoop* loop_;
|
||
|
|
TcpClient client_;
|
||
|
|
RpcChannelPtr channel_;
|
||
|
|
int got_;
|
||
|
|
int total_;
|
||
|
|
resolver::ResolverService::Stub stub_;
|
||
|
|
};
|
||
|
|
|
||
|
|
int main(int argc, char* argv[])
|
||
|
|
{
|
||
|
|
LOG_INFO << "pid = " << getpid();
|
||
|
|
if (argc > 1)
|
||
|
|
{
|
||
|
|
EventLoop loop;
|
||
|
|
InetAddress serverAddr(argv[1], 2053);
|
||
|
|
|
||
|
|
RpcClient rpcClient(&loop, serverAddr);
|
||
|
|
rpcClient.connect();
|
||
|
|
loop.loop();
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
printf("Usage: %s host_ip\n", argv[0]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|