tcpServer.h

Go to the documentation of this file.

#include <ctime>
#include <iostream>
#include <string>
#include <set>
#include <boost/bind/bind.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <boost/asio.hpp>
#include <boost/function.hpp>
#include <boost/thread.hpp>
#include "dpyError.h"

#define MAX_LENGHT  1024 


class TcpConnection: public boost::enable_shared_from_this<TcpConnection>
{
private:
    boost::function<void(const boost::system::error_code& error, const char* receiveData, std::string& sendData)> mReceiveSendHandler; 
    boost::function<void(const boost::system::error_code& error, std::size_t size)> mWriteStatusHandler; 
    boost::asio::ip::tcp::socket mSocket; 

    bool mDisconnecting; 
    char mReadBuffer[MAX_LENGHT]; 

private:
    explicit TcpConnection(BOOST_IO_CONTEXT &io_context);
    void handleWrite(const boost::system::error_code& error, size_t bytes_transferred);
    void handleReceive(const boost::system::error_code& error, std::size_t bytes_transferred);

public:

    typedef boost::shared_ptr<TcpConnection> pointer;

    static pointer create(BOOST_IO_CONTEXT& io_context)
    {
        return pointer(new TcpConnection(io_context));
    }
    
    boost::asio::ip::tcp::socket& socket(); 
public:


    bool isDisconnected();
    void disconnect();
    void write(std::string sendData);
    void receive();
    void setWriteStatusHandler(boost::function<void(const boost::system::error_code& error, std::size_t size)> writeStatusHandler);
    void setReceiveSendHandler(boost::function<void(const boost::system::error_code& error, const char* receiveData, std::string& sendData)> receiveSendHandler);

};

class TcpAsyncServer
{
private:

    boost::asio::io_context mIoContext;
    boost::shared_ptr<boost::asio::ip::tcp::acceptor> mAcceptor; 
    boost::thread mBt; 
    boost::function<void(const boost::system::error_code& error, const char* receiveData, std::string& sendData)> mReceiveSendHandler; 
    boost::function<void(const boost::system::error_code& error, std::size_t size)> mWriteStatusHandler; 

private:
    void handleAccept(const boost::system::error_code& error, boost::shared_ptr<TcpConnection> new_connection);

    std::set<TcpConnection::pointer> sConnections;

public:

    void startAccept();

    TcpAsyncServer(boost::asio::io_context& io_context, std::string ip, unsigned short port);

    virtual ~TcpAsyncServer();

    void close(boost::system::error_code& rEc);

    void stopAll();

    void publishMsg(const std::string &message);

    void setWriteStatusHandler(boost::function<void(const boost::system::error_code& error, std::size_t size)> writeStatusHandler);
    void setReceiveSendHandler(boost::function<void(const boost::system::error_code& error, const char* receiveData, std::string& sendData)> receiveSendHandler);
};