recode the program

Author: white0dew

CGImysql/README.md

@@ -0,0 +1,13 @@
+
+校验 & 数据库连接池
+===============
+数据库连接池
+> * 单例模式，保证唯一
+> * list实现连接池
+> * 连接池为静态大小
+> * 互斥锁实现线程安全
+
+校验  
+> * HTTP请求采用POST方式
+> * 登录用户名和密码校验
+> * 用户注册及多线程注册安全

CGImysql/sql_connection_pool.cpp

@@ -0,0 +1,143 @@
+#include <mysql/mysql.h>
+#include <stdio.h>
+#include <string>
+#include <string.h>
+#include <stdlib.h>
+#include <list>
+#include <pthread.h>
+#include <iostream>
+#include "sql_connection_pool.h"
+
+using namespace std;
+
+connection_pool::connection_pool()
+{
+	m_CurConn = 0;
+	m_FreeConn = 0;
+}
+
+connection_pool *connection_pool::GetInstance()
+{
+	static connection_pool connPool;
+	return &connPool;
+}
+
+//构造初始化
+void connection_pool::init(string url, string User, string PassWord, string DBName, int Port, int MaxConn, int close_log)
+{
+	m_url = url;
+	m_Port = Port;
+	m_User = User;
+	m_PassWord = PassWord;
+	m_DatabaseName = DBName;
+	m_close_log = close_log;
+
+	for (int i = 0; i < MaxConn; i++)
+	{
+		MYSQL *con = NULL;
+		con = mysql_init(con);
+
+		if (con == NULL)
+		{
+			LOG_ERROR("MySQL Error");
+			exit(1);
+		}
+		con = mysql_real_connect(con, url.c_str(), User.c_str(), PassWord.c_str(), DBName.c_str(), Port, NULL, 0);
+
+		if (con == NULL)
+		{
+			LOG_ERROR("MySQL Error");
+			exit(1);
+		}
+		connList.push_back(con);
+		++m_FreeConn;
+	}
+
+	reserve = sem(m_FreeConn);
+
+	m_MaxConn = m_FreeConn;
+}
+
+
+//当有请求时，从数据库连接池中返回一个可用连接，更新使用和空闲连接数
+MYSQL *connection_pool::GetConnection()
+{
+	MYSQL *con = NULL;
+
+	if (0 == connList.size())
+		return NULL;
+
+	reserve.wait();
+	
+	lock.lock();
+
+	con = connList.front();
+	connList.pop_front();
+
+	--m_FreeConn;
+	++m_CurConn;
+
+	lock.unlock();
+	return con;
+}
+
+//释放当前使用的连接
+bool connection_pool::ReleaseConnection(MYSQL *con)
+{
+	if (NULL == con)
+		return false;
+
+	lock.lock();
+
+	connList.push_back(con);
+	++m_FreeConn;
+	--m_CurConn;
+
+	lock.unlock();
+
+	reserve.post();
+	return true;
+}
+
+//销毁数据库连接池
+void connection_pool::DestroyPool()
+{
+
+	lock.lock();
+	if (connList.size() > 0)
+	{
+		list<MYSQL *>::iterator it;
+		for (it = connList.begin(); it != connList.end(); ++it)
+		{
+			MYSQL *con = *it;
+			mysql_close(con);
+		}
+		m_CurConn = 0;
+		m_FreeConn = 0;
+		connList.clear();
+	}
+
+	lock.unlock();
+}
+
+//当前空闲的连接数
+int connection_pool::GetFreeConn()
+{
+	return this->m_FreeConn;
+}
+
+connection_pool::~connection_pool()
+{
+	DestroyPool();
+}
+
+connectionRAII::connectionRAII(MYSQL **SQL, connection_pool *connPool){
+	*SQL = connPool->GetConnection();
+	
+	conRAII = *SQL;
+	poolRAII = connPool;
+}
+
+connectionRAII::~connectionRAII(){
+	poolRAII->ReleaseConnection(conRAII);
+}

CGImysql/sql_connection_pool.h

@@ -0,0 +1,63 @@
+#ifndef _CONNECTION_POOL_
+#define _CONNECTION_POOL_
+
+#include <stdio.h>
+#include <list>
+#include <mysql/mysql.h>
+#include <error.h>
+#include <string.h>
+#include <iostream>
+#include <string>
+#include "../lock/locker.h"
+#include "../log/log.h"
+
+using namespace std;
+
+class connection_pool
+{
+public:
+	MYSQL *GetConnection();				 //获取数据库连接
+	bool ReleaseConnection(MYSQL *conn); //释放连接
+	int GetFreeConn();					 //获取连接
+	void DestroyPool();					 //销毁所有连接
+
+	//单例模式 这个是利用局部静态变量懒汉模式实现单例（多线程不友好）
+	//对于多线程而言，多个线程可能同时访问到该静态变量，并发现其没有被初始化（C++实现机制是）
+	//（该看静态变量内部一标志位是为1，为1则说明已被初始化）
+	//多个线程同时判定其没有初始化而后均初始化就会造成错误（即它不是一个原子操作）
+	static connection_pool *GetInstance();
+
+	void init(string url, string User, string PassWord, string DataBaseName, int Port, int MaxConn, int close_log); 
+
+private:
+	connection_pool();
+	~connection_pool();
+
+	int m_MaxConn;  //最大连接数
+	int m_CurConn;  //当前已使用的连接数
+	int m_FreeConn; //当前空闲的连接数
+	locker lock;
+	list<MYSQL *> connList; //连接池
+	sem reserve;
+
+public:
+	string m_url;			 //主机地址
+	string m_Port;		 //数据库端口号
+	string m_User;		 //登陆数据库用户名
+	string m_PassWord;	 //登陆数据库密码
+	string m_DatabaseName; //使用数据库名
+	int m_close_log;	//日志开关
+};
+
+class connectionRAII{
+
+public:
+	connectionRAII(MYSQL **con, connection_pool *connPool);
+	~connectionRAII();
+	
+private:
+	MYSQL *conRAII;
+	connection_pool *poolRAII;
+};
+
+#endif