style: 修复black + isort格式问题(task_enqueue.py + 限流测试)

This commit is contained in:
用户CI Test
2026-07-11 18:24:35 +08:00
parent c48f8eef33
commit da1fd29a22
2 changed files with 7 additions and 13 deletions
+5 -11
View File
@@ -6,8 +6,8 @@ from app.core.celery_app import celery_app
logger = logging.getLogger(__name__)
# ── 限流阈值常量(全系统统一管理,不要在业务代码里硬编码) ──
USER_PENDING_LIMIT = 3 # 单用户 pending 上限
GLOBAL_PENDING_LIMIT = 20 # 全局 pending 上限
USER_PENDING_LIMIT = 3 # 单用户 pending 上限
GLOBAL_PENDING_LIMIT = 20 # 全局 pending 上限
class UserPendingLimitExceeded(Exception):
@@ -71,9 +71,7 @@ def check_queue_limits(
user_pending,
user_pending_limit,
)
raise UserPendingLimitExceeded(
user_id=user_id, pending_count=user_pending, limit=user_pending_limit
)
raise UserPendingLimitExceeded(user_id=user_id, pending_count=user_pending, limit=user_pending_limit)
def _mark_task_failed_safely(
@@ -157,9 +155,7 @@ def safe_enqueue_generation_task(
user_pending,
user_pending_limit,
)
exc = UserPendingLimitExceeded(
user_id=user_id, pending_count=user_pending, limit=user_pending_limit
)
exc = UserPendingLimitExceeded(user_id=user_id, pending_count=user_pending, limit=user_pending_limit)
_mark_task_failed_safely(task, generation_task_repository, log_prefix, str(exc))
raise exc
@@ -201,9 +197,7 @@ def safe_enqueue_generation_task(
exc: Exception = GlobalQueueFull(pending_count=global_after, limit=global_pending_limit)
else:
reason = f"用户 pending 超限(入队后): {user_after}/{user_pending_limit}"
exc = UserPendingLimitExceeded(
user_id=user_id, pending_count=user_after, limit=user_pending_limit
)
exc = UserPendingLimitExceeded(user_id=user_id, pending_count=user_after, limit=user_pending_limit)
logger.warning(
"[队列限流] %s, task_id=%s, user_id=%s — 回滚状态为 failed",