限制器多进程告警优化

services/apis.py

@@ -18,10 +18,10 @@ from fastapi import (
 from fastapi.requests import Request
 from fastapi.security import OAuth2PasswordRequestForm
 
-import services.limiter as limiter
+import setting
 from services.chaojiying import CJ
 from services.defults import FAKE_USERS_DB
-from services.limiter import LIMITER
+from services.limiter import limiter
 from services.utils import calculate
 
 images_router = APIRouter(prefix="/images")
@@ -162,15 +162,13 @@ async def cjy_postpic_base64(
 
 
 @images_router.post("/discern", summary="超级鹰识别")
-@LIMITER.limit(limiter.LM_VALUE)
+@limiter.limit(setting.LIMITER_VALUE)
 async def discern_complex_image(
-    request: Request,
+    request: Request,  # 必须设置，否则 limiter 无法起限制作用
     form_data: OAuth2PasswordRequestForm = Depends(),
     cjy_captcha: dict = Depends(cjy_postpic_base64),
 ):
     user_authentication(form_data)
-    limiter.increase_usages_count()
-    limiter.limiter_warring()
     return cjy_captcha
 
 
@@ -190,8 +188,8 @@ async def cjy_report_error(
 
 @images_router.get("/reset", summary="重置超级鹰日访问次数")
 async def reset_limiter():
-    LIMITER.reset()
-    limiter.flush_limiter()
+    limiter.reset()
+    limiter.reset_warn()
     return {
         "msg": "success",
         "code": 0,