|
#include "querybuilder.h"
|
|
#include "database.h"
|
|
#include <QMetaObject>
|
|
#include <QMetaProperty>
|
|
#include "entity.h"
|
|
#include <QRegularExpression>
|
|
#include "entityinstancefactory.h"
|
|
|
|
using namespace CuteEntityManager;
|
|
|
|
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"));
|
|
auto i = tableDefinition.constBegin();
|
|
while (i != tableDefinition.constEnd()) {
|
|
if (first) {
|
|
first = false;
|
|
} else {
|
|
s.append(", ");
|
|
}
|
|
s.append(this->schema.data()->quoteColumnName(i.key())).append(" " + this->getColumnType(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;
|
|
}
|
|
|
|
QString QueryBuilder::buildCreateQuery(QHash<QString, QVariant>::const_iterator i,
|
|
QHash<QString, QVariant>::const_iterator end,
|
|
QString &p1, QString &p2) const {
|
|
bool first = true;
|
|
while (i != end) {
|
|
if (!first) {
|
|
p1 += ",";
|
|
p2 += ",";
|
|
} else {
|
|
first = false;
|
|
}
|
|
p1 += this->schema.data()->quoteColumnName(i.key());
|
|
p2 += ":" + i.key();
|
|
++i;
|
|
}
|
|
p1 += ")";
|
|
p2 += ");";
|
|
return p1 + p2;
|
|
}
|
|
|
|
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) {
|
|
auto m = o->property(var);
|
|
if (m.name() != QString("objectName") && 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::MANY_TO_ONE) {
|
|
map.insert(QString(m.name()) + "_id", this->schema.data()->TYPE_BIGINT);
|
|
}
|
|
} 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::generateManyToManyTableName(const QSharedPointer<Entity> &firstEntity,
|
|
const QSharedPointer<Entity> &secondEntity) const {
|
|
return QString(firstEntity.data()->metaObject()->className()).toLower() + "_" + QString(
|
|
secondEntity.data()->metaObject()->className()).toLower();
|
|
}
|
|
|
|
QHash<QString, QHash<QString, QString>> QueryBuilder::generateRelationTables(const QSharedPointer<Entity> &entity)
|
|
const {
|
|
auto relations = QHash<QString, QHash<QString, QString>>();
|
|
QHash<QString, Relation> m = entity.data()->getRelations();
|
|
auto props = entity.data()->getMetaProperties();
|
|
for (auto i = m.begin(); i != m.end(); ++i) {
|
|
Relation r = i.value();
|
|
if (r.getType() == MANY_TO_MANY && r.getMappedBy().isEmpty()) {
|
|
QHash<QString, QString> h = QHash<QString, QString>();
|
|
h.insert("id", this->schema.data()->TYPE_BIGPK);
|
|
h.insert(QString(entity.data()->metaObject()->className()) + QString("_id"), this->schema.data()->TYPE_BIGINT);
|
|
auto m = props.value(r.getPropertyName());
|
|
Entity *e = EntityInstanceFactory::createInstance(m.type());
|
|
QSharedPointer<Entity> ptr = QSharedPointer<Entity>(e);
|
|
h.insert(QString(ptr.data()->metaObject()->className()) + QString("_id"),
|
|
this->schema.data()->TYPE_BIGINT);
|
|
if (r.getTableName().isEmpty()) {
|
|
relations.insert(this->generateManyToManyTableName(entity, ptr), h);
|
|
} else {
|
|
relations.insert(r.getTableName(), h);
|
|
}
|
|
}
|
|
}
|
|
return relations;
|
|
}
|
|
|
|
|
|
QString QueryBuilder::transformTypeToAbstractDbType(QString typeName) const {
|
|
QHash<QString, QString> *m = this->schema.data()->getAbstractTypeMap().data();
|
|
if (m->contains(typeName)) {
|
|
return m->value(typeName);
|
|
}
|
|
for (auto 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(const QString &type) const {
|
|
/**
|
|
* @WARNING
|
|
*/
|
|
QHash<QString, QString> *tMap = this->schema.data()->getTypeMap().data();
|
|
if (tMap->contains(type)) {
|
|
return this->transformAbstractTypeToRealDbType(type);
|
|
}
|
|
//cant believe that this could work in Qt
|
|
//https://github.com/yiisoft/yii2/blob/master/framework/db/QueryBuilder.php
|
|
QRegularExpression reg = QRegularExpression(QRegularExpression::escape("/^(\\w+)\((.+?)\\)(.*)$/"));
|
|
reg.optimize();
|
|
QRegularExpressionMatchIterator i = reg.globalMatch(type, 0, QRegularExpression::PartialPreferFirstMatch);
|
|
short s = 0;
|
|
bool ok = false;
|
|
QString before = "";
|
|
while (i.hasNext() && s < 2) {
|
|
before = i.next().captured();
|
|
if (tMap->contains(before)) {
|
|
ok = true;
|
|
}
|
|
if (ok) {
|
|
return before.replace(QRegularExpression::escape("/\\(.+\\)/"), "(" + i.next().captured() + ")");
|
|
}
|
|
s++;
|
|
}
|
|
reg = QRegularExpression(QRegularExpression::escape("/^(\\w+)\\s+/"));
|
|
i = reg.globalMatch(type, 0, QRegularExpression::PartialPreferFirstMatch);
|
|
if (i.hasNext()) {
|
|
return before.replace(QRegularExpression::escape("/^w+/"), i.next().captured());
|
|
}
|
|
return type;
|
|
}
|
|
|
|
/**
|
|
* @brief QueryBuilder::find
|
|
* @param id
|
|
* @param tableName
|
|
* @return
|
|
*/
|
|
QSqlQuery QueryBuilder::find(const qint64 &id, const QString &tableName) const {
|
|
QSqlQuery q = this->database.data()->getQuery("SELECT * FROM " + this->schema.data()->quoteTableName(
|
|
tableName) + " WHERE id= :id LIMIT 1;");
|
|
q.bindValue(":id", id);
|
|
return q;
|
|
}
|
|
|
|
/**
|
|
* @brief QueryBuilder::findByAttributes
|
|
* @param m
|
|
* @param tableName
|
|
* @param ignoreID
|
|
* @return
|
|
*/
|
|
QSqlQuery QueryBuilder::findByAttributes(const QHash<QString, QVariant> &m, const QString &tableName,
|
|
const bool &ignoreID) const {
|
|
QSqlQuery q = this->database.data()->getQuery("SELECT * FROM " + this->schema.data()->quoteTableName(
|
|
tableName) + this->where(m, "AND", ignoreID));
|
|
this->bindValues(m, q, ignoreID);
|
|
return q;
|
|
}
|
|
|
|
QSqlQuery QueryBuilder::findByAttributes(const QSharedPointer<Entity> &e, bool ignoreID) {
|
|
QHash<QString, QVariant> values = this->getEntityAttributes(e.data()->getMetaProperties(), e);
|
|
return this->findByAttributes(values, e.data()->getTablename(), ignoreID);
|
|
}
|
|
|
|
QSqlQuery QueryBuilder::findAll(const QString &tableName) const {
|
|
return this->database->getQuery("SELECT * FROM " + this->schema.data()->quoteTableName(
|
|
tableName) + ";");
|
|
}
|
|
|
|
QSqlQuery QueryBuilder::remove(const QSharedPointer<Entity> &entity) const {
|
|
QSqlQuery q = this->database->getQuery("DELETE FROM " + this->schema.data()->quoteTableName(
|
|
entity.data()->getTablename()) + " WHERE " + this->schema.data()->quoteColumnName("id") + "=:id;");
|
|
q.bindValue(":id", entity.data()->getId());
|
|
return q;
|
|
}
|
|
|
|
QSqlQuery QueryBuilder::findId(const QSharedPointer<Entity> &entity) const {
|
|
QHash<QString, QVariant> values = this->getEntityAttributes(entity.data()->getMetaProperties(), entity);
|
|
QSqlQuery q = this->database.data()->getQuery("SELECT " + this->schema.data()->quoteColumnName("id") + " FROM " +
|
|
this->schema.data()->quoteTableName(
|
|
entity.data()->getTablename()) + this->where(values,
|
|
"AND", true) + " LIMIT 1");
|
|
this->bindValues(values, q);
|
|
return q;
|
|
}
|
|
|
|
QSqlQuery QueryBuilder::count(const QSharedPointer<Entity> &entity, bool ignoreID) const {
|
|
QHash<QString, QVariant> values = this->getEntityAttributes(entity.data()->getMetaProperties(), entity);
|
|
QSqlQuery q = this->database.data()->getQuery("SELECT COUNT(*) FROM " + this->schema.data()->quoteTableName(
|
|
entity.data()->getTablename()) + this->where(
|
|
values, "AND", ignoreID));
|
|
this->bindValues(values, q, ignoreID);
|
|
return q;
|
|
}
|
|
|
|
QSqlQuery QueryBuilder::count(const QString &tableName) const {
|
|
QSqlQuery q = this->database.data()->getQuery("SELECT COUNT(*) FROM " + this->schema.data()->quoteTableName(
|
|
tableName) + ";");
|
|
return q;
|
|
}
|
|
|
|
QSqlQuery QueryBuilder::merge(const QSharedPointer<Entity> &entity) const {
|
|
QHash<QString, QVariant> values = this->saveAttributes(entity);
|
|
QSqlQuery q = this->database.data()->getQuery("UPDATE " + this->schema.data()->quoteTableName(
|
|
entity->getTablename()) + " SET " + this->attributes(
|
|
values) + " WHERE " + this->schema.data()->quoteColumnName("id") + "=:id;");
|
|
this->bindValues(values, q);
|
|
//q.bindValue(":idM", entity->getId());
|
|
return q;
|
|
}
|
|
|
|
QSqlQuery QueryBuilder::create(const QSharedPointer<Entity> &entity) const {
|
|
QHash<QString, QVariant> values = this->saveAttributes(entity);
|
|
values.remove("id");
|
|
QSqlQuery q = this->database.data()->getQuery();
|
|
QString p1 = "INSERT INTO " + this->schema.data()->quoteTableName(entity->getTablename()) + "(";
|
|
QString p2 = "VALUES(";
|
|
if (!values.isEmpty()) {
|
|
q.prepare(this->buildCreateQuery(values.constBegin(), values.constEnd(),
|
|
p1, p2));
|
|
}
|
|
this->bindValues(values, q);
|
|
return q;
|
|
}
|
|
|
|
QString QueryBuilder::limit(const qint8 limit, const qint64 offset) const {
|
|
return " LIMIT " + QString(limit) + (offset > 0 ? QString("," + offset) : "");
|
|
}
|
|
|
|
QSqlQuery QueryBuilder::getQuery() const {
|
|
return this->database.data()->getQuery();
|
|
}
|
|
|
|
QHash<QString, QVariant> QueryBuilder::saveAttributes(const QSharedPointer<Entity> &entity) const {
|
|
auto props = entity.data()->getMetaProperties();
|
|
auto values = this->getEntityAttributes(props, entity);
|
|
auto relValues = this->getManyToOneAttributes(props, entity);
|
|
auto iterator = relValues.constBegin();
|
|
while (iterator != relValues.constEnd()) {
|
|
values.insert(iterator.key(), iterator.value());
|
|
++iterator;
|
|
}
|
|
return values;
|
|
}
|
|
|
|
QHash<QString, QVariant> QueryBuilder::getEntityAttributes(const QHash<QString, QMetaProperty> &props,
|
|
const QSharedPointer<Entity> &entity) const {
|
|
Entity *e = entity.data();
|
|
auto map = QHash<QString, QVariant>();
|
|
auto transientAttrs = e->getTransientAttributes();
|
|
auto relations = e->getRelations();
|
|
auto i = props.constBegin();
|
|
while (i != props.constEnd()) {
|
|
if (!transientAttrs.contains(i.key()) && !relations.contains(i.key())) {
|
|
map.insert(i.key(), i.value().read(e));
|
|
}
|
|
++i;
|
|
}
|
|
return map;
|
|
}
|
|
|
|
QHash<QString, QVariant> QueryBuilder::getManyToOneAttributes(const QHash<QString, QMetaProperty> &props,
|
|
const QSharedPointer<Entity> &entity) const {
|
|
Entity *e = entity.data();
|
|
auto map = QHash<QString, QVariant>();
|
|
auto relations = e->getRelations();
|
|
auto i = relations.constBegin();
|
|
while (i != relations.constEnd()) {
|
|
Relation r = i.value();
|
|
if (r.getType() == MANY_TO_ONE && props.contains(i.key())) {
|
|
auto v = props.value(i.key()).read(e);
|
|
if (v.canConvert<Entity *>()) {
|
|
this->insertRelationId(qvariant_cast<Entity *>(v), map, i.key());
|
|
} else if (v.canConvert<QSharedPointer<Entity>>()) {
|
|
this->insertRelationId(qvariant_cast<QSharedPointer<Entity>>(v).data(), map, i.key());
|
|
} else if (v.canConvert<QPointer<Entity>>()) {
|
|
this->insertRelationId(qvariant_cast<QPointer<Entity>>(v).data(), map, i.key());
|
|
}
|
|
++i;
|
|
}
|
|
}
|
|
return map;
|
|
}
|
|
|
|
void QueryBuilder::insertRelationId(const Entity *e, QHash<QString, QVariant> &map, QString relName) const {
|
|
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;
|
|
}
|
|
|
|
void QueryBuilder::bindValues(const QHash<QString, QVariant> &h, QSqlQuery &q, bool ignoreID) const {
|
|
QHash<QString, QVariant>::const_iterator i = h.constBegin();
|
|
while (i != h.constEnd()) {
|
|
if (!ignoreID || (ignoreID && !(i.key() == "id"))) {
|
|
q.bindValue(":" + i.key(), i.value());
|
|
}
|
|
++i;
|
|
}
|
|
}
|
|
|
|
QString QueryBuilder::where(const QSharedPointer<Entity> &entity, QString conjunction, bool ignoreID) const {
|
|
return this->where(this->getEntityAttributes(entity.data()->getMetaProperties(), entity), conjunction, ignoreID);
|
|
}
|
|
|
|
QString QueryBuilder::where(const QHash<QString, QVariant> &m, const QString &conjunction, bool ignoreID) const {
|
|
if (m.size() == 0 || (ignoreID && m.contains("id") && m.size() == 1)) {
|
|
return "";
|
|
}
|
|
return " WHERE " + this->attributes(m, conjunction, ignoreID);
|
|
}
|
|
|
|
QString QueryBuilder::attributes(const QHash<QString, QVariant> &m, const QString &conjunction, bool ignoreID) const {
|
|
QString rc = "";
|
|
if (!m.isEmpty()) {
|
|
QHash<QString, QVariant>::const_iterator i = m.constBegin();
|
|
while (i != m.constEnd()) {
|
|
if (!ignoreID || (ignoreID && !(i.key() == "id"))) {
|
|
if (!(rc == "")) {
|
|
rc += " " + conjunction + " ";
|
|
}
|
|
rc += this->schema.data()->quoteColumnName(i.key()) + "=:" + i.key();
|
|
}
|
|
++i;
|
|
}
|
|
}
|
|
return rc;
|
|
}
|
|
|
|
QSharedPointer<Schema> QueryBuilder::getSchema() const {
|
|
return schema;
|
|
}
|
|
|
|
void QueryBuilder::setSchema(const QSharedPointer<Schema> &value) {
|
|
schema = value;
|
|
}
|