Herunterladen als
root/src/database.cpp @ 9cf4747e
81c23b56 | Christian Ehringfeld | /*
|
|
6899f814 | Christian Ehringfeld | * Copyright (C) 2015 Christian Ehringfeld <c.ehringfeld@t-online.de>
|
|
*
|
|||
* This program is free software; you can redistribute it and/or modify it
|
|||
* under the terms of the GNU Lesser General Public License as published by
|
|||
* the Free Software Foundation.
|
|||
*
|
|||
* This program is distributed in the hope that it will be useful, but
|
|||
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|||
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|||
* for more details.
|
|||
*
|
|||
* You should have received a copy of the GNU Lesser General Public License
|
|||
* along with this program; if not, see <http://www.gnu.org/licenses/>.
|
|||
*/
|
|||
81c23b56 | Christian Ehringfeld | ||
#include "database.h"
|
|||
9cf4747e | Christian Ehringfeld | using namespace CuteEntityManager;
|
|
81c23b56 | Christian Ehringfeld | ||
Database::Database(QSqlDatabase database) {
|
|||
this->database = database;
|
|||
this->init();
|
|||
this->connectionName = this->database.connectionName();
|
|||
}
|
|||
Database::Database(QString databaseType, QString connectionName, QString databasename) {
|
|||
426974c6 | Christian Ehringfeld | this->database = QSqlDatabase::addDatabase(databaseType, connectionName);
|
|
81c23b56 | Christian Ehringfeld | this->connectionName = connectionName;
|
|
this->database.setDatabaseName(databasename);
|
|||
this->init();
|
|||
}
|
|||
426974c6 | Christian Ehringfeld | Database::Database(QString databaseType, QString connectionName, QString hostname, QString databasename,
|
|
QString username, QString password, qint64 port) {
|
|||
this->database = QSqlDatabase::addDatabase(databaseType, connectionName);
|
|||
81c23b56 | Christian Ehringfeld | this->connectionName = connectionName;
|
|
this->database.setHostName(hostname);
|
|||
this->database.setDatabaseName(databasename);
|
|||
this->database.setUserName(username);
|
|||
this->database.setPassword(password);
|
|||
this->database.setPort(port);
|
|||
this->init();
|
|||
}
|
|||
void Database::init() {
|
|||
this->database.open();
|
|||
this->databasetype = this->getDatabaseType();
|
|||
this->supportTransactions = this->database.driver()->hasFeature(QSqlDriver::Transactions);
|
|||
9cf4747e | Christian Ehringfeld | this->tableList = QStringList();
|
|
81c23b56 | Christian Ehringfeld | this->getTableListFromDatabase();
|
|
}
|
|||
a604a5a2 | Christian Ehringfeld | DatabaseType Database::getDatabaseType() {
|
|
b0b8dac3 | Christian Ehringfeld | return CuteEntityManager::getDatabaseType(this->database.driverName());
|
|
81c23b56 | Christian Ehringfeld | }
|
|
void Database::getTableListFromDatabase() {
|
|||
426974c6 | Christian Ehringfeld | if (this->database.open()) {
|
|
81c23b56 | Christian Ehringfeld | QString q = "";
|
|
QSqlQuery query = QSqlQuery(this->database);
|
|||
query.prepare(q);
|
|||
this->select(query);
|
|||
this->setTableList(query);
|
|||
}
|
|||
}
|
|||
4d58ef6a | Christian Ehringfeld | //QString Database::mysqlTableList() {
|
|
// return "SHOW TABLES;";
|
|||
//}
|
|||
81c23b56 | Christian Ehringfeld | ||
4d58ef6a | Christian Ehringfeld | //QString Database::pgsqlTableList() {
|
|
// return "SELECT table_name FROM information_schema.tables WHERE table_catalog = '"+this->database.databaseName()+"';";
|
|||
//}
|
|||
81c23b56 | Christian Ehringfeld | ||
Database::~Database() {
|
|||
426974c6 | Christian Ehringfeld | if (this->database.isOpen()) {
|
|
81c23b56 | Christian Ehringfeld | this->database.close();
|
|
}
|
|||
QSqlDatabase::removeDatabase(this->connectionName);
|
|||
delete this->tableList;
|
|||
}
|
|||
QString Database::getConnectionName() {
|
|||
return this->connectionName;
|
|||
}
|
|||
caea9141 | Christian Ehringfeld | //QString Database::pgsqlSeqTable() {
|
|
// return "CREATE TABLE IF NOT EXISTS sequence (SEQ_NAME varchar(255) NOT NULL UNIQUE , SEQ_COUNT bigint NOT NULL);";
|
|||
//}
|
|||
81c23b56 | Christian Ehringfeld | ||
caea9141 | Christian Ehringfeld | //QString Database::mysqlSeqTable() {
|
|
// return "CREATE TABLE IF NOT EXISTS `sequence` (`SEQ_NAME` varchar(255) NOT NULL UNIQUE , `SEQ_COUNT` bigint(20) unsigned NOT NULL) CHARSET = utf8";
|
|||
//}
|
|||
81c23b56 | Christian Ehringfeld | ||
caea9141 | Christian Ehringfeld | //QString Database::sqliteSeqTable() {
|
|
// return "CREATE TABLE IF NOT EXISTS \"sequence\" (\"SEQ_NAME\" TEXT PRIMARY KEY NOT NULL , \"SEQ_COUNT\" INTEGER NOT NULL );";
|
|||
//}
|
|||
81c23b56 | Christian Ehringfeld | ||
QChar Database::escapeChar() {
|
|||
QChar c = QChar();
|
|||
426974c6 | Christian Ehringfeld | if (this->databasetype == CuteEntityManager::SQLITE) {
|
|
81c23b56 | Christian Ehringfeld | c = '\'';
|
|
426974c6 | Christian Ehringfeld | } else if (this->databasetype == CuteEntityManager::MYSQL) {
|
|
81c23b56 | Christian Ehringfeld | c = '`';
|
|
}
|
|||
return c;
|
|||
}
|
|||
bool Database::transaction(const QString &query) {
|
|||
bool rc = false;
|
|||
426974c6 | Christian Ehringfeld | if (supportTransactions) {
|
|
81c23b56 | Christian Ehringfeld | this->database.transaction();
|
|
QSqlQuery sqlquery = QSqlQuery(this->database);
|
|||
sqlquery.exec(query);
|
|||
426974c6 | Christian Ehringfeld | if (!this->database.commit()) {
|
|
81c23b56 | Christian Ehringfeld | this->database.rollback();
|
|
}
|
|||
} else {
|
|||
rc = this->exec(query);
|
|||
}
|
|||
return rc;
|
|||
}
|
|||
QSqlQuery Database::getQuery() {
|
|||
return QSqlQuery(this->database);
|
|||
}
|
|||
QSqlQuery Database::getQuery(const QString &prepare) {
|
|||
QSqlQuery q = QSqlQuery(this->database);
|
|||
q.prepare(prepare);
|
|||
return q;
|
|||
}
|
|||
bool Database::transaction(const QStringList &queries) {
|
|||
bool ok = false;
|
|||
426974c6 | Christian Ehringfeld | if (this->supportTransactions) {
|
|
81c23b56 | Christian Ehringfeld | this->database.transaction();
|
|
QSqlQuery sqlquery = QSqlQuery(this->database);
|
|||
for (int var = 0; var < queries.size(); ++var) {
|
|||
sqlquery.exec(queries.at(var));
|
|||
}
|
|||
426974c6 | Christian Ehringfeld | if (!this->database.commit()) {
|
|
81c23b56 | Christian Ehringfeld | this->database.rollback();
|
|
} else {
|
|||
ok = true;
|
|||
}
|
|||
} else {
|
|||
ok = this->exec(queries);
|
|||
}
|
|||
return ok;
|
|||
}
|
|||
bool Database::transaction(QSqlQuery &query) {
|
|||
this->database.transaction();
|
|||
query.exec();
|
|||
426974c6 | Christian Ehringfeld | if (!this->database.commit()) {
|
|
81c23b56 | Christian Ehringfeld | this->database.rollback();
|
|
return false;
|
|||
}
|
|||
return true;
|
|||
}
|
|||
bool Database::transaction(QList<QSqlQuery> &queries) {
|
|||
this->database.transaction();
|
|||
QSqlQuery q;
|
|||
for (int var = 0; var < queries.size(); ++var) {
|
|||
q = queries.at(var);
|
|||
q.exec();
|
|||
}
|
|||
426974c6 | Christian Ehringfeld | if (!this->database.commit()) {
|
|
81c23b56 | Christian Ehringfeld | this->database.rollback();
|
|
return false;
|
|||
}
|
|||
return true;
|
|||
}
|
|||
bool Database::exec(QString query) {
|
|||
this->database.transaction();
|
|||
QSqlQuery q = QSqlQuery(this->database);
|
|||
q.exec(query);
|
|||
426974c6 | Christian Ehringfeld | if (!this->database.commit()) {
|
|
81c23b56 | Christian Ehringfeld | this->database.rollback();
|
|
return false;
|
|||
}
|
|||
return true;
|
|||
}
|
|||
426974c6 | Christian Ehringfeld | bool Database::exec(QStringList queries) {
|
|
81c23b56 | Christian Ehringfeld | QSqlQuery q = QSqlQuery(this->database);
|
|
bool ok = true;
|
|||
for (int var = 0; var < queries.size() && ok; ++var) {
|
|||
ok = q.exec(queries.at(var));
|
|||
426974c6 | Christian Ehringfeld | if (!ok) {
|
|
81c23b56 | Christian Ehringfeld | break;
|
|
}
|
|||
}
|
|||
return ok;
|
|||
}
|
|||
bool Database::exec(QSqlQuery query) {
|
|||
return query.exec();
|
|||
}
|
|||
bool Database::exec(QList<QSqlQuery> queries) {
|
|||
bool ok = true;
|
|||
QSqlQuery q = QSqlQuery(this->database);
|
|||
for (int var = 0; var < queries.size() && ok; ++var) {
|
|||
q = queries.at(var);
|
|||
ok = q.exec();
|
|||
426974c6 | Christian Ehringfeld | if (!ok) {
|
|
81c23b56 | Christian Ehringfeld | break;
|
|
}
|
|||
}
|
|||
return ok;
|
|||
}
|
|||
bool Database::select(QSqlQuery &query) {
|
|||
query.setForwardOnly(true);
|
|||
return query.exec();
|
|||
}
|
|||
QSqlQuery Database::select(const QString &query) {
|
|||
QSqlQuery q = QSqlQuery(this->database);
|
|||
q.exec(query);
|
|||
return q;
|
|||
}
|
|||
QString Database::querySequenceCounter() {
|
|||
return "INSERT INTO sequence (SEQ_NAME, SEQ_COUNT) VALUES(\'id_count\',\'0\');";
|
|||
}
|
|||
void Database::refreshTableList() {
|
|||
this->tableList->clear();
|
|||
this->getTableListFromDatabase();
|
|||
}
|
|||
bool Database::containsTable(QString tblname) {
|
|||
return this->tableList->contains(tblname);
|
|||
}
|
|||
bool Database::updateSequenceCounter(QSqlQuery &q) {
|
|||
QList<QSqlQuery> l = QList<QSqlQuery>();
|
|||
426974c6 | Christian Ehringfeld | l.append(QSqlQuery("UPDATE sequence SET SEQ_COUNT=(SEQ_COUNT+1);", this->database));
|
|
81c23b56 | Christian Ehringfeld | l.append(q);
|
|
return this->transaction(l);
|
|||
}
|
|||
qint64 Database::getLastId() {
|
|||
qint64 id = -1;
|
|||
QSqlQuery q = this->select("SELECT SEQ_COUNT FROM sequence WHERE SEQ_NAME=\'id_count\';");
|
|||
426974c6 | Christian Ehringfeld | if (q.next()) {
|
|
81c23b56 | Christian Ehringfeld | id = q.value(0).toInt();
|
|
}
|
|||
return id;
|
|||
}
|
|||
QSqlDatabase Database::getDatabase() {
|
|||
return this->database;
|
|||
}
|