/*
 * TcpClientEventLoop_test.cpp
 *
 * @build make evpp
 * @server bin/TcpServer_test 1234
 * @client bin/TcpClientEventLoop_test 1234
 *
 */

#include <iostream>

#include "TcpClient.h"
#include "htime.h"

#define TEST_RECONNECT 1
#define TEST_TLS 0

using namespace hv;

class MyTcpClient : public TcpClient {
public:
 MyTcpClient(EventLoopPtr loop = NULL) : TcpClient(loop) {
 onConnection = [this](const SocketChannelPtr& channel) {
 std::string peeraddr = channel->peeraddr();
 if (channel->isConnected()) {
 printf("connected to %s! connfd=%d\n", peeraddr.c_str(), channel->fd());
 // send(time) every 3s
 setInterval(3000, [channel](TimerID timerID){
 if (channel->isConnected()) {
 if (channel->isWriteComplete()) {
 char str[DATETIME_FMT_BUFLEN] = {0};
 datetime_t dt = datetime_now();
 datetime_fmt(&dt, str);
 channel->write(str);
 }
 } else {
 killTimer(timerID);
 }
 });
 } else {
 printf("disconnected to %s! connfd=%d\n", peeraddr.c_str(), channel->fd());
 }
 if (isReconnect()) {
 printf("reconnect cnt=%d, delay=%d\n", reconn_setting->cur_retry_cnt, reconn_setting->cur_delay);
 }
 };

 onMessage = [](const SocketChannelPtr& channel, Buffer* buf) {
 printf("< %.*s\n", (int)buf->size(), (char*)buf->data());
 };
 }

 int connect(int port) {
 int connfd = createsocket(port);
 if (connfd < 0) {
 return connfd;
 }
#if TEST_RECONNECT
 // reconnect: 1,2,4,8,10,10,10...
 reconn_setting_t reconn;
 reconn_setting_init(&reconn);
 reconn.min_delay = 1000;
 reconn.max_delay = 10000;
 reconn.delay_policy = 2;
 setReconnect(&reconn);
#endif

#if TEST_TLS
 withTLS();
#endif
 printf("client connect to port %d, connfd=%d ...\n", port, connfd);
 start();
 return connfd;
 }
};
typedef std::shared_ptr<MyTcpClient> MyTcpClientPtr;

int TestMultiClientsRunInOneEventLoop(int port, int nclients) {
 auto loop_thread = std::make_shared<EventLoopThread>();
 loop_thread->start();

 std::map<int, MyTcpClient*> clients;
 for (int i = 0; i < nclients; ++i) {
 MyTcpClient* client = new MyTcpClient(loop_thread->loop());
 client->connect(port);
 clients[i] = client;
 }

 std::string str;
 while (std::getline(std::cin, str)) {
 if (str == "close") {
 for (auto& pair : clients) {
 MyTcpClient* client = pair.second;
 client->closesocket();
 }
 } else if (str == "delete") {
 for (auto& pair : clients) {
 MyTcpClient* client = pair.second;
 client->deleteInLoop();
 }
 break;
 } else {
 for (auto& pair : clients) {
 MyTcpClient* client = pair.second;
 client->send(str);
 }
 }
 }

 printf("Press Enter key to exit loop.\n");
 while (getchar() != '\n');
 loop_thread->stop();
 loop_thread->join();

 return 0;
}

int main(int argc, char* argv[]) {
 if (argc < 2) {
 printf("Usage: %s port [nclients]\n", argv[0]);
 return -10;
 }
 int port = atoi(argv[1]);

 int nclients = 100;
 if (argc > 2) {
 nclients = atoi(argv[2]);
 }

 return TestMultiClientsRunInOneEventLoop(port, nclients);
}