mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-05 05:16:03 +00:00
209f16af76
* style: remove necessary imports * style: fix F841 * style: fix F401 * style: fix F811 * style: fix E402 * style: fix E721 * style: fix E722 * style: fix E722 * style: fix F541 * style: ruff format * style: all passed * style: add ruff in deps * style: more ignores in ruff.toml * style: add pre-commit
68 lines
1.6 KiB
Python
68 lines
1.6 KiB
Python
from __future__ import annotations
|
||
import abc
|
||
import typing
|
||
|
||
from ...core import app, entities as core_entities
|
||
|
||
|
||
preregistered_algos: list[typing.Type[ReteLimitAlgo]] = []
|
||
|
||
|
||
def algo_class(name: str):
|
||
def decorator(cls: typing.Type[ReteLimitAlgo]) -> typing.Type[ReteLimitAlgo]:
|
||
cls.name = name
|
||
preregistered_algos.append(cls)
|
||
return cls
|
||
|
||
return decorator
|
||
|
||
|
||
class ReteLimitAlgo(metaclass=abc.ABCMeta):
|
||
"""限流算法抽象类"""
|
||
|
||
name: str = None
|
||
|
||
ap: app.Application
|
||
|
||
def __init__(self, ap: app.Application):
|
||
self.ap = ap
|
||
|
||
async def initialize(self):
|
||
pass
|
||
|
||
@abc.abstractmethod
|
||
async def require_access(
|
||
self,
|
||
query: core_entities.Query,
|
||
launcher_type: str,
|
||
launcher_id: typing.Union[int, str],
|
||
) -> bool:
|
||
"""进入处理流程
|
||
|
||
这个方法对等待是友好的，意味着算法可以实现在这里等待一段时间以控制速率。
|
||
|
||
Args:
|
||
launcher_type (str): 请求者类型 群聊为 group 私聊为 person
|
||
launcher_id (int): 请求者ID
|
||
|
||
Returns:
|
||
bool: 是否允许进入处理流程，若返回false，则直接丢弃该请求
|
||
"""
|
||
raise NotImplementedError
|
||
|
||
@abc.abstractmethod
|
||
async def release_access(
|
||
self,
|
||
query: core_entities.Query,
|
||
launcher_type: str,
|
||
launcher_id: typing.Union[int, str],
|
||
):
|
||
"""退出处理流程
|
||
|
||
Args:
|
||
launcher_type (str): 请求者类型 群聊为 group 私聊为 person
|
||
launcher_id (int): 请求者ID
|
||
"""
|
||
|
||
raise NotImplementedError
|