From a03157840221e92f00c17719e3421d94140ed5fc Mon Sep 17 00:00:00 2001 From: luoliangyi <87842688@qq.com> Date: Thu, 15 Dec 2022 18:21:28 +0800 Subject: [PATCH] =?UTF-8?q?=E5=BE=AE=E8=B0=83?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- db/HGPdtToolDb/HGPdtToolDbImpl.cpp | 9 ++++----- db/HGPdtToolDb/HGPdtToolDbImpl.hpp | 2 +- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/db/HGPdtToolDb/HGPdtToolDbImpl.cpp b/db/HGPdtToolDb/HGPdtToolDbImpl.cpp index f6c778b..b45bf95 100644 --- a/db/HGPdtToolDb/HGPdtToolDbImpl.cpp +++ b/db/HGPdtToolDb/HGPdtToolDbImpl.cpp @@ -178,7 +178,7 @@ HGResult HGPdtToolDbUserMgrImpl::Create(const HGChar* host, HGUShort port, const return HGBASE_ERR_FAIL; } - if (!SetUserLoginInfo(sql, userName, connectId, userName)) + if (0 != SetUserLoginInfo(sql, userName, connectId, userName)) { mysql_close(sql); return HGBASE_ERR_FAIL; @@ -290,7 +290,7 @@ HGResult HGPdtToolDbUserMgrImpl::RefuseUser(HGChar* userName) return HGBASE_ERR_FAIL; } - if (!SetUserLoginInfo(m_sql, userName, m_connectId, m_userName)) + if (0 != SetUserLoginInfo(m_sql, userName, m_connectId, m_userName)) { return HGBASE_ERR_FAIL; } @@ -1129,7 +1129,7 @@ int HGPdtToolDbUserMgrImpl::GetConnectId(MYSQL* sql) return connId; } -bool HGPdtToolDbUserMgrImpl::SetUserLoginInfo(MYSQL* sql, const HGChar* userName, int connectId, const std::string& takeUserName) +int HGPdtToolDbUserMgrImpl::SetUserLoginInfo(MYSQL* sql, const HGChar* userName, int connectId, const std::string& takeUserName) { assert(NULL != sql); assert(NULL != userName && 0 != *userName); @@ -1139,8 +1139,7 @@ bool HGPdtToolDbUserMgrImpl::SetUserLoginInfo(MYSQL* sql, const HGChar* userName char sqlCmd[1024]; sprintf(sqlCmd, "update user_list set connect_id = %d, take_user = '%s' where user = '%s';", connectId, takeUserName.c_str(), userName); - int ret = mysql_query(sql, sqlCmd); - return (0 == ret) ? true : false; + return mysql_query(sql, sqlCmd); } int HGPdtToolDbUserMgrImpl::GetUserLoginInfo(MYSQL* sql, const HGChar* userName, int& connectId, std::string& takeUserName) diff --git a/db/HGPdtToolDb/HGPdtToolDbImpl.hpp b/db/HGPdtToolDb/HGPdtToolDbImpl.hpp index dedf7fb..33fed14 100644 --- a/db/HGPdtToolDb/HGPdtToolDbImpl.hpp +++ b/db/HGPdtToolDb/HGPdtToolDbImpl.hpp @@ -45,7 +45,7 @@ private: static std::string GetPassword(MYSQL* sql, const HGChar* userName); static std::string GetAuthString(MYSQL* sql, const HGChar* pwd); static int GetConnectId(MYSQL* sql); - static bool SetUserLoginInfo(MYSQL* sql, const HGChar* userName, int connectId, const std::string& takeUserName); + static int SetUserLoginInfo(MYSQL* sql, const HGChar* userName, int connectId, const std::string& takeUserName); static int GetUserLoginInfo(MYSQL* sql, const HGChar* userName, int &connectId, std::string &takeUserName); int CreateFileTable(MYSQL* sql);