anti
f1e14280c0
Locust hit /health and /config on every @task(3), so each request was firing repo.get_total_logs() and two repo.get_state() calls against aiosqlite — filling the driver queue for data that changes on the order of seconds, not milliseconds. Both caches follow the shape already used by the existing Docker cache: - asyncio.Lock with double-checked TTL so concurrent callers collapse into one DB hit per 1s window. - _reset_* helpers called from tests/api/conftest.py::setup_db so the module-level cache can't leak across tests. tests/test_health_config_cache.py asserts 50 concurrent callers produce exactly 1 repo call, and the cache expires after TTL.
90 lines
2.7 KiB
Python
90 lines
2.7 KiB
Python
import asyncio
|
|
import time
|
|
from typing import Any, Optional
|
|
|
|
from fastapi import APIRouter, Depends
|
|
|
|
from decnet.env import DECNET_DEVELOPER
|
|
from decnet.telemetry import traced as _traced
|
|
from decnet.web.dependencies import require_viewer, repo
|
|
from decnet.web.db.models import UserResponse
|
|
|
|
router = APIRouter()
|
|
|
|
_DEFAULT_DEPLOYMENT_LIMIT = 10
|
|
_DEFAULT_MUTATION_INTERVAL = "30m"
|
|
|
|
# Cache config_limits / config_globals reads — these change on rare admin
|
|
# writes but get polled constantly by the UI and locust.
|
|
_STATE_TTL = 1.0
|
|
_state_cache: dict[str, tuple[Optional[dict[str, Any]], float]] = {}
|
|
_state_locks: dict[str, asyncio.Lock] = {}
|
|
|
|
|
|
def _reset_state_cache() -> None:
|
|
"""Reset cached config state — used by tests."""
|
|
_state_cache.clear()
|
|
|
|
|
|
async def _get_state_cached(name: str) -> Optional[dict[str, Any]]:
|
|
entry = _state_cache.get(name)
|
|
now = time.monotonic()
|
|
if entry is not None and now - entry[1] < _STATE_TTL:
|
|
return entry[0]
|
|
lock = _state_locks.setdefault(name, asyncio.Lock())
|
|
async with lock:
|
|
entry = _state_cache.get(name)
|
|
now = time.monotonic()
|
|
if entry is not None and now - entry[1] < _STATE_TTL:
|
|
return entry[0]
|
|
value = await repo.get_state(name)
|
|
_state_cache[name] = (value, time.monotonic())
|
|
return value
|
|
|
|
|
|
@router.get(
|
|
"/config",
|
|
tags=["Configuration"],
|
|
responses={
|
|
401: {"description": "Could not validate credentials"},
|
|
403: {"description": "Insufficient permissions"},
|
|
},
|
|
)
|
|
@_traced("api.get_config")
|
|
async def api_get_config(user: dict = Depends(require_viewer)) -> dict:
|
|
limits_state = await _get_state_cached("config_limits")
|
|
globals_state = await _get_state_cached("config_globals")
|
|
|
|
deployment_limit = (
|
|
limits_state.get("deployment_limit", _DEFAULT_DEPLOYMENT_LIMIT)
|
|
if limits_state
|
|
else _DEFAULT_DEPLOYMENT_LIMIT
|
|
)
|
|
global_mutation_interval = (
|
|
globals_state.get("global_mutation_interval", _DEFAULT_MUTATION_INTERVAL)
|
|
if globals_state
|
|
else _DEFAULT_MUTATION_INTERVAL
|
|
)
|
|
|
|
base = {
|
|
"role": user["role"],
|
|
"deployment_limit": deployment_limit,
|
|
"global_mutation_interval": global_mutation_interval,
|
|
}
|
|
|
|
if user["role"] == "admin":
|
|
all_users = await repo.list_users()
|
|
base["users"] = [
|
|
UserResponse(
|
|
uuid=u["uuid"],
|
|
username=u["username"],
|
|
role=u["role"],
|
|
must_change_password=u["must_change_password"],
|
|
).model_dump()
|
|
for u in all_users
|
|
]
|
|
if DECNET_DEVELOPER:
|
|
base["developer_mode"] = True
|
|
|
|
return base
|