亦余心之所善兮,虽九死其犹未悔。这篇文章主要讲述如何用C++自己实现mysql数据库的连接池?相关的知识,希望能为你提供帮助。
为什么是mysql?现在几乎所有的后台应用都要用到数据库,什么关系型的、非关系型的;正当关系的,不正当关系的;主流的和非主流的,大到Oracle,小到sqlite,以及包括现在逐渐流行的基于物联网的时序数据库,比如涛思的 TDengine,咱们中国人自己的开源时序数据库,性能杠杠滴。
凡此总总,即使没用过,也听说过,但大部分人或企业用的最多的就是白嫖型数据库:mysql。该数据库的特点就是无论是个人还是企业都能玩的起。像Oracle这种名媛型数据库基本就属于银行特供,银行需要花钱买平安,心里踏实。不买对的,只选贵的,因为人家确实不差钱。
如果你的后台应用连数据库都不需要,那跟咸鱼网站有什么区别呢?就是咸鱼二手网也要用到数据库的。如果一个IT民工一辈子没用过数据库就在35(~45)岁时“被退休”,那他的职业生涯是遗憾的,是不完美的,是不纯粹的。 好歹年轻是也要用一下非主流的Access吧,哪怕Execel也成。这种感觉就好比在大学时没谈过恋爱一样,光忙着羡慕别人就突然毕业了。
为什么要搞资源池?目前大部分后台程序都选择java开发或php,这两种语言的第三方库非常丰富,丰富到让开发人员的只要将精力放在具体业务上即可。比如数据库的资源池,只要选择好适当的jar包外加配置好相应的数据库参数,即可放心大胆的使用mysql。
当然,如果你命硬的话,也可以选择用C或C++开发后台应用。这时候你就需要自己DIY一个数据库资源池。
如果只是一个客户端程序,基本不需要连接池,但对于后台应用来说,高并发就意味着多线程,多线程程就意味着资源的竞争。内存访问如此,数据库访问也是如此。每次数据库的打开和关闭就是一次网络连接和关闭的过程,频繁的打开和关闭无疑会浪费大量的系统资源。这时候就需要提前建立好N个连接,并放在资源池中并提供给不同线程访问使用。
mysql资源池实现的案例源码【如何用C++自己实现mysql数据库的连接池()】我一直相信好的代码是不需要过的语言来解释的,代码即文档,要啥自行车。以下案例只是一个实现思路,供参考。
头文件:MysqlPool.h
#pragma warning(disable : 4786) #include <
windows.h>
#include <
winsock2.h>
#include <
mysql.h>
// 确保你的机器有mysql开发库
#include <
vector>
#include <
string>
using namespace std;
#define DEFAULT_POOL_SIZE20// 缺省mysql连接池中的数量
#define DEFAULT_POOL_TIMEOUT60// 获取池中mysql连接的超时// 自定义数据库查询回调函数
typedef BOOL (CALLBACK *LPFN_RetrieveRecordData)(MYSQL_ROW&
sqlRow, MYSQL_FIELD* pSqlFields, int iFieldCount, DWORD dwUserData);
// Mysql数据库连接类
class CMysqlConn
{
public:
CMysqlConn(const char* pszDBServer, UINT uDBPort, const char* pszDBName,
const char* pszDBUser, const char* pszDBPwd);
virtual ~CMysqlConn();
public:
// 打开/关闭一个mysql连接
BOOL Open();
void Close();
// ping连接是否已关闭
BOOL Ping();
// 重置字符集
BOOL ResetCharset();
public:
// ================SQL语句操作(简单实现几个)================
// 查询
BOOLSelect(const char* pszSql, LPFN_RetrieveRecordData lpfnRetrieveRecordData, DWORD dwUserData);
// 执行
BOOLExecute(const char* pszSql);
// 插入,如果主键是自增整型,返回插入后的主键值
__int64 Insert(const char* pszSql);
protected:
MYSQL*m_pMysql;
// mysql数据库操作对象// 以下是连接mysql需要的参数
stringm_strDBServer;
// mysql数据库所在服务器
UINTm_uDBPort;
// mysql数据库连接端口
stringm_strDBName;
// 数据库名称
stringm_strDBUser;
// 数据库账户
stringm_strDBPwd;
// 数据库密码};
// 数据库连接池实现
class CMysqlPool
{
public:
CMysqlPool();
virtual ~CMysqlPool();
// 创建mysql连接池
BOOL Create(const char* pszDBServer, UINT uDBPort, const char* pszDBName,
const char* pszDBUser, const char* pszDBPwd,
DWORD dwPoolSize = DEFAULT_POOL_SIZE,
DWORD dwTimeOut = DEFAULT_POOL_TIMEOUT);
// 销毁连接池
void Destroy();
public:// 获取一个mysql连接
CMysqlConn* Get();
// 释放一个mysql连接
void Release(CMysqlConn* pConn);
protected:
HANDLEm_hSemaphore;
// 信号量句柄
DWORDm_dwPoolSize;
// 连接池大小
DWORDm_dwTimeOut;
// 超时,单位秒
CRITICAL_SECTIONm_csPool;
// 连接池锁vector<
CMysqlConn*>
m_vecIdle;
// 闲队列
vector<
CMysqlConn*>
m_vecBusy;
// 忙队列
};
实现文件:MysqlPool.cpp
#include "stdafx.h"
#include "MysqlPool.h"
#include <
assert.h>
#include <
algorithm>
#pragma comment(lib, "libmysql.lib")//连接MysQL需要的库//////////////////////////////////////////////////////////////////////
// CMysqlConn: mysql数据库连接类
//////////////////////////////////////////////////////////////////////CMysqlConn::CMysqlConn(const char* pszDBServer, UINT uDBPort, const char* pszDBName,
const char* pszDBUser, const char* pszDBPwd)
{
assert(pszDBServer);
assert(pszDBName);
assert(pszDBUser);
assert(pszDBPwd);
m_pMysql = NULL;
m_strDBServer = pszDBServer;
m_uDBPort = uDBPort;
m_strDBName = pszDBName;
m_strDBUser = pszDBUser;
m_strDBPwd = pszDBPwd;
}CMysqlConn::~CMysqlConn()
{
Close();
}// 打开一个mysql数据库,即建立一个数据库连接
BOOL CMysqlConn::Open()
{
if(m_pMysql)
{
mysql_close(m_pMysql);
// 关闭连接
m_pMysql = NULL;
}m_pMysql = mysql_init(NULL);
if(!m_pMysql)
return FALSE;
// 连接数据库
if(!mysql_real_connect(m_pMysql, m_strDBServer.c_str(), m_strDBUser.c_str(),
m_strDBPwd.c_str(), m_strDBName.c_str(), m_uDBPort, NULL, 0))
{
int i = mysql_errno(m_pMysql);
const char * pszErr = mysql_error(m_pMysql);
return FALSE;
}// 设置重连
char chValue = https://www.songbingjia.com/android/1;
mysql_options(m_pMysql, MYSQL_OPT_RECONNECT, &
chValue);
mysql_query(m_pMysql,"set names \'gbk\'");
return TRUE;
}// 关闭数据库连接
void CMysqlConn::Close()
{
if(m_pMysql)
mysql_close(m_pMysql);
// 断开连接
m_pMysql = NULL;
}// ping一下mysql,看看连接还活着
BOOL CMysqlConn::Ping()
{
if(m_pMysql)
return (0 == mysql_ping(m_pMysql));
return FALSE;
}// 设置字符集为GBK
BOOL CMysqlConn::ResetCharset()
{
if(m_pMysql)
return (0 == mysql_query(m_pMysql, "set names \'gbk\'"));
return FALSE;
}// mysql执行:delete 或 update
BOOL CMysqlConn::Execute(const char* pszSql)
{
assert(pszSql);
if(!m_pMysql)
return FALSE;
MYSQL_STMT *myStmt = mysql_stmt_init(m_pMysql);
if(!myStmt)
{
return FALSE;
}if(0 != mysql_stmt_prepare(myStmt, pszSql, strlen(pszSql)))
{
mysql_stmt_close(myStmt);
return FALSE;
}
if(0 != mysql_stmt_execute(myStmt))
{
mysql_stmt_close(myStmt);
return FALSE;
}
mysql_stmt_close(myStmt);
return TRUE;
}// mysql插入
__int64 CMysqlConn::Insert(const char* pszSql)
{
assert(pszSql);
MYSQL_STMT *myStmt = mysql_stmt_init(m_pMysql);
if(!myStmt)
return 0;
if(0 != mysql_stmt_prepare(myStmt, pszSql, strlen(pszSql)))
{
int i = mysql_errno(m_pMysql);
const char * s = mysql_error(m_pMysql);
mysql_stmt_close(myStmt);
return 0;
}
if(0 != mysql_stmt_execute(myStmt))
{
mysql_stmt_close(myStmt);
return 0;
}
mysql_stmt_close(myStmt);
__int64 i64ID = mysql_insert_id(m_pMysql);
return i64ID;
}// mysql查询
BOOL CMysqlConn::Select(const char* pszSql, LPFN_RetrieveRecordData lpfnRetrieveRecordData, DWORD dwUserData)
{
if(!m_pMysql)
return FALSE;
if(NULL == lpfnRetrieveRecordData)
return FALSE;
if(0 != mysql_real_query(m_pMysql, pszSql, strlen(pszSql)))
{
return FALSE;
}MYSQL_RES *resRecord = mysql_store_result(m_pMysql);
int iFieldCount = resRecord->
field_count;
MYSQL_ROW sqlRow;
while (sqlRow = mysql_fetch_row(resRecord))
{
if(!lpfnRetrieveRecordData(sqlRow, resRecord->
fields, iFieldCount, dwUserData))
break;
}
mysql_free_result(resRecord);
return TRUE;
}//////////////////////////////////////////////////////////////////////
// CMysqlPool: mysql数据库连接池类
//////////////////////////////////////////////////////////////////////CMysqlPool::CMysqlPool()
{
::InitializeCriticalSection(&
m_csPool);
}CMysqlPool::~CMysqlPool()
{
Destroy();
::DeleteCriticalSection(&
m_csPool);
}// 创建mysql连接池
BOOL CMysqlPool::Create(const char* pszDBServer, UINT uDBPort, const char* pszDBName,
const char* pszDBUser, const char* pszDBPwd,
DWORD dwPoolSize, DWORD dwTimeOut)
{
m_dwTimeOut = dwTimeOut;
m_dwPoolSize = dwPoolSize;
// 创建信号量
m_hSemaphore = ::CreateSemaphore(NULL, dwPoolSize, dwPoolSize, NULL);
if (NULL == m_hSemaphore)
{
return FALSE;
}// 创建数据库连接池
for(DWORD i = 0;
i <
dwPoolSize;
++i)
{
// 创建一个mysql数据库连接
CMysqlConn *pConn = new CMysqlConn(pszDBServer, uDBPort, pszDBName, pszDBUser, pszDBPwd);
if(!pConn->
Open())
{
delete pConn;
continue;
}
m_vecIdle.push_back(pConn);
}return m_vecIdle.size() >
0;
}// 销毁mysql连接池
void CMysqlPool::Destroy()
{
::CloseHandle(m_hSemaphore);
m_hSemaphore = NULL;
// 释放idle队列
vector<
CMysqlConn*>
::iterator it;
for(it = m_vecIdle.begin();
it != m_vecIdle.end();
++it)
{
CMysqlConn* pConn =*it;
delete pConn;
}
m_vecIdle.clear();
// 释放busy队列
while(!m_vecBusy.empty())
{
CMysqlConn* pConn =m_vecBusy.back();
m_vecBusy.pop_back();
delete pConn;
}
}// 从mysql连接池获取一个连接
CMysqlConn* CMysqlPool::Get()
{
DWORD dwRet = ::WaitForSingleObject(m_hSemaphore, m_dwTimeOut*1000);
if (WAIT_OBJECT_0 != dwRet)// 超时,说明资源池没有可用mysql连接
{
printf("数据库没有可用连接。\\r\\n");
return NULL;
}// 从连接池中获取一个闲置连接
CMysqlConn* pConn = NULL;
::EnterCriticalSection(&
m_csPool);
if (!m_vecIdle.empty())
{
pConn = m_vecIdle.back();
// 移出idle队列
m_vecIdle.pop_back();
m_vecBusy.push_back(pConn);
// 加入busy队列
}
::LeaveCriticalSection(&
m_csPool);
if(NULL == pConn)
return NULL;
// 如果一个连接长时间无通信,可能被防火墙关闭,此时可以通过mysql_ping函数测试一下
// 本例中通过重新设置字符集
// 重新设置字符集,并判断数据库连接是否已断开
if(!pConn->
ResetCharset())
{
if(!pConn->
Open())
return NULL;
}printf("==》资源池:记得还我哦。\\r\\n");
return pConn;
}// 释放一个连接到mysql连接池
void CMysqlPool::Release(CMysqlConn* pConn)
{
if(NULL == pConn)
return;
// 释放一个信号量
::ReleaseSemaphore(m_hSemaphore, 1, NULL);
::EnterCriticalSection(&
m_csPool);
// 从Busy队列中释放该连接
vector<
CMysqlConn*>
::iterator it = find(m_vecBusy.begin(), m_vecBusy.end(), pConn);
if(it != m_vecBusy.end())
{
printf("POOL SIZE : %d, %d\\r\\n", m_vecIdle.size(), m_vecBusy.size());
m_vecBusy.erase(it);
// 移出busy队列
m_vecIdle.push_back(pConn);
// 加入idle队列
printf("POOL SIZE : %d, %d\\r\\n", m_vecIdle.size(), m_vecBusy.size());
}
::LeaveCriticalSection(&
m_csPool);
printf("《==资源池说:有借有还再借不难,常来玩啊。\\r\\n");
}
测试函数
void TestMysqlPool()
{
// 创建mysql连接资源池
CMysqlPool mysqlPool;
if(!mysqlPool.Create("127.0.0.1", 3306, "information_schema", "root", "123456"))
{
printf("Create mysql conneticon pool failed.\\r\\n");
return;
}// 从资源池中获取一个连接,连接池说:记得要还哦!
CMysqlConn* pConn = mysqlPool.Get();
// 假装做一次数据库操作
char* pszSQL ="SELECT * FROM CHARACTER_SETS";
pConn->
Select(pszSQL, RetrieveRecordData, 0);
// 将连接还给资源池并谢谢!连接池说:不客气!
mysqlPool.Release(pConn);
printf("Test over.\\r\\n");
}
输出打印
文章图片
推荐阅读
- Kubernetes + Spring Cloud 集成链路追踪 SkyWalking
- 超值分享为何写服务器程序需要自己管理内存,从改造std::string字符串操作说起。。。
- 一次 RocketMQ 顺序消费延迟的问题定位
- HBase Shell及其常用命令
- RocketMQ基础概念剖析&源码解析
- 简单了解一下K8S,并搭建自己的集群
- HBase概述与体系结构
- Zookeeper原理解析&使用场景详解
- 同步MySQL数据到ES神器mysqlmom介绍