TCP连接connect等待时长控制的另一种方法
原文地址:http://blog.csdn.net/slin000/article/details/3971644
在TCP网络应用开发中,作为客户端的程序经常需要主动连接服务器,这时你就需要建立一个Socket,然后调用connect函数连接到服务器地址。正常情况下,这并没有什么问题,但当服务器主机不存在的时候,connect函数可能会等待一分多钟才能返回。如果在主线程中调用connect函数,就会产生长时间无法响应的状况。
在现代的互联网硬件环境中,一分钟的等待有点太长了,我们需要缩短等待时间。
在Linux环境下,可以用alarm调用定时唤醒正在等待的线程,使connect函数从等待中返回,但在Windows下我没有找到类似的函数。如何让connect函数返回呢?
经过实验,找到一个简单的方法:直接关闭connect函数使用的那个socket套接字,connect函数就会立即返回。这个方法感觉土了点,但确实管用。该方法的工作过程描述如下:
1) 创建socket
2) 启动定时关闭该socket的线程
3) 调用connect函数连接服务器
4) 取消定时关闭线程的工作
5) 检查定时关闭线程的关闭操作是否已经执行
6) 检查connect返回值是否有效
摘录一段示例代码如下:
SOCKET CTCPConnector::ConnectTo(int toIp, int toPort, int localIp , int localPort, int timeOut) { SOCKET Socket = socket(AF_INET, SOCK_STREAM, 0); sockaddr_in InetAddr; InetAddr.sin_family = AF_INET; InetAddr.sin_addr.s_addr = htonl(localIp); InetAddr.sin_port = htons(localPort); if (localIp > 0 && localPort>0) { if (bind(Socket, (sockaddr *) &InetAddr, sizeof(InetAddr)) < 0) return INVALID_SOCKET; } InetAddr.sin_addr.s_addr = htonl(toIp); InetAddr.sin_port = htons(toPort); CTimeOutClose Closer(Socket); // 这个是超时关闭线程 if (timeOut > 0) Closer.SetTimeOut(timeOut);// 设定超时时长 int err = connect(Socket, (const sockaddr *)&InetAddr, sizeof(InetAddr)); if (timeOut >= 0) { Closer.Cancel(); // 取消超时关闭 if (Closer.HasDone() // 检查定时关闭线程的关闭操作是否已经执行 && err >= 0) // { err = -1; } } if (err < 0) // 检查connect返回值是否有效 { return INVALID_SOCKET; } return Socket; }
代码中CTimeOutClose类是启动关闭线程,等待一段时间后关闭指定的套接字。同时,该类还提供接口,用于取消操作和检查操作是否已经执行。
虽然该方法需要启动一个新的线程,但对于大多数的应用来说,主动发起建立TCP连接的量都不会太多,所以对程序的性能并不会产生明显影响。
(以上代码可以在我上传的资源“回城卷轴网络通讯架构源代码”中找到,下载地址为:http://download.csdn.net/source/1023342)
作者:苏林