node_connection.h

//========================================================================
#ifndef comfrm_cls_NodeConnectionH
#define comfrm_cls_NodeConnectionH
//========================================================================
#include "node_protocol.h"

#include <boost/asio.hpp>
#include <functional>
//========================================================================
namespace com {
    namespace interproc {
        class NodeConnection : public std::enable_shared_from_this< NodeConnection > // to allow the pointer to stay alive
        {
            enum NodeConnectionState
            {
                ncsNotConnected,
                ncsConnected,
                ncsReadyForMessage,
            };

        public:
            typedef std::function< void (std::shared_ptr< NodeConnection >) > OnSocketClosedClb;

        public:
            NodeConnection(const NodeConnection&) = delete;
            ~NodeConnection();

            static std::shared_ptr< NodeConnection > createConnection(const boost::asio::ip::address& p_ipAddr,
                                                                            boost::asio::io_service& p_ioService,
                                                                            std::function< void (std::shared_ptr< NodeConnection >, std::vector< char >&) > p_msgCallback,
                                                                            OnSocketClosedClb p_socketClosedClb);
            static std::shared_ptr< NodeConnection > createServerConnection(boost::asio::io_service& p_ioService,
                                                                                  std::function< void (std::shared_ptr< NodeConnection >, std::vector< char >&) > p_msgCallback,
                                                                                  OnSocketClosedClb p_socketClosedClb);

            void prepareToRead();
            void send(const std::vector< char >& p_data);
            NodeConnectionState getState() const;
            boost::asio::ip::address getIpAddr() const;
            boost::asio::ip::tcp::socket& getSocket();
            void setSocketClosedCallback(OnSocketClosedClb p_clb);

        private:
            NodeConnection(boost::asio::io_service& p_ioService,
                               std::function< void (std::shared_ptr< NodeConnection >, std::vector< char >&) > p_msgCallback,
                               OnSocketClosedClb p_socketClosedClb); // строить этот объект можно только через instance
            void connect(const boost::asio::ip::address& p_ipAddr);
            //void onConnect(const boost::system::error_code& p_error);
            void onRead(const boost::system::error_code& p_error, std::size_t p_bytesTransferred);
            //void onWrite(const boost::system::error_code& p_error, std::size_t p_bytesTransferred);
            void processReceivedBytes(std::vector< char >& p_bytes);
            bool fillMessageBuffer(std::vector< char >& p_bytes, size_t& o_restBytesCount); // возвращает false, если нужно вычитать больше байт
            void setState(NodeConnectionState p_state);
            void stop();

        protected:
            boost::asio::ip::tcp::socket m_socket;
            NodeConnectionState m_state;
            std::vector< char > m_readBuffer;
            //std::vector< char > m_writeBuffer;
            std::vector< char > m_messageBuffer;
        //    std::function< void (std::shared_ptr< cls_AnyMap >, std::string ) > m_anyMapCallback;
        //    std::function< void (cls_NodeRegistration) > m_regCallback;
            std::function< void (std::shared_ptr< NodeConnection >, std::vector< char >&) > m_msgCallback;
            OnSocketClosedClb m_socketClosedCallback;
            NodeProtocol m_protocol;
        };
    }
}

//========================================================================
#endif
//========================================================================