mirror of https://github.com/Qortal/Brooklyn
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
3.2 KiB
114 lines
3.2 KiB
/* |
|
SPDX-FileCopyrightText: 2007 Glenn Ergeerts <[email protected]> |
|
SPDX-FileCopyrightText: 2012 Marco Gulino <[email protected]> |
|
|
|
SPDX-License-Identifier: LGPL-2.0-or-later |
|
*/ |
|
|
|
#include "fetchsqlite.h" |
|
#include "bookmarks_debug.h" |
|
#include "bookmarksrunner_defs.h" |
|
#include <QDebug> |
|
#include <QFile> |
|
#include <QMutexLocker> |
|
#include <QSqlError> |
|
#include <QSqlQuery> |
|
#include <QSqlRecord> |
|
#include <sstream> |
|
#include <thread> |
|
|
|
FetchSqlite::FetchSqlite(const QString &databaseFile, QObject *parent) |
|
: QObject(parent) |
|
, m_databaseFile(databaseFile) |
|
{ |
|
} |
|
|
|
FetchSqlite::~FetchSqlite() |
|
{ |
|
} |
|
|
|
void FetchSqlite::prepare() |
|
{ |
|
} |
|
|
|
void FetchSqlite::teardown() |
|
{ |
|
const QString connectionPrefix = m_databaseFile + "-"; |
|
const auto connections = QSqlDatabase::connectionNames(); |
|
for (const auto &c : connections) { |
|
if (c.startsWith(connectionPrefix)) { |
|
qCDebug(RUNNER_BOOKMARKS) << "Closing connection" << c; |
|
QSqlDatabase::removeDatabase(c); |
|
} |
|
} |
|
} |
|
|
|
static QSqlDatabase openDbConnection(const QString &databaseFile) |
|
{ |
|
// create a thread unique connection name based on the DB filename and thread id |
|
auto connection = databaseFile + "-"; |
|
std::stringstream s; |
|
s << std::this_thread::get_id(); |
|
connection += QString::fromStdString(s.str()); |
|
|
|
// Try to reuse the previous connection |
|
auto db = QSqlDatabase::database(connection); |
|
if (db.isValid()) { |
|
qCDebug(RUNNER_BOOKMARKS) << "Reusing connection" << connection; |
|
return db; |
|
} |
|
|
|
// Otherwise, create, configure and open a new one |
|
db = QSqlDatabase::addDatabase(QStringLiteral("QSQLITE"), connection); |
|
db.setHostName(QStringLiteral("localhost")); |
|
db.setDatabaseName(databaseFile); |
|
db.open(); |
|
qCDebug(RUNNER_BOOKMARKS) << "Opened connection" << connection; |
|
|
|
return db; |
|
} |
|
|
|
QList<QVariantMap> FetchSqlite::query(const QString &sql, QMap<QString, QVariant> bindObjects) |
|
{ |
|
QMutexLocker lock(&m_mutex); |
|
|
|
auto db = openDbConnection(m_databaseFile); |
|
if (!db.isValid()) { |
|
return QList<QVariantMap>(); |
|
} |
|
|
|
// qDebug() << "query: " << sql; |
|
QSqlQuery query(db); |
|
query.prepare(sql); |
|
for (auto entry = bindObjects.constKeyValueBegin(); entry != bindObjects.constKeyValueEnd(); ++entry) { |
|
query.bindValue((*entry).first, (*entry).second); |
|
// qDebug() << "* Bound " << variableName << " to " << query.boundValue(variableName); |
|
} |
|
|
|
if (!query.exec()) { |
|
QSqlError error = db.lastError(); |
|
// qDebug() << "query failed: " << error.text() << " (" << error.type() << ", " << error.number() << ")"; |
|
// qDebug() << query.lastQuery(); |
|
} |
|
|
|
QList<QVariantMap> result; |
|
while (query.next()) { |
|
QVariantMap recordValues; |
|
QSqlRecord record = query.record(); |
|
for (int field = 0; field < record.count(); field++) { |
|
QVariant value = record.value(field); |
|
recordValues.insert(record.fieldName(field), value); |
|
} |
|
result << recordValues; |
|
} |
|
|
|
return result; |
|
} |
|
|
|
QStringList FetchSqlite::tables(QSql::TableType type) |
|
{ |
|
QMutexLocker lock(&m_mutex); |
|
|
|
auto db = openDbConnection(m_databaseFile); |
|
return db.tables(type); |
|
}
|
|
|