This commit is contained in:
luoliangyi 2022-12-15 18:21:28 +08:00
parent 96ba574f68
commit a031578402
2 changed files with 5 additions and 6 deletions

View File

@ -178,7 +178,7 @@ HGResult HGPdtToolDbUserMgrImpl::Create(const HGChar* host, HGUShort port, const
return HGBASE_ERR_FAIL; return HGBASE_ERR_FAIL;
} }
if (!SetUserLoginInfo(sql, userName, connectId, userName)) if (0 != SetUserLoginInfo(sql, userName, connectId, userName))
{ {
mysql_close(sql); mysql_close(sql);
return HGBASE_ERR_FAIL; return HGBASE_ERR_FAIL;
@ -290,7 +290,7 @@ HGResult HGPdtToolDbUserMgrImpl::RefuseUser(HGChar* userName)
return HGBASE_ERR_FAIL; 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; return HGBASE_ERR_FAIL;
} }
@ -1129,7 +1129,7 @@ int HGPdtToolDbUserMgrImpl::GetConnectId(MYSQL* sql)
return connId; 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 != sql);
assert(NULL != userName && 0 != *userName); assert(NULL != userName && 0 != *userName);
@ -1139,8 +1139,7 @@ bool HGPdtToolDbUserMgrImpl::SetUserLoginInfo(MYSQL* sql, const HGChar* userName
char sqlCmd[1024]; char sqlCmd[1024];
sprintf(sqlCmd, "update user_list set connect_id = %d, take_user = '%s' where user = '%s';", sprintf(sqlCmd, "update user_list set connect_id = %d, take_user = '%s' where user = '%s';",
connectId, takeUserName.c_str(), userName); connectId, takeUserName.c_str(), userName);
int ret = mysql_query(sql, sqlCmd); return mysql_query(sql, sqlCmd);
return (0 == ret) ? true : false;
} }
int HGPdtToolDbUserMgrImpl::GetUserLoginInfo(MYSQL* sql, const HGChar* userName, int& connectId, std::string& takeUserName) int HGPdtToolDbUserMgrImpl::GetUserLoginInfo(MYSQL* sql, const HGChar* userName, int& connectId, std::string& takeUserName)

View File

@ -45,7 +45,7 @@ private:
static std::string GetPassword(MYSQL* sql, const HGChar* userName); static std::string GetPassword(MYSQL* sql, const HGChar* userName);
static std::string GetAuthString(MYSQL* sql, const HGChar* pwd); static std::string GetAuthString(MYSQL* sql, const HGChar* pwd);
static int GetConnectId(MYSQL* sql); 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); static int GetUserLoginInfo(MYSQL* sql, const HGChar* userName, int &connectId, std::string &takeUserName);
int CreateFileTable(MYSQL* sql); int CreateFileTable(MYSQL* sql);