ConnectionHandler.h

/*
This program has been developed by students from the bachelor Computer Science at
Utrecht University within the Software Project course.
© Copyright Utrecht University (Department of Information and Computing Sciences)
*/
 
#pragma once
#include "RequestHandler.h"
#include "RAFTConsensus.h"
#include "Statistics.h"
 
#include <boost/shared_ptr.hpp>
#include <boost/enable_shared_from_this.hpp>
#include <boost/asio.hpp>
 
#define PORT 8003
#define CONNECTION_TIMEOUT 10000000 // Timeout in microseconds.
 
using boost::asio::ip::tcp;
 
class TcpServer;
 
class ConnectionHandler
{
public:
    ~ConnectionHandler();
    void startListen(DatabaseHandler* databaseHandler, 
        DatabaseConnection* databaseConnection,
        RAFTConsensus *raft,
        Statistics *stats,
        int port = PORT, 
        RequestHandler *handler = nullptr);
 
    RequestHandler* getRequestHandler() { return handler; };
private:
    RequestHandler* handler;
    TcpServer* server;
};
 
class TcpConnection
    : public boost::enable_shared_from_this<TcpConnection>
{
public:
    typedef boost::shared_ptr<TcpConnection> pointer;
 
    static pointer create(boost::asio::io_context& ioContext);
 
    tcp::socket& socket()
    {
        return socket_;
    }
 
    virtual void sendData(const std::string &data, boost::system::error_code &error);
 
    virtual void start(RequestHandler *handler, pointer thisPointer, Statistics *stats);
 
protected:
    TcpConnection(boost::asio::io_context& ioContext)
        : socket_(ioContext)
    {
    }
 
private:
    void readExpectedData(int &size, std::vector<char> &data, std::string &totalData, 
                          boost::system::error_code &error);
 
 
    tcp::socket socket_;
    std::string message_;
};
 
class TcpServer
{
public:
    TcpServer(boost::asio::io_context &ioContext, DatabaseHandler *databaseHandler,
            DatabaseConnection *databaseConnection, RAFTConsensus *raft, RequestHandler *handler, int port,
            Statistics *stats);
 
    void stop();
 
private:
    void startAccept();
 
    void handleAccept(TcpConnection::pointer newConnection, const boost::system::error_code& error);
 
    boost::asio::io_context& ioContext_;
    tcp::acceptor acceptor_;
    RequestHandler* handler;
    Statistics *stats;
 
    bool stopped = false;
};