03 连接池实现

1. 连接池对象

class ConnectionPool
{
public:
	// 获取连接池对象实例   
	static ConnectionPool* getConnectionPool();
	//给外部提供接口，从连接池中获取一个可用的空闲连接  消费者线程   智能指针出作用域自动析构 我们重新定义一下智能指针的的析构方式 设置为 连接返回队列
	std::shared_ptr<Connection> getConnection();

private:
	ConnectionPool(); // 单例#1 构造函数私有化

	bool loadConfigFile();// 从配置文件中加载配置项

	//运行在独立的线程中，专门负责生产新连接
	void produceConnectionTask();

	//启动一个新的定时线程，扫描超过maxIdleTime时间的空闲连接，进行对于的连接回收
	void scannerConnectionTask();

	string _ip; // mysql的ip地址
	unsigned short _port; // mysql的端口号 3306
	string _username; // mysql登录用户名
	string _password; // mysql登录密码
	string _dbname; // 连接的数据库名称
	int _initSize; // 连接池的初始连接量
	int _maxSize; // 连接池的最大连接量
	int _maxIdleTime; // 连接池最大空闲时间
	int _connectionTimeout; // 连接池获取连接的超时时间

	queue<Connection*> _connectionQue; // 存储mysql连接的队列
	mutex _queueMutex; // 维护连接队列的线程安全互斥锁
	atomic_int _connectionCnt; // 记录连接所创建的connection连接的总数量 
	condition_variable cv; // 设置条件变量，用于连接生产线程和连接消费线程的通信
};

2. 创建懒汉式的单例函数接口

        第一次调用的时候进行ConntionPool 的实例化 pool

// 线程安全的懒汉单例函数接口 
ConnectionPool* ConnectionPool::getConnectionPool()
{
	static ConnectionPool pool; // lock和unlock 静态的初始化第一次运行才会初始化 线程安全
	return &pool;
}

3. 从配置文件中加载配置项

新建.ini文件

#数据库连接池的配置文件
ip=127.0.0.1
port=3306
username=root
password=1234
dbname=chat
initSize=10
maxSize=1024
#最大空闲时间默认单位是秒
maxIdleTime=60
#连接超时时间单位是毫秒
connectionTimeOut=100

// 从配置文件中加载配置项
bool ConnectionPool::loadConfigFile()
{
	FILE* pf = fopen("mysql.ini", "r");
	if (pf == nullptr)
	{
		LOG("mysql.ini file is not exist!"); //日志信息
		return false;
	}
	while (!feof(pf)) //feof 用于查询文件指针是否以达到末尾
	{
		char line[1024] = { 0 };
		fgets(line, 1024, pf);
		string str = line;

		//找等号
		int idx = str.find('=', 0);
		if (idx == -1) {	//无效配置项
			continue;
		}

		//找回车
		//password=1234\n
		int endidx = str.find('\n', idx);
		//解析文本内容
		string key = str.substr(0, idx);
		string value = str.substr(idx + 1, endidx - idx - 1);

		if (key == "ip")
		{
			_ip = value;
		}
		else if (key == "port")
		{
			_port = atoi(value.c_str());
		}
		else if (key == "username")
		{
			_username = value;
		}
		else if (key == "password")
		{
			_password = value;
		}
		else if (key == "dbname")
		{
			_dbname = value;
		}
		else if (key == "initSize")
		{
			_initSize = atoi(value.c_str());
		}
		else if (key == "maxSize")
		{
			_maxSize = atoi(value.c_str());
		}
		else if (key == "maxIdleTime")
		{
			_maxIdleTime = atoi(value.c_str());
		}
		else if (key == "connectionTimeOut")
		{
			_connectionTimeout = atoi(value.c_str());
		}
	}
	return true;
}

 4. 连接池构造函数

        在这里要完成连接池中_initSize数量的连接创建，

        以及启动一个新的线程进行生产新的连接，

        启动一个线程定时线程，扫描超过maxIdleTime时间的空闲连接，进行对于的连接回收

//连接池的构造函数
ConnectionPool::ConnectionPool() {
	// 加载配置项了
	if (!loadConfigFile())
	{
		return;
	}
	// 创建初始数量的连接
	for (int i = 0; i < _initSize; ++i)
	{
		Connection* p = new Connection();
		p->connect(_ip, _port, _username, _password, _dbname);
		p->refreshAliveTime();	// 刷新一下开始空闲的起始时间
		_connectionQue.push(p); 
		_connectionCnt++; 
	}
	// 绑定器
	// 启动一个新的线程，作为连接的生产者 linux thread => pthread_create 
	thread produce(std::bind(&ConnectionPool::produceConnectionTask, this));
	produce.detach();//分离线程

	// 启动一个新的定时线程，扫描超过maxIdleTime时间的空闲连接，进行对于的连接回收
	thread scanner(std::bind(&ConnectionPool::scannerConnectionTask, this));
	scanner.detach();//分离线程
}