Function RoutineConnectWithTimeout

phxsqlproxy/routineutil.cpp:22–61

namespace phxsqlproxy {

int RoutineConnectWithTimeout(int fd, const struct sockaddr *address, socklen_t address_len,
                                         int timeout_ms) {
    assert(IsNonBlock(fd));
    //1.sys call
    int ret = connect(fd, address, address_len);

    //2.wait
    int pollret = 0;
    struct pollfd pf = { 0 };

    for (int i = 0; i < 3; i++)  //25s * 3 = 75s
            {
        memset(&pf, 0, sizeof(pf));
        pf.fd = fd;
        pf.events = (POLLOUT | POLLERR | POLLHUP);

        pollret = poll(&pf, 1, timeout_ms);

        if (1 == pollret) {
            break;
        }
    }

    if (pf.revents & POLLOUT)  //connect succ
            {
        errno = 0;
        return 0;
    }

    //3.set errno
    int err = 0;
    socklen_t errlen = sizeof(err);
    getsockopt(fd, SOL_SOCKET, SO_ERROR, &err, &errlen);
    if (err) {
        errno = err;
    } else {
        errno = ETIMEDOUT;
    }
    return ret;
}

int RoutineReadWithTimeout(int source_fd, char * buf, int buf_size, int timeout_ms) {
    assert(IsNonBlock(source_fd));