From 906a8d08378ce6a957fd079b21c068ad255ab7e3 Mon Sep 17 00:00:00 2001 From: HuangHai <10402852@qq.com> Date: Fri, 28 Feb 2025 15:14:01 +0800 Subject: [PATCH] 'commit' --- AI/AiService/Model/TaskModel.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/AI/AiService/Model/TaskModel.py b/AI/AiService/Model/TaskModel.py index 195e2309..b127c3c8 100644 --- a/AI/AiService/Model/TaskModel.py +++ b/AI/AiService/Model/TaskModel.py @@ -1,10 +1,18 @@ import mysql.connector from AiService.Config.Config import * +import mysql.connector.pooling class TaskModel: + # 创建连接池 + connection_pool = mysql.connector.pooling.MySQLConnectionPool( + pool_name="task_pool", + pool_size=5, # 连接池大小 + **MYSQL_CONFIG + ) + def __init__(self): - # 初始化 MySQL 连接 - self.connection = mysql.connector.connect(**MYSQL_CONFIG) + # 从连接池获取连接 + self.connection = self.connection_pool.get_connection() self.cursor = self.connection.cursor() def insert_task(self, task_id: str, keyword: str): @@ -24,7 +32,7 @@ class TaskModel: def close(self): """ - 关闭 MySQL 连接 + 将连接归还到连接池 """ self.cursor.close() - self.connection.close() + self.connection.close() \ No newline at end of file