DatabaseConnection.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 "JobTypes.h"
 
#include <string>
#include <cassandra.h>
 
#define IP "cassandra"
#define DBPORT 8002
#define UPDATE_JOBS_TIMEOUT 300000000 // 5 minutes.
#define MAX_JOB_RETRIES 1
#define RECOUNT_WAIT_TIME 600
 
using namespace jobTypes;
 
class DatabaseConnection
{
public:
    virtual void connect(std::string ip, int port);
 
    virtual void uploadJob(Job job, bool newJob);
 
    virtual Job getTopJob();
 
    virtual Job getCurrentJob(std::string jobid);
 
    virtual long long getCurrentJobTime(std::string jobid);
 
    virtual long long addCurrentJob(Job job);
 
    virtual void addFailedJob(FailedJob job);
 
    virtual int getNumberOfJobs();
 
    virtual int getCrawlID();
 
    virtual void setCrawlID(int id);
 
    virtual void updateCurrentJobs();
 
private:
    void deleteTopJob(CassUuid id, cass_int64_t priority);
 
    void deleteCurrentJob(CassUuid id);
 
    long long addCurrentJob(CassUuid id, Job job);
 
    Job retrieveCurrentJob(const CassRow *row);
 
    void setPreparedStatements();
 
    virtual void updateCurrentJob(CassUuid jobid, Job job, long long currentTime);
 
    CassSession *connection;
 
    int numberOfJobs;
    long long timeLastRecount = -1;
 
    const CassPrepared *preparedGetTopJob;
    const CassPrepared *preparedDeleteTopJob;
    const CassPrepared *preparedAddCurrentJob;
    const CassPrepared *preparedGetCurrentJob;
    const CassPrepared *preparedGetCurrentJobs;
    const CassPrepared *preparedDeleteCurrentJob;
    const CassPrepared *preparedAddFailedJob;
    const CassPrepared *preparedAmountOfJobs;
    const CassPrepared *preparedUploadJob;
    const CassPrepared *preparedUploadRetryJob;
    const CassPrepared *preparedCrawlID;
    const CassPrepared *preparedUpdateCrawlID;
};