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("数据库连接池未初始化,插件无法正常工作")