Python数据库链接池DBUtils

出处：http://www.cnblogs.com/wupeiqi/

DBUtils是Python的一个用于实现数据库链接池的模块。

此链接池有两种链接模式：

• 模式一：为每一个线程建立一个链接，线程即便调用了close方法，也不会关闭，只是把链接从新放到链接池，供本身线程再次使用。当线程终止时，链接自动关闭。
  
  POOL = PersistentDB(
      creator=pymysql,  # 使用连接数据库的模块 maxusage=None, # 一个连接最多被重复使用的次数，None表示无限制 setsession=[], # 开始会话前执行的命令列表。如：["set datestyle to ...", "set time zone ..."] ping=0, # ping MySQL服务端，检查是否服务可用。# 如：0 = None = never, 1 = default = whenever it is requested, 2 = when a cursor is created, 4 = when a query is executed, 7 = always closeable=False, # 若是为False时， conn.close() 实际上被忽略，供下次使用，再线程关闭时，才会自动关闭连接。若是为True时， conn.close()则关闭连接，那么再次调用pool.connection时就会报错，由于已经真的关闭了链接（pool.steady_connection()能够获取一个新的连接） threadlocal=None, # 本线程独享值得对象，用于保存连接对象，若是连接对象被重置 host='127.0.0.1', port=3306, user='root', password='123', database='pooldb', charset='utf8' ) def func(): conn = POOL.connection(shareable=False) cursor = conn.cursor() cursor.execute('select * from tb1') result = cursor.fetchall() cursor.close() conn.close() func()
• 模式二：建立一批链接到链接池，供全部线程共享使用。
  PS：因为pymysql、MySQLdb等threadsafety值为1，因此该模式链接池中的线程会被全部线程共享。
  
  import time import pymysql import threading from DBUtils.PooledDB import PooledDB, SharedDBConnection POOL = PooledDB( creator=pymysql, # 使用连接数据库的模块 maxconnections=6, # 链接池容许的最大链接数，0和None表示不限制链接数 mincached=2, # 初始化时，连接池中至少建立的空闲的连接，0表示不建立 maxcached=5, # 连接池中最多闲置的连接，0和None不限制 maxshared=3, # 连接池中最多共享的连接数量，0和None表示所有共享。PS: 无用，由于pymysql和MySQLdb等模块的 threadsafety都为1，全部值不管设置为多少，_maxcached永远为0，因此永远是全部连接都共享。 blocking=True, # 链接池中若是没有可用链接后，是否阻塞等待。True，等待；False，不等待而后报错 maxusage=None, # 一个连接最多被重复使用的次数，None表示无限制 setsession=[], # 开始会话前执行的命令列表。如：["set datestyle to ...", "set time zone ..."] ping=0, # ping MySQL服务端，检查是否服务可用。# 如：0 = None = never, 1 = default = whenever it is requested, 2 = when a cursor is created, 4 = when a query is executed, 7 = always host='127.0.0.1', port=3306, user='root', password='123', database='pooldb', charset='utf8' ) def func(): # 检测当前正在运行链接数的是否小于最大连接数，若是不小于则：等待或报raise TooManyConnections异常 # 不然 # 则优先去初始化时建立的连接中获取连接 SteadyDBConnection。 # 而后将SteadyDBConnection对象封装到PooledDedicatedDBConnection中并返回。 # 若是最开始建立的连接没有连接，则去建立一个SteadyDBConnection对象，再封装到PooledDedicatedDBConnection中并返回。 # 一旦关闭连接后，链接就返回到链接池让后续线程继续使用。 conn = POOL.connection() # print(th, '连接被拿走了', conn1._con) # print(th, '池子里目前有', pool._idle_cache, '\r\n')  cursor = conn.cursor() cursor.execute('select * from tb1') result = cursor.fetchall() conn.close() func()

 若是没有链接池，使用pymysql来链接数据库时，单线程应用彻底没有问题，但若是涉及到多线程应用那么就须要加锁，一旦加锁那么链接势必就会排队等待，当请求比较多时，性能就会下降了。

#!/usr/bin/env python # -*- coding:utf-8 -*- import pymysql import threading from threading import RLock LOCK = RLock() CONN = pymysql.connect(host='127.0.0.1', port=3306, user='root', password='123', database='pooldb', charset='utf8') def task(arg): with LOCK: cursor = CONN.cursor() cursor.execute('select * from tb1') result = cursor.fetchall() cursor.close() print(result) for i in range(10): t = threading.Thread(target=task, args=(i,)) t.start()

#!/usr/bin/env python # -*- coding:utf-8 -*- import pymysql import threading CONN = pymysql.connect(host='127.0.0.1', port=3306, user='root', password='123', database='pooldb', charset='utf8') def task(arg): cursor = CONN.cursor() cursor.execute('select * from tb1') result = cursor.fetchall() cursor.close() print(result) for i in range(10): t = threading.Thread(target=task, args=(i,)) t.start()