Projekt

Allgemein

Profil

Herunterladen als
Herunterladen (6,35 KB) Statistiken
| Zweig: | Revision:
c22391b2 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/>.
*/
#include "schema.h"
9cf4747e Christian Ehringfeld
#include <QRegularExpression>
#include <QSqlRecord>
#include <QSqlQuery>
14f9beed Christian Ehringfeld
#include "database.h"
4d58ef6a Christian Ehringfeld
using namespace CuteEntityManager;
aa44e7d1 Christian Ehringfeld
d2c13492 Christian Ehringfeld
Schema::Schema(QSharedPointer<Database> database,
QSharedPointer<QueryBuilder> builder) {
9cf4747e Christian Ehringfeld
this->database = database;
e0e1ead8 Christian Ehringfeld
this->abstractTypeMap = QSharedPointer<QHash<QString, QString>>
(new QHash<QString, QString>());
696666eb Christian Ehringfeld
this->queryBuilder = builder;
e0e1ead8 Christian Ehringfeld
this->typeMap = QSharedPointer<QHash<QString, QString>>(new
QHash<QString, QString>());
24425325 Christian Ehringfeld
this->initAbstractDatabaseTypes();
aa44e7d1 Christian Ehringfeld
}

4d58ef6a Christian Ehringfeld
Schema::~Schema() {
aa44e7d1 Christian Ehringfeld
}

24425325 Christian Ehringfeld
void Schema::initAbstractDatabaseTypes() {
696666eb Christian Ehringfeld
this->abstractTypeMap = QSharedPointer<QHash<QString, QString>>::create();
24425325 Christian Ehringfeld
this->abstractTypeMap.data()->insert("bool", TYPE_SMALLINT);
this->abstractTypeMap.data()->insert("short", TYPE_SMALLINT);
this->abstractTypeMap.data()->insert("int", TYPE_INTEGER);
this->abstractTypeMap.data()->insert("long", TYPE_INTEGER);
this->abstractTypeMap.data()->insert("long long", TYPE_INTEGER);
this->abstractTypeMap.data()->insert("qlonglong", TYPE_INTEGER);
this->abstractTypeMap.data()->insert("float", TYPE_FLOAT);
this->abstractTypeMap.data()->insert("double", TYPE_FLOAT);
this->abstractTypeMap.data()->insert("long double", TYPE_FLOAT);
this->abstractTypeMap.data()->insert("qint", TYPE_INTEGER);
this->abstractTypeMap.data()->insert("quint", TYPE_INTEGER);
this->abstractTypeMap.data()->insert("quuid", TYPE_INTEGER);
this->abstractTypeMap.data()->insert("qfloat", TYPE_FLOAT);
this->abstractTypeMap.data()->insert("unsigned short", TYPE_SMALLINT);
this->abstractTypeMap.data()->insert("unsigned int", TYPE_INTEGER);
this->abstractTypeMap.data()->insert("unsigned long", TYPE_INTEGER);
this->abstractTypeMap.data()->insert("unsigned long long", TYPE_INTEGER);
this->abstractTypeMap.data()->insert("char", TYPE_CHAR);
this->abstractTypeMap.data()->insert("std::string", TYPE_TEXT);
this->abstractTypeMap.data()->insert("std::wstring", TYPE_TEXT);
this->abstractTypeMap.data()->insert("QString", TYPE_TEXT);
this->abstractTypeMap.data()->insert("QVariant", TYPE_TEXT);
this->abstractTypeMap.data()->insert("QUuid", TYPE_TEXT);
this->abstractTypeMap.data()->insert("QDate", TYPE_DATE);
this->abstractTypeMap.data()->insert("QTime", TYPE_TIME);
this->abstractTypeMap.data()->insert("QDateTime", TYPE_DATETIME);
this->abstractTypeMap.data()->insert("QByteArray", TYPE_BINARY);
this->abstractTypeMap.data()->insert("QBitArray", TYPE_BINARY);
b0bf458e Christian Ehringfeld
}

a604a5a2 Christian Ehringfeld
QString Schema::quoteSimpleTableName(QString name) {
caea9141 Christian Ehringfeld
return name.indexOf("`") ? name : "`" + name + "`";
a604a5a2 Christian Ehringfeld
}

426974c6 Christian Ehringfeld
QString Schema::quoteTableName(QString name) {
9cf4747e Christian Ehringfeld
if (name.indexOf("(") || name.indexOf("{{")) {
return name;
}
if (name.indexOf(".") == -1) {
return this->quoteSimpleTableName(name);
}
QStringList parts = name.split(".");
for (int i = 0; i < parts.size(); ++i) {
parts.replace(i, this->quoteSimpleTableName(parts.at(i)));
}
return parts.join(".");
426974c6 Christian Ehringfeld
}

QString Schema::quoteColumnName(QString name) {
9cf4747e Christian Ehringfeld
if (name.indexOf("(") || name.indexOf("[[") || name.indexOf("{{")) {
return name;
}
int pos = name.indexOf(".");
QString prefix = "";
if (pos) {
prefix = this->quoteTableName(name.mid(0, pos)) + ".";
name = name.mid(pos + 1);
}
return prefix + this->quoteSimpleColumnName(name);
426974c6 Christian Ehringfeld
}

QString Schema::quoteSimpleColumnName(QString name) {
caea9141 Christian Ehringfeld
return name.indexOf("`") || name == "*" ? name : "`" + name + "`";
426974c6 Christian Ehringfeld
}

e0e1ead8 Christian Ehringfeld
QHash<QString, QSharedPointer<TableSchema> > Schema::getTableSchemas(
QString schema, bool refresh) {
9cf4747e Christian Ehringfeld
QStringList names = this->getTableNames();
for (int i = 0; i < names.size(); ++i) {
QString name;
if (schema != "") {
name = schema + "." + names.at(i);
}
b0bf458e Christian Ehringfeld
this->getTableSchema(name, refresh);
9cf4747e Christian Ehringfeld
}
return this->tables;
426974c6 Christian Ehringfeld
}

caea9141 Christian Ehringfeld
QStringList Schema::getTableNames(QString schema) {
14f9beed Christian Ehringfeld
return this->database.data()->getDatabase().tables();
426974c6 Christian Ehringfeld
}

9cf4747e Christian Ehringfeld
QVariant Schema::getLastInsertID() {
14f9beed Christian Ehringfeld
QSqlQuery q(this->database.data()->getDatabase());
6d91d381 Christian Ehringfeld
auto lastId = q.lastInsertId();
return lastId;
426974c6 Christian Ehringfeld
}

void Schema::refresh() {
9cf4747e Christian Ehringfeld
this->tables.clear();
426974c6 Christian Ehringfeld
}

QString Schema::getRawTable(QString name) {
9cf4747e Christian Ehringfeld
if (name.indexOf("{{")) {
9c2f773f Christian Ehringfeld
QRegularExpression re(QRegularExpression::escape("/\\{\{(.*?)\\}\\}/"));
return name.replace(re, QRegularExpression::escape("\\1"));
9cf4747e Christian Ehringfeld
}
return name;
}

14f9beed Christian Ehringfeld
bool Schema::containsTable(QString tblname) {
return this->tables.contains(tblname);
}

b0bf458e Christian Ehringfeld
QSharedPointer<TableSchema> Schema::getTableSchema(QString name, bool refresh) {
9cf4747e Christian Ehringfeld
if (refresh) {
this->refresh();
}
if (this->tables.contains(name)) {
b0bf458e Christian Ehringfeld
return this->tables.value(name);
9cf4747e Christian Ehringfeld
}
QString realName = this->getRawTable(name);
14f9beed Christian Ehringfeld
auto ts = this->loadTableSchema(realName);
9cf4747e Christian Ehringfeld
if (ts.data()) {
this->tables.insert(name, ts);
}
b0bf458e Christian Ehringfeld
return ts;
14f9beed Christian Ehringfeld
}

QSharedPointer<Database> Schema::getDatabase() const {
return database;
}
9cf4747e Christian Ehringfeld
14f9beed Christian Ehringfeld
void Schema::setDatabase(const QSharedPointer<Database> &value) {
database = value;
9cf4747e Christian Ehringfeld
}
b0bf458e Christian Ehringfeld
QSharedPointer<QueryBuilder> Schema::getQueryBuilder() const {
return queryBuilder;
}
14f9beed Christian Ehringfeld
24425325 Christian Ehringfeld
QSharedPointer<QHash<QString, QString> > Schema::getAbstractTypeMap() const {
return abstractTypeMap;
}

e0e1ead8 Christian Ehringfeld
void Schema::setAbstractTypeMap(const QSharedPointer<QHash<QString, QString> >
&value) {
24425325 Christian Ehringfeld
abstractTypeMap = value;
}


9cf4747e Christian Ehringfeld
QHash<QString, QSharedPointer<TableSchema> > Schema::getTables() const {
return this->tables;
}

e0e1ead8 Christian Ehringfeld
void Schema::setTables(const QHash<QString, QSharedPointer<TableSchema> >
&value) {
9cf4747e Christian Ehringfeld
tables = value;
a604a5a2 Christian Ehringfeld
}