TcpClient.h

Go to the documentation of this file.

 
#ifndef TcpClient_H
#define TcpClient_H
 
#include "AsioContext.h"
#include "CscCommon.h"
 
using namespace conscience_utils::logging;
 
namespace conscience_core::communication {
 
    class TcpClientRuntimeError : public runtime_error {
    public:
        TcpClientRuntimeError(const string &message) : runtime_error(message.c_str()) {}
    };
 
    class TcpClient {
    public:
        TcpClient(const string &ipAddress, unsigned int port);
 
        struct ExistingSocket {
            void * socket = nullptr;
            AsioContext *asioContext = nullptr;
 
            ExistingSocket(void *socket, AsioContext *asioContext);
        };
        TcpClient(const ExistingSocket &existingSocket, bool releaseAsioContext = false);
        ~TcpClient();
 
        string getIpAddress() const;
        unsigned int getPort() const;
        optional<vector<unsigned char>*> send(const vector<unsigned char> *bytes, bool noResponse = false, bool close = false);
        optional<vector<unsigned char>*> receive();
 
        void release();
        bool isReleased() const;
 
        optional<unsigned int> getTimeout() const;
 
        void setTimeout(unsigned timeoutSeconds);
 
        void scheduleCloseAfterTimeout();
        void cancelCloseAfterTimeout();
 
        class AsioState {
        public:
            bool closing = false;
            void *socket = nullptr;
            void *closeAfterTimeoutTimer = nullptr;
            void *strand = nullptr;
            AsioContext *asioContext = nullptr;
            bool releaseAsioContext = true;
            string loggerCategory;
            AsioState(void *socket, AsioContext *asioContext, bool releaseAsioContext, const string &loggerCategory);
            ~AsioState();
        };
    private:
        inline static std::atomic<unsigned long long> counter = 0;
        string loggerCategory = "TcpClient_" + to_string(++counter);
        unique_ptr<CscLogger> logger = CscLogger::getForCategory(loggerCategory);
 
 
        // IMPORTANT: do not touch this shared ptr. Boost ASIO lifecycle is very tricky and we NEED this shared ptr to clean close
        ptr<AsioState> asioState = nullptr;
        const string ipAddress = "";
        optional<unsigned> timeoutSeconds = {};
 
        bool released = false;
 
        vector<unsigned char> bufferedBytes;
 
 
 
 
    };
}
#endif