try to use params (broken)

This commit is contained in:
Evgenii Alekseev 2018-03-25 01:03:03 +03:00
parent ba9a1c5cdd
commit 9f57f9db88
5 changed files with 117 additions and 41 deletions

View File

@ -38,6 +38,22 @@ class QueuedDatabase : public QObject
Q_PROPERTY(QString path READ path) Q_PROPERTY(QString path READ path)
public: public:
/**
* @struct QueuedDatabaseCondition
* @brief structure to define database condition payload
* @var key
* payload key
* @var value
* payload value
* @var operation
* comparison operation
*/
struct QueuedDatabaseCondition {
QString key;
QVariant value;
QString operation;
};
/** /**
* @brief QueuedDatabase class constructor * @brief QueuedDatabase class constructor
* @param parent * @param parent
@ -88,7 +104,8 @@ public:
* optional condition string * optional condition string
* @return list of records from table * @return list of records from table
*/ */
QList<QVariantHash> get(const QString &_table, const QString &_condition = ""); QList<QVariantHash> get(const QString &_table, const QList<QueuedDatabaseCondition> &_condition
= QList<QueuedDatabaseCondition>());
/** /**
* @brief get record from table with given id * @brief get record from table with given id
* @param _table * @param _table
@ -181,6 +198,13 @@ private:
* @return list of columns in table * @return list of columns in table
*/ */
QStringList getColumnsInRecord(const QSqlRecord &_record) const; QStringList getColumnsInRecord(const QSqlRecord &_record) const;
/**
* @brief parse condition map to sql query
* @param _condition
* condition map
* @return sql query string part
*/
QString getCondition(const QList<QueuedDatabaseCondition> &_condition) const;
/** /**
* @brief last insertion ID * @brief last insertion ID
* @param _table * @param _table

View File

@ -370,7 +370,7 @@ QueuedResult<QVariantHash> QueuedCorePrivate::pluginSettings(const QString &_plu
return QueuedError("Not allowed", QueuedEnums::ReturnStatus::InsufficientPermissions); return QueuedError("Not allowed", QueuedEnums::ReturnStatus::InsufficientPermissions);
auto dbSettings = m_database->get(QueuedDB::SETTINGS_TABLE, auto dbSettings = m_database->get(QueuedDB::SETTINGS_TABLE,
QString("WHERE key LIKE 'Plugin.%1.%'").arg(_plugin)); {{"key", QString("Plugin.%1.%").arg(_plugin), "LIKE"}});
QVariantHash settings; QVariantHash settings;
std::for_each(dbSettings.cbegin(), dbSettings.cend(), std::for_each(dbSettings.cbegin(), dbSettings.cend(),
[&settings, &_plugin](const QVariantHash &value) { [&settings, &_plugin](const QVariantHash &value) {

View File

@ -131,7 +131,7 @@ void QueuedCorePrivate::initProcesses()
m_processes = m_helper->initObject(m_processes); m_processes = m_helper->initObject(m_processes);
m_processes->setExitAction(onExitAction); m_processes->setExitAction(onExitAction);
auto dbProcesses = m_database->get(QueuedDB::TASKS_TABLE, "WHERE endTime IS NULL"); auto dbProcesses = m_database->get(QueuedDB::TASKS_TABLE, {{"endTime", "NULL", "IS"}});
m_processes->loadProcesses(dbProcesses); m_processes->loadProcesses(dbProcesses);
m_connections += connect(m_processes, &QueuedProcessManager::taskStartTimeReceived, m_connections += connect(m_processes, &QueuedProcessManager::taskStartTimeReceived,
@ -207,8 +207,7 @@ void QueuedCorePrivate::initUsers()
m_users->setSalt(m_settings->admin().salt); m_users->setSalt(m_settings->admin().salt);
m_users->setTokenExpiration(expiry); m_users->setTokenExpiration(expiry);
QString now = QDateTime::currentDateTimeUtc().toString(Qt::ISODateWithMs); QString now = QDateTime::currentDateTimeUtc().toString(Qt::ISODateWithMs);
auto dbTokens = m_database->get( auto dbTokens = m_database->get(QueuedDB::TOKENS_TABLE, {{"validUntil", now, ">"}});
QueuedDB::TOKENS_TABLE, QString("WHERE datetime(validUntil) > datetime('%1')").arg(now));
m_users->loadTokens(dbTokens); m_users->loadTokens(dbTokens);
auto dbUsers = m_database->get(QueuedDB::USERS_TABLE); auto dbUsers = m_database->get(QueuedDB::USERS_TABLE);
m_users->loadUsers(dbUsers); m_users->loadUsers(dbUsers);

View File

@ -88,8 +88,12 @@ void QueuedDatabase::createAdministrator(const QString &_user, const QString &_p
{ {
qCDebug(LOG_LIB) << "Check for user" << _user; qCDebug(LOG_LIB) << "Check for user" << _user;
QSqlQuery query = m_database.exec( QSqlQuery query(m_database);
QString("SELECT * FROM '%1' WHERE name='%2'").arg(QueuedDB::USERS_TABLE).arg(_user)); query.prepare("SELECT * FROM :table WHERE name=:username");
query.bindValue(":table", QueuedDB::USERS_TABLE);
query.bindValue(":username", _user);
query.exec();
QSqlError error = query.lastError(); QSqlError error = query.lastError();
if (error.isValid()) if (error.isValid())
qCWarning(LOG_LIB) << "Could not get record" << _user << "from" << QueuedDB::USERS_TABLE qCWarning(LOG_LIB) << "Could not get record" << _user << "from" << QueuedDB::USERS_TABLE
@ -126,10 +130,12 @@ void QueuedDatabase::createSchema(const QString &_table)
if (columns.contains(column)) if (columns.contains(column))
continue; continue;
QueuedDB::QueuedDBField field = QueuedDB::DBSchema[_table][column]; QueuedDB::QueuedDBField field = QueuedDB::DBSchema[_table][column];
QSqlQuery query = m_database.exec(QString("ALTER TABLE '%1' ADD `%2` %3") QSqlQuery query(m_database);
.arg(_table) query.prepare(
.arg(column) QString("ALTER TABLE :table ADD `%1` %2").arg(column).arg(field.sqlDescription));
.arg(field.sqlDescription)); query.bindValue(":table", _table);
query.exec();
QSqlError error = query.lastError(); QSqlError error = query.lastError();
if (error.isValid()) if (error.isValid())
qCCritical(LOG_LIB) << "Could not insert column" << column << "to table" << _table qCCritical(LOG_LIB) << "Could not insert column" << column << "to table" << _table
@ -145,8 +151,11 @@ void QueuedDatabase::createTable(const QString &_table)
{ {
qCDebug(LOG_LIB) << "Create table" << _table; qCDebug(LOG_LIB) << "Create table" << _table;
QSqlQuery query = m_database.exec( QSqlQuery query(m_database);
QString("CREATE TABLE '%1' (`_id` INTEGER PRIMARY KEY AUTOINCREMENT)").arg(_table)); query.prepare("CREATE TABLE :table (`_id` INTEGER PRIMARY KEY AUTOINCREMENT)");
query.bindValue(":table", _table);
query.exec();
QSqlError error = query.lastError(); QSqlError error = query.lastError();
if (error.isValid()) if (error.isValid())
qCCritical(LOG_LIB) << "Could not create table" << _table << "error:" << error.text(); qCCritical(LOG_LIB) << "Could not create table" << _table << "error:" << error.text();
@ -156,13 +165,18 @@ void QueuedDatabase::createTable(const QString &_table)
/** /**
* @fn get * @fn get
*/ */
QList<QVariantHash> QueuedDatabase::get(const QString &_table, const QString &_condition) QList<QVariantHash> QueuedDatabase::get(const QString &_table,
const QList<QueuedDatabaseCondition> &_condition)
{ {
qCDebug(LOG_LIB) << "Get records in table" << _table << "with condition" << _condition; qCDebug(LOG_LIB) << "Get records in table" << _table;
QList<QVariantHash> output; QList<QVariantHash> output;
QSqlQuery query = m_database.exec( QSqlQuery query(m_database);
QString("SELECT * FROM '%1' %2 ORDER BY _id ASC").arg(_table).arg(_condition)); query.prepare("SELECT * FROM :table " + getCondition(_condition) + " ORDER BY _id ASC");
query.bindValue(":table", _table);
for (auto &cond : _condition)
query.bindValue(QString(":%1").arg(cond.key), cond.value);
query.exec();
QSqlError error = query.lastError(); QSqlError error = query.lastError();
if (error.isValid()) { if (error.isValid()) {
@ -190,7 +204,7 @@ QVariantHash QueuedDatabase::get(const QString &_table, const long long _id)
{ {
qCDebug(LOG_LIB) << "Get record" << _id << "in table" << _table; qCDebug(LOG_LIB) << "Get record" << _id << "in table" << _table;
auto output = get(_table, QString("WHERE _id=%1").arg(_id)); auto output = get(_table, {{"_id", _id, "="}});
if (output.count() == 0) { if (output.count() == 0) {
qCWarning(LOG_LIB) << "Could not find records for" << _id; qCWarning(LOG_LIB) << "Could not find records for" << _id;
return QVariantHash(); return QVariantHash();
@ -242,10 +256,15 @@ long long QueuedDatabase::add(const QString &_table, const QVariantHash &_value)
auto payload = getQueryPayload(_table, _value); auto payload = getQueryPayload(_table, _value);
// build query // build query
QSqlQuery query = m_database.exec(QString("INSERT INTO %1 (%2) VALUES (%3)") QSqlQuery query(m_database);
.arg(_table) query.prepare(QString("INSERT INTO :table (%1) VALUES (%2)")
.arg(payload.keys().join(',')) .arg(payload.keys().join(','))
.arg(payload.values().join(','))); .arg(payload.values().join(',')));
query.bindValue(":table", _table);
for (auto &key : _value.keys())
query.bindValue(QString(":%1").arg(key), _value[key]);
query.exec();
QSqlError error = query.lastError(); QSqlError error = query.lastError();
if (error.isValid()) { if (error.isValid()) {
qCCritical(LOG_LIB) << "Could not add record" << _value << "to table" << _table << "message" qCCritical(LOG_LIB) << "Could not add record" << _value << "to table" << _table << "message"
@ -267,10 +286,16 @@ bool QueuedDatabase::modify(const QString &_table, const long long _id, const QV
auto payload = getQueryPayload(_table, _value); auto payload = getQueryPayload(_table, _value);
QStringList stringPayload; QStringList stringPayload;
for (auto &key : payload.keys()) for (auto &key : payload.keys())
stringPayload.append(QString("%1=%2").arg(key).arg(payload[key])); stringPayload.append(QString("%1=:%1").arg(key));
// build query // build query
QSqlQuery query = m_database.exec( QSqlQuery query(m_database);
QString("UPDATE %1 SET %2 WHERE _id=%3").arg(_table).arg(stringPayload.join(',')).arg(_id)); query.prepare(QString("UPDATE :table SET %1 WHERE _id=:_id").arg(stringPayload.join(',')));
query.bindValue(":table", _table);
query.bindValue(":_id", _id);
for (auto &key : _value.keys())
query.bindValue(QString(":%1").arg(key), _value[key]);
query.exec();
QSqlError error = query.lastError(); QSqlError error = query.lastError();
if (error.isValid()) { if (error.isValid()) {
qCCritical(LOG_LIB) << "Could not modify record" << _value << "in table" << _table qCCritical(LOG_LIB) << "Could not modify record" << _value << "in table" << _table
@ -289,7 +314,12 @@ bool QueuedDatabase::remove(const QString &_table, const long long _id)
{ {
qCDebug(LOG_LIB) << "Remove row" << _id << "from" << _table; qCDebug(LOG_LIB) << "Remove row" << _id << "from" << _table;
QSqlQuery query = m_database.exec(QString("DELETE FROM %1 WHERE _id=%2").arg(_table).arg(_id)); QSqlQuery query(m_database);
query.prepare("DELETE FROM :table WHERE _id=:_id");
query.bindValue(":table", _table);
query.bindValue(":_id", _id);
query.exec();
QSqlError error = query.lastError(); QSqlError error = query.lastError();
if (error.isValid()) { if (error.isValid()) {
qCCritical(LOG_LIB) << "Could not remove record" << _id << "in table" << _table << "message" qCCritical(LOG_LIB) << "Could not remove record" << _id << "in table" << _table << "message"
@ -308,10 +338,11 @@ void QueuedDatabase::removeTasks(const QDateTime &_endTime)
{ {
qCDebug(LOG_LIB) << "Remove all tasks which are older than" << _endTime; qCDebug(LOG_LIB) << "Remove all tasks which are older than" << _endTime;
QSqlQuery query QSqlQuery query(m_database);
= m_database.exec(QString("DELETE FROM %1 WHERE datetime(endTime) < datetime('%2')") query.prepare("DELETE FROM :table WHERE datetime(endTime) < datetime(:time)");
.arg(QueuedDB::TASKS_TABLE) query.bindValue(":table", QueuedDB::TASKS_TABLE);
.arg(_endTime.toString(Qt::ISODateWithMs))); query.bindValue(":time", _endTime.toString(Qt::ISODateWithMs));
query.exec();
QSqlError error = query.lastError(); QSqlError error = query.lastError();
if (error.isValid()) if (error.isValid())
@ -326,10 +357,11 @@ void QueuedDatabase::removeTasks(const QDateTime &_endTime)
void QueuedDatabase::removeTokens() void QueuedDatabase::removeTokens()
{ {
QString now = QDateTime::currentDateTimeUtc().toString(Qt::ISODateWithMs); QString now = QDateTime::currentDateTimeUtc().toString(Qt::ISODateWithMs);
QSqlQuery query QSqlQuery query(m_database);
= m_database.exec(QString("DELETE FROM %1 WHERE datetime(validUntil) > datetime('%2')") query.prepare("DELETE FROM :table WHERE datetime(validUntil) < datetime(:time)");
.arg(QueuedDB::TOKENS_TABLE) query.bindValue(":table", QueuedDB::TOKENS_TABLE);
.arg(now)); query.bindValue(":time", now);
query.exec();
QSqlError error = query.lastError(); QSqlError error = query.lastError();
if (error.isValid()) if (error.isValid())
@ -345,10 +377,11 @@ void QueuedDatabase::removeUsers(const QDateTime &_lastLogin)
{ {
qCDebug(LOG_LIB) << "Remove all users which logged older than" << _lastLogin; qCDebug(LOG_LIB) << "Remove all users which logged older than" << _lastLogin;
QSqlQuery query QSqlQuery query(m_database);
= m_database.exec(QString("DELETE FROM %1 WHERE datetime(lastLogin) < datetime('%2')") query.prepare("DELETE FROM :table WHERE datetime(lastLogin) < datetime(:time)");
.arg(QueuedDB::USERS_TABLE) query.bindValue(":table", QueuedDB::USERS_TABLE);
.arg(_lastLogin.toString(Qt::ISODateWithMs))); query.bindValue(":time", _lastLogin.toString(Qt::ISODateWithMs));
query.exec();
QSqlError error = query.lastError(); QSqlError error = query.lastError();
if (error.isValid()) if (error.isValid())
@ -372,6 +405,22 @@ QStringList QueuedDatabase::getColumnsInRecord(const QSqlRecord &_record) const
} }
/**
* @fn getCondition
*/
QString QueuedDatabase::getCondition(const QList<QueuedDatabaseCondition> &_condition) const
{
if (_condition.isEmpty())
return "";
QStringList query;
for (auto &cond : _condition)
query += QString("%1 %2 :%1").arg(cond.key).arg(cond.operation);
return "WHERE " + query.join(',');
}
/** /**
* @fn lastInsertionId * @fn lastInsertionId
*/ */
@ -379,7 +428,11 @@ long long QueuedDatabase::lastInsertionId(const QString &_table) const
{ {
qCDebug(LOG_LIB) << "Get last row ID from" << _table; qCDebug(LOG_LIB) << "Get last row ID from" << _table;
QSqlQuery query = m_database.exec(QString("SELECT max(_id) FROM '%1'").arg(_table)); QSqlQuery query(m_database);
query.prepare("SELECT max(_id) FROM :table");
query.bindValue(":table", _table);
query.exec();
QSqlError error = query.lastError(); QSqlError error = query.lastError();
if (error.isValid()) { if (error.isValid()) {
qCCritical(LOG_LIB) << "Could not get last insertion ID"; qCCritical(LOG_LIB) << "Could not get last insertion ID";
@ -414,7 +467,7 @@ QHash<QString, QString> QueuedDatabase::getQueryPayload(const QString &_table,
qCWarning(LOG_LIB) << "Modifying record ID is not allowed"; qCWarning(LOG_LIB) << "Modifying record ID is not allowed";
continue; continue;
} }
output[key] = QString("'%1'").arg(_value[key].toString()); output[key] = QString(":%1").arg(key);
} }
return output; return output;

View File

@ -60,7 +60,7 @@ QList<QVariantHash> QueuedReportManager::performance(const QueuedCore *_core,
{ {
qCDebug(LOG_LIB) << "Build performance report from" << _from << "to" << _to; qCDebug(LOG_LIB) << "Build performance report from" << _from << "to" << _to;
QStringList conditions; QList<QueuedDatabaseCondition> conditions;
if (_from.isValid()) if (_from.isValid())
conditions += QString("((datetime(startTime) > datetime('%1')) OR " conditions += QString("((datetime(startTime) > datetime('%1')) OR "
"(startTime IS NULL))") "(startTime IS NULL))")