|
#include "querybuilder.h"
|
|
#include "database.h"
|
|
#include <QMetaObject>
|
|
#include <QMetaProperty>
|
|
#include "entity.h"
|
|
using namespace CuteEntityManager;
|
|
|
|
//bool QueryBuilder::createTable(QString tablename, QHash<QString, QString> tableDefinition) {
|
|
//// QHash<QString, QString> Artikel::getProperties(DatabaseType type) {
|
|
//// QHash<QString, QString> h = QHash<QString, QString>();
|
|
//// h.insert("id",this->idColumnSQL());
|
|
//// h.insert("preis","DOUBLE");
|
|
//// h.insert("name","TEXT");
|
|
//// return h;
|
|
//// }
|
|
//}
|
|
|
|
QueryBuilder::QueryBuilder(QSharedPointer<Schema> schema, QSharedPointer<Database> database) {
|
|
this->schema = schema;
|
|
this->database = database;
|
|
}
|
|
|
|
QueryBuilder::~QueryBuilder() {
|
|
|
|
}
|
|
|
|
bool QueryBuilder::createTable(const QSharedPointer<Entity> &entity) const {
|
|
return this->createTable(entity.data()->getTablename(), this->generateTableDefinition(entity));
|
|
}
|
|
|
|
bool QueryBuilder::createTable(const QString &tableName, const QHash<QString, QString> &tableDefinition) const {
|
|
bool rc = false;
|
|
this->schema.data()->containsTable(tableName) ? rc = true : rc = false;
|
|
if (!rc) {
|
|
QSqlQuery q = this->database.data()->getQuery(this->createTableQuery(tableName, tableDefinition));
|
|
if (this->database.data()->transaction(q)) {
|
|
this->schema.data()->getTableSchema(tableName);
|
|
rc = true;
|
|
}
|
|
}
|
|
return rc;
|
|
}
|
|
|
|
QString QueryBuilder::createTableQuery(const QString &tableName, const QHash<QString, QString> &tableDefinition) const {
|
|
bool first = true;
|
|
QString s = "CREATE TABLE ";
|
|
s.append(this->schema.data()->quoteTableName(tableName).append(" (\n"));
|
|
QHash<QString, QString>::const_iterator i = tableDefinition.constBegin();
|
|
while (i != tableDefinition.constEnd()) {
|
|
if (first) {
|
|
first = false;
|
|
} else {
|
|
s.append(',');
|
|
}
|
|
s.append(this->schema.data()->quoteColumnName(i.key())).append(" " + i.value());
|
|
++i;
|
|
}
|
|
s.append("\n);");
|
|
return s;
|
|
}
|
|
|
|
QString QueryBuilder::renameTable(QString tableName, QString newName) const {
|
|
return "RENAME TABLE " + this->schema.data()->quoteTableName(tableName) + " TO " + this->schema.data()->quoteTableName(
|
|
newName);
|
|
}
|
|
|
|
QString QueryBuilder::dropTable(QString tableName) const {
|
|
return "DROP TABLE " + this->schema.data()->quoteTableName(tableName);
|
|
}
|
|
|
|
QString QueryBuilder::truncateTable(QString tableName) const {
|
|
return "TRUNCATE TABLE " + this->schema.data()->quoteTableName(tableName);
|
|
}
|
|
|
|
QString QueryBuilder::addColumn(QString tableName, QString columnName, QString columnType) const {
|
|
return "ALTER TABLE " + this->schema.data()->quoteTableName(tableName) + " ADD " + this->schema.data()->quoteColumnName(
|
|
columnName) + " " + this->getColumnType(columnType);
|
|
}
|
|
|
|
QString QueryBuilder::dropColumn(QString tableName, QString columName) const {
|
|
return "ALTER TABLE " + this->schema.data()->quoteTableName(tableName) + " DROP COLUMN " +
|
|
this->schema.data()->quoteColumnName(columName);
|
|
}
|
|
|
|
QString QueryBuilder::renameColumn(QString tableName, QString oldName, QString newName) const {
|
|
return "ALTER TABLE " + this->schema.data()->quoteTableName(tableName) + " RENAME COLUMN " +
|
|
this->schema.data()->quoteColumnName(oldName) + " TO " + this->schema.data()->quoteColumnName(newName);
|
|
}
|
|
|
|
QString QueryBuilder::alterColumn(QString tableName, QString columnName, QString newType) const {
|
|
return "ALTER TABLE " + this->schema.data()->quoteTableName(tableName) + " CHANGE " +
|
|
this->schema.data()->quoteColumnName(columnName) + " " +
|
|
this->schema.data()->quoteColumnName(columnName) + this->getColumnType(newType);
|
|
}
|
|
|
|
QString QueryBuilder::addPrimaryKey(QString name, QString tableName, QStringList columns) const {
|
|
return "ALTER TABLE " + this->schema.data()->quoteTableName(tableName) + " ADD CONSTRAINT " +
|
|
this->schema.data()->quoteColumnName(name) + "PRIMARY KEY (" + this->buildColumns(columns) + " )";
|
|
}
|
|
|
|
QString QueryBuilder::dropPrimaryKey(QString name, QString tableName) const {
|
|
return "ALTER TABLE " + this->schema.data()->quoteTableName(tableName) + " DROP CONSTRAINT " +
|
|
this->schema.data()->quoteColumnName(name);
|
|
}
|
|
|
|
QString QueryBuilder::addForeignKey(QString name, QString tableName, QStringList columns, QString refTableName,
|
|
QStringList refColumns, QString deleteConstraint, QString updateConstraint) const {
|
|
QString r = "ALTER TABLE " + this->schema.data()->quoteTableName(tableName) + "ADD CONSTRAINT " +
|
|
this->schema.data()->quoteColumnName(name)
|
|
+ " FOREIGN KEY (" + this->buildColumns(columns) + ")" + " REFERENCES " + this->schema.data()->quoteTableName(
|
|
refTableName) +
|
|
" (" + this->buildColumns(refColumns) + ")";
|
|
if (!deleteConstraint.isEmpty()) {
|
|
r.append(" ON DELETE " + deleteConstraint);
|
|
}
|
|
if (!updateConstraint.isEmpty()) {
|
|
r.append(" ON UPDATE " + updateConstraint);
|
|
}
|
|
return r;
|
|
}
|
|
|
|
QString QueryBuilder::dropForeignKey(QString name, QString tableName) const {
|
|
return "ALTER TABLE " + this->schema.data()->quoteTableName(tableName) + " DROP CONSTRAINT " +
|
|
this->schema.data()->quoteColumnName(name);
|
|
}
|
|
|
|
QString QueryBuilder::createIndex(QString name, QString tableName, QStringList columns, bool unique) const {
|
|
QString s = (unique ? "CREATE UNIQUE INDEX " : "CREATE INDEX ") + this->schema.data()->quoteTableName(
|
|
name) + " ON " + this->schema.data()->quoteTableName(tableName) + " (";
|
|
s.append(this->buildColumns(columns));
|
|
s.append(");");
|
|
return s;
|
|
}
|
|
|
|
QString QueryBuilder::dropIndex(QString name, QString tableName) const {
|
|
return "DROP INDEX " + this->schema.data()->quoteTableName(name) + " ON " + this->schema.data()->quoteTableName(
|
|
tableName);
|
|
}
|
|
|
|
QSharedPointer<Database> QueryBuilder::getDatabase() const {
|
|
return database;
|
|
}
|
|
|
|
void QueryBuilder::setDatabase(const QSharedPointer<Database> &value) {
|
|
database = value;
|
|
}
|
|
|
|
QHash<QString, QString> QueryBuilder::generateTableDefinition(const QSharedPointer<Entity> &entity) const {
|
|
auto map = QHash<QString, QString>();
|
|
auto o = entity.data()->metaObject();
|
|
QHash<QString, Relation> relations = entity.data()->getRelations();
|
|
for (int var = 0; var < o->propertyCount(); ++var) {
|
|
o->property(var);
|
|
auto m = o->property(var);
|
|
if (m.isReadable() && !entity.data()->getTransientAttributes().contains(m.name())) {
|
|
if (m.isEnumType()) {
|
|
map.insert(m.name(), this->schema.data()->getTypeMap().data()->value(this->schema.data()->TYPE_INTEGER));
|
|
} else if (relations.contains(m.name())) {
|
|
Relation r = relations.value(m.name());
|
|
if (r.getType() == RelationType::BELONGS_TO) {
|
|
//@TODO detect if id is BIGINT or only Integer
|
|
map.insert(QString(m.name()) + "_id", this->schema.data()->TYPE_INTEGER);
|
|
}
|
|
} else if (entity.data()->getBLOBColumns().contains(m.name())) {
|
|
map.insert(m.name(), this->schema.data()->getTypeMap().data()->value(this->schema.data()->TYPE_BINARY));
|
|
} else {
|
|
map.insert(m.name(), this->transformAbstractTypeToRealDbType(this->transformTypeToAbstractDbType(m.typeName())));
|
|
}
|
|
}
|
|
}
|
|
QString pkType = map.value(entity.data()->getPrimaryKey());
|
|
if (pkType == this->schema.data()->TYPE_BIGINT) {
|
|
map.insert(entity.data()->getPrimaryKey(), this->schema.data()->TYPE_BIGPK);
|
|
} else {
|
|
map.insert(entity.data()->getPrimaryKey(), this->schema.data()->TYPE_PK);
|
|
}
|
|
return map;
|
|
}
|
|
|
|
QString QueryBuilder::transformTypeToAbstractDbType(QString typeName) const {
|
|
QHash<QString, QString> *m = this->schema.data()->getAbstractTypeMap().data();
|
|
if (m->contains(typeName)) {
|
|
return m->value(typeName);
|
|
}
|
|
QHash<QString, QString>::iterator i;
|
|
for (i = m->begin(); i != m->end(); ++i) {
|
|
if (i.key().contains(typeName)) {
|
|
return i.value();
|
|
}
|
|
}
|
|
return this->schema.data()->TYPE_TEXT;
|
|
}
|
|
|
|
QString QueryBuilder::transformAbstractTypeToRealDbType(QString typeName) const {
|
|
return this->schema.data()->getTypeMap().data()->value(typeName);
|
|
}
|
|
|
|
QString QueryBuilder::getColumnType(QString type) const {
|
|
/**
|
|
* @TODO
|
|
*/
|
|
return this->transformAbstractTypeToRealDbType(type);
|
|
}
|
|
|
|
QHash<QString, QVariant> QueryBuilder::getEntityAttributes(const QSharedPointer<Entity> &entity) {
|
|
Entity *e = entity.data();
|
|
auto map = QHash<QString, QVariant>();
|
|
auto metaObject = e->metaObject();
|
|
auto transientAttrs = e->getTransientAttributes();
|
|
for (int var = 0; var < metaObject->propertyCount(); ++var) {
|
|
auto p = metaObject->property(var);
|
|
QString name = QString(p.name());
|
|
if (p.isValid() && !transientAttrs.contains(name)) {
|
|
QVariant v = p.read(e);
|
|
//Relation
|
|
if (v.canConvert<Entity *>()) {
|
|
this->insertRelationId(qvariant_cast<Entity *>(v), map, name);
|
|
} else if (v.canConvert<QSharedPointer<Entity>>()) {
|
|
this->insertRelationId(qvariant_cast<QSharedPointer<Entity>>(v).data(), map, name);
|
|
} else if (QString(p.typeName()).contains("QList")) {
|
|
/**
|
|
@TODO
|
|
//List and/or ManyToManyRelation
|
|
*/
|
|
auto n = static_cast<QList<CuteEntityManager::Entity *>*>(v.data());
|
|
for (int var = 0; var < n->size(); ++var) {
|
|
CuteEntityManager::Entity *entity = n->at(var);
|
|
qDebug() << entity->toString();
|
|
}
|
|
} else {
|
|
map.insert(name, v);
|
|
}
|
|
}
|
|
}
|
|
return map;
|
|
}
|
|
|
|
void QueryBuilder::insertRelationId(const Entity *e, QHash<QString, QVariant> &map, QString relName) {
|
|
if (e && e->getId() > -1) {
|
|
map.insert(relName + "_id", e->getId());
|
|
}
|
|
}
|
|
|
|
QString QueryBuilder::buildColumns(const QStringList &columns) const {
|
|
QString r = "";
|
|
bool first = true;
|
|
for (int var = 0; var < columns.size(); ++var) {
|
|
if (!first) {
|
|
r.append(",");
|
|
} else {
|
|
first = false;
|
|
}
|
|
r.append(columns.at(var));
|
|
}
|
|
return r;
|
|
}
|
|
|
|
QSharedPointer<Schema> QueryBuilder::getSchema() const {
|
|
return schema;
|
|
}
|
|
|
|
void QueryBuilder::setSchema(const QSharedPointer<Schema> &value) {
|
|
schema = value;
|
|
}
|
|
|