52 lines
1.0 KiB
C++
52 lines
1.0 KiB
C++
|
#include "examples/cdns/Resolver.h"
|
||
|
|
#include "muduo/net/EventLoop.h"
|
||
|
|
#include <stdio.h>
|
||
|
|
|
||
|
|
using namespace muduo;
|
||
|
|
using namespace muduo::net;
|
||
|
|
using namespace cdns;
|
||
|
|
|
||
|
|
EventLoop* g_loop;
|
||
|
|
int count = 0;
|
||
|
|
int total = 0;
|
||
|
|
|
||
|
|
void quit()
|
||
|
|
{
|
||
|
|
g_loop->quit();
|
||
|
|
}
|
||
|
|
|
||
|
|
void resolveCallback(const string& host, const InetAddress& addr)
|
||
|
|
{
|
||
|
|
printf("resolveCallback %s -> %s\n", host.c_str(), addr.toIpPort().c_str());
|
||
|
|
if (++count == total)
|
||
|
|
quit();
|
||
|
|
}
|
||
|
|
|
||
|
|
void resolve(Resolver* res, const string& host)
|
||
|
|
{
|
||
|
|
res->resolve(host, std::bind(&resolveCallback, host, _1));
|
||
|
|
}
|
||
|
|
|
||
|
|
int main(int argc, char* argv[])
|
||
|
|
{
|
||
|
|
EventLoop loop;
|
||
|
|
loop.runAfter(10, quit);
|
||
|
|
g_loop = &loop;
|
||
|
|
Resolver resolver(&loop,
|
||
|
|
argc == 1 ? Resolver::kDNSonly : Resolver::kDNSandHostsFile);
|
||
|
|
if (argc == 1)
|
||
|
|
{
|
||
|
|
total = 3;
|
||
|
|
resolve(&resolver, "www.chenshuo.com");
|
||
|
|
resolve(&resolver, "www.example.com");
|
||
|
|
resolve(&resolver, "www.google.com");
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
total = argc-1;
|
||
|
|
for (int i = 1; i < argc; ++i)
|
||
|
|
resolve(&resolver, argv[i]);
|
||
|
|
}
|
||
|
|
loop.loop();
|
||
|
|
}
|