132 lines
3.7 KiB
C++
132 lines
3.7 KiB
C++
/*
|
|
* 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);
|
|
}
|