mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
[F013] release_proposal: Redis SET NX mutex guards the ~40min execute against concurrent approves
This commit is contained in:
@@ -11,12 +11,16 @@ executor is fail-closed and the proposal stays open unless a publish succeeds.
|
|||||||
|
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import logging
|
||||||
from typing import TYPE_CHECKING
|
from typing import TYPE_CHECKING
|
||||||
|
|
||||||
|
import redis.asyncio as redis
|
||||||
|
|
||||||
|
from roboco.config import settings
|
||||||
from roboco.foundation.policy.content import markers
|
from roboco.foundation.policy.content import markers
|
||||||
from roboco.models.base import TaskStatus
|
from roboco.models.base import TaskStatus
|
||||||
from roboco.services.base import BaseService
|
from roboco.services.base import BaseService
|
||||||
from roboco.services.release_executor import get_release_executor
|
from roboco.services.release_executor import ReleaseResult, get_release_executor
|
||||||
from roboco.services.release_readiness import report_from_dict
|
from roboco.services.release_readiness import report_from_dict
|
||||||
from roboco.services.task import RELEASE_MANAGER_SOURCE, get_task_service
|
from roboco.services.task import RELEASE_MANAGER_SOURCE, get_task_service
|
||||||
|
|
||||||
@@ -26,7 +30,15 @@ if TYPE_CHECKING:
|
|||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
|
||||||
from roboco.db.tables import TaskTable
|
from roboco.db.tables import TaskTable
|
||||||
from roboco.services.release_executor import ReleaseResult
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
# F013: a Redis mutex guarding the ~40min release execute against concurrent
|
||||||
|
# approves (CEO double-click / panel retry). TTL is a backstop above the CI
|
||||||
|
# poll ceiling (ReleaseExecutor ~40min) so a crashed process can't hold the
|
||||||
|
# release hostage forever; the lock is released explicitly on completion.
|
||||||
|
_RELEASE_LOCK_PREFIX = "roboco:release_proposal:"
|
||||||
|
_RELEASE_LOCK_TTL_SECONDS = 3000 # 50 min > 40 min CI ceiling
|
||||||
|
|
||||||
|
|
||||||
class ReleaseProposalService(BaseService):
|
class ReleaseProposalService(BaseService):
|
||||||
@@ -46,6 +58,13 @@ class ReleaseProposalService(BaseService):
|
|||||||
proposal / has no stored report. The proposal is marked COMPLETED only
|
proposal / has no stored report. The proposal is marked COMPLETED only
|
||||||
when the release actually publishes — a gate/CI failure leaves it open so
|
when the release actually publishes — a gate/CI failure leaves it open so
|
||||||
the CEO can retry after the cause is fixed.
|
the CEO can retry after the cause is fixed.
|
||||||
|
|
||||||
|
F013: a Redis ``SET NX`` mutex keyed by the proposal id guards the
|
||||||
|
~40min execute against concurrent approves (double-click / panel retry)
|
||||||
|
that would race on the shared, ``rm -rf``'d writable release clone. A
|
||||||
|
second approve while the lock is held returns ``already_in_progress``
|
||||||
|
without running the executor. Fail-closed on Redis outage — a release is
|
||||||
|
rare and CEO-gated, and the race it prevents corrupts the release.
|
||||||
"""
|
"""
|
||||||
task = await get_task_service(self.session).get(task_id)
|
task = await get_task_service(self.session).get(task_id)
|
||||||
if task is None or task.source != RELEASE_MANAGER_SOURCE:
|
if task is None or task.source != RELEASE_MANAGER_SOURCE:
|
||||||
@@ -53,13 +72,61 @@ class ReleaseProposalService(BaseService):
|
|||||||
report_dict = markers.get_release_report(task)
|
report_dict = markers.get_release_report(task)
|
||||||
if report_dict is None:
|
if report_dict is None:
|
||||||
return None
|
return None
|
||||||
report = report_from_dict(report_dict)
|
|
||||||
executor = await get_release_executor(self.session)
|
lock_key = f"{_RELEASE_LOCK_PREFIX}{task_id}"
|
||||||
result = await executor.execute(report)
|
lock = await self._acquire_release_lock(lock_key)
|
||||||
if result.status == "published":
|
if lock is None:
|
||||||
task.status = TaskStatus.COMPLETED
|
report = report_from_dict(report_dict)
|
||||||
await self.session.flush()
|
return ReleaseResult(
|
||||||
return result
|
status="already_in_progress",
|
||||||
|
version=report.proposed_version,
|
||||||
|
files_changed=[],
|
||||||
|
commit_sha=None,
|
||||||
|
release_url=None,
|
||||||
|
detail=(
|
||||||
|
"A release execute is already in progress for this proposal "
|
||||||
|
"(concurrent approve refused). Wait for it to finish and retry."
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
report = report_from_dict(report_dict)
|
||||||
|
executor = await get_release_executor(self.session)
|
||||||
|
result = await executor.execute(report)
|
||||||
|
if result.status == "published":
|
||||||
|
task.status = TaskStatus.COMPLETED
|
||||||
|
await self.session.flush()
|
||||||
|
return result
|
||||||
|
finally:
|
||||||
|
await self._release_release_lock(lock_key)
|
||||||
|
|
||||||
|
async def _acquire_release_lock(self, lock_key: str) -> bool | None:
|
||||||
|
"""``SET NX EX`` the release mutex. Returns True if acquired, None if held
|
||||||
|
or Redis is unavailable (fail-closed → treat as held)."""
|
||||||
|
try:
|
||||||
|
conn = redis.from_url(settings.redis_url)
|
||||||
|
try:
|
||||||
|
acquired = await conn.set(
|
||||||
|
lock_key, "release", nx=True, ex=_RELEASE_LOCK_TTL_SECONDS
|
||||||
|
)
|
||||||
|
# redis-py returns True on SET NX success, None on conflict.
|
||||||
|
return True if acquired else None
|
||||||
|
finally:
|
||||||
|
await conn.aclose()
|
||||||
|
except Exception as exc:
|
||||||
|
logger.warning("release lock acquire failed (redis): %s", exc)
|
||||||
|
return None
|
||||||
|
|
||||||
|
async def _release_release_lock(self, lock_key: str) -> None:
|
||||||
|
"""Best-effort ``DEL`` the release mutex (the TTL is the backstop)."""
|
||||||
|
try:
|
||||||
|
conn = redis.from_url(settings.redis_url)
|
||||||
|
try:
|
||||||
|
await conn.delete(lock_key)
|
||||||
|
finally:
|
||||||
|
await conn.aclose()
|
||||||
|
except Exception as exc:
|
||||||
|
logger.warning("release lock release failed (redis): %s", exc)
|
||||||
|
|
||||||
async def reject(self, task_id: UUID, required_changes: str) -> TaskTable | None:
|
async def reject(self, task_id: UUID, required_changes: str) -> TaskTable | None:
|
||||||
"""Record the CEO's required changes; keep the proposal held for revision."""
|
"""Record the CEO's required changes; keep the proposal held for revision."""
|
||||||
|
|||||||
@@ -0,0 +1,214 @@
|
|||||||
|
"""F013 — concurrent approve races on the shared release clone.
|
||||||
|
|
||||||
|
The approve flow ran the ~40min ``ReleaseExecutor.execute`` with no guard, so
|
||||||
|
two concurrent CEO ``POST /proposal/approve`` calls (double-click, panel retry)
|
||||||
|
both found the same held proposal and raced on the shared, ``rm -rf``'d writable
|
||||||
|
release clone — interleaving ``git add``/``commit``/``push`` and corrupting the
|
||||||
|
release. The fix acquires a Redis ``SET NX`` mutex keyed by the proposal id
|
||||||
|
before execute (TTL > the 40min CI ceiling) and releases it on completion; a
|
||||||
|
second concurrent approve sees the lock held and refuses instead of racing.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from unittest.mock import AsyncMock, MagicMock, patch
|
||||||
|
from uuid import uuid4
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
from roboco.models.base import TaskStatus
|
||||||
|
from roboco.services.release_executor import ReleaseResult
|
||||||
|
from roboco.services.release_proposal import ReleaseProposalService
|
||||||
|
|
||||||
|
|
||||||
|
def _task(*, source: str = "release_manager") -> MagicMock:
|
||||||
|
t = MagicMock()
|
||||||
|
t.id = uuid4()
|
||||||
|
t.source = source
|
||||||
|
t.status = TaskStatus.AWAITING_CEO_APPROVAL.value
|
||||||
|
return t
|
||||||
|
|
||||||
|
|
||||||
|
def _session() -> MagicMock:
|
||||||
|
s = MagicMock()
|
||||||
|
s.flush = AsyncMock()
|
||||||
|
return s
|
||||||
|
|
||||||
|
|
||||||
|
class _FakeRedis:
|
||||||
|
"""Single-key SET NX / DEL recorder for the release-proposal lock."""
|
||||||
|
|
||||||
|
def __init__(self, *, held: bool = False) -> None:
|
||||||
|
self._held = held
|
||||||
|
self.set_calls: list[tuple[str, str, bool, int]] = []
|
||||||
|
self.del_calls: list[str] = []
|
||||||
|
|
||||||
|
async def set(
|
||||||
|
self, name: str, value: str, *, nx: bool = False, ex: int = 0
|
||||||
|
) -> bool:
|
||||||
|
self.set_calls.append((name, value, nx, ex))
|
||||||
|
if nx and self._held:
|
||||||
|
return False
|
||||||
|
self._held = True
|
||||||
|
return True
|
||||||
|
|
||||||
|
async def delete(self, name: str) -> int:
|
||||||
|
self.del_calls.append(name)
|
||||||
|
self._held = False
|
||||||
|
return 1
|
||||||
|
|
||||||
|
async def aclose(self) -> None:
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def _wire(
|
||||||
|
task: MagicMock,
|
||||||
|
report_dict: dict,
|
||||||
|
executor_result: ReleaseResult,
|
||||||
|
fake_redis: _FakeRedis,
|
||||||
|
) -> dict[str, object]:
|
||||||
|
"""Patch every collaborator ``approve`` touches. Returns the mocks."""
|
||||||
|
task_svc = MagicMock()
|
||||||
|
task_svc.get = AsyncMock(return_value=task)
|
||||||
|
|
||||||
|
executor = MagicMock()
|
||||||
|
executor.execute = AsyncMock(return_value=executor_result)
|
||||||
|
|
||||||
|
markers_mod = MagicMock()
|
||||||
|
markers_mod.get_release_report = MagicMock(return_value=report_dict)
|
||||||
|
markers_mod.set_release_required_changes = MagicMock()
|
||||||
|
|
||||||
|
report = MagicMock()
|
||||||
|
return {
|
||||||
|
"task_svc": task_svc,
|
||||||
|
"executor": executor,
|
||||||
|
"markers": markers_mod,
|
||||||
|
"report": report,
|
||||||
|
"patches": [
|
||||||
|
patch(
|
||||||
|
"roboco.services.release_proposal.get_task_service",
|
||||||
|
return_value=task_svc,
|
||||||
|
),
|
||||||
|
patch(
|
||||||
|
"roboco.services.release_proposal.get_release_executor",
|
||||||
|
AsyncMock(return_value=executor),
|
||||||
|
),
|
||||||
|
patch("roboco.services.release_proposal.markers", markers_mod),
|
||||||
|
patch(
|
||||||
|
"roboco.services.release_proposal.report_from_dict", return_value=report
|
||||||
|
),
|
||||||
|
patch(
|
||||||
|
"roboco.services.release_proposal.redis.from_url",
|
||||||
|
return_value=fake_redis,
|
||||||
|
),
|
||||||
|
],
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
_REPORT = {"proposed_version": "0.13.0", "version_bump_plan": ["pyproject.toml"]}
|
||||||
|
# The CI poll ceiling is ~40 min (ReleaseExecutor._CI_MAX_POLLS * 30s); the
|
||||||
|
# lock TTL must exceed it so a crashed execute can't hold the release hostage.
|
||||||
|
_FORTY_MIN_SECONDS = 2400
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_approve_acquires_lock_then_runs_executor_and_releases() -> None:
|
||||||
|
"""A single approve acquires the NX lock, runs execute, releases on success."""
|
||||||
|
task = _task()
|
||||||
|
fake_redis = _FakeRedis()
|
||||||
|
published = ReleaseResult(
|
||||||
|
status="published",
|
||||||
|
version="0.13.0",
|
||||||
|
files_changed=["pyproject.toml"],
|
||||||
|
commit_sha="abc",
|
||||||
|
release_url="https://x",
|
||||||
|
detail="ok",
|
||||||
|
)
|
||||||
|
w = _wire(task, _REPORT, published, fake_redis)
|
||||||
|
svc = ReleaseProposalService(_session())
|
||||||
|
|
||||||
|
with (
|
||||||
|
w["patches"][0],
|
||||||
|
w["patches"][1],
|
||||||
|
w["patches"][2],
|
||||||
|
w["patches"][3],
|
||||||
|
w["patches"][4],
|
||||||
|
):
|
||||||
|
result = await svc.approve(task.id)
|
||||||
|
|
||||||
|
assert result is not None
|
||||||
|
assert result.status == "published"
|
||||||
|
assert task.status == TaskStatus.COMPLETED.value
|
||||||
|
# Lock acquired with NX + a TTL > the 40min CI ceiling, then released.
|
||||||
|
assert len(fake_redis.set_calls) == 1
|
||||||
|
name, _value, nx, ex = fake_redis.set_calls[0]
|
||||||
|
assert nx is True
|
||||||
|
assert ex >= _FORTY_MIN_SECONDS # > the 40 min CI ceiling
|
||||||
|
assert name.endswith(str(task.id))
|
||||||
|
assert fake_redis.del_calls == [name]
|
||||||
|
w["executor"].execute.assert_awaited_once()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_concurrent_approve_refused_while_lock_held() -> None:
|
||||||
|
"""A second approve while the lock is held refuses and never runs execute."""
|
||||||
|
task = _task()
|
||||||
|
fake_redis = _FakeRedis(held=True) # another approve already holds the lock
|
||||||
|
published = ReleaseResult(
|
||||||
|
status="published",
|
||||||
|
version="0.13.0",
|
||||||
|
files_changed=[],
|
||||||
|
commit_sha=None,
|
||||||
|
release_url=None,
|
||||||
|
detail="ok",
|
||||||
|
)
|
||||||
|
w = _wire(task, _REPORT, published, fake_redis)
|
||||||
|
svc = ReleaseProposalService(_session())
|
||||||
|
|
||||||
|
with (
|
||||||
|
w["patches"][0],
|
||||||
|
w["patches"][1],
|
||||||
|
w["patches"][2],
|
||||||
|
w["patches"][3],
|
||||||
|
w["patches"][4],
|
||||||
|
):
|
||||||
|
result = await svc.approve(task.id)
|
||||||
|
|
||||||
|
assert result is not None
|
||||||
|
assert result.status == "already_in_progress"
|
||||||
|
# The executor MUST NOT run — that's the whole point of the guard.
|
||||||
|
w["executor"].execute.assert_not_awaited()
|
||||||
|
# Nothing committed the task to COMPLETED.
|
||||||
|
assert task.status != TaskStatus.COMPLETED.value
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_failed_execute_releases_lock_so_ceo_can_retry() -> None:
|
||||||
|
"""A gate/CI failure leaves the proposal open AND releases the lock for retry."""
|
||||||
|
task = _task()
|
||||||
|
fake_redis = _FakeRedis()
|
||||||
|
gate_failed = ReleaseResult(
|
||||||
|
status="gate_failed",
|
||||||
|
version="0.13.0",
|
||||||
|
files_changed=["pyproject.toml"],
|
||||||
|
commit_sha=None,
|
||||||
|
release_url=None,
|
||||||
|
detail="make quality failed",
|
||||||
|
)
|
||||||
|
w = _wire(task, _REPORT, gate_failed, fake_redis)
|
||||||
|
svc = ReleaseProposalService(_session())
|
||||||
|
|
||||||
|
with (
|
||||||
|
w["patches"][0],
|
||||||
|
w["patches"][1],
|
||||||
|
w["patches"][2],
|
||||||
|
w["patches"][3],
|
||||||
|
w["patches"][4],
|
||||||
|
):
|
||||||
|
result = await svc.approve(task.id)
|
||||||
|
|
||||||
|
assert result is not None
|
||||||
|
assert result.status == "gate_failed"
|
||||||
|
# Proposal stays open (not COMPLETED) for retry...
|
||||||
|
assert task.status != TaskStatus.COMPLETED.value
|
||||||
|
# ...and the lock is released so the retry can acquire it.
|
||||||
|
assert len(fake_redis.del_calls) == 1
|
||||||
Reference in New Issue
Block a user