DBUtils是Python的一個用于實作資料庫連接配接池的子產品。
此連接配接池有兩種連接配接模式:
- 模式一:為每個線程建立一個連接配接,線程即使調用了close方法,也不會關閉,隻是把連接配接重新放到連接配接池,供自己線程再次使用。當線程終止時,連接配接自動關閉。
-
模式二:建立一批連接配接到連接配接池,供所有線程共享使用。
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='rui123',
database='db1',
charset='utf8'
)
def func():
# 檢測目前正在運作連接配接數的是否小于最大連結數,如果不小于則:等待或報raise TooManyConnections異常
# 否則
# 則優先去初始化時建立的連結中擷取連結 SteadyDBConnection。
# 然後将SteadyDBConnection對象封裝到PooledDedicatedDBConnection中并傳回。
# 如果最開始建立的連結沒有連結,則去建立一個SteadyDBConnection對象,再封裝到PooledDedicatedDBConnection中并傳回。
# 一旦關閉連結後,連接配接就傳回到連接配接池讓後續線程繼續使用。
conn = POOL.connection()
cursor = conn.cursor()
cursor.execute('select * from tb1')
result = cursor.fetchall()
conn.close()
func()
使用多線程進行資料庫連接配接
import threading
for i in range(10):
t = threading.Thread(target=func)
t.start()