socket使用非阻塞connect

在使用tcp的connect调用时，默认是使用阻塞方式，当服务器当前不可用时，connect会等待(内部在重试？)直到超时时间到达，而这个超时时间是系统内核规定的，不能使用setSocketOpt来设置。

在碰到服务器不可用，上层逻辑进行重试时，如果超时时间过长，会产生卡死的感觉，用户体验也不佳，所以需要控制connect的超时时间。

参考网络上的资料，这里使用select。实现方式是：将socket设置为非阻塞方式，使用select来轮询socket，在select里指定超时时间，根据socket来判断连接状态。最后恢复socket的阻塞方式。

代码如下(linux)：

int connect_with_timeout(int socket, const struct sockaddr *address, socklen_t address_len, int time_out)

{

    int flag, old_flag;

    old_flag = flag = fcntl(socket, F_GETFL, );

    flag |= O_NONBLOCK;

    fcntl(socket, F_SETFL, flag);

    int ret = -;

    ret = ::connect(socket, (struct sockaddr*)address, address_len);

    if (ret != ) {

        if (errno != EINPROGRESS) {

            LOG("connect failed,err(%d)", errno);

        } else {

            struct timeval tm;

            tm.tv_sec = time_out;

            tm.tv_usec = ;

            fd_set set,rset;

            FD_ZERO(&set);

            FD_ZERO(&rset);

            FD_SET(socket, &set);

            FD_SET(socket, &rset);

            int res;

            res = ::select(socket+, &rset, &set, NULL, &tm);

            if (res < ) {

                LOG("select:network error in connect.errno:%d", errno);

            } else if(res == ) {

                LOG("select:connect timeout.errno:%d", errno);

            } else if (res == ) {

                if (FD_ISSET(socket, &set)) {

                    LOG("select success");

                    ret = ;

                }

            } else {

                LOG("other error when select: %s", strerror(errno));

            }

        }

    }

    fcntl(socket, F_SETFL, old_flag);

    return ret;

}

::connect在非阻塞模式下会立即返回，如果没有其他错误，返回值等于0。

当::connect不能立即建立连接时，会返回EINPROGRESS，表示正在连接的过程中，这时可以使用select去轮询套接口，而select超时时间由参数指定。

select返回值小于0，表明connect出错；等于0，表明connect超时；等于1，并且套接口的状态是可写，则表明connect已经成功建立。

最后恢复socket的阻塞属性。

参考：

http://olive101.blog.163.com/blog/static/2051263201011221915696/

巴特西

socket使用非阻塞connect

最新文章

热门文章