blob: 203cfdaa161bc299b5510d6d0d9f606c569df456 (
plain)
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
|
#include "sql_qt_wrapper/sql_database.h"
#include <QSqlError>
namespace {
const QString kDatabaseExtension = ".sqlite";
} // namespace
namespace utils {
namespace dbms {
SQLDatabase::SQLDatabase() : database_path_() {}
SQLDatabase::SQLDatabase(const std::string& database_path,
const std::string& connection_name)
: database_path_((database_path + kDatabaseExtension.toStdString()).c_str())
, connection_name_(connection_name.c_str()) {
db_ = QSqlDatabase::addDatabase("QSQLITE", connection_name_);
}
SQLDatabase::~SQLDatabase() {
Close();
sync_primitives::AutoLock auto_lock(conn_lock_);
/*
* All database queries and connections should be destroyed
* before database removing. See
* http://doc.qt.io/qt-5/qsqldatabase.html#removeDatabase
*/
db_ = QSqlDatabase();
QSqlDatabase::removeDatabase(connection_name_);
}
bool SQLDatabase::Open() {
sync_primitives::AutoLock auto_lock(conn_lock_);
db_.setDatabaseName(database_path_);
return db_.open();
}
void SQLDatabase::Close() {
sync_primitives::AutoLock auto_lock(conn_lock_);
if (db_.isOpen()) {
db_.close();
}
}
bool SQLDatabase::BeginTransaction() {
sync_primitives::AutoLock auto_lock(conn_lock_);
return db_.transaction();
}
bool SQLDatabase::CommitTransaction() {
sync_primitives::AutoLock auto_lock(conn_lock_);
return db_.commit();
}
bool SQLDatabase::RollbackTransaction() {
sync_primitives::AutoLock auto_lock(conn_lock_);
return db_.rollback();
}
SQLError SQLDatabase::LastError() const {
sync_primitives::AutoLock auto_lock(conn_lock_);
return SQLError(db_.lastError());
}
bool SQLDatabase::HasErrors() const {
sync_primitives::AutoLock auto_lock(conn_lock_);
return db_.lastError().type() != QSqlError::NoError;
}
std::string SQLDatabase::get_path() const {
return database_path_.toStdString();
}
bool SQLDatabase::IsReadWrite() {
return true;
}
SQLDatabase::operator QSqlDatabase() const {
return db_;
}
bool SQLDatabase::Backup() {
return true;
}
} // namespace dbms
} // namespace utils
|
