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

Update MySQLConn

Browse Source
This commit is contained in:
Kirigaya Kazuto 2017-08-07 10:55:28 +08:00
parent d13e4042bd
commit 7598ed496d
2 changed files with 34 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
MySQLWrapper.cpp
View File

@ -1,5 +1,11 @@
#include "MySQLWrapper.h" #include "MySQLWrapper.h"
#include "MySQLInclude.h" #include "MySQLInclude.h"
#include <string>
using namespace std;
/**********************
MySQL Result Class.
**********************/
struct MySQLResult::impl struct MySQLResult::impl
{ {
@ -48,9 +54,15 @@ void MySQLResult::stepRow(const std::function<void(char** RowPtr, unsigned long*
} }
} }
/**********************
MySQL Connection Class.
**********************/
struct MySQLConn::impl struct MySQLConn::impl
{ {
MYSQL m; MYSQL m;
string lasterr;
}; };
MySQLConn::MySQLConn() : pimpl(new impl) MySQLConn::MySQLConn() : pimpl(new impl)
@ -65,7 +77,15 @@ MySQLConn::~MySQLConn()
int MySQLConn::connect(const char* host, const char* user, const char* passwd, const char* db, unsigned int port) int MySQLConn::connect(const char* host, const char* user, const char* passwd, const char* db, unsigned int port)
{ {
return (mysql_real_connect(&(pimpl->m),host,user,passwd,db,port,NULL,0)==NULL)?0:-1; if(mysql_real_connect(&(pimpl->m),host,user,passwd,db,port,NULL,0)==NULL)
{
pimpl->lasterr=mysql_error(&(pimpl->m));
return -1;
}
else
{
return 0;
}
} }
int MySQLConn::exec(const std::string& SQLCommand, const std::function<void(MySQLResult&)>& func) int MySQLConn::exec(const std::string& SQLCommand, const std::function<void(MySQLResult&)>& func)
@ -73,8 +93,10 @@ 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));
return -1; return -1;
} }
MYSQL_RES* pres=mysql_store_result(&(pimpl->m)); MYSQL_RES* pres=mysql_store_result(&(pimpl->m));
if(pres==nullptr) if(pres==nullptr)
{ {
@ -87,6 +109,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));
return -2; return -2;
} }
} }
@ -94,7 +117,7 @@ int MySQLConn::exec(const std::string& SQLCommand, const std::function<void(MySQ
MySQLResult res; MySQLResult res;
res.pimpl->res=pres; res.pimpl->res=pres;
func(res); if(func) func(res);
/// MySQL_RES will be released normally by MySQLResult::~MySQLResult() /// MySQL_RES will be released normally by MySQLResult::~MySQLResult()
return 0; return 0;
@ -109,3 +132,8 @@ uint64_t MySQLConn::getAffectedRows()
{ {
return mysql_affected_rows(&(pimpl->m)); return mysql_affected_rows(&(pimpl->m));
} }
const std::string& MySQLConn::getError()
{
return pimpl->lasterr;
}

6
MySQLWrapper.h
View File

@ -32,13 +32,15 @@ public:
~MySQLConn(); ~MySQLConn();
/// 数据库连接 /// 数据库连接
/// 返回值: 0:连接正常, -1:连接错误
int connect(const char* host,const char* user,const char* passwd,const char* db,unsigned int port); int connect(const char* host,const char* user,const char* passwd,const char* db,unsigned int port);
/// 执行SQL语句 /// 执行SQL语句
int exec(const std::string& SQLCommand,const std::function<void(MySQLResult&)>& res); /// 返回值: 1: 语句执行正常,无返回结果(func未执行) 0:语句执行正常,返回结果集交付func执行. -1:语句执行错误.(mysql_real_query失败) -2: 语句执行正确,获取结果出现错误
int exec(const std::string& SQLCommand,const std::function<void(MySQLResult&)>& func);
/// 获取错误信息 /// 获取错误信息
const char* getError(); const std::string& getError();
/// 当结果为空时用于判断数据库运行正常与否 /// 当结果为空时用于判断数据库运行正常与否
unsigned int getFieldCount(); unsigned int getFieldCount();