注：笔者的Django版本为 1.9，Python版本为 2.7

一直想在Django中实现MySQL连接池，在网上寻找良久无果，偶然之间，看到了一个帖子"让Django支持数据库长连接(可以提高不少性能哦)"，受到启发，既然可以实现长连接，那么也可以实现连接池了，动手！

原理与链接帖子是一样的，只是将数据库的连接实例改为连接池的连接实例。连接池使用DBUtils实现，但在Django中，需要对DBUtils进行一点小改动。

一部分内容是与链接帖子一样的，在此为了完整性，只做一次搬运工。

第一步，对DBUtils动个小手术。

DBUtils是一个实现数据库连接池的库，支持DB-API 2接口规范，点击下载。解压下载后的压缩包，将其中的DBUtils目录加入应用程序目录中，笔者将其放在应用程序根目录"database"中，Docs、Examples、Tests这3个目录可以删除，修改SteadyDB.py文件，在类 SteadyDBConnection 中添加3个成员方法，如下：

def autocommit(self, *args, **kwargs):

self._con.autocommit(*args, **kwargs)

def get_server_info(self):

return self._con.get_server_info()

@property

def encoders(self):

return self._con.encoders

修改 PooledDB.py 和 PersistentDB.py 两个文件，将

from DBUtils.SteadyDB import connect

改为

from .SteadyDB import connect

第二步，"狸猫换太子"。

将Django中的mysql模块加入应用程序目录中，笔者将Django安装在虚拟环境中，位置是 Django虚拟环境路径/lib/python2.7/site-packages/django/db/backends，将backends目录下的 __init__.py文件和mysql目录复制到database目录下的db目录中，目录结构如下:

应用程序根目录

|

| -- database

|

| -- DBUtils

| -- db

|

| -- __init__.py

| -- mysql

在mysql目录下添加pool.py文件，如下:

# coding:utf-8

'''A connection pool of MySQL in Django 1.5 based on DBUtils.'''

import MySQLdb

from database.DBUtils.PooledDB import PooledDB

class ConnectionWrapper(object):

def __init__(self, connection):

self._conn = connection

def __getattr__(self, method):

''' 代理数据库连接的属性方法 '''

return getattr(self._conn, method)

def close(self):

''' 代理Django的关闭数据库连接 '''

self._conn.close()

class DBWrapper(object):

def __init__(self, module):

self._connection = None

self._db = module

self._pool = {}

def __getattr__(self, item):

return getattr(self._db, item)

def _clear_connections(self, **kwargs):

''' 关闭已有连接 '''

conn = MySQLdb.connect(**kwargs)

cursor = conn.cursor()

sql = ''

cursor.execute('show full processlist;')

processlist = cursor.fetchall()

for th in processlist:

if th[3] == kwargs.get('db') and th[0] != conn.thread_id():

sql += 'kill %s;' % th[0]

if len(sql.split(';')) > 1:

cursor.execute(sql)

cursor.close()

conn.close()

def connect(self, *args, **kwargs):

''' 创建连接 '''

db = kwargs.get('db')

if db not in self._pool:

size = kwargs.get('size')

if 'size' in kwargs:

kwargs.pop('size')

self._clear_connections(**kwargs)

self._pool[db] = PooledDB(self._db, mincached=size, maxcached=size, *args, **kwargs)

self._connection = self._pool[db].connection()

return ConnectionWrapper(self._connection)

修改mysql目录下的 base.py 文件，在

import MySQLdb as Database

下添加两行

from .pool import DBWrapper

Database = DBWrapper(Database)

修改类 DatabaseWrapper 中的方法 get_connection_params，在

if settings_dict['PORT']:

kwargs['port'] = int(settings_dict['PORT'])

下添加如下代码，使连接池支持"SIZE"参数

if settings_dict.get('SIZE'):

kwargs['size'] = int(settings_dict['SIZE'])

在settings.py配置文件的数据库项中，可以通过配置"SIZE"参数来指定连接池中某数据库的连接数量，如：

DATABASES = {

'default': {

# 'ENGINE': 'django.db.backends.mysql',

'ENGINE': 'database.db.mysql',

'NAME': 'cartoon',

'USER': 'root',

'PASSWORD': 'root',

'HOST': '',

'PORT': '',

# 指定10个连接到cartoon库

'SIZE': '10', # Default '0' means unlimit connection pool size

'OPTIONS': {

'init_command': 'SET default_storage_engine=INNODB',

},

}

}

至此，连接池改造完成。

运行程序，进入mysql命令行，执行

show full processlist;

可以看到已有10个cartoon库的连接

Paste_Image.png