HCTech/MySQLWrapper
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update MySQL Error handling

Browse Source
This commit is contained in:
Kirigaya Kazuto 2017-11-14 14:43:25 +08:00
parent 7598ed496d
commit a5dcc01d2f
2 changed files with 26 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
MySQLWrapper.cpp
View File

@ -63,11 +63,13 @@ struct MySQLConn::impl
{ {
MYSQL m; MYSQL m;
string lasterr; string lasterr;
unsigned int lasterrcode;
}; };
MySQLConn::MySQLConn() : pimpl(new impl) MySQLConn::MySQLConn() : pimpl(new impl)
{ {
mysql_init(&(pimpl->m)); mysql_init(&(pimpl->m));
pimpl->lasterrcode=0;
} }
MySQLConn::~MySQLConn() MySQLConn::~MySQLConn()
@ -75,11 +77,11 @@ MySQLConn::~MySQLConn()
mysql_close(&(pimpl->m)); mysql_close(&(pimpl->m));
} }
int MySQLConn::connect(const char* host, const char* user, const char* passwd, const char* db, unsigned int port) int MySQLConn::connect(const std::string& host, const std::string& user, const std::string& passwd, const std::string& db, unsigned int port)
{ {
if(mysql_real_connect(&(pimpl->m),host,user,passwd,db,port,NULL,0)==NULL) if(mysql_real_connect(&(pimpl->m),host.c_str(),user.c_str(),passwd.c_str(),db.c_str(),port,nullptr,0)==nullptr)
{ {
pimpl->lasterr=mysql_error(&(pimpl->m)); _doUpdateError();
return -1; return -1;
} }
else else
@ -93,7 +95,7 @@ int MySQLConn::exec(const std::string& SQLCommand, const std::function<void(MySQ
if(mysql_real_query(&(pimpl->m),SQLCommand.c_str(),SQLCommand.size())!=0) if(mysql_real_query(&(pimpl->m),SQLCommand.c_str(),SQLCommand.size())!=0)
{ {
/// Failed to Query /// Failed to Query
pimpl->lasterr=mysql_error(&(pimpl->m)); _doUpdateError();
return -1; return -1;
} }
@ -109,7 +111,7 @@ int MySQLConn::exec(const std::string& SQLCommand, const std::function<void(MySQ
else else
{ {
/// Failed to store result. /// Failed to store result.
pimpl->lasterr=mysql_error(&(pimpl->m)); _doUpdateError();
return -2; return -2;
} }
} }
@ -137,3 +139,15 @@ const std::string& MySQLConn::getError()
{ {
return pimpl->lasterr; return pimpl->lasterr;
} }
unsigned int MySQLConn::getErrCode()
{
return pimpl->lasterrcode;
}
///protected
void MySQLConn::_doUpdateError()
{
pimpl->lasterr=mysql_error(&(pimpl->m));
pimpl->lasterrcode=mysql_errno(&(pimpl->m));
}

8
MySQLWrapper.h
View File

@ -33,7 +33,7 @@ public:
/// 数据库连接 /// 数据库连接
/// 返回值: 0:连接正常, -1:连接错误 /// 返回值: 0:连接正常, -1:连接错误
int connect(const char* host,const char* user,const char* passwd,const char* db,unsigned int port); int connect(const std::string& host, const std::string& user, const std::string& passwd, const std::string& db, unsigned int port);
/// 执行SQL语句 /// 执行SQL语句
/// 返回值: 1: 语句执行正常,无返回结果(func未执行) 0:语句执行正常,返回结果集交付func执行. -1:语句执行错误.(mysql_real_query失败) -2: 语句执行正确,获取结果出现错误 /// 返回值: 1: 语句执行正常,无返回结果(func未执行) 0:语句执行正常,返回结果集交付func执行. -1:语句执行错误.(mysql_real_query失败) -2: 语句执行正确,获取结果出现错误
@ -41,10 +41,16 @@ public:
/// 获取错误信息 /// 获取错误信息
const std::string& getError(); const std::string& getError();
unsigned int getErrCode();
/// 当结果为空时用于判断数据库运行正常与否 /// 当结果为空时用于判断数据库运行正常与否
unsigned int getFieldCount(); unsigned int getFieldCount();
uint64_t getAffectedRows(); uint64_t getAffectedRows();
protected:
/// 更新错误信息
void _doUpdateError();
private: private:
struct impl; struct impl;
impl* pimpl; impl* pimpl;