xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-07-16 09:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: 独立ratelimiter包

Browse Source
This commit is contained in:
RockChinQ
2024-01-25 22:35:15 +08:00
parent f4ead5ec5c
commit b43882aad0
9 changed files with 147 additions and 102 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/qqbot/manager.py
+4
View File
@@ -22,6 +22,7 @@ from .resprule import resprule
from .bansess import bansess
from .cntfilter import cntfilter
from .longtext import longtext
from .ratelim import ratelim
from ..boot import app
@@ -44,6 +45,7 @@ class QQBotManager:
cntfilter_mgr: cntfilter.ContentFilterManager = None
longtext_pcs: longtext.LongTextProcessor = None
resprule_chkr: resprule.GroupRespondRuleChecker = None
ratelimiter: ratelim.RateLimiter = None
def __init__(self, first_time_init=True, ap: app.Application = None):
config = context.get_config_manager().data
@@ -53,6 +55,7 @@ class QQBotManager:
self.cntfilter_mgr = cntfilter.ContentFilterManager(ap)
self.longtext_pcs = longtext.LongTextProcessor(ap)
self.resprule_chkr = resprule.GroupRespondRuleChecker(ap)
self.ratelimiter = ratelim.RateLimiter(ap)
self.timeout = config['process_message_timeout']
self.retry = config['retry_times']
@@ -62,6 +65,7 @@ class QQBotManager:
await self.cntfilter_mgr.initialize()
await self.longtext_pcs.initialize()
await self.resprule_chkr.initialize()
await self.ratelimiter.initialize()
config = context.get_config_manager().data