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()