suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2025-06-02 09:26:12 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✨ 添加 cookie_choose_counter

Browse Source
This commit is contained in:
suyiiyii 2024-12-12 14:54:12 +08:00
parent 80f924123d
commit 4c29cf10e4
Signed by: suyiiyii
GPG Key ID: 044704CB29B8AD85
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nonebot_bison/metrics.py
View File

@ -10,6 +10,10 @@ request_counter = Counter(
# Sent counter # Sent counter
sent_counter = Counter("bison_sent_counter", "The number of sent messages", ["site_name", "platform_name", "target"]) sent_counter = Counter("bison_sent_counter", "The number of sent messages", ["site_name", "platform_name", "target"])
cookie_choose_counter = Counter(
"bison_cookie_choose_counter", "The number of cookie choose", ["site_name", "target", "cookie_id"]
)
metrics_router = APIRouter(prefix="/api/metrics", tags=["metrics"]) metrics_router = APIRouter(prefix="/api/metrics", tags=["metrics"])

2
nonebot_bison/utils/site.py
View File

@ -13,6 +13,7 @@ from ..types import Target
from ..config import config from ..config import config
from .http import http_client from .http import http_client
from ..config.db_model import Cookie from ..config.db_model import Cookie
from ..metrics import cookie_choose_counter
class ClientManager(ABC): class ClientManager(ABC):
@ -131,6 +132,7 @@ class CookieClientManager(ClientManager):
"""获取 client，根据 target 选择 cookie""" """获取 client，根据 target 选择 cookie"""
client = http_client() client = http_client()
cookie = await self._choose_cookie(target) cookie = await self._choose_cookie(target)
cookie_choose_counter.labels(site_name=self._site_name, target=target, cookie_id=cookie.id).inc()
if cookie.is_universal: if cookie.is_universal:
logger.trace(f"平台 {self._site_name} 未获取到用户cookie, 使用匿名cookie") logger.trace(f"平台 {self._site_name} 未获取到用户cookie, 使用匿名cookie")
else: else: