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);