Python学习一：基于DBUtils和pymysql的MySQL数据库连接池封装

前言

Python安装及配置略，本文前默认已经配置好Python环境。
打开cmd，输入python，出现以下画面表示Python安装成功。

需要库

1、DBUtils 下载地址：DBUtils
2、pymysql 下载地址：pymysql
说明：DBUtils 暂时不支持Python3
下载解压后，cmd进入库根目录，执行 ：python setup.py install 进行安装。

另外可以使用pip命令在线安装，打开cmd，输入 ：pip install DBUtils，即可进行自动安装。

说明：

DBUtils是一套Python数据库连接池包，并允许对非线程安全的数据库接口进行线程安全包装。DBUtils来自Webware for Python。

DBUtils提供两种外部接口：
* PersistentDB ：提供线程专用的数据库连接，并自动管理连接。
* PooledDB ：提供线程间可共享的数据库连接，并自动管理连接。

实现

在实际数据库操作中，使用完毕后需要关闭游标（cursor）和连接（connection），为了简化代码，获取数据库连接采用 With open as …方式，执行完毕后自动关闭连接， 无需主动关闭。

1、数据库参数配置
DB_config.py

#-*- coding: UTF-8 -*-  
'''
@描述：数据库配置信息
@作者：CYH
@版本：V1.0
@创建时间：2016-11-24 上午9:23:06
'''

#TEST数据库信息
DB_TEST_HOST="127.0.0.1";
DB_TEST_PORT=3306;
DB_TEST_DBNAME="sy";
DB_TEST_USER="root";
DB_TEST_PASSWORD="123456";


#数据库连接编码
DB_CHARSET="utf8";

#mincached : 启动时开启的闲置连接数量(缺省值 0 以为着开始时不创建连接)
DB_MIN_CACHED=10;

#maxcached : 连接池中允许的闲置的最多连接数量(缺省值 0 代表不闲置连接池大小)
DB_MAX_CACHED=10;

#maxshared : 共享连接数允许的最大数量(缺省值 0 代表所有连接都是专用的)如果达到了最大数量,被请求为共享的连接将会被共享使用
DB_MAX_SHARED=20;

#maxconnecyions : 创建连接池的最大数量(缺省值 0 代表不限制)
DB_MAX_CONNECYIONS=100;

#blocking : 设置在连接池达到最大数量时的行为(缺省值 0 或 False 代表返回一个错误; 其他代表阻塞直到连接数减少,连接被分配)
DB_BLOCKING=True;

#maxusage : 单个连接的最大允许复用次数(缺省值 0 或 False 代表不限制的复用).当达到最大数时,连接会自动重新连接(关闭和重新打开)
DB_MAX_USAGE=0;

#setsession : 一个可选的SQL命令列表用于准备每个会话，如["set datestyle to german", ...]
DB_SET_SESSION=None;

2、数据库连接创建、连接获取

# -*- coding: UTF-8 -*-
"""
@描述：数据库连接池管理模块
@作者：CYH
@版本：V1.0
@创建时间：2016-11-24 上午8:43:14
"""

import pymysql;
from DBUtils.PooledDB import PooledDB;

import DB_config as Config;

'''
@功能：PT数据库连接池
'''   
class PTConnectionPool(object):
    __pool = None;
    def __enter__(self):
        self.conn = self.__getConn();
        self.cursor = self.conn.cursor();
        print "PT数据库创建con和cursor";
        return self;

    def getConn(self):
        if self.__pool is None:
            self.__pool = PooledDB(creator=pymysql, mincached=Config.DB_MIN_CACHED , maxcached=Config.DB_MAX_CACHED, 
                                   maxshared=Config.DB_MAX_SHARED, maxconnections=Config.DB_MAX_CONNECYIONS, 
                                   blocking=Config.DB_BLOCKING, maxusage=Config.DB_MAX_USAGE, 
                                   setsession=Config.DB_SET_SESSION,
                                   host=Config.DB_TEST_HOST , port=Config.DB_TEST_PORT , 
                                   user=Config.DB_TEST_USER , passwd=Config.DB_TEST_PASSWORD ,
                                   db=Config.DB_TEST_DBNAME , use_unicode=False, charset=Config.DB_CHARSET);

        return self.__pool.connection()

    """
    @summary: 释放连接池资源
    """
    def __exit__(self, type, value, trace):
        self.cursor.close()
        self.conn.close()

        print "PT连接池释放con和cursor";

'''
@功能：获取PT数据库连接
'''  
def getPTConnection():
    return PTConnectionPool();

3、获取连接操作数据库

#-*- coding: UTF-8 -*-  
'''
@描述：
@作者：CYH
@版本：V1.0
@创建时间：2016-11-24 上午9:34:54
'''
from DB_connetion_pool import getPTConnection, PTConnectionPool;

def TestMySQL():  
    #申请资源  
    with getPTConnection() as db:
        # SQL 查询语句;
        sql = "SELECT * FROM sy_user";
        try:
            # 获取所有记录列表
            db.cursor.execute(sql)
            results = db.cursor.fetchall();
            for row in results:
                userId = row[0]
                name= row[1]         
                sex= row[2]
                createTime = row[3]
                # 打印结果
                print ("userId=%d,name=%s,sex=%s,createTime=%s" %\
                     (userId, name, sex, createTime ))
        except:
            print ("Error: unable to fecth data")

TestMySQL()

总结

采用DBUtils和pymyql实现数据库连接池操作，使用Python内置的With open as db 实现数据库连接使用完后自动关闭。

说明：本文提供内容及代码仅作为学习、测试使用，未做详细性能测试和评估！