sql.cpp

/*
 * SPDX-FileCopyrightText: (C) 2015-2022 Daniel Nicoletti <dantti12@gmail.com>
 * SPDX-License-Identifier: BSD-3-Clause
 */
#include "sql.h"
 
#include <QJsonArray>
#include <QJsonObject>
#include <QJsonValue>
#include <QLoggingCategory>
#include <QThread>
#include <QtSql/QSqlError>
#include <QtSql/QSqlQuery>
#include <QtSql/QSqlRecord>
 
Q_LOGGING_CATEGORY(C_SQL, "cutelyst.utils.sql", QtWarningMsg)
 
using namespace Cutelyst;
 
QVariantHash Sql::queryToHashObject(QSqlQuery &query)
{
    QVariantHash ret;
    if (query.next()) {
        const QSqlRecord record = query.record();
        const int columns       = record.count();
        for (int i = 0; i < columns; ++i) {
            ret.insert(record.fieldName(i), query.value(i));
        }
    }
    return ret;
}
 
QVariantList Sql::queryToHashList(QSqlQuery &query)
{
    QVariantList ret;
    const QSqlRecord record = query.record();
    const int columns       = record.count();
    QStringList cols;
    for (int i = 0; i < columns; ++i) {
        cols.append(record.fieldName(i));
    }
 
    while (query.next()) {
        QVariantHash line;
        for (int i = 0; i < columns; ++i) {
            line.insert(cols.at(i), query.value(i));
        }
        ret.append(line);
    }
    return ret;
}
 
QVariantMap Sql::queryToMapObject(QSqlQuery &query)
{
    QVariantMap ret;
    if (query.next()) {
        const QSqlRecord record = query.record();
        const int columns       = record.count();
        for (int i = 0; i < columns; ++i) {
            ret.insert(record.fieldName(i), query.value(i));
        }
    }
    return ret;
}
 
QJsonObject Sql::queryToJsonObject(QSqlQuery &query)
{
    QJsonObject ret;
    if (query.next()) {
        const QSqlRecord record = query.record();
        const int columns       = record.count();
        for (int i = 0; i < columns; ++i) {
            ret.insert(record.fieldName(i), QJsonValue::fromVariant(query.value(i)));
        }
    }
    return ret;
}
 
QVariantList Sql::queryToMapList(QSqlQuery &query)
{
    QVariantList ret;
    const QSqlRecord record = query.record();
    const int columns       = record.count();
    QStringList cols;
    for (int i = 0; i < columns; ++i) {
        cols.append(record.fieldName(i));
    }
 
    while (query.next()) {
        QVariantMap line;
        for (int i = 0; i < columns; ++i) {
            line.insert(cols.at(i), query.value(i));
        }
        ret.append(line);
    }
    return ret;
}
 
QJsonArray Sql::queryToJsonObjectArray(QSqlQuery &query)
{
    QJsonArray ret;
    const QSqlRecord record = query.record();
    const int columns       = record.count();
    QStringList cols;
    for (int i = 0; i < columns; ++i) {
        cols.append(record.fieldName(i));
    }
 
    while (query.next()) {
        QJsonObject line;
        for (int i = 0; i < columns; ++i) {
            line.insert(cols.at(i), QJsonValue::fromVariant(query.value(i)));
        }
        ret.append(line);
    }
    return ret;
}
 
QVariantList Sql::queryToList(QSqlQuery &query)
{
    QVariantList ret;
 
    const int columns = query.record().count();
    while (query.next()) {
        QVariantList line;
        for (int i = 0; i < columns; ++i) {
            line.append(query.value(i));
        }
        ret.append(QVariant::fromValue(line));
    }
 
    return ret;
}
 
QJsonArray Sql::queryToJsonArray(QSqlQuery &query)
{
    QJsonArray ret;
 
    const int columns = query.record().count();
    while (query.next()) {
        QJsonArray array;
        for (int i = 0; i < columns; ++i) {
            array.append(QJsonValue::fromVariant(query.value(i)));
        }
        ret.append(array);
    }
 
    return ret;
}
 
QVariantHash Sql::queryToIndexedHash(QSqlQuery &query, const QString &key)
{
    QVariantHash ret;
 
    const QSqlRecord record = query.record();
    int index               = record.indexOf(key);
    if (index == -1) {
        qCCritical(C_SQL) << "Field Name " << key << " not found in result set";
        return ret;
    }
 
    const int columns = record.count();
    QStringList cols;
 
    for (int i = 0; i < columns; ++i) {
        cols.append(record.fieldName(i));
    }
 
    while (query.next()) {
        QVariantHash line;
        for (int i = 0; i < columns; ++i) {
            if (i != index) {
                line.insert(cols.at(i), query.value(i));
            }
        }
 
        ret.insert(query.value(index).toString(), line);
    }
 
    return ret;
}
 
QJsonObject Sql::queryToIndexedJsonObject(QSqlQuery &query, const QString &key)
{
    QJsonObject ret;
 
    const QSqlRecord record = query.record();
    int index               = record.indexOf(key);
    if (index == -1) {
        qCCritical(C_SQL) << "Field Name " << key << " not found in result set";
        return ret;
    }
 
    const int columns = record.count();
    QStringList cols;
 
    for (int i = 0; i < columns; ++i) {
        cols.append(record.fieldName(i));
    }
 
    while (query.next()) {
        QJsonObject obj;
        for (int i = 0; i < columns; ++i) {
            if (i != index) {
                obj.insert(cols.at(i), QJsonValue::fromVariant(query.value(i)));
            }
        }
 
        ret.insert(query.value(index).toString(), obj);
    }
 
    return ret;
}
 
void Sql::bindParamsToQuery(QSqlQuery &query,
                            const Cutelyst::ParamsMultiMap &params,
                            bool htmlEscaped)
{
    auto it = params.constBegin();
    if (htmlEscaped) {
        while (it != params.constEnd()) {
            if (it.value().isNull()) {
                query.bindValue(u':' + it.key(), QVariant());
            } else {
                query.bindValue(u':' + it.key(), it.value().toHtmlEscaped());
            }
            ++it;
        }
    } else {
        while (it != params.constEnd()) {
            if (it.value().isNull()) {
                query.bindValue(u':' + it.key(), QVariant());
            } else {
                query.bindValue(u':' + it.key(), it.value());
            }
            ++it;
        }
    }
}
 
QSqlQuery Sql::preparedQuery(const QString &query, QSqlDatabase db, bool forwardOnly)
{
    QSqlQuery sqlQuery(db);
    sqlQuery.setForwardOnly(forwardOnly);
    if (!sqlQuery.prepare(query)) {
        qCCritical(C_SQL) << "Failed to prepare query:" << query
                          << sqlQuery.lastError().databaseText();
    }
    return sqlQuery;
}
 
QSqlQuery Sql::preparedQueryThread(const QString &query, const QString &dbName, bool forwardOnly)
{
    QSqlQuery sqlQuery(QSqlDatabase::database(databaseNameThread(dbName)));
    sqlQuery.setForwardOnly(forwardOnly);
    if (!sqlQuery.prepare(query)) {
        qCCritical(C_SQL) << "Failed to prepare query:" << query
                          << sqlQuery.lastError().databaseText();
    }
    return sqlQuery;
}
 
QString Sql::databaseNameThread(const QString &dbName)
{
    return dbName + u'-' + QThread::currentThread()->objectName();
}
 
QSqlDatabase Sql::databaseThread(const QString &dbName)
{
    return QSqlDatabase::database(databaseNameThread(dbName));
}
 
Sql::Transaction::Transaction(const QString &databaseName)
    : m_db(databaseThread(databaseName))
{
    m_transactionRunning = m_db.transaction();
}
 
Sql::Transaction::Transaction(const QSqlDatabase &database)
    : m_db(database)
{
    m_transactionRunning = m_db.transaction();
}
 
Sql::Transaction::~Transaction()
{
    if (m_transactionRunning) {
        m_db.rollback();
    }
}
 
bool Sql::Transaction::transaction() const
{
    return m_transactionRunning;
}
 
bool Sql::Transaction::commit()
{
    // In case we fail to commit we will still call rollback, not sure it's really needed
    m_transactionRunning = !m_db.commit();
    return !m_transactionRunning;
}
 
bool Sql::Transaction::rollback()
{
    m_transactionRunning = false;
    return m_db.rollback();
}