mirror of
https://github.com/hyperion-project/hyperion.ng.git
synced 2025-03-01 10:33:28 +00:00
Refactor config API
This commit is contained in:
LordGrey
@@ -1,38 +1,48 @@
|
||||
#include "utils/settings.h"
|
||||
#include <db/DBManager.h>
|
||||
|
||||
#include <QSqlDatabase>
|
||||
#include <QSqlError>
|
||||
#include <QSqlQuery>
|
||||
#include <QSqlRecord>
|
||||
#include <QThreadStorage>
|
||||
#include <QUuid>
|
||||
#include <QDir>
|
||||
#include <QMetaType>
|
||||
#include <QJsonObject>
|
||||
|
||||
#ifdef _WIN32
|
||||
#include <stdexcept>
|
||||
#endif
|
||||
|
||||
// not in header because of linking
|
||||
static QString _rootPath;
|
||||
static QThreadStorage<QSqlDatabase> _databasePool;
|
||||
#define NO_SQLQUERY_LOGGING
|
||||
|
||||
// Constants
|
||||
namespace {
|
||||
const char DATABASE_DIRECTORYNAME[] = "db";
|
||||
const char DATABASE_FILENAME[] = "hyperion.db";
|
||||
|
||||
} //End of constants
|
||||
|
||||
|
||||
QDir DBManager::_dataDirectory;
|
||||
QDir DBManager::_databaseDirectory;
|
||||
QFileInfo DBManager::_databaseFile;
|
||||
QThreadStorage<QSqlDatabase> DBManager::_databasePool;
|
||||
bool DBManager::_isReadOnly {false};
|
||||
|
||||
DBManager::DBManager(QObject* parent)
|
||||
: QObject(parent)
|
||||
, _log(Logger::getInstance("DB"))
|
||||
, _readonlyMode (false)
|
||||
{
|
||||
}
|
||||
|
||||
DBManager::~DBManager()
|
||||
void DBManager::initializeDatabase(const QDir& dataDirectory, bool isReadOnly)
|
||||
{
|
||||
}
|
||||
|
||||
void DBManager::setRootPath(const QString& rootPath)
|
||||
{
|
||||
_rootPath = rootPath;
|
||||
// create directory
|
||||
QDir().mkpath(_rootPath+"/db");
|
||||
_dataDirectory = dataDirectory;
|
||||
_databaseDirectory.setPath(_dataDirectory.absoluteFilePath(DATABASE_DIRECTORYNAME));
|
||||
QDir().mkpath(_databaseDirectory.absolutePath());
|
||||
_databaseFile.setFile(_databaseDirectory,DATABASE_FILENAME);
|
||||
_isReadOnly = isReadOnly;
|
||||
}
|
||||
|
||||
void DBManager::setTable(const QString& table)
|
||||
@@ -43,38 +53,39 @@ void DBManager::setTable(const QString& table)
|
||||
QSqlDatabase DBManager::getDB() const
|
||||
{
|
||||
if(_databasePool.hasLocalData())
|
||||
return _databasePool.localData();
|
||||
else
|
||||
{
|
||||
auto db = QSqlDatabase::addDatabase("QSQLITE", QUuid::createUuid().toString());
|
||||
_databasePool.setLocalData(db);
|
||||
db.setDatabaseName(_rootPath+"/db/"+_dbn+".db");
|
||||
if(!db.open())
|
||||
return _databasePool.localData();
|
||||
}
|
||||
auto database = QSqlDatabase::addDatabase("QSQLITE", QUuid::createUuid().toString());
|
||||
|
||||
if (isReadOnly())
|
||||
{
|
||||
Error(_log, "%s", QSTRING_CSTR(db.lastError().text()));
|
||||
database.setConnectOptions("QSQLITE_OPEN_READONLY");
|
||||
}
|
||||
Debug(Logger::getInstance("DB"), "Database is opened in %s mode", _isReadOnly ? "read-only" : "read/write");
|
||||
|
||||
_databasePool.setLocalData(database);
|
||||
database.setDatabaseName(_databaseFile.absoluteFilePath());
|
||||
if(!database.open())
|
||||
{
|
||||
Error(_log, "%s", QSTRING_CSTR(database.lastError().text()));
|
||||
throw std::runtime_error("Failed to open database connection!");
|
||||
}
|
||||
return db;
|
||||
}
|
||||
|
||||
return database;
|
||||
}
|
||||
|
||||
bool DBManager::createRecord(const VectorPair& conditions, const QVariantMap& columns) const
|
||||
{
|
||||
if ( _readonlyMode )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
if(recordExists(conditions))
|
||||
{
|
||||
// if there is no column data, return
|
||||
if(columns.isEmpty())
|
||||
{
|
||||
return true;
|
||||
}
|
||||
|
||||
if(!updateRecord(conditions, columns))
|
||||
return false;
|
||||
|
||||
return true;
|
||||
return updateRecord(conditions, columns);
|
||||
}
|
||||
|
||||
QSqlDatabase idb = getDB();
|
||||
@@ -84,14 +95,15 @@ bool DBManager::createRecord(const VectorPair& conditions, const QVariantMap& co
|
||||
QVariantList cValues;
|
||||
QStringList prep;
|
||||
QStringList placeh;
|
||||
|
||||
// prep merge columns & condition
|
||||
QVariantMap::const_iterator i = columns.constBegin();
|
||||
while (i != columns.constEnd()) {
|
||||
prep.append(i.key());
|
||||
cValues += i.value();
|
||||
QVariantMap::const_iterator columnIter = columns.constBegin();
|
||||
while (columnIter != columns.constEnd()) {
|
||||
prep.append(columnIter.key());
|
||||
cValues += columnIter.value();
|
||||
placeh.append("?");
|
||||
|
||||
++i;
|
||||
++columnIter;
|
||||
}
|
||||
for(const auto& pair : conditions)
|
||||
{
|
||||
@@ -101,21 +113,19 @@ bool DBManager::createRecord(const VectorPair& conditions, const QVariantMap& co
|
||||
cValues << pair.second;
|
||||
placeh.append("?");
|
||||
}
|
||||
query.prepare(QString("INSERT INTO %1 ( %2 ) VALUES ( %3 )").arg(_table,prep.join(", ")).arg(placeh.join(", ")));
|
||||
query.prepare(QString("INSERT INTO %1 ( %2 ) VALUES ( %3 )").arg(_table,prep.join(", "), placeh.join(", ")));
|
||||
// add column & condition values
|
||||
doAddBindValue(query, cValues);
|
||||
if(!query.exec())
|
||||
{
|
||||
Error(_log, "Failed to create record: '%s' in table: '%s' Error: %s", QSTRING_CSTR(prep.join(", ")), QSTRING_CSTR(_table), QSTRING_CSTR(idb.lastError().text()));
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
addBindValues(query, cValues);
|
||||
|
||||
return executeQuery(query);
|
||||
}
|
||||
|
||||
bool DBManager::recordExists(const VectorPair& conditions) const
|
||||
{
|
||||
if(conditions.isEmpty())
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
QSqlDatabase idb = getDB();
|
||||
QSqlQuery query(idb);
|
||||
@@ -127,35 +137,28 @@ bool DBManager::recordExists(const VectorPair& conditions) const
|
||||
|
||||
for(const auto& pair : conditions)
|
||||
{
|
||||
prepCond << pair.first+"=?";
|
||||
prepCond << pair.first+"= ?";
|
||||
bindVal << pair.second;
|
||||
}
|
||||
query.prepare(QString("SELECT * FROM %1 %2").arg(_table,prepCond.join(" ")));
|
||||
doAddBindValue(query, bindVal);
|
||||
if(!query.exec())
|
||||
addBindValues(query, bindVal);
|
||||
|
||||
if (!executeQuery(query))
|
||||
{
|
||||
Error(_log, "Failed recordExists(): '%s' in table: '%s' Error: %s", QSTRING_CSTR(prepCond.join(" ")), QSTRING_CSTR(_table), QSTRING_CSTR(idb.lastError().text()));
|
||||
return false;
|
||||
}
|
||||
|
||||
int entry = 0;
|
||||
while (query.next()) {
|
||||
while (query.next())
|
||||
{
|
||||
entry++;
|
||||
}
|
||||
|
||||
if(entry)
|
||||
return true;
|
||||
|
||||
return false;
|
||||
return entry > 0;
|
||||
}
|
||||
|
||||
bool DBManager::updateRecord(const VectorPair& conditions, const QVariantMap& columns) const
|
||||
{
|
||||
if ( _readonlyMode )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
QSqlDatabase idb = getDB();
|
||||
QSqlQuery query(idb);
|
||||
query.setForwardOnly(true);
|
||||
@@ -164,88 +167,75 @@ bool DBManager::updateRecord(const VectorPair& conditions, const QVariantMap& co
|
||||
QStringList prep;
|
||||
|
||||
// prepare columns valus
|
||||
QVariantMap::const_iterator i = columns.constBegin();
|
||||
while (i != columns.constEnd()) {
|
||||
prep += i.key()+"=?";
|
||||
values += i.value();
|
||||
QVariantMap::const_iterator columnIter = columns.constBegin();
|
||||
while (columnIter != columns.constEnd()) {
|
||||
prep += columnIter.key()+"= ?";
|
||||
values += columnIter.value();
|
||||
|
||||
++i;
|
||||
++columnIter;
|
||||
}
|
||||
|
||||
// prepare condition values
|
||||
QStringList prepCond;
|
||||
QVariantList prepBindVal;
|
||||
if(!conditions.isEmpty())
|
||||
if(!conditions.isEmpty()) {
|
||||
prepCond << "WHERE";
|
||||
}
|
||||
|
||||
for(const auto& pair : conditions)
|
||||
{
|
||||
prepCond << pair.first+"=?";
|
||||
prepCond << pair.first+"= ?";
|
||||
prepBindVal << pair.second;
|
||||
}
|
||||
|
||||
query.prepare(QString("UPDATE %1 SET %2 %3").arg(_table,prep.join(", ")).arg(prepCond.join(" ")));
|
||||
query.prepare(QString("UPDATE %1 SET %2 %3").arg(_table,prep.join(", "), prepCond.join(" ")));
|
||||
// add column values
|
||||
doAddBindValue(query, values);
|
||||
addBindValues(query, values);
|
||||
// add condition values
|
||||
doAddBindValue(query, prepBindVal);
|
||||
if(!query.exec())
|
||||
{
|
||||
Error(_log, "Failed to update record: '%s' in table: '%s' Error: %s", QSTRING_CSTR(prepCond.join(" ")), QSTRING_CSTR(_table), QSTRING_CSTR(idb.lastError().text()));
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
addBindValues(query, prepBindVal);
|
||||
|
||||
return executeQuery(query);
|
||||
}
|
||||
|
||||
bool DBManager::getRecord(const VectorPair& conditions, QVariantMap& results, const QStringList& tColumns, const QStringList& tOrder) const
|
||||
{
|
||||
QSqlDatabase idb = getDB();
|
||||
QSqlQuery query(idb);
|
||||
query.setForwardOnly(true);
|
||||
|
||||
QString sColumns("*");
|
||||
if(!tColumns.isEmpty())
|
||||
sColumns = tColumns.join(", ");
|
||||
|
||||
QString sOrder("");
|
||||
if(!tOrder.isEmpty())
|
||||
{
|
||||
sOrder = " ORDER BY ";
|
||||
sOrder.append(tOrder.join(", "));
|
||||
QVector<QVariantMap> resultVector{};
|
||||
bool success = getRecords(conditions, resultVector, tColumns, tOrder);
|
||||
if (success && !resultVector.isEmpty()) {
|
||||
results = resultVector.first();
|
||||
}
|
||||
// prep conditions
|
||||
QStringList prepCond;
|
||||
QVariantList bindVal;
|
||||
if(!conditions.isEmpty())
|
||||
prepCond << " WHERE";
|
||||
|
||||
for(const auto& pair : conditions)
|
||||
{
|
||||
prepCond << pair.first+"=?";
|
||||
bindVal << pair.second;
|
||||
}
|
||||
query.prepare(QString("SELECT %1 FROM %2%3%4").arg(sColumns,_table).arg(prepCond.join(" ")).arg(sOrder));
|
||||
doAddBindValue(query, bindVal);
|
||||
|
||||
if(!query.exec())
|
||||
{
|
||||
Error(_log, "Failed to get record: '%s' in table: '%s' Error: %s", QSTRING_CSTR(prepCond.join(" ")), QSTRING_CSTR(_table), QSTRING_CSTR(idb.lastError().text()));
|
||||
return false;
|
||||
}
|
||||
|
||||
// go to first row
|
||||
query.next();
|
||||
|
||||
QSqlRecord rec = query.record();
|
||||
for(int i = 0; i<rec.count(); i++)
|
||||
{
|
||||
results[rec.fieldName(i)] = rec.value(i);
|
||||
}
|
||||
|
||||
return true;
|
||||
return success;
|
||||
}
|
||||
|
||||
bool DBManager::getRecords(QVector<QVariantMap>& results, const QStringList& tColumns, const QStringList& tOrder) const
|
||||
{
|
||||
return getRecords({}, results, tColumns, tOrder);
|
||||
}
|
||||
|
||||
bool DBManager::getRecords(const VectorPair& conditions, QVector<QVariantMap>& results, const QStringList& tColumns, const QStringList& tOrder) const
|
||||
{
|
||||
// prep conditions
|
||||
QStringList conditionList;
|
||||
QVariantList bindValues;
|
||||
|
||||
for(const auto& pair : conditions)
|
||||
{
|
||||
conditionList << pair.first;
|
||||
if (pair.second.isNull())
|
||||
{
|
||||
conditionList << "IS NULL";
|
||||
}
|
||||
else
|
||||
{
|
||||
conditionList << "= ?";
|
||||
bindValues << pair.second;
|
||||
}
|
||||
}
|
||||
|
||||
return getRecords(conditionList.join((" ")), bindValues, results, tColumns, tOrder);
|
||||
}
|
||||
|
||||
bool DBManager::getRecords(const QString& condition, const QVariantList& bindValues, QVector<QVariantMap>& results, const QStringList& tColumns, const QStringList& tOrder) const
|
||||
{
|
||||
QSqlDatabase idb = getDB();
|
||||
QSqlQuery query(idb);
|
||||
@@ -253,20 +243,28 @@ bool DBManager::getRecords(QVector<QVariantMap>& results, const QStringList& tCo
|
||||
|
||||
QString sColumns("*");
|
||||
if(!tColumns.isEmpty())
|
||||
{
|
||||
sColumns = tColumns.join(", ");
|
||||
}
|
||||
|
||||
QString sOrder("");
|
||||
if(!tOrder.isEmpty())
|
||||
{
|
||||
sOrder = " ORDER BY ";
|
||||
sOrder = "ORDER BY ";
|
||||
sOrder.append(tOrder.join(", "));
|
||||
}
|
||||
|
||||
query.prepare(QString("SELECT %1 FROM %2%3").arg(sColumns,_table,sOrder));
|
||||
|
||||
if(!query.exec())
|
||||
// prep conditions
|
||||
QString prepCond;
|
||||
if(!condition.isEmpty())
|
||||
{
|
||||
prepCond = QString("WHERE %1").arg(condition);
|
||||
}
|
||||
|
||||
query.prepare(QString("SELECT %1 FROM %2 %3 %4").arg(sColumns,_table, prepCond, sOrder));
|
||||
addBindValues(query, bindValues);
|
||||
if (!executeQuery(query))
|
||||
{
|
||||
Error(_log, "Failed to get records: '%s' in table: '%s' Error: %s", QSTRING_CSTR(sColumns), QSTRING_CSTR(_table), QSTRING_CSTR(idb.lastError().text()));
|
||||
return false;
|
||||
}
|
||||
|
||||
@@ -288,11 +286,6 @@ bool DBManager::getRecords(QVector<QVariantMap>& results, const QStringList& tCo
|
||||
|
||||
bool DBManager::deleteRecord(const VectorPair& conditions) const
|
||||
{
|
||||
if ( _readonlyMode )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
if(conditions.isEmpty())
|
||||
{
|
||||
Error(_log, "Oops, a deleteRecord() call wants to delete the entire table (%s)! Denied it", QSTRING_CSTR(_table));
|
||||
@@ -310,29 +303,20 @@ bool DBManager::deleteRecord(const VectorPair& conditions) const
|
||||
|
||||
for(const auto& pair : conditions)
|
||||
{
|
||||
prepCond << pair.first+"=?";
|
||||
prepCond << pair.first+"= ?";
|
||||
bindValues << pair.second;
|
||||
}
|
||||
|
||||
query.prepare(QString("DELETE FROM %1 %2").arg(_table,prepCond.join(" ")));
|
||||
doAddBindValue(query, bindValues);
|
||||
if(!query.exec())
|
||||
{
|
||||
Error(_log, "Failed to delete record: '%s' in table: '%s' Error: %s", QSTRING_CSTR(prepCond.join(" ")), QSTRING_CSTR(_table), QSTRING_CSTR(idb.lastError().text()));
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
addBindValues(query, bindValues);
|
||||
|
||||
return executeQuery(query);
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
bool DBManager::createTable(QStringList& columns) const
|
||||
{
|
||||
if ( _readonlyMode )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
if(columns.isEmpty())
|
||||
{
|
||||
Error(_log,"Empty tables aren't supported!");
|
||||
@@ -347,9 +331,9 @@ bool DBManager::createTable(QStringList& columns) const
|
||||
// empty tables aren't supported by sqlite, add one column
|
||||
QString tcolumn = columns.takeFirst();
|
||||
// default CURRENT_TIMESTAMP is not supported by ALTER TABLE
|
||||
if(!query.exec(QString("CREATE TABLE %1 ( %2 )").arg(_table,tcolumn)))
|
||||
query.prepare(QString("CREATE TABLE %1 ( %2 )").arg(_table,tcolumn));
|
||||
if (!executeQuery(query))
|
||||
{
|
||||
Error(_log, "Failed to create table: '%s' Error: %s", QSTRING_CSTR(_table), QSTRING_CSTR(idb.lastError().text()));
|
||||
return false;
|
||||
}
|
||||
}
|
||||
@@ -358,8 +342,8 @@ bool DBManager::createTable(QStringList& columns) const
|
||||
int err = 0;
|
||||
for(const auto& column : columns)
|
||||
{
|
||||
QStringList id = column.split(' ');
|
||||
if(rec.indexOf(id.at(0)) == -1)
|
||||
QStringList columName = column.split(' ');
|
||||
if(rec.indexOf(columName.at(0)) == -1)
|
||||
{
|
||||
if(!createColumn(column))
|
||||
{
|
||||
@@ -367,79 +351,91 @@ bool DBManager::createTable(QStringList& columns) const
|
||||
}
|
||||
}
|
||||
}
|
||||
if(err)
|
||||
return false;
|
||||
|
||||
return true;
|
||||
return err == 0;
|
||||
}
|
||||
|
||||
bool DBManager::createColumn(const QString& column) const
|
||||
{
|
||||
if ( _readonlyMode )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
QSqlDatabase idb = getDB();
|
||||
QSqlQuery query(idb);
|
||||
if(!query.exec(QString("ALTER TABLE %1 ADD COLUMN %2").arg(_table,column)))
|
||||
{
|
||||
Error(_log, "Failed to create column: '%s' in table: '%s' Error: %s", QSTRING_CSTR(column), QSTRING_CSTR(_table), QSTRING_CSTR(idb.lastError().text()));
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
|
||||
query.prepare(QString("ALTER TABLE %1 ADD COLUMN %2").arg(_table,column));
|
||||
return executeQuery(query);
|
||||
}
|
||||
|
||||
bool DBManager::tableExists(const QString& table) const
|
||||
{
|
||||
QSqlDatabase idb = getDB();
|
||||
QStringList tables = idb.tables();
|
||||
if(tables.contains(table))
|
||||
return true;
|
||||
return false;
|
||||
return tables.contains(table);
|
||||
}
|
||||
|
||||
bool DBManager::deleteTable(const QString& table) const
|
||||
{
|
||||
if ( _readonlyMode )
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
if(tableExists(table))
|
||||
{
|
||||
QSqlDatabase idb = getDB();
|
||||
QSqlQuery query(idb);
|
||||
if(!query.exec(QString("DROP TABLE %1").arg(table)))
|
||||
{
|
||||
Error(_log, "Failed to delete table: '%s' Error: %s", QSTRING_CSTR(table), QSTRING_CSTR(idb.lastError().text()));
|
||||
return false;
|
||||
}
|
||||
|
||||
query.prepare(QString("DROP TABLE %1").arg(table));
|
||||
return executeQuery(query);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
void DBManager::doAddBindValue(QSqlQuery& query, const QVariantList& variants) const
|
||||
void DBManager::addBindValues(QSqlQuery& query, const QVariantList& bindValues) const
|
||||
{
|
||||
for(const auto& variant : variants)
|
||||
if (!bindValues.isEmpty())
|
||||
{
|
||||
auto t = variant.userType();
|
||||
switch(t)
|
||||
for(const auto& value : bindValues)
|
||||
{
|
||||
case QMetaType::UInt:
|
||||
case QMetaType::Int:
|
||||
case QMetaType::Bool:
|
||||
query.addBindValue(variant.toInt());
|
||||
break;
|
||||
case QMetaType::Double:
|
||||
query.addBindValue(variant.toFloat());
|
||||
break;
|
||||
case QMetaType::QByteArray:
|
||||
query.addBindValue(variant.toByteArray());
|
||||
break;
|
||||
default:
|
||||
query.addBindValue(variant.toString());
|
||||
break;
|
||||
query.addBindValue(value);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
QString DBManager::constructExecutedQuery(const QSqlQuery& query) const
|
||||
{
|
||||
QString executedQuery = query.executedQuery();
|
||||
|
||||
// Check if the query uses positional placeholders
|
||||
if (executedQuery.contains('?')) {
|
||||
QVariantList boundValues = query.boundValues(); // Get bound values as a list
|
||||
// Iterate through the bound values and replace placeholders
|
||||
for (const QVariant &value : boundValues) {
|
||||
// Replace the first occurrence of '?' with the actual value
|
||||
QString valueStr;
|
||||
if (value.canConvert<QString>())
|
||||
{
|
||||
valueStr = value.toString();
|
||||
}
|
||||
else
|
||||
{
|
||||
valueStr = "Unkown";
|
||||
}
|
||||
executedQuery.replace(executedQuery.indexOf('?'), 1, valueStr);
|
||||
}
|
||||
}
|
||||
return executedQuery;
|
||||
}
|
||||
|
||||
bool DBManager::executeQuery(QSqlQuery& query) const
|
||||
{
|
||||
if(!query.exec())
|
||||
{
|
||||
QString finalQuery = constructExecutedQuery(query);
|
||||
QString errorText = query.lastError().text();
|
||||
|
||||
Debug(_log, "Database Error: '%s', SqlQuery: '%s'", QSTRING_CSTR(errorText), QSTRING_CSTR(finalQuery));
|
||||
Error(_log, "Database Error: '%s'", QSTRING_CSTR(errorText));
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
#ifdef SQLQUERY_LOGGING
|
||||
QString finalQuery = constructExecutedQuery(query);
|
||||
Debug(_log, "SqlQuery executed: '%s'", QSTRING_CSTR(finalQuery));
|
||||
#endif
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user