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;
    std::function<void(pointer)> mOnDisconnect;
    void setDisconnectHandler(std::function<void(pointer)> h);
    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);
};