From 7f3008fc541d3ee5ec5c43dfd8230c95b0b6b050 Mon Sep 17 00:00:00 2001 From: liuwei Date: Wed, 19 Mar 2025 17:58:49 +0800 Subject: [PATCH] =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=BA=93=E8=BF=9E=E6=8E=A5?= =?UTF-8?q?=E4=BC=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- plugins/message_sign/main.py | 7 ++----- plugins/point_trade/main.py | 5 ++++- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/plugins/message_sign/main.py b/plugins/message_sign/main.py index f12a7a9..44ea24c 100644 --- a/plugins/message_sign/main.py +++ b/plugins/message_sign/main.py @@ -5,6 +5,7 @@ from typing import Dict, Any, List, Optional, Tuple from wcferry import Wcf, WxMsg +from db.connection import DBConnectionManager from plugin_common.message_plugin_interface import MessagePluginInterface from plugin_common.plugin_interface import PluginStatus from plugins.stats_collector.decorators import plugin_stats_decorator @@ -59,12 +60,8 @@ class MessageSignPlugin(MessagePluginInterface): self.message_util = context.get("message_util") self.gbm = context.get("gbm") self.all_contacts = context.get("all_contacts", {}) - self.db_manager = context.get("db_manager") - - if not self.db_manager: - self.LOG.error("数据库连接管理器未初始化，插件无法正常工作") - return False + self.db_manager = DBConnectionManager.get_instance() # 初始化数据库操作类 self.sign_in_db = SignInDB(self.db_manager) self.sign_in_redis = SignInRedisDB(self.db_manager) diff --git a/plugins/point_trade/main.py b/plugins/point_trade/main.py index c26f28e..71efa95 100644 --- a/plugins/point_trade/main.py +++ b/plugins/point_trade/main.py @@ -8,6 +8,7 @@ import xml.etree.ElementTree as ET from wcferry import Wcf +from db.connection import DBConnectionManager from plugin_common.message_plugin_interface import MessagePluginInterface from plugin_common.plugin_interface import PluginStatus from plugins.stats_collector.decorators import plugin_stats_decorator @@ -57,7 +58,9 @@ class PointTradePlugin(MessagePluginInterface): self.event_system = context.get("event_system") self.message_util = context.get("message_util") self.gbm = context.get("gbm") - self.db_pool = context.get("db_pool") + self.db_manager = DBConnectionManager.get_instance() + + self.db_pool = self.db_manager.mysql_pool if not self.db_pool: self.LOG.error("数据库连接池未初始化，插件无法正常工作")