DatabaseHandler.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 "Types.h"
 
#include <tuple>
#include <cassandra.h>
 
#define IP "cassandra"
#define DBPORT 8002
#define HASHES_INSERT_MAX 1000
 
using namespace types;
 
class DatabaseHandler
{
public:
    virtual void connect(std::string ip, int port);
 
    virtual bool addProject(ProjectIn project);
 
    virtual void addHashToProject(ProjectIn project, int index);
 
    virtual ProjectOut searchForProject(ProjectID projectID, Version version);
 
    virtual ProjectOut prevProject(ProjectID projectID);
 
    virtual void addMethod(MethodIn method, ProjectIn project, long long prevVersion, long long parserVersion,
                           bool newProject);
 
    virtual std::vector<Hash> updateUnchangedFiles(std::vector<Hash> hashes, std::vector<std::string> files,
                                                   ProjectIn project, long long prevVersion);
 
    virtual std::vector<MethodOut> hashToMethods(std::string hash);
 
    virtual Author idToAuthor(std::string id);
 
    virtual std::vector<MethodID> authorToMethods(std::string authorID);
 
private:
    void addMethodByAuthor(CassUuid authorID, MethodIn method, ProjectIn project);
 
    ProjectOut getProject(const CassRow *row);
 
    MethodOut getMethod(const CassRow *row);
 
    MethodID getMethodID(const CassRow *row);
 
    void addNewMethod(MethodIn method, ProjectIn project, long long parserVersion);
 
    void updateMethod(MethodIn method, ProjectIn project, long long startVersion);
 
    void handleSelectMethodQueryResult(CassFuture *queryFuture, MethodIn method, ProjectIn project,
                                       long long prevVersion, long long parserVersion, bool &newProject);
 
    std::vector<Hash> handleSelectUnchangedMethodsResult(CassFuture *queryFuture, ProjectIn project,
                                                         long long prevVersion);
 
    CassFuture *executeSelectUnchangedMethodsQuery(std::vector<Hash> hashes, std::vector<std::string> files,
                                                   ProjectIn project);
    CassUuid createAuthorIfNotExists(Author author);
 
    void setPreparedStatements();
 
    CassSession *connection;
 
    const CassPrepared *selectMethods;
    const CassPrepared *selectProject;
    const CassPrepared *selectPrevProject;
    const CassPrepared *insertProject;
    const CassPrepared *addHashesToProject;
    const CassPrepared *insertMethod;
    const CassPrepared *insertVulnMethod;
    const CassPrepared *updateMethods;
    const CassPrepared *updateUnchangedMethods;
    const CassPrepared *selectMethod;
    const CassPrepared *selectUnchangedMethods;
    const CassPrepared *insertMethodByAuthor;
    const CassPrepared *selectMethodByAuthor;
    const CassPrepared *insertAuthorByID;
    const CassPrepared *selectAuthorByID;
};