132 lines
3.5 KiB
C++
132 lines
3.5 KiB
C++
//
|
|
// Created by Krad on 2021/10/11.
|
|
//
|
|
|
|
#include "SQLHelper.h"
|
|
#include <QSqlQuery>
|
|
#include <QSqlDatabase>
|
|
#include <QSqlQuery>
|
|
#include <QSqlError>
|
|
#include <QSqlRecord>
|
|
#include <qdebug.h>
|
|
QSqlDatabase* SQLHelper::defaultDatabase= nullptr;
|
|
QHash<QString,QSqlQueryModel*>* SQLHelper::cache= new QHash<QString,QSqlQueryModel*>;
|
|
bool SQLHelper::Open() {
|
|
if (defaultDatabase) return true;
|
|
defaultDatabase= new QSqlDatabase(QSqlDatabase::addDatabase("QSQLITE", "USCTDB"));
|
|
defaultDatabase->setDatabaseName("USCTDB.data");
|
|
return defaultDatabase->open();
|
|
}
|
|
|
|
bool SQLHelper::Open(QSqlDatabase *base) {
|
|
if(base && base != defaultDatabase){
|
|
defaultDatabase = base;
|
|
}
|
|
return defaultDatabase->open();
|
|
}
|
|
|
|
void SQLHelper::Close() {
|
|
for(auto item : cache->values())
|
|
{
|
|
delete item;
|
|
}
|
|
cache->clear();
|
|
if(defaultDatabase->isValid() && defaultDatabase->isOpen())
|
|
{
|
|
defaultDatabase->close();
|
|
delete defaultDatabase;
|
|
defaultDatabase = nullptr;
|
|
}
|
|
}
|
|
|
|
void prepareSQL(QSqlQuery& query,QMap<QString,QVariant>* params)
|
|
{
|
|
if (!params) return;
|
|
for (auto key : params->keys())
|
|
{
|
|
query.bindValue(key,params->value(key));
|
|
}
|
|
}
|
|
|
|
|
|
QSqlTableModel* SQLHelper::getTable(const QString &tableName) {
|
|
if (cache->contains(tableName)) return (QSqlTableModel*)(*cache)[tableName];
|
|
CenterAlignSqlTableModel *model = new CenterAlignSqlTableModel(nullptr,*defaultDatabase);
|
|
model->setTable(tableName);
|
|
(*cache)[tableName] = model;
|
|
return (QSqlTableModel*)(*cache)[tableName];
|
|
}
|
|
|
|
int SQLHelper::QueryCount(QString sql, QMap<QString,QVariant>* params) {
|
|
QSqlQuery query(*defaultDatabase);
|
|
prepareSQL(query, params);
|
|
if (query.exec(sql))
|
|
{
|
|
query.last();
|
|
return query.at()+1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int SQLHelper::ExecuteNoQuery(QString sql, QMap<QString,QVariant>* params) {
|
|
QSqlQuery query(*defaultDatabase);
|
|
query.prepare(sql);
|
|
prepareSQL(query, params);
|
|
if (query.exec())
|
|
{
|
|
return query.numRowsAffected();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
void SQLHelper::QueryFirst(QString sql, QMap<QString,QVariant>& result, QMap<QString,QVariant>* params) {
|
|
QSqlQuery query(*defaultDatabase);
|
|
query.prepare(sql);
|
|
prepareSQL(query, params);
|
|
if (query.exec())
|
|
{
|
|
if(query.next())
|
|
{
|
|
for(int i=0; i<query.record().count(); i++)
|
|
{
|
|
result[query.record().fieldName(i)]=query.record().value(i);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
QSqlQueryModel *SQLHelper::QueryModel(QString queryName, QString sql, QMap<QString, QVariant> *params) {
|
|
if (cache->contains(queryName)) return (*cache)[queryName];
|
|
QSqlQueryModel* model = new QSqlQueryModel;
|
|
|
|
QSqlQuery query(*defaultDatabase);
|
|
query.prepare(sql);
|
|
prepareSQL(query, params);
|
|
model->setQuery(query);
|
|
if (model->lastError().isValid()){
|
|
qDebug() << model->lastError();
|
|
delete model;
|
|
return nullptr;
|
|
}
|
|
(*cache)[queryName] = model;
|
|
return model;
|
|
}
|
|
|
|
QSqlQueryModel *SQLHelper::QueryModel(QString queryName) {
|
|
if (cache->contains(queryName)) return (*cache)[queryName];
|
|
return nullptr;
|
|
}
|
|
|
|
void SQLHelper::QueryMap(QString sql, QMap<QString, QVariant> &result, QMap<QString, QVariant> *params) {
|
|
QSqlQuery query(*defaultDatabase);
|
|
query.prepare(sql);
|
|
prepareSQL(query, params);
|
|
if (query.exec())
|
|
{
|
|
while(query.next())
|
|
{
|
|
result[query.record().value(0).toString()]=query.record().value(1);
|
|
}
|
|
}
|
|
}
|