New method for User

This commit is contained in:
Krad
2021-12-10 10:16:41 +08:00
parent a24ff260a3
commit aed07807a6
2 changed files with 107 additions and 7 deletions

View File

@@ -5,6 +5,7 @@
#include "User.h"
#include "db/SQLHelper.h"
#include <QMetaProperty>
#include <QUuid>
#include <qdebug.h>
User* User::currentUser = nullptr;
@@ -66,15 +67,109 @@ bool User::QueryUser(QString userID, QString Pwd) {
return false;
}
bool User::existsUser(QString userCode) {
QString sql = QString("select * from Account where UserCode=:userID");
QMap<QString,QVariant> map;
QMap<QString,QVariant> params;
params[":userID"] = userCode;
SQLHelper::QueryFirst(sql, map, &params);
return !map.isEmpty();
}
bool User::getUser(QString userUID, User& user) {
QString sql = QString("select * from Account where UserID=:userUID");
QMap<QString,QVariant> map;
QMap<QString,QVariant> params;
params[":userUID"] = userUID;
SQLHelper::QueryFirst(sql, map, &params);
if(!map.isEmpty())
{
#define USER_READONLY_PROPERTY(name) user.m_##name = map[#name].toString();
#define USER_PROPERTY(name) user.mf_##name = false;\
USER_READONLY_PROPERTY(name)
USER_PROPERTIES_MACRO()
#undef USER_PROPERTY
#undef USER_READONLY_PROPERTY
return true;
}
return false;
}
bool User::isAdmin() {
if (isAdminFlag<0)
{
QString sql = QString("select RoleName from Role where RoleID=:RoleID");
QMap<QString,QVariant> map;
QMap<QString,QVariant> params;
params[":RoleID"] = this->m_RoleID;
SQLHelper::QueryFirst(sql, map, &params);
isAdminFlag = map["RoleName"].toString() == "admin"?1:0 ;
auto RoleName = getRoleName(this->m_RoleID);
isAdminFlag = RoleName == "admin"?1:0 ;
}
return isAdminFlag;
}
static bool LOAD_ALL_ROLE = false;
static QMap<QString,QString> roleCache;
QString User::getRoleName(QString RoleID) {
if (roleCache.contains(RoleID)) return roleCache[RoleID];
QString sql = QString("select RoleName from Role where RoleID=:RoleID");
QMap<QString,QVariant> map;
QMap<QString,QVariant> params;
params[":RoleID"] = RoleID;
SQLHelper::QueryFirst(sql, map, &params);
roleCache[RoleID] = map["RoleName"].toString();
return map["RoleName"].toString();
}
QString User::getRoleID(QString RoleName) {
if (roleCache.values().contains(RoleName)) {
int index = roleCache.values().indexOf(RoleName);
return roleCache.keys()[index];
}
QString sql = QString("select RoleID from Role where RoleName=:RoleName");
QMap<QString,QVariant> map;
QMap<QString,QVariant> params;
params[":RoleName"] = RoleName;
SQLHelper::QueryFirst(sql, map, &params);
roleCache[map["RoleID"].toString()] = RoleName;
return map["RoleID"].toString();
}
QStringList User::getAllRoleName() {
if (!LOAD_ALL_ROLE)
{
auto table = SQLHelper::getTable("Role");
table->select();
for (int i = 0; i < table->rowCount(); i++)
{
roleCache[table->data(table->index(i, 0)).toString()] =
table->data(table->index(i,1)).toString();
qDebug()<<i<<", "<< table->data(table->index(i, 0)).toString()<<", "
<<table->data(table->index(i,1)).toString();
}
}
return roleCache.values();
}
bool User::insertUser(QString UserCode, User &user) {
user.m_UserID = QUuid::createUuid().toString();
user.m_UserCode = UserCode;
static QString updateSQL = "insert into Account (%1) values (%2)";
QString namesString = "";
QString valuesString = "";
QMap<QString,QVariant> params;
#define USER_READONLY_PROPERTY(name)\
namesString += QString((!namesString.isEmpty())?", %1":"%1").arg(#name);\
valuesString += QString((!valuesString.isEmpty())?", :%1":":%1").arg(#name);\
params[":" #name] = user.m_##name;
#define USER_PROPERTY(name)\
USER_READONLY_PROPERTY(name)
USER_PROPERTIES_MACRO()
#undef USER_PROPERTY
#undef USER_READONLY_PROPERTY
bool result = 1 == SQLHelper::ExecuteNoQuery(updateSQL.arg(namesString,valuesString),&params);
return result;
}

View File

@@ -21,6 +21,7 @@ class User:public QObject {
Q_OBJECT
public:
static bool QueryUser(QString userID, QString Pwd);
static bool existsUser(QString userCode);
static QString getEncryptedPassword(const QString& password)
{
QByteArray bytePwd = password.toLatin1();
@@ -30,6 +31,8 @@ public:
static User* Current(){
return currentUser;
}
static bool getUser(QString userUID,User& user);
static bool insertUser(QString UserCode,User& user);
explicit User(QObject *parent=nullptr);
~User();
QString getIndexName(){
@@ -60,9 +63,11 @@ public:
}
bool submitChange();
bool isAdmin();
static QString getRoleName(QString RoleID);
static QString getRoleID(QString RoleName);
static QStringList getAllRoleName();
private:
static User* currentUser;
// add properties store and modify flag
#define USER_READONLY_PROPERTY(name) QString m_##name;
#define USER_PROPERTY(name)\