簡單實現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，則不會導致信號退出。