C++网络编程实战：如何优雅地控制TCP连接断开与重连机制

1次阅读

没有评论

共计 2122 个字符，预计需要花费 6 分钟才能阅读完成。

在网络编程中，TCP 连接的异常断开是开发者经常面临的棘手问题。本文将深入探讨 C ++ 中如何通过 socket 选项设置、心跳包机制以及非阻塞 IO 等技术手段，实现稳定可靠的连接断开控制。

网络抖动导致的意外断开 ：在网络不稳定的情况下，连接可能会意外断开，而应用程序可能无法立即感知。
半开连接检测 ：当一方已经断开连接，但另一方仍认为连接有效时，会导致数据发送失败或超时。
资源泄漏风险 ：未正确关闭的连接会导致文件描述符泄漏，进而影响系统性能。

阻塞式 IO：简单易用，但在连接断开时可能导致线程阻塞，难以实现超时控制。
非阻塞式 IO：灵活性高，可以通过 select/poll 等机制实现超时检测，但代码复杂度较高。

默认行为 ：TCP 的 SO_KEEPALIVE 选项可以检测死连接，但默认时间较长（通常为 2 小时）。
调优建议 ：可以通过设置 TCP_KEEPIDLE、TCP_KEEPINTVL 等参数来缩短检测时间。

心跳包机制 ：通过定期发送小数据包来检测连接是否存活。
优点：灵活性高，可以根据业务需求调整心跳间隔。
缺点：增加了网络流量和 CPU 开销。

#include <fcntl.h>
#include <unistd.h>
#include <sys/socket.h>
#include <cerrno>

int set_nonblocking(int fd) {int flags = fcntl(fd, F_GETFL, 0);
    if (flags == -1) {return -1;}
    flags |= O_NONBLOCK;
    if (fcntl(fd, F_SETFL, flags) == -1) {return -1;}
    return 0;
}

#include <sys/select.h>
#include <sys/time.h>

bool is_connected(int fd, int timeout_sec) {
    fd_set read_set;
    FD_ZERO(&read_set);
    FD_SET(fd, &read_set);

    fd_set write_set;
    FD_ZERO(&write_set);
    FD_SET(fd, &write_set);

    timeval timeout;
    timeout.tv_sec = timeout_sec;
    timeout.tv_usec = 0;

    int result = select(fd + 1, &read_set, &write_set, nullptr, &timeout);
    if (result == -1) {
        // 处理错误
        return false;
    } else if (result == 0) {
        // 超时
        return false;
    }

    // 检查 socket 是否可读或可写
    if (FD_ISSET(fd, &read_set) || FD_ISSET(fd, &write_set)) {
        int error = 0;
        socklen_t len = sizeof(error);
        if (getsockopt(fd, SOL_SOCKET, SO_ERROR, &error, &len) == -1) {return false;}
        return error == 0;
    }
    return false;
}

class TcpClient {
public:
    TcpClient() : fd_(-1) {}
    ~TcpClient() {if (fd_ != -1) {close(fd_);
            fd_ = -1;
        }
    }

    bool connect(const std::string& host, int port, int timeout_sec) {if (fd_ != -1) {close(fd_);
            fd_ = -1;
        }

        fd_ = socket(AF_INET, SOCK_STREAM, 0);
        if (fd_ == -1) {return false;}

        // 设置非阻塞
        if (set_nonblocking(fd_) == -1) {close(fd_);
            fd_ = -1;
            return false;
        }

        // 连接逻辑
        // ...

        // 检测连接状态
        return is_connected(fd_, timeout_sec);
    }

private:
    int fd_;
};