liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

数据库连接优化

Browse Source
This commit is contained in:
liuwei
2025-03-19 17:58:49 +08:00
parent 6902919cda
commit 7f3008fc54
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
plugins/message_sign/main.py
View File

@@ -5,6 +5,7 @@ from typing import Dict, Any, List, Optional, Tuple
from wcferry import Wcf, WxMsg from wcferry import Wcf, WxMsg
from db.connection import DBConnectionManager
from plugin_common.message_plugin_interface import MessagePluginInterface from plugin_common.message_plugin_interface import MessagePluginInterface
from plugin_common.plugin_interface import PluginStatus from plugin_common.plugin_interface import PluginStatus
from plugins.stats_collector.decorators import plugin_stats_decorator from plugins.stats_collector.decorators import plugin_stats_decorator
@@ -59,12 +60,8 @@ class MessageSignPlugin(MessagePluginInterface):
self.message_util = context.get("message_util") self.message_util = context.get("message_util")
self.gbm = context.get("gbm") self.gbm = context.get("gbm")
self.all_contacts = context.get("all_contacts", {}) 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_db = SignInDB(self.db_manager)
self.sign_in_redis = SignInRedisDB(self.db_manager) self.sign_in_redis = SignInRedisDB(self.db_manager)

5
plugins/point_trade/main.py
View File

@@ -8,6 +8,7 @@ import xml.etree.ElementTree as ET
from wcferry import Wcf from wcferry import Wcf
from db.connection import DBConnectionManager
from plugin_common.message_plugin_interface import MessagePluginInterface from plugin_common.message_plugin_interface import MessagePluginInterface
from plugin_common.plugin_interface import PluginStatus from plugin_common.plugin_interface import PluginStatus
from plugins.stats_collector.decorators import plugin_stats_decorator from plugins.stats_collector.decorators import plugin_stats_decorator
@@ -57,7 +58,9 @@ class PointTradePlugin(MessagePluginInterface):
self.event_system = context.get("event_system") self.event_system = context.get("event_system")
self.message_util = context.get("message_util") self.message_util = context.get("message_util")
self.gbm = context.get("gbm") 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: if not self.db_pool:
self.LOG.error("数据库连接池未初始化，插件无法正常工作") self.LOG.error("数据库连接池未初始化，插件无法正常工作")