【服务器编程】多线程安全数据库连接池

某风平浪静的晚上，我和多线程搅上劲，如何在多线程环境安全创建连接池、释放、使用，想了很久，写了很久，写出了如下代码，感觉也很多bug吧。

发表此文，谨表示我今晚不知道为什么如此 较真，很想做到完美，可是能力还是不到那个高度，写不出很完美的代码。

如果有想法的同志路过，留点提示给我，谢谢谢谢。

/* 析构不知道怎么在多线程保证安全，程序员责任，我的脑袋要炸了 */
/* 销毁由一个线程完成，而创建和使用是线程安全的连接池 */
class MYSQLConnPool
{
public:
	typedef enum {
		E_MALLOC, 	/* 分配线程池时失败 */
		E_SUM_TO_MAX,	/* 连接池的连接数过大 */
		E_EXIST,		/* 连接池已存在 */
		E_NO_SOURCE,	/* 没有空闲连接 */
		E_INIT_SEM,	/* 初始化信号量错误（也许不该这样） */
		NONERR
	}ERRNO;
public:
	/* Singleton模式*/
	static MYSQLConnPool* CreateConnPool(const char *host, const char *user, const char *psw, const char *db, int conn_num)
	{
		
		pthread_mutex_lock(&m_MutexCreat);

		/* 连接数太大 */
		if (conn_num > MAX_CONN_NUM)
		{
			pthread_mutex_lock(&m_MutexMap);
			m_ErrMap[pthread_self()] = E_SUM_TO_MAX;
			pthread_mutex_unlock(&m_MutexMap);
			pthread_mutex_unlock(&m_MutexCreat);
			return nullptr;
		}
		if (m_ConnPool == nullptr)
		{
			m_ConnPool = new MYSQLConnPool(host, user, psw, db, conn_num);
			if (m_ErrMap[pthread_self()] == E_INIT_SEM)
			{/* 初始化信号量时出错 */
				delete m_ConnPool;
				pthread_mutex_unlock(&m_MutexCreat);
				return nullptr;
			}
			if (m_ConnPool)
			{/* 建立连接池成功 */
				pthread_mutex_unlock(&m_MutexCreat);
				return m_ConnPool;
			}
			/* new失败 */
			pthread_mutex_lock(&m_MutexMap);
			m_ErrMap[pthread_self()] = E_MALLOC;
			pthread_mutex_unlock(&m_MutexMap);
			pthread_mutex_unlock(&m_MutexCreat);
			return nullptr;
		}
		/* 该进程已经生成连接池 */
		pthread_mutex_lock(&m_MutexMap);
		m_ErrMap[pthread_self()] = E_EXIST;
		pthread_mutex_unlock(&m_MutexMap);
		pthread_mutex_unlock(&m_MutexCreat);
		return nullptr;

	}

	/*
	*	试想：一个监听主线程有几个服务线程，而每个服务线程（不同的服务封装了一个类）都想用到这个连接池（一个进程一个连接池），
	*	如果又上层（主线程）传到参数（连接池）到下层（服务线程）方式，那么太破坏服务类的封装了，故定义一个返回池指针的函数。
	*	（针对本开发的项目）
	*/
	static MYSQLConnPool* GetConnPoolPointer()
	{
		return m_ConnPool;
	}

	static ERRNO GetLastError()
	{
		pthread_mutex_lock(&m_MutexMap);
		std::map::iterator ite;
		ite = map.find(pthread_self());
		if (ite == map.end())
		{
			pthread_mutex_unlock(&m_MutexMap);
			return NONERR;
		}
		ERRNO e = m_ErrMap[pthread_self()];
		pthread_mutex_unlock(&m_MutexMap);
		return e;
	}

	MYSQL *GetConnection()
	{/* 连接池没有空闲连接就阻塞 */	
		sem_wait(&m_ConnNum);
		pthread_mutex_lock(&m_MutexConnVec);
		MYSQL *conn = m_ConnVec.back();
		m_ConnVec.pop_back();
		pthread_mutex_unlock(&m_MutexConnVec);
		return conn;
	}

	void RecoverConnection(MYSQL *conn)
	{
		if (conn)
		{
			pthread_mutex_lock(&m_MutexConnVec);
			m_ConnVec.push_back(conn);
			pthread_mutex_unlock(&m_MutexConnVec);
			sem_post(&m_ConnNum);
		}
	}

	/* 目前没想到如何做到在多线程环境析构而绝对不冲突 */
	/* 虚函数不是原子操作，在多线程环境有可能在析构同时，另外一个线程在使用连接池 */
	~MYSQLConnPool()
	{
		/* 回收所有资源（连接）后，销毁信号量，就不可能再有线程非法操作了 */
		for (int i = 0; i < m_count; ++i)
		{/* 所以，使用了一定要归还啊，不然死的很惨+_+ */
			sem_wait(&m_ConnNum);
		}
		sem_destroy(&m_ConnNum);
		pthread_mutex_destroy(&m_MutexConnVec);
		for (int i = 0; i < m_ConnNum; ++i)
		{
			mysql_close(m_ConnVec[i]);
		}
		m_ConnPool = nullptr;
	}

private:
	MYSQLConnPool(const char *host, const char *user, const char *psw, const char *db, int conn_num);

private:
	/* 根据线程存储了已发生什么错误 */
	pthread_mutex_t m_MutexMap;
	std::map m_ErrMap;

	/* 连接池最大连接数 */
	static const int MAX_CONN_NUM = 16;

	/* 连接池的连接数 */
	int m_count;
	/* 可用资源的信号量表示 */
	sem_t m_ConnNum;
	pthread_mutex_t m_MutexConnVec;

	/* 连接列表 */
	std::vector m_ConnVec;

	/* 多个线程一起创建时，使用互斥防止创建多个 */
	static pthread_mutex_t m_MutexCreat;

	/* 实例指针*/
	static MYSQLConnPool *m_ConnPool;
};

MYSQLConnPool* MYSQLConnPool::m_ConnPool = nullptr;

pthread_mutex_t MYSQLConnPool::m_MutexCreat = PTHREAD_MUTEX_INITIALIZER;


MYSQLConnPool::MYSQLConnPool(const char *host, const char *user, const char *psw, const char *db, int conn_num)
{
	m_count = 0;
	for (int i = 0; i < conn_num; ++i)
	{
		MYSQL *conn = mysql_init(nullptr);
		if (!conn)
			continue;
		conn = mysql_real_connect(conn, host, user, psw, db, 0, nullptr, 0);
		if (!conn)
			continue;
		m_count++;
		m_ConnVec.push_back(conn);
	}
	m_ErrMap[pthread_self()] = NONERR;
	int ret = sem_init(&m_ConnNum, 0, m_ConnVec.size());
	if (ret == -1)
	{
		pthread_mutex_lock(&m_MutexMap);
		m_ErrMap[pthread_self()] = E_INIT_SEM;
		pthread_mutex_unlock(&m_MutexMap);
	}
	m_MutexConnVec = PTHREAD_MUTEX_INITIALIZER;
}