TCP连接connect等待时长控制的另一种方法

在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.youkuaiyun.com/source/1023342）

作者：苏林