用C++封装MySQL的API的教程
作者：goldensun 发布时间：2024-01-25 07:51:51
标签：MySQL
其实相信每个和mysql打过交道的程序员都应该会尝试去封装一套mysql的接口，这一次的封装已经记不清是我第几次了，但是每一次我希望都能做的比上次更好，更容易使用。
先来说一下这次的封装，遵守了几个原则，其中部分思想是从python借鉴过来的：
1.简单
简单，意味着不为了微小的效率提升，而去把接口搞的复杂。因为本身数据库存储效率的瓶颈并不是那一两次内存copy，代码中随处可以看到以这个为依据的设计。
2.低学习成本
使用一套新库通常意味着投入学习成本，而这次的封装并没有像django那样实现一套完整的模型系统，也没有做soci那样的语法分析器，我选择最简单易懂的方式：做sql语句拼接器，所以对习惯了使用原生mysql api的朋友，学习成本很低
3.模块化
代码实际包括了两个模块，一个是mysql client端的封装，一个是sql的拼接器，这两个模块是完全独立的，调用者可以任意组合或者独立使用。
4.尽量使用STL以及模板，简化代码编写
最大的特点就是大量使用了stringstream进行类型转化，减少了大量的重复代码。
OK，基于以上的简单介绍，我们先来看一下
一.mysql client端的封装:
class CMYSQLWrapper
{
/**
* @brief 获取错误信息
*
* @return 错误信息
*/
char* GetErrMsg();
/**
* @brief 连接MYSQL，已经支持了自动重连模式，即mysql server关闭链接会自动重连
*
* @param ip IP
* @param user 用户名
* @param pwd 密码(没有则传NULL)
* @param db 库(没有则传NULL)
*
* @return 0 succ
* else fail
*/
int Open(const char* ip,const char* user,const char* pwd,const char* strDb);
/**
* @brief 关闭链接并释放result
*/
void Close();
/**
* @brief 执行SQL语句
*
* @param strSql 执行语句
* @param result 执行结果
*
* @return 0 succ
* else fail
*/
int Query(const char* strSql);
/**
* @brief 针对Read(select)相关的的Query，可以支持blob了
*
* @param strSql sql语句
* @param vecData rows
*
* @return 0 succ
* else fail
*/
int Query(const char* strSql, vector<map<string, MYSQLValue> > &vecData);
/**
* @brief 针对Write(insert,update,delete)相关的Query
*
* @param strSql sql语句
* @param affectRowsCount 影响的行的个数
*
* @return 0 succ
* else fail
*/
int Query(const char* strSql, int& affectRowsCount);
/**
* @brief Select时获取数据，记得手工析构，或者用StMYSQLRes
*
* @param result 执行结果
*
* @return 0 succ
* else fail
*/
int Result(MYSQL_RES *&result);
/**
* @brief 返回影响行数
*
* @return >0 succ
* 0 没有更新
* <0 fail
*/
int AffectedRows();
/**
* @brief 主要是将blob转成字符串
*
* @param src blob源
* @param len 长度
*
* @return 转化后的字符串
*/
string EscStr(const char* src,uint32_t len);
/**
* @brief 将字符串中的某些字符转化(如')
*
* @param src 字符串
*
* @return 转化后的字符串
*/
string EscStr(const char* src);
};
class CMYSQLWrapper
{
/**
* @brief 获取错误信息
*
* @return 错误信息
*/
char* GetErrMsg();
/**
* @brief 连接MYSQL，已经支持了自动重连模式，即mysql server关闭链接会自动重连
*
* @param ip IP
* @param user 用户名
* @param pwd 密码(没有则传NULL)
* @param db 库(没有则传NULL)
*
* @return 0 succ
* else fail
*/
int Open(const char* ip,const char* user,const char* pwd,const char* strDb);
/**
* @brief 关闭链接并释放result
*/
void Close();
/**
* @brief 执行SQL语句
*
* @param strSql 执行语句
* @param result 执行结果
*
* @return 0 succ
* else fail
*/
int Query(const char* strSql);
/**
* @brief 针对Read(select)相关的的Query，可以支持blob了
*
* @param strSql sql语句
* @param vecData rows
*
* @return 0 succ
* else fail
*/
int Query(const char* strSql, vector<map<string, MYSQLValue> > &vecData);
/**
* @brief 针对Write(insert,update,delete)相关的Query
*
* @param strSql sql语句
* @param affectRowsCount 影响的行的个数
*
* @return 0 succ
* else fail
*/
int Query(const char* strSql, int& affectRowsCount);
/**
* @brief Select时获取数据，记得手工析构，或者用StMYSQLRes
*
* @param result 执行结果
*
* @return 0 succ
* else fail
*/
int Result(MYSQL_RES *&result);
/**
* @brief 返回影响行数
*
* @return >0 succ
* 0 没有更新
* <0 fail
*/
int AffectedRows();
/**
* @brief 主要是将blob转成字符串
*
* @param src blob源
* @param len 长度
*
* @return 转化后的字符串
*/
string EscStr(const char* src,uint32_t len);
/**
* @brief 将字符串中的某些字符转化(如')
*
* @param src 字符串
*
* @return 转化后的字符串
*/
string EscStr(const char* src);
};
代码中的注释已经描述的很清楚了，语言描述不清楚，我们直接来看一下gtest的代码:
select:
string g_name = "good";
int g_sex = 1;
string g_name_up = "update";
int g_sex_up = 2;
TEST(mysql_wrapper_easy, select)
{
vector<map<string,MYSQLValue> > vecData;
string sql = "select * from tb_test where name = '"+g_name_up+"'";
int ret = g_client.Query(sql.c_str(),vecData);
ASSERT_EQ(ret, 0) << g_client.GetErrMsg();
foreach(vecData, it_vec)
{
foreach(*it_vec, it_map)
{
cout << it_map->first << ",";
if (it_map->first == "sex")
{
cout << it_map->second.as<uint32_t>();
}
else
{
cout << it_map->second.data();
}
cout << "," << it_map->second.size() << endl;
}
}
}
int main(int argc, char **argv)
{
int ret = g_client.Open("localhost","dantezhu",NULL,"soci");
//int ret = g_client.Open("127.0.0.1","dantezhu",NULL,"soci");
if (ret)
{
cout << ret << "," << g_client.GetErrMsg() << endl;
return -1;
}
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}
string g_name = "good";
int g_sex = 1;
string g_name_up = "update";
int g_sex_up = 2;
TEST(mysql_wrapper_easy, select)
{
vector<map<string,MYSQLValue> > vecData;
string sql = "select * from tb_test where name = '"+g_name_up+"'";
int ret = g_client.Query(sql.c_str(),vecData);
ASSERT_EQ(ret, 0) << g_client.GetErrMsg();
foreach(vecData, it_vec)
{
foreach(*it_vec, it_map)
{
cout << it_map->first << ",";
if (it_map->first == "sex")
{
cout << it_map->second.as<uint32_t>();
}
else
{
cout << it_map->second.data();
}
cout << "," << it_map->second.size() << endl;
}
}
}
int main(int argc, char **argv)
{
int ret = g_client.Open("localhost","dantezhu",NULL,"soci");
//int ret = g_client.Open("127.0.0.1","dantezhu",NULL,"soci");
if (ret)
{
cout << ret << "," << g_client.GetErrMsg() << endl;
return -1;
}
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}
insert:
TEST(mysql_wrapper_easy, insert)
{
clear_data();
stringstream ss;
ss
<< "insert into tb_test(name,sex) values('"
<< g_client.EscStr(g_name.c_str())
<< "',"
<< g_sex
<< ");";
int affectRowsNum;
int ret = g_client.Query(ss.str().c_str(), affectRowsNum);
ASSERT_EQ(ret, 0) << g_client.GetErrMsg();
EXPECT_GE(affectRowsNum,0) << g_client.GetErrMsg();
}
TEST(mysql_wrapper_easy, insert)
{
clear_data();
stringstream ss;
ss
<< "insert into tb_test(name,sex) values('"
<< g_client.EscStr(g_name.c_str())
<< "',"
<< g_sex
<< ");";
int affectRowsNum;
int ret = g_client.Query(ss.str().c_str(), affectRowsNum);
ASSERT_EQ(ret, 0) << g_client.GetErrMsg();
EXPECT_GE(affectRowsNum,0) << g_client.GetErrMsg();
}
可以看出，对于mysql的收发包已经很简洁了，但是sql语句的拼装却显得十分臃肿。所以这个时候sql语句拼装器-SQLJoin闪亮登场！
二.sql语句拼装器-SQLJoin
class SQLJoin
{
public:
/**
* @brief 用流处理的方式，添加一个列名
*
* @param key 列名
*
* @return 0
*/
SQLJoin& operator << (const string& key);
/**
* @brief 用流处理的方式，添加一个SQLPair对象
*
* @param pair_data SQLPair对象
*
* @return 0
*/
SQLJoin& operator << (const SQLPair& pair_data);
/**
* @brief 输出所有列名（如name, sex, age）
*
* @return 所有列名
*/
string keys();
/**
* @brief 输出所有列值（如'dante', 1, 25）
*
* @return 所有列值
*/
string values();
/**
* @brief 输入所有列名-列值，并用指定分隔符分割（如name='dante', sex=1, age=25）
*
* @param split_str 分割符，默认是用','，也可以用and、or之类
*
* @return 所有列名-列值
*/
string pairs(const string& split_str = ",");
/**
* @brief 清空所有数据
*/
void clear();
};
class SQLJoin
{
public:
/**
* @brief 用流处理的方式，添加一个列名
*
* @param key 列名
*
* @return 0
*/
SQLJoin& operator << (const string& key);
/**
* @brief 用流处理的方式，添加一个SQLPair对象
*
* @param pair_data SQLPair对象
*
* @return 0
*/
SQLJoin& operator << (const SQLPair& pair_data);
/**
* @brief 输出所有列名（如name, sex, age）
*
* @return 所有列名
*/
string keys();
/**
* @brief 输出所有列值（如'dante', 1, 25）
*
* @return 所有列值
*/
string values();
/**
* @brief 输入所有列名-列值，并用指定分隔符分割（如name='dante', sex=1, age=25）
*
* @param split_str 分割符，默认是用','，也可以用and、or之类
*
* @return 所有列名-列值
*/
string pairs(const string& split_str = ",");
/**
* @brief 清空所有数据
*/
void clear();
};
看看我们用了SQLJoin之后的代码应该如何:
TEST(mysql_wrapper_join, insert)
{
clear_data();
SQLJoin sql_join;
sql_join
<< SQLPair("name", g_client.EscapeRealString(g_name.c_str()))
<< SQLPair("sex", g_sex);
stringstream ss;
ss
<< "insert into tb_test("
<< sql_join.keys()
<< ") values("
<< sql_join.values()
<< ")";
int affectRowsNum;
int ret = g_client.ExecuteWrite(ss.str().c_str(), affectRowsNum);
ASSERT_EQ(ret, 0) << g_client.GetErrMsg();
EXPECT_GE(affectRowsNum,0) << g_client.GetErrMsg();
}
TEST(mysql_wrapper_join, update)
{
SQLJoin sql_join;
sql_join
<< SQLPair("name", g_name_up)
<< SQLPair("sex", g_sex_up);
stringstream ss;
ss
<< "update tb_test set "
<< sql_join.pairs()
<< " where name='"
<< g_name
<<"';";
int affectRowsNum;
int ret = g_client.ExecuteWrite(ss.str().c_str(),affectRowsNum);
ASSERT_EQ(ret, 0) << g_client.GetErrMsg();
EXPECT_GE(affectRowsNum,0) << g_client.GetErrMsg();
}
TEST(mysql_wrapper_join, insert)
{
clear_data();
SQLJoin sql_join;
sql_join
<< SQLPair("name", g_client.EscapeRealString(g_name.c_str()))
<< SQLPair("sex", g_sex);
stringstream ss;
ss
<< "insert into tb_test("
<< sql_join.keys()
<< ") values("
<< sql_join.values()
<< ")";
int affectRowsNum;
int ret = g_client.ExecuteWrite(ss.str().c_str(), affectRowsNum);
ASSERT_EQ(ret, 0) << g_client.GetErrMsg();
EXPECT_GE(affectRowsNum,0) << g_client.GetErrMsg();
}
TEST(mysql_wrapper_join, update)
{
SQLJoin sql_join;
sql_join
<< SQLPair("name", g_name_up)
<< SQLPair("sex", g_sex_up);
stringstream ss;
ss
<< "update tb_test set "
<< sql_join.pairs()
<< " where name='"
<< g_name
<<"';";
int affectRowsNum;
int ret = g_client.ExecuteWrite(ss.str().c_str(),affectRowsNum);
ASSERT_EQ(ret, 0) << g_client.GetErrMsg();
EXPECT_GE(affectRowsNum,0) << g_client.GetErrMsg();
}
从上面的代码可以看出，代码的可维护性和健壮性得到了很大的提升。
OK，简单的介绍就是这样，说的比较简略，大家有兴趣可以直接看代码，也欢迎给我提意见和建议。代码下载路径如下:
mysql_wrapper
明天这份代码就会作为数据库访问层正式进入生产环境的代码中，因此有什么bug我也会及时在这里更新。
0
投稿猜你喜欢
- MySQL有多种存储引擎：MyISAM、InnoDB、MERGE、MEMORY(HEAP)、BDB(BerkeleyDB)、EXAMPLE、
- 前言为了保证数据的一致完整性，任何一个数据库都存在锁定机制。锁定机制的优劣直接应想到一个数据库系统的并发处理能力和性能，所以锁定机制的实现也
- 引言除非您正在对服务进行原型设计，否则您可能会关心应用程序的内存使用情况。内存占用更小，基础设施成本降低，扩展变得更容易/延迟。尽管 Go
- 最近做IM的时候遇到一个问题，同时用到了这三个关键字。就是查询一个人的离线消息详情，我们服务端返回给客户端显示的这个详情包括了三个内容，第一
今天，在在使用 pycharm 的使用，进行创建 python的时候，发现使用默认的创建的选项使用的python 3环境 。而我系统默认的p- 阅读上一篇：垂直栅格与渐进式行距(上) 新问题来也匆匆，去也“冲冲”。距上次发布垂直栅格与渐进式行距(上)发布，已经不知不觉过去了
- 现在同类型的网站数不胜数，网站的功能或服务日趋同质化，大的方面看不出什么差别，差别就体现在细节上。“窥斑见豹”，细节成为网站最有力的表现形式
- 什么是迭代器能被 next 指针调用，并不断返回下一个值的对象，叫做迭代器。表示为Iterator，迭代器是一个对象类型数据。概念迭代器指的
- 本文实例讲述了php以post形式发送xml的方法。分享给大家供大家参考。具体方法如下：方法一，使用curl：$xml_data = <
- 本文是将yolo3目标检测框架训练出来的ckpt文件固化成pb文件，主要利用了GitHub上的该项目。为什么要最终生成pb文件呢？简单来说就
- 在Python中，语法错误可以被Python解释器发现，但逻辑上错误或变量使用错误却不容易发现，如果结果没有符合预期，则需要调试，一个很好的
- 基础知识在关系型数据库中每一个数据表相当于一个文件，而不同的存储引擎则会构建出不同的表类型。存储引擎的作用是规定数据表如何存储数据，如何为存
- 与抓取预定义好的页面集合不同，抓取一个网站的所有内链会带来一个 挑战，即你不知道会获得什么。好在有几种基本的方法可以识别页面类型。通过URL
- 使用python的subprocess模块实现对SVN的相关操作。设置GitSvn类，在该类下自定义执行SVN常规操作的方法。SVN的常规操
- Pycharm中SQL语句提示SQL Dialect is Not Configured使用Pycharm进行数据库查询时，SQL语句经常提
- 过滤html代码的函数，当然也可以使用正则表达式。<％Function FilterHTML(strToFilter)&nb
- 从开始认识CSS(DW4)那时起,我就知道了CSS的强大,但从未用CSS排版过,因为我曾经尝试过学习,但感觉太难了而且用DW的表格,所见及所
- $r = new ReflectionClass($this); Zend_Debug::dump($r->getConstants(
- 前天不小心把硬盘格式化了，丢了好多照片，后来用Recuva这款软件成功把文件恢复过来，可是恢复的文件中有好多重复的文件和无法打开的图片，所以
- 1.项目目录及文件说明：manage.pydjango中的一个命令行工具，管理django项目；__init__.py空文件，告诉pytho
