linux mysql 连接池

/*=============================================================================
#     FileName: Lock.h
#         Desc: delare Lock Class
#       Author: Pugna
#        Email: 860009944@qq.com
#     HomePage: http://www.cnblogs.com/pugna/
#      Version: 0.0.1
#   LastChange: 2014-10-21 05:11:54
#      History:
=============================================================================*/
#ifndef _LOCAL_LOCK_H_
#define _LOCAL_LOCK_H_

#include <pthread.h>

class RWLock{
    public:
        RWLock(){pthread_rwlock_init(&rwlock,NULL);}
        ~RWLock(){pthread_rwlock_destroy(&rwlock);}
        void Lock(){pthread_rwlock_rdlock(&rwlock);}
        void Unlock(){pthread_rwlock_wrlock(&rwlock);}
    private:
        pthread_rwlock_t rwlock;
};

class MutexLock{
    public:
        MutexLock(){pthread_mutex_init(&lock,NULL);}
        ~MutexLock(){pthread_mutex_destroy(&lock);}
        void Lock(){pthread_mutex_lock(&lock);}
        void Unlock(){pthread_mutex_unlock(&lock);}
    private:
        pthread_mutex_t lock;
};

template<class ClockObj>class LocalLock{
    public:
        LocalLock(ClockObj& obj):m_lock(obj){m_lock.Lock();}
        ~LocalLock(){m_lock.Unlock();}
    private:
        ClockObj & m_lock;
};

typedef LocalLock<RWLock> LocalRWLock;
typedef LocalLock<MutexLock> LocalMutexLock;

#endif

/*=============================================================================
#     FileName: mysqlconnpool.h
#         Desc: declare mysqlconnpool class
#       Author: Pugna
#        Email: 860009944@qq.com
#     HomePage: http://www.cnblogs.com/pugna/
#      Version: 0.0.1
#   LastChange: 2014-10-21 15:14:31
#      History:
=============================================================================*/
#ifndef _MYSQLCONN_POOL_H
#define _MYSQLCONN_POOL_H

#include <mysql/mysql.h>
#include <pthread.h>
#include <string>
#include <queue>
#include "../common/Lock.h"

using namespace std;

class mysqlconnpool{
    public:
        mysqlconnpool(string url,int port,string user,string pwd,string db,string encode,int minSize=10,int maxSize=100);
        ~mysqlconnpool();
    
        bool Init();
        void Uninit();

        
        MYSQL * CreateConn();
        bool DeleteConn(MYSQL *conn);
        bool PopConn(MYSQL * &conn);
        bool PushConn(MYSQL * &conn);
        void getCurSize(int & n);
    private:
        int maxSize;
        int curSize;
        int minSize;
        string username;
        string password;
        string url;
        int port;
        string db;
        string encode;
        MutexLock  mutexLock;
        queue<MYSQL*>connQueue;
};

template<class ConnPool>class MysqlConn{
    public:
        MysqlConn(ConnPool & obj):conn(obj){flag=conn.PopConn(pMysql)?true:false;}
        ~MysqlConn(){if(flag)conn.PushConn(pMysql);}
    private:
        bool flag;
        ConnPool & conn;
    public:
        MYSQL * pMysql;
};

typedef MysqlConn<mysqlconnpool> LocalMysqlConn;

#endif

/*=============================================================================
#     FileName: mysqlconnpool.cpp
#         Desc: Functions of MysqlConnPool
#       Author: Pugna
#        Email: 860009944@qq.com
#     HomePage: http://www.cnblogs.com/pugna/
#      Version: 0.0.1
#   LastChange: 2014-10-21 19:48:53
#      History:
=============================================================================*/
#include <stdexcept>
#include <exception>
#include <stdio.h>
#include "mysqlconnpool.h"
#include "../common/Lock.h"


mysqlconnpool::mysqlconnpool(string url,int port,string username,string pwd,string db,string encode,int minSize,int maxSize):
    url(url)
    ,port(port)
    ,maxSize(maxSize)
    ,minSize(minSize)
    ,curSize(0)
    ,username(username)
    ,password(pwd)
    ,db(db)
    ,encode(encode)
{
}

mysqlconnpool::~mysqlconnpool(){
}

bool mysqlconnpool::Init(){
    LocalMutexLock lock(mutexLock);
    if(minSize<=0)
        return false;
    for(int i=0;i<minSize;i++){
        MYSQL * pMysql=CreateConn();
        if(pMysql==NULL)
            return false;
        connQueue.push(pMysql);
        curSize++;
    }
    if(curSize==minSize)
        return true;
    else
        return false;
}    

void mysqlconnpool::Uninit(){
    LocalMutexLock lock(mutexLock);
    if(!connQueue.empty()){
        do{
            MYSQL * pMysql=connQueue.front();
            mysql_close(pMysql);
            connQueue.pop();
        }while(!connQueue.empty());
    }
}

MYSQL * mysqlconnpool::CreateConn(){
    MYSQL * pMysql=mysql_init(NULL);
    if(pMysql==NULL){
        return NULL;
    }
    char value=1;
    if(0==mysql_options(pMysql,MYSQL_OPT_RECONNECT,&value)){
        if(mysql_real_connect(pMysql , url.c_str(), username.c_str(), password.c_str(), db.c_str(), port, NULL, 0)){
            if(0==encode.length()&&!mysql_set_character_set(pMysql,encode.c_str())){
                printf("set character false\r\n");
            }
            else{
                return pMysql;
            }
        }
        else{
            printf("connect error\r\n");
        }
    }
    else{
        printf("optione error\r\n");
    }
    mysql_close(pMysql);
    return NULL;
}

bool mysqlconnpool::PushConn(MYSQL * &conn){
    LocalMutexLock lock(mutexLock);
    if(curSize>=maxSize)
    {
        mysql_close(conn);
        conn=NULL;
        return false;
    }
    connQueue.push(conn);
    curSize++;
    return true;
}

bool mysqlconnpool::PopConn(MYSQL * &conn){
    {
        LocalMutexLock lock(mutexLock);
        if(curSize>0){
            conn=connQueue.front();
            connQueue.pop();
            curSize--;
            return true;
        }
    }
    conn=CreateConn();
    if(NULL!=conn)
        return true;
    else
        return false;
}

void mysqlconnpool::getCurSize(int & n){
    LocalMutexLock lock(mutexLock);
    n=curSize;
}

/*=============================================================================
#     FileName: client.cpp
#         Desc: test mysqlconnpool
#       Author: Pugna
#        Email: 860009944@qq.com
#     HomePage: http://www.cnblogs.com/pugna/
#      Version: 0.0.1
#   LastChange: 2014-10-21 22:13:35
#      History:
=============================================================================*/
#include <stdio.h>
#include <pthread.h>
#include "../mysqlconnpool/mysqlconnpool.h"

static const string url="127.0.0.1";
static const string username="root";
static const string password="123456";
static const string dbname="test";
static const int port=3306;
static const string encodetype="";

static const int threadNum=100;


mysqlconnpool conn(url,port,username,password,dbname,encodetype,10,1000);

void * TestThread(void * LpVoid){
    LocalMysqlConn connPool(conn);
    if(NULL==connPool.pMysql){
        printf("thread:%p,conn null\r\n");
        connPool.pMysql=conn.CreateConn();
        if(NULL==connPool.pMysql){
            printf("thread:%p,no mysql\r\n");
            pthread_exit(NULL);
        }
        else{
            printf("thread:%p,get new mysql\r\n");
        }
    }
    else{
        printf("mysql:%p,thread:%p\r\n",connPool.pMysql,pthread_self());
        if(0==mysql_query(connPool.pMysql,"update user set point=point+1 where name=\‘lantun\‘;")){
            printf("thread:%p,query ok!\r\n",pthread_self());
        }
        else{
            printf(mysql_error(connPool.pMysql));
        }
    }
    pthread_exit(NULL);
}

int main(){
    pthread_t threadID[threadNum];

    conn.Init();
    int tdCreated=0;
    for(int i=0;i<threadNum;i++){
        if(0!=pthread_create(&threadID[i],NULL,TestThread,NULL)){
            printf("create thread false\r\n");
            break;
        }
        else
            tdCreated++;
    }
    //wate for all threads exit.
    for(int i=0;i<tdCreated;i++){
        pthread_join(threadID[i],NULL);
    }
    int mysqlNum=0;
    conn.getCurSize(mysqlNum);
    printf("mysql Num:%d\r\n",mysqlNum);
    conn.Uninit();
    return 0;
}

#=============================================================================
#     FileName: Makefile
#         Desc: :
#       Author: Pugna
#        Email: 860009944@qq.com
#     HomePage: http://www.cnblogs.com/pugna/
#      Version: 0.0.1
#   LastChange: 2014-10-21 17:00:26
#      History:
#=============================================================================
all: 
    g++ -g ../mysqlconnpool/*.* *.cpp -L /usr/lib64/mysql/ -lmysqlclient_r -lpthread -o client.o
clean:
    rm *.o