#include <stdio.h>
#include <unistd.h>

#include <utility>

#include "muduo/base/Logging.h"
#include "muduo/net/EventLoop.h"
#include "muduo/net/InetAddress.h"
#include "muduo/net/TcpClient.h"

using namespace muduo;
using namespace muduo::net;

class ChargenClient : noncopyable {
public:
    ChargenClient(EventLoop* loop, const InetAddress& listenAddr)
        : loop_(loop), client_(loop, listenAddr, "ChargenClient")
    {
        client_.setConnectionCallback(
            std::bind(&ChargenClient::onConnection, this, _1));
        client_.setMessageCallback(
            std::bind(&ChargenClient::onMessage, this, _1, _2, _3));
        // client_.enableRetry();
    }

    void connect() { client_.connect(); }

private:
    void onConnection(const TcpConnectionPtr& conn)
    {
        LOG_INFO << conn->localAddress().toIpPort() << " -> "
                 << conn->peerAddress().toIpPort() << " is "
                 << (conn->connected() ? "UP" : "DOWN");

        if (!conn->connected()) loop_->quit();
    }

    void onMessage(const TcpConnectionPtr& conn, Buffer* buf,
                   Timestamp receiveTime)
    {
        buf->retrieveAll();
    }

    EventLoop* loop_;
    TcpClient  client_;
};

int main(int argc, char* argv[])
{
    LOG_INFO << "pid = " << getpid();
    if (argc > 1) {
        EventLoop   loop;
        InetAddress serverAddr(argv[1], 2019);

        ChargenClient chargenClient(&loop, serverAddr);
        chargenClient.connect();
        loop.loop();
    } else {
        printf("Usage: %s host_ip\n", argv[0]);
    }
}