简单实现TCP客户端重连机制

实现在服务端可能不定时离线的情况下，客户端自动连接服务端

#ifndef _TCP_CLIENT
#define _TCP_CLIENT

#include <errno.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>

#include <atomic>
#include <thread>

#include "log/log.h"

class TcpClient {
   public:
    TcpClient()
        : mThread([this]() {
              while (this->sockInit() == -1) {
                  LOG_INFO("try to sockInit again");
                  sleep(1);
              }
              mIsSockInit = true;
          }) {}

    int sockInit() {
        // create sock fd
        if ((mFd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
            LOG_ERROR("create socket failed");
            return -1;
        }

        // connect to server
        bzero(&mSin, sizeof(mSin));  //初始值置零
        mSin.sin_family = AF_INET;
        mSin.sin_port = htons(mPort);  //转化为NBD

        if (inet_pton(AF_INET, mIpAddr.c_str(), (void*)&mSin.sin_addr.s_addr) !=
            1) {
            LOG_ERROR("inet_pton");
            return -1;
        }

        if (connect(mFd, (struct sockaddr*)&mSin, sizeof(mSin)) < 0) {
            LOG_ERROR("connect failed");
            return -1;
        }
        LOG_INFO("Client starting ...");
        return 0;
    }

    int sendData(const char* data, size_t len) {
        if (mIsSockInit) {
            // judge if the connection is broken
            int infoLen = sizeof(mInfo);
            getsockopt(mFd, IPPROTO_TCP, TCP_INFO, &mInfo,
                       (socklen_t*)&infoLen);
            if ((mInfo.tcpi_state == TCP_ESTABLISHED)) {
                int ret = 0;
                if (data != nullptr && len != 0) {
                    int sendLen = 0;
                    while (sendLen != len) {
                        int remainLen = len - sendLen;
                        ret =
                            send(mFd, data + sendLen, remainLen, MSG_NOSIGNAL);
                        sendLen += ret;
                        LOG_INFO("send len {}", ret);
                    }
                }
            } else {
                LOG_WARN("socket Reconnecting....");
                close(mFd);
                sleep(1);
                sockInit();
                LOG_INFO("Socket Reconnnecting success!");
            }
        }
    }

   private:
    int mFd{-1};

    std::thread mThread;
    std::string mIpAddr{"192.168.0.174"};
    int mPort{8001};
    struct sockaddr_in mSin;
    struct tcp_info mInfo;
    std::atomic<bool> mIsSockInit{false};
};

#endif

 在构造函数中，启动一个线程，用于初始化socket连接，如果服务端不在线或有其他故障就一直重复初始化知道connect成功。第一次连接成功后，将mIsSockInit标志置为真。发送数据sendData一般会在另一个数据源线程中循环发送，判断第一次连接成功后就尝试发送数据。在每次发送数据时也要检测连接是否保持，断开的话就重新连接。

send函数中有两点需要注意：1

1. 当IO缓冲区的大小bufSize小于发送的长度len时，send函数实际发送的长度可能会小于len。因此有必要对send的返回值进行校验，将每一次数据包的数据都发送出去。

2. 在linux下编写TCP socket程序时，如果某一端突然退出，导致连接中断，这个时候另一端端如果继续调用send函数发送数据的话，会引发一个信号SIGPIPE，此信号会导致整个进程退出。把flags设置为MSG_NOSIGNAL，则不会导致信号退出。