RequestHandler.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 "DatabaseHandler.h"
#include "JobRequestHandler.h"
#include "DatabaseRequestHandler.h"
#include "RAFTConsensus.h"
#include "DatabaseConnection.h"
#include "Statistics.h"
 
#include <boost/shared_ptr.hpp>
 
class TcpConnection;
 
enum ERequestType
{
    eUpload,
    eCheck,
    eCheckUpload,
    eConnect,
    eGetIPs,
    eUploadJob,
    eUploadCrawlData,
    eGetTopJob,
    eUpdateJob,
    eFinishJob,
    eExtractProjects,
    eGetAuthor,
    eGetMethodByAuthor,
    eGetPrevProjectsRequest,
    eUnknown
};
 
 
class RequestHandler
{
public:
    virtual void initialize(DatabaseHandler *databaseHandler, DatabaseConnection *databaseConnection,
                            RAFTConsensus *raft, Statistics *stats, std::string ip = IP, int port = DBPORT);
 
    virtual std::string handleRequest(std::string requestType, std::string client, std::string request,
                                      boost::shared_ptr<TcpConnection> connection);
 
    JobRequestHandler *getJobRequestHandler()
    {
        return jrh;
    }
 
private:
    std::string handleUnknownRequest();
 
    std::string handleNotImplementedRequest();
 
    ERequestType getERequestType(std::string requestType);
 
    DatabaseRequestHandler *dbrh;
    JobRequestHandler *jrh;
};