fix(api): лимит одновременных задач считать по БД, не по Redis-счётчику
All checks were successful
Deploy / test (push) Successful in 2m54s
Deploy / deploy (push) Successful in 11s

Живая проверка: юзер сделал один поиск (давно завершился, status='done'),
второй поиск сразу упёрся в "Превышен лимит одновременных задач" — на
free-тарифе лимит 1.

Причина: acquire_concurrent_slot() инкрементирует Redis-счётчик
concurrent:{user_id} при создании КАЖДОЙ задачи (search.py, documents.py),
а release_concurrent_slot() — которая должна его декрементировать по
завершении — НЕ ВЫЗЫВАЛАСЬ НИГДЕ В КОДЕ (grep подтвердил: только
определение, ни одного вызова). Счётчик только рос, лимит превышался
навсегда для практически любого юзера после первой же задачи — до
часового TTL-автосброса.

Фикс — не "доставить забытый release()" (это лечит симптом, но оставляет
класс бага: счётчик и реальность могут разойтись любым другим путём), а
убрать сам отдельный счётчик. check_concurrent_limit() считает активные
задачи (status IN queued/processing) напрямую в Postgres — Task.status уже
корректно обновляется во всех воркерах (проверено многократно в этой
сессии), рассинхронизация невозможна по конструкции. Redis-лимиты
(дневные/месячные, Lua-скрипт) не тронуты — там свой, рабочий, механизм.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
jze9
2026-08-24 18:42:29 +05:00
parent 61c903ae0e
commit 712dc383ea
3 changed files with 34 additions and 49 deletions

View File

@@ -1,16 +1,27 @@
"""Redis-based rate limiter с атомарными Lua-скриптами.
"""Rate limiter: дневные/месячные лимиты — в Redis, лимит одновременных задач — в БД.
Дневные/месячные лимиты (Redis, атомарные Lua-скрипты):
Проблема наивного подхода (GET → проверка → INCR):
- Race condition: 10 конкурентных запросов могут одновременно пройти GET,
увидеть значение ниже лимита и все инкрементировать.
Решение: один Lua-скрипт выполняется атомарно на стороне Redis.
Redis гарантирует, что между командами внутри скрипта нет других операций.
Лимит одновременных задач — НЕ Redis-счётчик. Раньше был acquire/release-счётчик
(INCR при создании задачи, DECR при завершении) — но release_concurrent_slot()
никогда не вызывался ни из одного воркера, так что счётчик только рос и лимит
превышался навсегда (до часового TTL-автосброса), даже когда все задачи юзера
давно завершены. Вместо ручного счётчика, который может рассинхронизироваться
с реальностью, считаем активные задачи прямо по Task.status в Postgres —
рассинхронизации тогда не может быть в принципе.
"""
import logging
from datetime import UTC, datetime
from sqlalchemy import func, select
from sqlalchemy.ext.asyncio import AsyncSession
from app.core.redis_client import get_redis
logger = logging.getLogger(__name__)
@@ -74,24 +85,6 @@ end
return {new_val, 1}
"""
# Атомарная проверка + инкремент счётчика одновременных задач.
# Возвращает 1 если слот получен, 0 если превышен лимит.
_LUA_ACQUIRE_CONCURRENT = """
local key = KEYS[1]
local limit = tonumber(ARGV[1])
local ttl = tonumber(ARGV[2])
local current = tonumber(redis.call('GET', key) or '0')
if current >= limit then
return 0
end
redis.call('INCR', key)
redis.call('EXPIRE', key, ttl)
return 1
"""
def _period_suffix(period: str) -> str:
now = datetime.now(UTC)
return now.strftime("%Y-%m-%d") if period == "day" else now.strftime("%Y-%m")
@@ -136,33 +129,27 @@ async def check_and_increment_limit(
}
async def acquire_concurrent_slot(user_id: int, plan: str) -> bool:
async def check_concurrent_limit(db: AsyncSession, user_id: int, plan: str) -> bool:
"""
Атомарно захватить слот одновременной задачи.
Проверить лимит одновременных задач по факту в БД (queued/processing).
Возможен редкий race (два запроса одновременно оба видят N-1 активных и
оба проходят) — на практике не критично для этого лимита (защита от
злоупотребления, не от превышения на единицу), а взамен исключён класс
багов "счётчик разошёлся с реальностью и завис навсегда".
Returns:
True — слот получен (задачу можно создавать).
False — все слоты заняты.
True — лимит не превышен, задачу можно создавать.
"""
from app.models.task import Task # избегаем circular import на уровне модуля
limits = PLAN_LIMITS.get(plan, PLAN_LIMITS["free"])
max_concurrent = limits.get("concurrent", 1)
r = get_redis()
result = await r.eval(
_LUA_ACQUIRE_CONCURRENT,
1,
f"concurrent:{user_id}",
max_concurrent,
3_600, # TTL 1 час — автосброс если воркер упал не освободив слот
result = await db.execute(
select(func.count())
.select_from(Task)
.where(Task.user_id == user_id, Task.status.in_(("queued", "processing")))
)
return bool(result)
async def release_concurrent_slot(user_id: int) -> None:
"""Освободить слот одновременной задачи после завершения."""
r = get_redis()
key = f"concurrent:{user_id}"
# DECR безопасен: Redis не уходит в отрицательные значения если мы контролируем acquire
current = await r.get(key)
if current and int(current) > 0:
await r.decr(key)
current = result.scalar_one()
return current < max_concurrent