加入限流策略。防止过度消耗资源

This commit is contained in:
liuwei
2025-06-25 17:06:27 +08:00
parent 11661906c4
commit ebcd8d4d55
5 changed files with 10 additions and 0 deletions

View File

@@ -14,6 +14,7 @@ from base.plugin_common.message_plugin_interface import MessagePluginInterface
from base.plugin_common.plugin_interface import PluginStatus
from utils.decorator.plugin_decorators import plugin_stats_decorator
from utils.decorator.points_decorator import plugin_points_cost
from utils.decorator.rate_limit_decorator import group_feature_rate_limit
from utils.robot_cmd.robot_command import Feature, PermissionStatus, GroupBotManager
from wechat_ipad import WechatAPIClient
@@ -108,6 +109,7 @@ class VideoManPlugin(MessagePluginInterface):
@plugin_stats_decorator(plugin_name="猛男视频")
@plugin_points_cost(2, "猛男视频消耗积分", FEATURE_KEY)
@group_feature_rate_limit(max_per_minute=3, feature_key=FEATURE_KEY)
async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
"""处理消息"""
content = str(message.get("content", "")).strip()