跳至主要內容

加载配置项

张威大约 2 分钟mysqlmysql连接池

加载配置项

#ifndef _CONNECTIONPOOL_H
#define _CONNECTIONPOOL_H

#include <string>
#include <queue>
#include <mutex>
#include <atomic>   //atomic_int  原子类型
#include "connection.h"

using std::string;
using std::queue;
using std::mutex;

class ConnectionPool {
public:
    static ConnectionPool* getConnectionPool();
private:
    ConnectionPool();
    ~ConnectionPool();
public: //测试的时候可以先变成共有的
    bool loadConfigFile();

private:
	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;  // 维护连接队列的线程安全互斥锁
    std::atomic_int _connectionCnt; // 记录连接所创建的connection连接的总数量 
};

#endif //_CONNECTIONPOOL_H
#include "connectionPool.h"
#include <iostream>
#include "public.h"

using std::cout;
using std::endl;


// 线程安全的懒汉单例函数接口
ConnectionPool* ConnectionPool::getConnectionPool() { //静态函数的实现,不写static
    static ConnectionPool pool; //静态局部变量的初始化,编译器会生成lock和unlock
    return &pool;
}
// 从配置文件中加载配置项
bool ConnectionPool::loadConfigFile() {
    FILE* pf = fopen("mysql.cnf","r");
    if(pf == nullptr) {
        LOG("mysql.cnf file is not exit!");
        return false;
    }

    while(!feof(pf)) {  //文件不为空
        char line[1024] = {0};
        fgets(line, sizeof(line), pf);
        string str = line;
        int idx = str.find('=', 0); //从第0位开始找'='的下标
        if(idx == -1) {  //无效的配置项
            continue;
        }

        // password=123456\n
        int endidx = str.find('\n', idx); //从第idx位开始找'\n'的下标

        string key = str.substr(0,idx); //从第0位开始idx个字符
        string value = str.substr(idx + 1, endidx - (idx + 1));
        //cout << key << '=' << value << endl;
        if(key == "ip") {
            _ip = value;
        }
        else if(key == "port") {
            _port = atoi(value.c_str());    //string->const char* ->int
        }
        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;
}

ConnectionPool::ConnectionPool() {
    if(!loadConfigFile()) { //配置文件加载失败
        return;
    }

    //创建初始的数量连接
    //这一块是在连接池启动的时候做的,不用考虑线程安全
    for(int i = 0; i < _initSize; ++i) {
        Connection* conn = new Connection();
        conn->connection(_ip, _port, _username, _password, _dbname);
        _connectionQue.push(conn);
        _connectionCnt++;
    }

    //启动一个新的线程作为生产者线程

}
ConnectionPool::~ConnectionPool() {}