Projekt

Allgemein

Profil

Herunterladen als
Herunterladen (30,9 KB) Statistiken
| Zweig: | Revision:
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 "entitymanager.h"
7e233492 Christian Ehringfeld
#include "enums/databasetype.h"
1e213c09 Christian Ehringfeld
#include "databasemigration.h"
4d58ef6a Christian Ehringfeld
using namespace CuteEntityManager;
81c23b56 Christian Ehringfeld
QStringList EntityManager::connectionNames = QStringList();

afde9013 Christian Ehringfeld
QStringList EntityManager::getConnectionNames() {
return EntityManager::connectionNames;
}

EntityManager::EntityManager(QSqlDatabase database) : QObject() {
auto db = new Database(database);
this->db = QSharedPointer<Database>(db);
this->init();
}

EntityManager::EntityManager(const QString &databaseType, QString databasename ,
QString hostname,
QString username,
QString password, QString port) : QObject() {
auto db = new Database(databaseType, this->createConnection(), hostname,
databasename, username,
password,
port.toInt());
this->db = QSharedPointer<Database>(db);
this->init();
}

7e233492 Christian Ehringfeld
void EntityManager::init() {
3820ae33 Christian Ehringfeld
auto schema = Database::getSchema(Database::getDatabaseType(
abb9e8c5 Christian Ehringfeld
this->db->getDatabase().driverName()), this->db);
7e233492 Christian Ehringfeld
this->schema = QSharedPointer<Schema>(schema);
abb9e8c5 Christian Ehringfeld
this->schema->setTables(this->schema->getTableSchemas());
7e233492 Christian Ehringfeld
}

afde9013 Christian Ehringfeld
EntityManager::~EntityManager() {
EntityManager::removeConnectionName(this->db->getConnectionName());
7e233492 Christian Ehringfeld
}

1e213c09 Christian Ehringfeld
bool EntityManager::startup(QString version, QStringList toInitialize) {
d568923d Christian Ehringfeld
DatabaseMigration *dbm = new DatabaseMigration();
b7446f4c Christian Ehringfeld
QSharedPointer<Entity> ptrDbm = QSharedPointer<DatabaseMigration>(dbm);
1e213c09 Christian Ehringfeld
QHash<QString, QVariant> map = QHash<QString, QVariant>();
d84c91e7 Christian Ehringfeld
bool ok = true;
1e213c09 Christian Ehringfeld
map.insert("version", version);
abb9e8c5 Christian Ehringfeld
if (!this->schema->getTableNames().contains(dbm->getTablename())) {
b7446f4c Christian Ehringfeld
this->createTable(ptrDbm, true);
e86c23a2 Christian Ehringfeld
}
d568923d Christian Ehringfeld
if (this->findAllByAttributes(map, dbm->getTablename()).isEmpty()) {
for (int var = 0; var < toInitialize.size(); ++var) {
d84c91e7 Christian Ehringfeld
if (ok) {
QString c = toInitialize.at(var);
e0e1ead8 Christian Ehringfeld
ok = this->createTable(QSharedPointer<Entity>
(EntityInstanceFactory::createInstance(c)));
d84c91e7 Christian Ehringfeld
} else {
break;
}
}
if (ok) {
dbm->setVersion(version);
dbm->setApplyTime(QDateTime::currentDateTime());
this->create(ptrDbm);
d568923d Christian Ehringfeld
}
abb9e8c5 Christian Ehringfeld
this->schema->setTables(this->schema->getTableSchemas());
d568923d Christian Ehringfeld
}
d84c91e7 Christian Ehringfeld
return ok;
1e213c09 Christian Ehringfeld
}

e0e1ead8 Christian Ehringfeld
bool EntityManager::executeQuery(const QString &query) {
abb9e8c5 Christian Ehringfeld
return this->db->transaction(query);
e0e1ead8 Christian Ehringfeld
}

b0bf458e Christian Ehringfeld
bool EntityManager::checkTable(const QSharedPointer<Entity> &entity) {
bool rc = true;
df1e56bd Christian Ehringfeld
if (!this->schema->containsTable(entity->getTablename())
&& this->schema->getQueryBuilder()->createTable(entity)) {
this->schema->getTableSchema(entity->getTablename(), true);
rc = this->schema->getTables().contains(entity->getTablename());
b0bf458e Christian Ehringfeld
}
return rc;
}
9d05e414 Christian Ehringfeld
7e233492 Christian Ehringfeld
QSharedPointer<Database> EntityManager::getDb() const {
return db;
}

void EntityManager::setDb(const QSharedPointer<Database> &value) {
db = value;
}

QSharedPointer<Schema> EntityManager::getSchema() const {
return schema;
}

4f3b13f3 Christian Ehringfeld
void EntityManager::refresh(QSharedPointer<Entity> &entity) {
b5d490c7 Christian Ehringfeld
entity = this->findById(entity->getProperty(
abb9e8c5 Christian Ehringfeld
entity->getPrimaryKey()).toLongLong(),
QString(entity->getClassname()));
4f3b13f3 Christian Ehringfeld
}

7e233492 Christian Ehringfeld
void EntityManager::setSchema(const QSharedPointer<Schema> &value) {
schema = value;
}

QString EntityManager::createConnection() {
QStringList l = EntityManager::getConnectionNames();
QString conName = "";
bool ok = false;
qint16 i = 0;
while (!ok) {
if (l.contains("con" + QString::number(i))) {
++i;
} else {
l.append("con" + QString::number(i));
ok = true;
conName = "con" + QString::number(i);
EntityManager::setConnectionNames(l);
}
}
return conName;
}

void EntityManager::removeConnectionName(const QString &name) {
EntityManager::connectionNames.removeOne(name);
}

f5087482 Christian Ehringfeld
QSharedPointer<Entity> EntityManager::findById(const qint64 &id,
QSharedPointer<Entity> &e,
4f3b13f3 Christian Ehringfeld
const bool refresh) {
2d9fab10 Christian Ehringfeld
QSharedPointer<Entity> r;
df1e56bd Christian Ehringfeld
if (!e.isNull() && (refresh
|| !(r = this->cache.get(id, QString(e->getClassname()))))) {
e->setId(id);
auto map = this->findByPk(id, e);
r = this->convert(map, e->getClassname(), refresh);
2d9fab10 Christian Ehringfeld
}
return r;
}

e0e1ead8 Christian Ehringfeld
QSharedPointer<Entity> EntityManager::findById(const qint64 &id,
const QString &classname) {
f5087482 Christian Ehringfeld
QSharedPointer<Entity> e = QSharedPointer<Entity>
(EntityInstanceFactory::createInstance(classname));
2d9fab10 Christian Ehringfeld
return this->findById(id, e);
}

e0e1ead8 Christian Ehringfeld
QSharedPointer<Entity> EntityManager::convert(const QHash<QString, QVariant>
&map,
4f3b13f3 Christian Ehringfeld
const char *classname, const bool refresh) {
e0e1ead8 Christian Ehringfeld
auto ptr = QSharedPointer<Entity>(EntityInstanceFactory::createInstance(
classname, map));
9070a496 Christian Ehringfeld
this->cache.insert(ptr);
2ce163c3 Christian Ehringfeld
this->resolveRelations(ptr, map, refresh);
9070a496 Christian Ehringfeld
return ptr;
f4e3904b Christian Ehringfeld
}

e0e1ead8 Christian Ehringfeld
QList<QSharedPointer<Entity> > EntityManager::convert(
QList<QHash<QString, QVariant> > maps,
4f3b13f3 Christian Ehringfeld
const char *classname, const bool refresh) {
f4e3904b Christian Ehringfeld
auto list = QList<QSharedPointer<Entity> >();
for (int var = 0; var < maps.size(); ++var) {
4f3b13f3 Christian Ehringfeld
auto ptr = this->convert(maps.at(var), classname, refresh);
9070a496 Christian Ehringfeld
list.append(ptr);
this->cache.insert(ptr);
f4e3904b Christian Ehringfeld
}
return list;
586bb527 Christian Ehringfeld
}

e0e1ead8 Christian Ehringfeld
void EntityManager::manyToOne(const QSharedPointer<Entity> &entity,
const QVariant &id,
4f3b13f3 Christian Ehringfeld
const QMetaProperty &property, const bool refresh) {
a06633f7 Christian Ehringfeld
qint64 convertedId = -1;
bool ok = false;
df1e56bd Christian Ehringfeld
if ((convertedId = id.toLongLong(&ok)) && ok && convertedId > -1) {
e0e1ead8 Christian Ehringfeld
QString className = EntityInstanceFactory::extractEntityType(
property.typeName());
706de2e8 Christian Ehringfeld
QSharedPointer<Entity> ptr = QSharedPointer<Entity>();
4f3b13f3 Christian Ehringfeld
if (refresh || !(this->cache.contains(convertedId, className)
afde9013 Christian Ehringfeld
&& (ptr = this->cache.get(convertedId, className)) && ptr)) {
706de2e8 Christian Ehringfeld
ptr = this->findById(convertedId, className);
}
c599658a Christian Ehringfeld
this->setProperty(entity, ptr, property);
a06633f7 Christian Ehringfeld
}
}

e0e1ead8 Christian Ehringfeld
void EntityManager::oneToMany(const QSharedPointer<Entity> &entity,
const Relation &r,
4f3b13f3 Christian Ehringfeld
const QMetaProperty &property, const bool refresh) {
abb9e8c5 Christian Ehringfeld
if (entity.data() && entity->getId() > -1) {
df1e56bd Christian Ehringfeld
auto e = QSharedPointer<Entity>(EntityInstanceFactory::createInstance(
EntityInstanceFactory::extractEntityType(
property.typeName())));
if (e) {
QSqlQuery q = this->schema->getQueryBuilder()->oneToMany(e->getTablename(),
this->schema->getQueryBuilder()->generateColumnNameID(r.getMappedBy()),
entity->getId());
auto listMap = this->convertQueryResult(q);
auto entities = this->convert(listMap, e->getClassname(), refresh);
this->setListProperty(entity, entities, property);
}
706de2e8 Christian Ehringfeld
}
}
a06633f7 Christian Ehringfeld
c599658a Christian Ehringfeld
e0e1ead8 Christian Ehringfeld
void EntityManager::oneToOne(const QSharedPointer<Entity> &entity,
const Relation &r,
4f3b13f3 Christian Ehringfeld
const QMetaProperty &property, const bool refresh,
c599658a Christian Ehringfeld
const QVariant &id) {
if (r.getMappedBy().isEmpty()) {
this->manyToOne(entity, id, property);
} else {
df1e56bd Christian Ehringfeld
auto e = QSharedPointer<Entity>(EntityInstanceFactory::createInstance(
EntityInstanceFactory::extractEntityType(property.typeName())));
if (e) {
QSqlQuery q = this->schema->getQueryBuilder()->oneToMany(
e->getTablename(),
this->schema->getQueryBuilder()->generateColumnNameID(
r.getMappedBy()),
entity->getProperty(entity->getPrimaryKey()).toLongLong(), 1);
auto listMap = this->convertQueryResult(q);
auto entities = this->convert(listMap, e->getClassname(), refresh);
if (!entities.isEmpty()) {
QSharedPointer<Entity> ptr = entities.at(0);
this->setProperty(entity, ptr, property);
}
c599658a Christian Ehringfeld
}
}
}

98b5b08d Christian Ehringfeld
bool EntityManager::canPersistRelation(const Relation &relation,
const RelationType &r, const QVariant &var) const {
f6a3fe0a Christian Ehringfeld
return relation.getType() == r && var.canConvert<QVariantList>();
4f3b13f3 Christian Ehringfeld
}

e0e1ead8 Christian Ehringfeld
void EntityManager::setListProperty(const QSharedPointer<Entity> &entity,
QList<QSharedPointer<Entity> > &list,
706de2e8 Christian Ehringfeld
const QMetaProperty &property) const {
QVariant var;
var.setValue<QList<QSharedPointer<Entity>>>(list);
property.write(entity.data(), var);
a06633f7 Christian Ehringfeld
}

91ed1164 Christian Ehringfeld
void EntityManager::addEntityToListProperty(const QSharedPointer<Entity>
&entity, QSharedPointer<Entity> add, const QMetaProperty &property) {
QVariant var = property.read(entity.data());
d7727319 Christian Ehringfeld
if (!var.isNull() && var.canConvert<QList<QVariant>>()) {
auto list = EntityInstanceFactory::castQVariantList(var);
91ed1164 Christian Ehringfeld
if (!list.contains(add)) {
list.append(add);
35cf13b7 Christian Ehringfeld
this->setListProperty(entity, list, property);
91ed1164 Christian Ehringfeld
}
}
}

df1e56bd Christian Ehringfeld
void EntityManager::setProperty(const QSharedPointer<Entity> &entity,
e0e1ead8 Christian Ehringfeld
QSharedPointer<Entity> value,
c599658a Christian Ehringfeld
const QMetaProperty &property) const {
b5d490c7 Christian Ehringfeld
if (value && value->getProperty(value->getPrimaryKey()).toLongLong()
99140fe9 Christian Ehringfeld
> -1) {
df1e56bd Christian Ehringfeld
QVariant var;
var.setValue<QSharedPointer<Entity>>(value);
property.write(entity.data(), var);
c599658a Christian Ehringfeld
}
}

df1e56bd Christian Ehringfeld
void EntityManager::savePrePersistedRelations(const QSharedPointer<Entity>
&entity) {
abb9e8c5 Christian Ehringfeld
auto relations = entity->getRelationProperties();
4f3b13f3 Christian Ehringfeld
auto iterator = relations.constBegin();
while (iterator != relations.constEnd()) {
const Relation r = iterator.key();
auto var = iterator.value().read(entity.data());
df1e56bd Christian Ehringfeld
if (!var.isNull()) {
if (r.getType() == MANY_TO_ONE) {
auto e = EntityInstanceFactory::castQVariant(var);
if (this->shouldBeSaved(e, r)) {
this->save(e);
}
} else if (r.getType() == ONE_TO_ONE && r.getMappedBy().isEmpty()) {
auto entity = EntityInstanceFactory::castQVariant(var);
this->save(entity);
e24791d7 Christian Ehringfeld
}
df1e56bd Christian Ehringfeld
}
++iterator;
}
}

void EntityManager::savePostPersistedRelations(const QSharedPointer<Entity>
&entity) {
auto relations = entity->getRelationProperties();
auto iterator = relations.constBegin();
while (iterator != relations.constEnd()) {
const Relation r = iterator.key();
auto var = iterator.value().read(entity.data());
if (!var.isNull()) {
if (this->canPersistRelation(r, MANY_TO_MANY, var)) {
this->persistManyToMany(entity, r, var);
} else if (this->canPersistRelation(r, ONE_TO_MANY, var)) {
QList<QSharedPointer<Entity>> list = EntityInstanceFactory::castQVariantList(
var);
for (int var = 0; var < list.size(); ++var) {
auto entity = list.at(var);
if (this->shouldBeSaved(entity, r)) {
this->save(entity);
}
e24791d7 Christian Ehringfeld
}
df1e56bd Christian Ehringfeld
} else if (r.getType() == ONE_TO_ONE && !r.getMappedBy().isEmpty()) {
auto entity = EntityInstanceFactory::castQVariant(var);
this->save(entity);
4f3b13f3 Christian Ehringfeld
}
}
396a60d7 Christian Ehringfeld
++iterator;
4f3b13f3 Christian Ehringfeld
}
91ed1164 Christian Ehringfeld
}

void EntityManager::persistMappedByRelation(const QList<QSharedPointer<Entity> >
3fd96253 Christian Ehringfeld
&list, QSqlQuery &q, const QSharedPointer<Entity> &entity,
const QSharedPointer<Entity> &ptr, const Relation &r,
91ed1164 Christian Ehringfeld
const QString &tblName) {
q.clear();
e24791d7 Christian Ehringfeld
QList<QSharedPointer<Entity>> saved = r.getCascadeType().contains(ALL)
|| r.getCascadeType().contains(MERGE)
|| r.getCascadeType().contains(PERSIST) ? this->saveRelationEntities(list,
r) : list;
719a0ba1 Christian Ehringfeld
this->db->startTransaction();
abb9e8c5 Christian Ehringfeld
auto builder = this->schema->getQueryBuilder();
q = builder->manyToManyInsert(tblName,
builder->generateManyToManyColumnName(entity),
builder->generateManyToManyColumnName(ptr));
b5d490c7 Christian Ehringfeld
q.bindValue(0, entity->getProperty(entity->getPrimaryKey()));
35cf13b7 Christian Ehringfeld
auto prop = this->mappedProperty(r, ptr);
3fd96253 Christian Ehringfeld
QSharedPointer<Entity> item;
e24791d7 Christian Ehringfeld
for (int var = 0; var < saved.size(); ++var) {
3fd96253 Christian Ehringfeld
item = list.at(var);
b5d490c7 Christian Ehringfeld
if (item->getProperty(item->getPrimaryKey()).toLongLong() > -1) {
q.bindValue(1, item->getProperty(ptr->getPrimaryKey()));
e24791d7 Christian Ehringfeld
q.exec();
3fd96253 Christian Ehringfeld
if (prop.isReadable()) {
e24791d7 Christian Ehringfeld
this->addEntityToListProperty(entity, ptr, prop);
3fd96253 Christian Ehringfeld
} else {
qDebug() << "Query exec for many to many relation failed." <<
q.lastError().text();
qDebug() << "Involved entities: " << entity->getClassname() <<
"(MainEntitiy) and " << entity->getClassname();
qDebug() << "Relation:" << r.getType() << r.getPropertyName();
e24791d7 Christian Ehringfeld
}
91ed1164 Christian Ehringfeld
}
}
719a0ba1 Christian Ehringfeld
if (!this->db->commitTransaction()) {
this->db->rollbackTransaction();
}
4f3b13f3 Christian Ehringfeld
}
a06633f7 Christian Ehringfeld
d7727319 Christian Ehringfeld
35cf13b7 Christian Ehringfeld
QMetaProperty EntityManager::mappedProperty(const Relation &r,
const QSharedPointer<Entity> &foreignEntity) const {
d7727319 Christian Ehringfeld
QMetaProperty prop;
auto props = foreignEntity->getMetaProperties();
df1e56bd Christian Ehringfeld
if (!r.getMappedBy().isEmpty() && props.contains(r.getMappedBy())) {
prop = props.value(r.getMappedBy());
d7727319 Christian Ehringfeld
} else {
auto relations = foreignEntity->getRelations();
35cf13b7 Christian Ehringfeld
for (auto iterator = relations.constBegin(); iterator != relations.constEnd();
++iterator) {
d7727319 Christian Ehringfeld
auto rel = iterator.value();
35cf13b7 Christian Ehringfeld
if (rel.getMappedBy() == r.getPropertyName()) {
d7727319 Christian Ehringfeld
prop = props.value(rel.getPropertyName());
break;
}
}
}
return prop;
}

e24791d7 Christian Ehringfeld
bool EntityManager::shouldBeSaved(QSharedPointer<Entity> &entity,
const Relation &r) {
afde9013 Christian Ehringfeld
return entity && (r.getCascadeType().contains(ALL)
b5d490c7 Christian Ehringfeld
|| (entity->getProperty(entity->getPrimaryKey()) > -1
afde9013 Christian Ehringfeld
&& r.getCascadeType().contains(MERGE))
b5d490c7 Christian Ehringfeld
|| (entity->getProperty(entity->getPrimaryKey()) <= -1
afde9013 Christian Ehringfeld
&& r.getCascadeType().contains(PERSIST)));
e24791d7 Christian Ehringfeld
}

void EntityManager::removeRelations(const QSharedPointer<Entity> &entity) {
abb9e8c5 Christian Ehringfeld
auto props = entity->getRelationProperties();
e24791d7 Christian Ehringfeld
auto iterator = props.constBegin();
while (iterator != props.constEnd()) {
const Relation r = iterator.key();
2075db87 Christian Ehringfeld
auto property = iterator.value();
auto var = property.read(entity.data());
if (r.getType() == MANY_TO_MANY) {
this->removeManyToManyEntityList(entity, r, var);
} else if (r.getType() == ONE_TO_MANY) {
if (r.getCascadeType().contains(REMOVE) || r.getCascadeType().contains(ALL)) {
this->removeEntityList(var);
} else {
this->setNullOneToManyRelation(var, r);
}
} else if (r.getType() == MANY_TO_ONE || r.getType() == MANY_TO_ONE) {
this->setNullEntityPropertyRelation(var, r);
e24791d7 Christian Ehringfeld
}
++iterator;
}
}

2075db87 Christian Ehringfeld
void EntityManager::setNullOneToManyRelation(QVariant &var, const Relation &r) {
df1e56bd Christian Ehringfeld
if (!r.getMappedBy().isEmpty() && !var.isNull()
&& var.canConvert<QVariantList>()) {
auto list = EntityInstanceFactory::castQVariantList(var);
if (!list.isEmpty()) {
auto metas = list.at(0)->getMetaProperties();
if (metas.contains(r.getMappedBy())) {
for (int var = 0; var < list.size(); ++var) {
auto entity = list.at(var);
this->setProperty(entity, QSharedPointer<Entity>(),
metas.value(r.getMappedBy()));
this->save(entity);
2075db87 Christian Ehringfeld
}
}
}
}
}

void EntityManager::setNullEntityPropertyRelation(QVariant &var,
const Relation &r) {
if (r.getCascadeType().contains(REMOVE) || r.getCascadeType().contains(ALL)) {
this->removeEntity(var);
df1e56bd Christian Ehringfeld
} else if (!r.getMappedBy().isEmpty() && !var.isNull()) {
auto e = EntityInstanceFactory::castQVariant(var);
auto metas = e->getMetaProperties();
if (metas.contains(r.getMappedBy())) {
this->setProperty(e, QSharedPointer<Entity>(), metas.value(r.getMappedBy()));
this->save(e);
2075db87 Christian Ehringfeld
}
}
}

d7727319 Christian Ehringfeld
2075db87 Christian Ehringfeld
void EntityManager::removeEntity(QVariant &var) {
df1e56bd Christian Ehringfeld
if (!var.isNull()) {
auto e = EntityInstanceFactory::castQVariant(var);
2075db87 Christian Ehringfeld
this->remove(e);
}
}

void EntityManager::removeEntityList(QVariant &var) {
df1e56bd Christian Ehringfeld
if (var.canConvert<QVariantList>()) {
auto list = EntityInstanceFactory::castQVariantList(var);
2075db87 Christian Ehringfeld
for (int var = 0; var < list.size(); ++var) {
auto entity = list.at(var);
this->remove(entity);
}
}
}

void EntityManager::removeManyToManyEntityList(const QSharedPointer<Entity> &e,
const Relation &r,
QVariant &var) {
df1e56bd Christian Ehringfeld
if (!var.isNull() && var.canConvert<QVariantList>()) {
auto list = EntityInstanceFactory::castQVariantList(var);
2075db87 Christian Ehringfeld
if (!list.isEmpty()) {
abb9e8c5 Christian Ehringfeld
auto builder = this->schema->getQueryBuilder();
2075db87 Christian Ehringfeld
auto ptr = list.at(0);
abb9e8c5 Christian Ehringfeld
QString tblName = builder->generateManyToManyTableName(e, ptr);
if (this->schema->getTables().contains(tblName)) {
QSqlQuery q = builder->manyToManyDelete(
tblName, builder->generateManyToManyColumnName(e),
b5d490c7 Christian Ehringfeld
e->getProperty(e->getPrimaryKey()).toLongLong());
2075db87 Christian Ehringfeld
bool refresh = r.getCascadeType().contains(REFRESH)
|| r.getCascadeType().contains(ALL);
bool remove = r.getCascadeType().contains(REMOVE)
|| r.getCascadeType().contains(ALL);
if (q.exec()) {
for (int var = 0; var < list.size(); ++var) {
auto entity = list.at(var);
if (remove) {
this->remove(entity);
} else if (refresh) {
/**
not really with good performance, alternatively iterate over relation attribute and delete pointer from list
**/
this->refresh(entity);
}
}
}
}
}
}
}

e24791d7 Christian Ehringfeld
QList<QSharedPointer<Entity>> EntityManager::saveRelationEntities(
91ed1164 Christian Ehringfeld
const QList<QSharedPointer<Entity> > &list, const Relation &r) {
e24791d7 Christian Ehringfeld
QList<QSharedPointer<Entity>> saved = QList<QSharedPointer<Entity>>();
91ed1164 Christian Ehringfeld
QSharedPointer<Entity> ptr;
for (int var = 0; var < list.size(); ++var) {
ptr = list.at(var);
e24791d7 Christian Ehringfeld
if (this->shouldBeSaved(ptr, r) && this->save(ptr)) {
saved.append(ptr);
91ed1164 Christian Ehringfeld
}
}
e24791d7 Christian Ehringfeld
return saved;
91ed1164 Christian Ehringfeld
}
98b5b08d Christian Ehringfeld
void EntityManager::persistManyToMany(const QSharedPointer<Entity> &entity,
abb9e8c5 Christian Ehringfeld
const Relation &r, QVariant &property) {
Q_UNUSED(r)
f6a3fe0a Christian Ehringfeld
auto list = property.value<QList<QVariant>>();
df1e56bd Christian Ehringfeld
if (!list.isEmpty() && !(list.at(0).isNull())) {
auto var = list.at(0);
auto ptr = EntityInstanceFactory::castQVariant(var);
auto builder = this->schema->getQueryBuilder();
QString tblName = builder->generateManyToManyTableName(entity, ptr);
if (this->schema->getTables().contains(tblName)) {
QSqlQuery q = builder->manyToManyDelete(
tblName, builder->generateManyToManyColumnName(entity),
entity->getProperty(entity->getPrimaryKey()).toLongLong());
if (this->db->transaction(q)) {
auto nList = EntityInstanceFactory::castQVariantList(property);
this->persistMappedByRelation(nList, q, entity, ptr, r, tblName);
98b5b08d Christian Ehringfeld
}
df1e56bd Christian Ehringfeld
} else {
qDebug() << "MANY_TO_MANY Table " << tblName << " not exists";
98b5b08d Christian Ehringfeld
}
}
}



e0e1ead8 Christian Ehringfeld
void EntityManager::manyToMany(const QSharedPointer<Entity> &entity,
4f3b13f3 Christian Ehringfeld
const QMetaProperty &property, const bool refresh) {
df1e56bd Christian Ehringfeld
QSharedPointer<Entity> secEntityPtr = QSharedPointer<Entity>
(EntityInstanceFactory::createInstance(EntityInstanceFactory::extractEntityType(
QString(property.typeName()))));
abb9e8c5 Christian Ehringfeld
auto builder = this->schema->getQueryBuilder();
df1e56bd Christian Ehringfeld
if (secEntityPtr) {
QString tblName = builder->generateManyToManyTableName(entity, secEntityPtr);
abb9e8c5 Christian Ehringfeld
if (this->schema->getTables().contains(tblName)) {
QSqlQuery q = builder->manyToMany(tblName,
builder->generateManyToManyColumnName(entity),
2ce163c3 Christian Ehringfeld
entity->getProperty(entity->getPrimaryKey()).toLongLong());
c599658a Christian Ehringfeld
auto listMap = this->convertQueryResult(q);
2ce163c3 Christian Ehringfeld
auto entities = QList<QSharedPointer<Entity> >();
for (int var = 0; var < listMap.size(); ++var) {
auto id = listMap.at(var).value(builder->generateManyToManyColumnName(
secEntityPtr));
if (!refresh
&& this->cache.contains(id.toLongLong(), secEntityPtr->getClassname())) {
entities.append(this->cache.get(id.toLongLong(), secEntityPtr->getClassname()));
} else {
entities.append(this->findById(id.toLongLong(), secEntityPtr->getClassname()));
}
}
c599658a Christian Ehringfeld
this->setListProperty(entity, entities, property);
} else {
qDebug() << "MANY_TO_MANY Table " << tblName << " not exists";
}
706de2e8 Christian Ehringfeld
}
a06633f7 Christian Ehringfeld
}

e0e1ead8 Christian Ehringfeld
QList<QSharedPointer<Entity> > EntityManager::findEntityByAttributes(
const QSharedPointer<Entity>
&entity,
bool ignoreID) {
1e213c09 Christian Ehringfeld
auto maps = this->findAllByAttributes(entity, ignoreID);
abb9e8c5 Christian Ehringfeld
return this->convert(maps, entity->getClassname());
7e233492 Christian Ehringfeld
}

3820ae33 Christian Ehringfeld
/**
244c6d53 Christian Ehringfeld
* @todo should be an insert statement with many values
3820ae33 Christian Ehringfeld
* @brief EntityManager::create
* @param entities
* @return
*/
244c6d53 Christian Ehringfeld
bool EntityManager::create(QList<QSharedPointer<Entity> > entities,
const bool persistRelations) {
3820ae33 Christian Ehringfeld
bool ok = true;
foreach (QSharedPointer<Entity> ent, entities) {
244c6d53 Christian Ehringfeld
ok = this->create(ent, persistRelations);
3820ae33 Christian Ehringfeld
if (!ok) {
break;
}
}
return ok;
d84c91e7 Christian Ehringfeld
}

244c6d53 Christian Ehringfeld
bool EntityManager::create(QSharedPointer<Entity> &entity,
696666eb Christian Ehringfeld
const bool persistRelations, const bool checkDuplicate) {
dc6b13b4 Christian Ehringfeld
bool rc = false;
696666eb Christian Ehringfeld
if (this->checkTable(entity) && !(checkDuplicate && this->count(entity) == 0)) {
df1e56bd Christian Ehringfeld
if (persistRelations) {
this->savePrePersistedRelations(entity);
}
this->db->startTransaction();
abb9e8c5 Christian Ehringfeld
QList<QSqlQuery> q = this->schema->getQueryBuilder()->create(
99140fe9 Christian Ehringfeld
entity);
bool first = true;
for (int var = 0; var < q.size(); ++var) {
auto query = q.at(var);
abb9e8c5 Christian Ehringfeld
rc = this->db->exec(query);
afde9013 Christian Ehringfeld
if (!rc) {
b7446f4c Christian Ehringfeld
qDebug() << "Query failed:" << query.lastError().text() << " of class " <<
abb9e8c5 Christian Ehringfeld
entity->getClassname();
afde9013 Christian Ehringfeld
break;
}
99140fe9 Christian Ehringfeld
if (first) {
b5d490c7 Christian Ehringfeld
entity->setProperty(entity->getPrimaryKey(), query.lastInsertId());
99140fe9 Christian Ehringfeld
first = false;
}
}
afde9013 Christian Ehringfeld
if (!rc || !this->db->commitTransaction()) {
99140fe9 Christian Ehringfeld
this->db->rollbackTransaction();
abb9e8c5 Christian Ehringfeld
entity->setId(-1);
99140fe9 Christian Ehringfeld
rc = false;
} else {
2ce163c3 Christian Ehringfeld
this->cache.insert(entity);
244c6d53 Christian Ehringfeld
if (persistRelations) {
df1e56bd Christian Ehringfeld
this->savePostPersistedRelations(entity);
244c6d53 Christian Ehringfeld
}
99140fe9 Christian Ehringfeld
rc = true;
7e233492 Christian Ehringfeld
}
}
dc6b13b4 Christian Ehringfeld
return rc;
7e233492 Christian Ehringfeld
}
9d05e414 Christian Ehringfeld
afde9013 Christian Ehringfeld
bool EntityManager::merge(QSharedPointer<Entity> &entity, bool withRelations) {
if (entity->getId() > -1 && this->count(entity) == 1) {
df1e56bd Christian Ehringfeld
if (withRelations) {
this->savePrePersistedRelations(entity);
}
afde9013 Christian Ehringfeld
this->db->startTransaction();
abb9e8c5 Christian Ehringfeld
QList<QSqlQuery> q = this->schema->getQueryBuilder()->merge(
afde9013 Christian Ehringfeld
entity);
bool ok = true;
for (int var = 0; var < q.size(); ++var) {
auto query = q.at(var);
abb9e8c5 Christian Ehringfeld
ok = this->db->exec(query);
afde9013 Christian Ehringfeld
if (!ok) {
qDebug() << query.lastError().text();
break;
}
}
if (!ok || !this->db->commitTransaction()) {
this->db->rollbackTransaction();
return false;
} else if (ok && withRelations) {
df1e56bd Christian Ehringfeld
this->savePostPersistedRelations(entity);
afde9013 Christian Ehringfeld
}
}
return false;
dc6b13b4 Christian Ehringfeld
}
b0b8dac3 Christian Ehringfeld
6d91d381 Christian Ehringfeld
QHash<QString, QVariant> EntityManager::findByPk(qint64 id,
const QSharedPointer<Entity>
f5087482 Christian Ehringfeld
&e) {
abb9e8c5 Christian Ehringfeld
QSqlQuery q = this->schema->getQueryBuilder()->find(id, e, 0,
e->getPrimaryKey());
706de2e8 Christian Ehringfeld
auto listMap = this->convertQueryResult(q);
if (!listMap.isEmpty()) {
return listMap.at(0);
7e233492 Christian Ehringfeld
}
706de2e8 Christian Ehringfeld
return QHash<QString, QVariant>();
7e233492 Christian Ehringfeld
}

f4e3904b Christian Ehringfeld
e0e1ead8 Christian Ehringfeld
QList<QHash<QString, QVariant> > EntityManager::findAllByAttributes(
const QSharedPointer<Entity>
&entity,
bool ignoreID) {
abb9e8c5 Christian Ehringfeld
QSqlQuery q = this->schema->getQueryBuilder()->findByAttributes(
e0e1ead8 Christian Ehringfeld
entity, ignoreID);
dc6b13b4 Christian Ehringfeld
return this->convertQueryResult(q);
}

e0e1ead8 Christian Ehringfeld
QList<QHash <QString, QVariant> > EntityManager::findAllByAttributes(
e24791d7 Christian Ehringfeld
const QHash<QString, QVariant> &m, const QString &tblname, bool ignoreID) {
abb9e8c5 Christian Ehringfeld
QSqlQuery q = this->schema->getQueryBuilder()->findByAttributes(m,
dc6b13b4 Christian Ehringfeld
tblname, ignoreID);
7e233492 Christian Ehringfeld
return this->convertQueryResult(q);
}

e0e1ead8 Christian Ehringfeld
QList<QHash<QString, QVariant> > EntityManager::convertQueryResult(
QSqlQuery &q) {
7e233492 Christian Ehringfeld
QList<QHash <QString, QVariant> > listmap = QList<QHash <QString, QVariant> >();
this->db->select(q);
QSqlRecord rec = q.record();
QStringList l = QStringList();
qint16 field_count = rec.count();
for (int var = 0; var < field_count; ++var) {
l.append(rec.fieldName(var));
}
QHash<QString, QVariant> map = QHash<QString, QVariant>();
while (q.next()) {
for (int var = 0; var < field_count; ++var) {
map.insert(l.at(var), q.value(rec.indexOf(l.at(var))));
}
listmap.append(map);
}
return listmap;
}

f5087482 Christian Ehringfeld
QList<QHash <QString, QVariant> > EntityManager::findAll(
const QSharedPointer<Entity> &e) {
abb9e8c5 Christian Ehringfeld
QSqlQuery q = this->schema->getQueryBuilder()->findAll(e);
7e233492 Christian Ehringfeld
return this->convertQueryResult(q);
}
b0b8dac3 Christian Ehringfeld
e0e1ead8 Christian Ehringfeld
void EntityManager::resolveRelations(const QSharedPointer<Entity> &entity,
4f3b13f3 Christian Ehringfeld
const QHash<QString, QVariant> &map, const bool refresh) {
abb9e8c5 Christian Ehringfeld
auto props = entity->getRelationProperties();
2d9fab10 Christian Ehringfeld
auto iterator = props.constBegin();
while (iterator != props.constEnd()) {
const Relation r = iterator.key();
const QMetaProperty property = iterator.value();
df1e56bd Christian Ehringfeld
QString colName = this->schema->getQueryBuilder()->generateColumnNameID(
r.getPropertyName());
a06633f7 Christian Ehringfeld
switch (r.getType()) {
case MANY_TO_ONE:
df1e56bd Christian Ehringfeld
if (map.contains(colName)) {
this->manyToOne(entity, map.value(colName), property, refresh);
a06633f7 Christian Ehringfeld
}
break;
case MANY_TO_MANY:
afde9013 Christian Ehringfeld
this->manyToMany(entity, property, refresh);
a06633f7 Christian Ehringfeld
break;
case ONE_TO_MANY:
4f3b13f3 Christian Ehringfeld
this->oneToMany(entity, r, property, refresh);
a06633f7 Christian Ehringfeld
break;
case ONE_TO_ONE:
df1e56bd Christian Ehringfeld
this->oneToOne(entity, r, property, refresh, map.value(colName));
a06633f7 Christian Ehringfeld
break;
}
++iterator;
}
}

244c6d53 Christian Ehringfeld
bool EntityManager::save(QSharedPointer<Entity> &entity,
const bool persistRelations) {
b5d490c7 Christian Ehringfeld
if (entity->getProperty(entity->getPrimaryKey()) > -1) {
244c6d53 Christian Ehringfeld
return this->merge(entity, persistRelations);
7e233492 Christian Ehringfeld
} else {
244c6d53 Christian Ehringfeld
return this->create(entity, persistRelations);
7e233492 Christian Ehringfeld
}
}

dc6b13b4 Christian Ehringfeld
qint64 EntityManager::findId(QSharedPointer<Entity> &entity) {
qint64 r = -1;
abb9e8c5 Christian Ehringfeld
QSqlQuery q = this->schema->getQueryBuilder()->findId(entity);
dc6b13b4 Christian Ehringfeld
this->db->select(q);
if (q.next()) {
r = q.value(0).toLongLong();
}
return r;
}

586bb527 Christian Ehringfeld
bool EntityManager::remove(QSharedPointer<Entity> &entity) {
7e233492 Christian Ehringfeld
bool rc = false;
e24791d7 Christian Ehringfeld
this->db->startTransaction();
this->removeRelations(entity);
abb9e8c5 Christian Ehringfeld
auto queries = this->schema->getQueryBuilder()->remove(entity);
95b60eb2 Christian Ehringfeld
bool ok = true;
for (int var = 0; var < queries.size(); ++var) {
auto q = queries.at(var);
if (!q.exec()) {
this->db->rollbackTransaction();
ok = false;
break;
}
}
if (ok && this->db->commitTransaction()) {
9070a496 Christian Ehringfeld
this->cache.remove(entity);
586bb527 Christian Ehringfeld
entity.clear();
7e233492 Christian Ehringfeld
rc = true;
}
return rc;
}
b0b8dac3 Christian Ehringfeld
8306a974 Christian Ehringfeld
bool EntityManager::removeAll(QString tblname) {
abb9e8c5 Christian Ehringfeld
return this->schema->getQueryBuilder()->removeAll(tblname).exec();
8306a974 Christian Ehringfeld
}

b7446f4c Christian Ehringfeld
bool EntityManager::createTable(const QSharedPointer<Entity> &entity,
bool createRelationTables) {
abb9e8c5 Christian Ehringfeld
return this->schema->getQueryBuilder()->createTable(entity,
b7446f4c Christian Ehringfeld
createRelationTables);
9c2f773f Christian Ehringfeld
}

e0e1ead8 Christian Ehringfeld
qint8 EntityManager::count(const QSharedPointer<Entity> &entity,
bool ignoreID) {
dc6b13b4 Christian Ehringfeld
qint8 rc = -1;
abb9e8c5 Christian Ehringfeld
QSqlQuery q = this->schema->getQueryBuilder()->count(entity,
e0e1ead8 Christian Ehringfeld
ignoreID);
dc6b13b4 Christian Ehringfeld
this->db->select(q);
if (q.next()) {
rc = q.value(0).toInt();
}
return rc;
}

qint8 EntityManager::count(const QString &tableName) {
qint8 rc = -1;
abb9e8c5 Christian Ehringfeld
QSqlQuery q = this->schema->getQueryBuilder()->count(tableName);
dc6b13b4 Christian Ehringfeld
this->db->select(q);
if (q.next()) {
rc = q.value(0).toInt();
}
return rc;
}

586bb527 Christian Ehringfeld
void EntityManager::setConnectionNames(QStringList list) {
EntityManager::connectionNames = list;
}