From 2c83a26f0609bf9c772f82328c65814ce4428c1f Mon Sep 17 00:00:00 2001
From: Krad <kradchen123@163.com>
Date: Thu, 11 Nov 2021 14:08:44 +0800
Subject: [PATCH] Query and executeNoQuery method for SQLHelper

---
 src/db/SQLHelper.cpp | 24 ++++++++++++++++++++++++
 src/db/SQLHelper.h   |  2 ++
 2 files changed, 26 insertions(+)

diff --git a/src/db/SQLHelper.cpp b/src/db/SQLHelper.cpp
index 62e30f5..e15c269 100644
--- a/src/db/SQLHelper.cpp
+++ b/src/db/SQLHelper.cpp
@@ -57,3 +57,27 @@ int SQLHelper::QueryCount(QString sql) {
     }
     return 0;
 }
+
+int SQLHelper::ExecuteNoQuery(QString sql) {
+    QSqlQuery query(*defaultDatabase);
+    qDebug()<<sql;
+    if (query.exec(sql))
+    {
+        return query.numRowsAffected();
+    }
+    return 0;
+}
+
+void SQLHelper::Query(QString sql,QMap<QString,QVariant>& result) {
+    QSqlQuery query(*defaultDatabase);
+    if (query.exec(sql))
+    {
+        if(query.next())
+        {
+            for(int i=0; i<query.record().count(); i++)
+            {
+                result[query.record().fieldName(i)]=query.record().value(i);
+            }
+        }
+    }
+}
diff --git a/src/db/SQLHelper.h b/src/db/SQLHelper.h
index 9573623..e820f26 100644
--- a/src/db/SQLHelper.h
+++ b/src/db/SQLHelper.h
@@ -16,7 +16,9 @@ public:
     static bool Open();
     static bool Open(QSqlDatabase* base);
     static void Close();
+    static void Query(QString sql,QMap<QString,QVariant>& result);
     static int QueryCount(QString sql);
+    static int ExecuteNoQuery(QString sql);
     static QSqlTableModel* getTable(const QString & tableName);
 private:
     static QSqlDatabase*  defaultDatabase;