91 lines
2.2 KiB
C
91 lines
2.2 KiB
C
|
#pragma once
|
|||
|
|
#ifndef CCONNECT_POOL_H_2021
|
|||
|
|
#define CCONNECT_POOL_H_2021
|
|||
|
|
|
|||
|
|
#include <stdio.h>
|
|||
|
|
#include <stdlib.h>
|
|||
|
|
#include <queue>
|
|||
|
|
#include <set>
|
|||
|
|
#include <pthread.h>
|
|||
|
|
#include <unistd.h>
|
|||
|
|
#include "mysql/mysql.h"
|
|||
|
|
|
|||
|
|
class CConnectPool
|
|||
|
|
{
|
|||
|
|
public:
|
|||
|
|
struct Connect_Info
|
|||
|
|
{
|
|||
|
|
int MinConnectCount;
|
|||
|
|
int MaxConnectCount;
|
|||
|
|
const char *Host;
|
|||
|
|
int Port;
|
|||
|
|
const char *Account;
|
|||
|
|
const char *Passwd;
|
|||
|
|
const char* KName;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
Connect_Info My_Info;
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
/**
|
|||
|
|
* @brief 创建连接池 \n
|
|||
|
|
* 在主线程中创建连接池，创建后需要调用FreePool予以释放
|
|||
|
|
* 创建失败程序会打印错误信息并终止运行
|
|||
|
|
*/
|
|||
|
|
static void
|
|||
|
|
CreatePool(int MinConnectCount, int MaxConnectCount, const char *Host, int Port, const char *Account, const char *Passwd);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 释放连接池 \n
|
|||
|
|
* 需要在主线程中调用该接口
|
|||
|
|
*/
|
|||
|
|
static void FreePool();
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 获取连接 \n
|
|||
|
|
* 支持多线程，使用完成后需调用PutConnect，归还连接，本函数为阻塞函数，直到有可用连接才调用返回
|
|||
|
|
* @return 返回数据库连接
|
|||
|
|
* --NULL表示调用失败
|
|||
|
|
*/
|
|||
|
|
static MYSQL *GetConnect();
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 获取连接 \n
|
|||
|
|
* 支持多线程，使用完成后需调用PutConnect，归还连接，本函数调用后立刻返回，若没有空闲连接则返回NULL
|
|||
|
|
* @return 返回数据库连接
|
|||
|
|
* --NULL表示调用失败
|
|||
|
|
* */
|
|||
|
|
static MYSQL *TryGetConnect();
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 向连接池归还连接 \n
|
|||
|
|
* 支持多线程
|
|||
|
|
* @param[in] pConn 需要归还的数据库连接
|
|||
|
|
* */
|
|||
|
|
static void PutConnect(MYSQL *pConn);
|
|||
|
|
|
|||
|
|
private:
|
|||
|
|
CConnectPool();
|
|||
|
|
~CConnectPool();
|
|||
|
|
MYSQL *_GetConnect(bool bTry);
|
|||
|
|
void _PutConnect(MYSQL *pConn);
|
|||
|
|
// 真正创建连接
|
|||
|
|
MYSQL *_CreateConnect();
|
|||
|
|
|
|||
|
|
static CConnectPool *s_pInstance;
|
|||
|
|
std::queue<MYSQL *> m_queueFree; // 空闲队列
|
|||
|
|
std::set<MYSQL *> m_setBusy; // 繁忙队列
|
|||
|
|
pthread_mutex_t m_mutex;
|
|||
|
|
pthread_cond_t m_cond;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
使用说明
|
|||
|
|
CConnectPool::CreatePool();
|
|||
|
|
MYSQL *sql = CConnectPool::GetConnect();
|
|||
|
|
CConnectPool::PutConnect(sql);
|
|||
|
|
CConnectPool::FreePool();
|
|||
|
|
*/
|