sqldatabasestorage.h

/*
 Copyright 2017 Herik Lima de Castro and Marcelo Medeiros Eler
 Distributed under MIT license, or public domain if desired and
 recognized in your jurisdiction.
 See file LICENSE for detail.
*/

#ifndef SQLDATABASESTORAGE_H
#define SQLDATABASESTORAGE_H

#include <QUuid>
#include <QDebug>
#include <QSqlError>
#include <QSqlDatabase>
#include <QThreadStorage>
#include "cppwebframework_global.h"

CWF_BEGIN_NAMESPACE
class CPPWEBFRAMEWORKSHARED_EXPORT SqlDatabaseStorage
{    
    class Database
    {
        friend class SqlDatabaseStorage;
        QSqlDatabase *db = nullptr;
    public:
        Database() = default;
        Database(Database &other)
        {
            db = other.db;
            other.db = nullptr;
        }
        ~Database()
        {
            if(db)
            {
                const QString conName(db->connectionName());
                db->close();
                delete db;
                QSqlDatabase::removeDatabase(conName);
            }
        }
    };
    QString type;
    QString hostName;
    QString databaseName;
    QString userName;
    QString password;
    int port;
    QThreadStorage<Database> pool;
public:
    SqlDatabaseStorage(const QString &type = "", const QString &hostName = "",
                       const QString &databaseName = "", const QString &userName = "",
                       const QString &password = "", int port = 0) : type(type), hostName(hostName),
                                                                    databaseName(databaseName), userName(userName),
                                                                    password(password), port(port)
    {
    }

    inline QString getType() const { return type; }

    inline QString getPassword() const { return password; }

    inline QString getHostName() const { return hostName; }

    inline QString getDatabaseName() const { return databaseName; }

    inline QString getUserName() const { return userName; }

    inline int getPort() const { return port; }

    QSqlDatabase &getDatabase()
    {
        if(!pool.hasLocalData())
        {
            Database database;
            database.db = new QSqlDatabase(QSqlDatabase::addDatabase(type, QUuid::createUuid().toString()));
            database.db->setHostName(hostName);
            database.db->setDatabaseName(databaseName);
            database.db->setPort(port);
            database.db->setUserName(userName);
            database.db->setPassword(password);
            if(!database.db->open())
                qDebug() << database.db->lastError().text();
            pool.setLocalData(database);
        }
        return *pool.localData().db;
    }
};

CWF_END_NAMESPACE

#endif // SQLDATABASESTORAGE_H