方案一:之前有设计中转服务器,用于转发redis、url等消息。
在这里面,专门开线程负责套接字的连接与重连。使用阻塞等待式的方式直到连接真正连上,效率低下。代码如下:
bool Connect(int &sock, sockaddr *addr_ptr, int wait_time)
{
int connect_res = ::connect(sock, addr_ptr, sizeof(sockaddr));
if(0 != connect_res && errno != EINPROGRESS)
return false;
int ep_fd = epoll_create(10240);
struct epoll_event ev = {0};
ev.data.fd = sock;
ev.events = EPOLLOUT;
if(epoll_ctl(ep_fd, EPOLL_CTL_ADD, sock, &ev) == -1)
{
return false;
}
struct epoll_event events[10];
while(wait_time--)
{
int ret = epoll_wait(ep_fd, events, 10, 100);
if(0 < ret)
{
if(events[0].events & (EPOLLERR | EPOLLHUP))
{
Destroy(ep_fd);
return false;
}
else if(events[0].events & EPOLLOUT)
{
connect_res = connect(sock, addr_ptr, sizeof(sockaddr));
if(0 == connect_res || errno == EISCONN)
{
Destroy(ep_fd);
return true;
}
else if(errno == EALREADY || errno == EINPROGRESS)
{
continue;
}
else
{
Destroy(ep_fd);
return false;
}
}
}
else if(0 == ret)
{
continue;
}
else
{
Destroy(ep_fd);
return false;
}
}
Destroy(ep_fd);
return false;
}
方案二:
由于网络的不可预测性,connect可能会出现微秒到几十秒的连接时长,也可能出现连接拒绝连接不可达等现象。如何尽可能少的占用资源,且兼顾所有这些情况。代码如下:
bool Connect(int &sock, sockaddr *addr_ptr)
{
int connect_res = ::connect(sock, (sockaddr *)&addr_ptr, sizeof(sockaddr));
if(0 != connect_res && errno != EINPROGRESS && errno != EISCONN)
{
close(sock);
return false;
}
return true;
}
看似很简单,为什么可以这样做?
正常connect后,不管对方的地址是否可达,你的网络地址断开或正常情况等,返回的都是EINPROGRESS,连接正在处理中。
这时直接返回true认为它成功了,实际上很有可能是不成功的,如何验证:
当有数据需要发送时,需要判断此套接字是否可写,这时如拔掉网线,套接字将会一直不可写(本地测试大约十几秒后套接字变成可写的)。当可写时,可通过send返回值判定该套接字是否是有效套接字。如果异常再重新连接。这样connect的异步返回和需要发送时判定可写状态以及发送返回值就可以确定是否真正连接上,占用的资源很少,效率很高。
测试connect连接主要代码如图:
当正常时:
当拔掉网线时:
当拔掉又重新插上时:有可能会连接成功,也有可能连接不成功