DatabaseDao.cpp

Go to the documentation of this file.

//SpikeStream includes
#include "DatabaseDao.h"
#include "SpikeStreamDBException.h"
#include "SpikeStreamIOException.h"
using namespace spikestream;

//Qt includes
#include <QTextStream>

//Other includes
#include <iostream>
using namespace std;


DatabaseDao::DatabaseDao(const DBInfo& dbInfo) : AbstractDao(dbInfo){
}


DatabaseDao::DatabaseDao() : AbstractDao(){
}


DatabaseDao::~DatabaseDao(){
        closeDatabaseConnection();
}


/*----------------------------------------------------------*/
/*-----                 PUBLIC METHODS                 -----*/
/*----------------------------------------------------------*/

void DatabaseDao::connectToDatabase(const DBInfo& dbInfo){
        //Store the information about the database
        setDBInfo(dbInfo);

        //Connect to the database
        connectToDatabase();
}


QList<QString> DatabaseDao::getDatabaseNames(){
        QSqlQuery query = getQuery("SHOW DATABASES");
        executeQuery(query);
        QList<QString> dbList;
        while ( query.next() ) {
                dbList.append(query.value(0).toString());
        }
        return dbList;
}


void DatabaseDao::executeSQLFile(const QString& fileName){
        QFile dbFile(fileName);
        if(!dbFile.exists()){
                throw SpikeStreamDBException("Attempting to load file " + fileName + " that does not exist");
        }
        QString fileStr;
        loadFileIntoString(dbFile, fileStr);

        /* There is a limit to the length of string that can be executed, so have to
                break string down into individual commands */
        QStringList cmdList = fileStr.split(";", QString::SkipEmptyParts);
        foreach(QString sqlCmd, cmdList){
                executeQuery(sqlCmd);
        }
}


void DatabaseDao::executeSQLFile(const QString& dbName, const QString& fileName){
        QFile dbFile(fileName);
        if(!dbFile.exists()){
                throw SpikeStreamDBException("Attempting to load file " + fileName + " that does not exist");
        }
        QString fileStr = "USE " + dbName + ";\n";
        loadFileIntoString(dbFile, fileStr);

        /* There is a limit to the length of string that can be executed, so have to
                break string down into individual commands */
        QStringList cmdList = fileStr.split(";", QString::SkipEmptyParts);
        foreach(QString sqlCmd, cmdList){
                executeQuery(sqlCmd);
        }
}


/*----------------------------------------------------------*/
/*-----                 PRIVATE METHODS                -----*/
/*----------------------------------------------------------*/

void DatabaseDao::loadFileIntoString(QFile& file, QString& fileString){
        if (!file.open(QIODevice::ReadOnly | QIODevice::Text))
                throw SpikeStreamIOException("Cannot open file: " + file.fileName());

        QTextStream in(&file);
        while (!in.atEnd()) {
                fileString += in.readLine();
        }
}