Herunterladen als
root/src/schema/mysqlschema.cpp @ 49762cb7
| 829c3e69 | Christian Ehringfeld | #include "mysqlschema.h"
|
|
#include "../database.h"
|
|||
#include <QSqlRecord>
|
|||
#include <QSqlResult>
|
|||
| ed9ffe4a | Christian Ehringfeld | #include "mysqlquerybuilder.h"
|
|
| 829c3e69 | Christian Ehringfeld | using namespace CuteEntityManager;
|
|
| b0b8dac3 | Christian Ehringfeld | ||
| ed9ffe4a | Christian Ehringfeld | MysqlSchema::MysqlSchema(QSharedPointer<Database> database) : Schema(
|
|
| 57d6da31 | Christian Ehringfeld | database, QSharedPointer<QueryBuilder>(new MysqlQueryBuilder(this, database))) {
|
|
| ed9ffe4a | Christian Ehringfeld | }
|
|
| 829c3e69 | Christian Ehringfeld | QSharedPointer<QHash<QString, QString> > MysqlSchema::getTypeMap() {
|
|
if (this->typeMap->empty()) {
|
|||
this->typeMap->insert(TYPE_PK,
|
|||
"integer PRIMARY KEY AUTOINCREMENT NOT NULL");
|
|||
this->typeMap->insert(TYPE_BIGPK,
|
|||
"bigint PRIMARY KEY AUTOINCREMENT NOT NULL");
|
|||
this->typeMap->insert(TYPE_BOOLEAN, "boolean");
|
|||
this->typeMap->insert(TYPE_SMALLINT, "smallint");
|
|||
this->typeMap->insert(TYPE_INTEGER, "integer");
|
|||
this->typeMap->insert(TYPE_BIGINT, "bigint");
|
|||
this->typeMap->insert(TYPE_FLOAT, "float");
|
|||
this->typeMap->insert(TYPE_DOUBLE, "double");
|
|||
this->typeMap->insert(TYPE_FLOAT, "real");
|
|||
this->typeMap->insert(TYPE_DECIMAL, "decimal(10,0)");
|
|||
this->typeMap->insert(TYPE_TEXT, "text");
|
|||
this->typeMap->insert(TYPE_STRING, "text");
|
|||
this->typeMap->insert(TYPE_CHAR, "char");
|
|||
this->typeMap->insert(TYPE_BINARY, "blob");
|
|||
this->typeMap->insert(TYPE_DATETIME, "datetime");
|
|||
this->typeMap->insert(TYPE_DATE, "date");
|
|||
this->typeMap->insert(TYPE_TIME, "time");
|
|||
this->typeMap->insert(TYPE_TIMESTAMP, "timestamp");
|
|||
this->typeMap->insert(TYPE_MONEY, "decimal(19,4)");
|
|||
}
|
|||
return this->typeMap;
|
|||
}
|
|||
| b0b8dac3 | Christian Ehringfeld | ||
| 829c3e69 | Christian Ehringfeld | QStringList MysqlSchema::findTableNames(QString schema) {
|
|
QString sql = "SHOW TABLES";
|
|||
if (!schema.isEmpty()) {
|
|||
sql += " FROM " + this->quoteSimpleTableName(schema);
|
|||
}
|
|||
auto l = QStringList();
|
|||
auto q = this->database->getQuery();
|
|||
q.prepare(sql);
|
|||
this->database->select(q);
|
|||
while (q.next()) {
|
|||
l.append(q.value(0).toString());
|
|||
}
|
|||
return l;
|
|||
}
|
|||
| b0b8dac3 | Christian Ehringfeld | ||
| 829c3e69 | Christian Ehringfeld | QHash<QString, QStringList> MysqlSchema::findUniqueIndexes(
|
|
const QSharedPointer<TableSchema> &table) {
|
|||
QHash<QString, QStringList> uniqueIndexes = QHash<QString, QStringList>();
|
|||
QSqlQuery q = this->database->getQuery();
|
|||
q.setForwardOnly(true);
|
|||
q.exec("SHOW INDEX FROM " + this->quoteSimpleTableName(
|
|||
table->getName()));
|
|||
while (q.next()) {
|
|||
QString indexName = q.value("Key_name").toString();
|
|||
if (!(q.value("Non_unique").toBool())) {
|
|||
QStringList indexInfo = QStringList();
|
|||
indexInfo.append(q.value("Column_name").toString());
|
|||
uniqueIndexes.insert(indexName, indexInfo);
|
|||
}
|
|||
}
|
|||
return uniqueIndexes;
|
|||
}
|
|||
| b0b8dac3 | Christian Ehringfeld | ||
| 829c3e69 | Christian Ehringfeld | void MysqlSchema::findConstraints(const QSharedPointer<TableSchema> &ts) {
|
|
| 5b3f11b2 | Christian Ehringfeld | QString sql =
|
|
| ed9ffe4a | Christian Ehringfeld | "SELECT " + this->quoteColumnName("table_name") + ", " +
|
|
this->quoteColumnName("column_name") + ", " +
|
|||
this->quoteColumnName("referenced_table_name") + ", " +
|
|||
this->quoteColumnName("referenced_column_name");
|
|||
| 5b3f11b2 | Christian Ehringfeld | sql += " FROM " + this->quoteTableName("information_schema.key_column_usage");
|
|
sql += "WHERE " + this->quoteColumnName("referenced_table_name") +
|
|||
" is not null";
|
|||
| ed9ffe4a | Christian Ehringfeld | sql += " AND " + this->quoteColumnName("table_schema") + " = ?";
|
|
sql += " AND " + this->quoteColumnName("table_name") + " = ?";
|
|||
| 5b3f11b2 | Christian Ehringfeld | QSqlQuery q = this->database->getQuery();
|
|
| ed9ffe4a | Christian Ehringfeld | q.prepare(sql);
|
|
q.bindValue(0, this->getDatabase()->getDatabase().databaseName());
|
|||
q.bindValue(1, ts->getName());
|
|||
this->database->select(q);
|
|||
| 829c3e69 | Christian Ehringfeld | auto foreignKeys = ts->getRelations();
|
|
| 5b3f11b2 | Christian Ehringfeld | int id = 0;
|
|
while (q.next()) {
|
|||
| ed9ffe4a | Christian Ehringfeld | auto rel = new QSqlRelation(q.value("referenced_table_name").toString(),
|
|
q.value("column_name").toString(),
|
|||
q.value("referenced_column_name").toString());
|
|||
auto ptr = QSharedPointer<QSqlRelation>(rel);
|
|||
foreignKeys.insert(QString::number(id), ptr);
|
|||
++id;
|
|||
| 5b3f11b2 | Christian Ehringfeld | }
|
|
| 829c3e69 | Christian Ehringfeld | ts->setRelations(foreignKeys);
|
|
}
|
|||
| 14f9beed | Christian Ehringfeld | ||
| 829c3e69 | Christian Ehringfeld | QSharedPointer<TableSchema> MysqlSchema::loadTableSchema(QString name) {
|
|
auto ptr = QSharedPointer<TableSchema>(new TableSchema());
|
|||
this->resolveTableNames(ptr, name);
|
|||
if (this->findColumns(ptr)) {
|
|||
this->findConstraints(ptr);
|
|||
} else {
|
|||
ptr.clear();
|
|||
}
|
|||
return ptr;
|
|||
}
|
|||
void MysqlSchema::resolveTableNames(const QSharedPointer<TableSchema> &ts,
|
|||
QString name) {
|
|||
QStringList parts = name.replace("`", "").split(".");
|
|||
if (parts.size() > 1) {
|
|||
ts->setSchemaName(parts.at(0));
|
|||
ts->setName(parts.at(1));
|
|||
ts->setFullName(ts->getSchemaName() + "." + ts->getName());
|
|||
} else {
|
|||
ts->setFullName(parts.at(0));
|
|||
ts->setName(parts.at(0));
|
|||
}
|
|||
}
|