23 lines
897 B
Python
23 lines
897 B
Python
import pymysql
|
||
from dbutils.pooled_db import PooledDB
|
||
from config import DB_CONFIG
|
||
|
||
class MySQLPool:
|
||
def __init__(self):
|
||
# 初始化时建立数据库连接池
|
||
self.pool = PooledDB(
|
||
creator=pymysql, # 使用链接数据库的模块
|
||
maxconnections=6, # 连接池允许的最大连接数
|
||
mincached=2, # 初始化时，连接池中至少创建的空闲的连接
|
||
maxcached=5, # 连接池中最多闲置的连接
|
||
blocking=True, # 连接池中如果没有可用连接后是否阻塞等待
|
||
maxusage=None, # 一个连接最多被重复使用的次数，None表示无限制
|
||
setsession=[], # 开始会话前执行的命令列表
|
||
ping=0,
|
||
**DB_CONFIG
|
||
)
|
||
|
||
def get_connection(self):
|
||
# 从连接池中获取一个连接
|
||
return self.pool.connection()
|