JobRequestHandler.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 "DatabaseConnection.h"
#include "RAFTConsensus.h"
 
#include <mutex>
#include <boost/shared_ptr.hpp>
 
#define MIN_AMOUNT_JOBS 500
#define MAX_RETRIES 3
#define CRAWL_TIMEOUT_SECONDS 150
#define NO_RETRY_REASONS {10}
 
class TcpConnection;
 
class JobRequestHandler
{
public:
    JobRequestHandler(RAFTConsensus *raft, RequestHandler *requestHandler, DatabaseConnection *database, Statistics *stats,
                    std::string ip, int port);
 
    std::string handleConnectRequest(boost::shared_ptr<TcpConnection> connection, std::string request);
 
    std::string handleGetIPsRequest(std::string request, std::string client, std::string data);
 
    std::string handleUploadJobRequest(std::string request, std::string client, std::string data);
 
    std::string handleUploadJobRequest(std::string request, std::string client, std::vector<std::string> data);
 
    std::string handleGetJobRequest(std::string request, std::string client, std::string data);
 
    std::string handleUpdateJobRequest(std::string request, std::string client, std::string data);
 
    std::string handleFinishJobRequest(std::string request, std::string client, std::string data);
 
    std::string handleCrawlDataRequest(std::string request, std::string client, std::string data);
 
    int crawlID;
    long long timeLastCrawl = -1;
 
    void updateCrawlID(int id);
 
    DatabaseConnection *getDatabaseConnection()
    {
        return database;
    }
 
private:
    RAFTConsensus *raft;
    RequestHandler *requestHandler;
    DatabaseConnection *database;
    Statistics *stats;
    std::mutex jobmtx;
 
    void connectWithRetry(std::string ip, int port);
 
    Job getTopJobWithRetry();
 
    void tryUploadJobWithRetry(Job job, bool newMethod);
 
    long long getCurrentJobTimeWithRetry(std::string jobid);
 
    Job getCurrentJobWithRetry(std::string jobid);
 
    long long addCurrentJobWithRetry(Job job);
 
    void addFailedJobWithRetry(FailedJob job);
};