1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
/*
SPDX-FileCopyrightText: 2007 Glenn Ergeerts <glenn.ergeerts@telenet.be>
SPDX-FileCopyrightText: 2012 Marco Gulino <marco.gulino@xpeppers.com>
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);
}
|
