#include "examples/protobuf/resolver/resolver.pb.h"

#include "muduo/base/Logging.h"
#include "muduo/net/EventLoop.h"
#include "muduo/net/protorpc/RpcServer.h"
#include "examples/cdns/Resolver.h"

#include <unistd.h>

using namespace muduo;
using namespace muduo::net;

namespace resolver
{

class ResolverServiceImpl : public ResolverService
{
 public:
  ResolverServiceImpl(EventLoop* loop)
    : resolver_(loop, cdns::Resolver::kDNSonly)
  {
  }

  virtual void Resolve(::google::protobuf::RpcController* controller,
                       const ::resolver::ResolveRequest* request,
                       ::resolver::ResolveResponse* response,
                       ::google::protobuf::Closure* done)
  {
    LOG_INFO << "ResolverServiceImpl::Resolve " << request->address();

    bool succeed = resolver_.resolve(request->address(),
                                     std::bind(&ResolverServiceImpl::doneCallback,
                                                 this,
                                                 request->address(),
                                                 _1,
                                                 response,
                                                 done));
    if (!succeed)
    {
      response->set_resolved(false);
      done->Run();
    }
  }

 private:

  void doneCallback(const std::string& host,
                    const muduo::net::InetAddress& address,
                    ::resolver::ResolveResponse* response,
                    ::google::protobuf::Closure* done)

  {
    LOG_INFO << "ResolverServiceImpl::doneCallback " << host;
    int32_t ip = address.ipNetEndian();
    if (ip)
    {
      response->set_resolved(true);
      response->add_ip(ip);
      response->add_port(address.portNetEndian());
    }
    else
    {
      response->set_resolved(false);
    }
    done->Run();
  }

  cdns::Resolver resolver_;
};

}  // namespace resolver

int main()
{
  LOG_INFO << "pid = " << getpid();
  EventLoop loop;
  InetAddress listenAddr(2053);
  resolver::ResolverServiceImpl impl(&loop);
  RpcServer server(&loop, listenAddr);
  server.registerService(&impl);
  server.start();
  loop.loop();
}