fix(automation): recover stale outbox delivering leases
_claim_due_events flipped rows to 'delivering' and committed before the HTTP call; if the process died between that commit and the outcome- recording transaction, the row stayed 'delivering' forever with no reclaim path -- a real gap, not previously documented as an accepted limitation. Give each claim a lease deadline (reusing next_attempt_at, since it's only otherwise meaningful for pending-status backoff scheduling) and sweep expired leases back to pending at the start of every dispatch cycle, before claiming new work. attempts is preserved so the count still reflects true history. Only leases past their deadline are touched, so a still-alive worker mid-delivery is never disturbed or double-processed.
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import uuid
|
||||
from datetime import UTC, datetime
|
||||
from datetime import UTC, datetime, timedelta
|
||||
from types import SimpleNamespace
|
||||
|
||||
from sqlalchemy import select
|
||||
@@ -161,6 +161,83 @@ def test_deliver_one_handles_malformed_payload_without_getting_stuck(monkeypatch
|
||||
assert "Malformed outbox payload" in event.last_error
|
||||
|
||||
|
||||
def test_claim_sets_a_lease_deadline():
|
||||
event_id = _make_pending_event("MO-006")
|
||||
settings = get_settings()
|
||||
before = datetime.now(UTC)
|
||||
dispatcher._claim_due_events()
|
||||
|
||||
event = _get_event(event_id)
|
||||
assert event.delivery_status == "delivering"
|
||||
assert event.next_attempt_at is not None
|
||||
lease = settings.n8n_delivery_lease_seconds
|
||||
assert event.next_attempt_at > before + timedelta(seconds=lease - 5)
|
||||
|
||||
|
||||
def test_reclaim_ignores_an_active_unexpired_lease():
|
||||
# A worker that is still within its lease window must not be disturbed -- this is
|
||||
# what prevents double delivery of an event another (still-alive) worker is handling.
|
||||
event_id = _make_pending_event("MO-007")
|
||||
dispatcher._claim_due_events()
|
||||
|
||||
reclaimed = dispatcher._reclaim_stale_deliveries()
|
||||
assert reclaimed == 0
|
||||
assert _get_event(event_id).delivery_status == "delivering"
|
||||
|
||||
|
||||
def test_reclaim_recovers_an_expired_lease_and_preserves_attempts(monkeypatch):
|
||||
# Simulates a process crash: the row was claimed (delivering) but no outcome was ever
|
||||
# recorded, and its lease has since expired.
|
||||
event_id = _make_pending_event("MO-008")
|
||||
dispatcher._claim_due_events()
|
||||
|
||||
db = SessionLocal()
|
||||
try:
|
||||
event = db.scalar(select(OutboxEvent).where(OutboxEvent.event_id == event_id))
|
||||
event.attempts = 2
|
||||
event.next_attempt_at = datetime.now(UTC) - timedelta(seconds=1)
|
||||
db.commit()
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
reclaimed = dispatcher._reclaim_stale_deliveries()
|
||||
assert reclaimed == 1
|
||||
|
||||
event = _get_event(event_id)
|
||||
assert event.delivery_status == "pending"
|
||||
assert event.next_attempt_at is None
|
||||
assert event.attempts == 2
|
||||
assert "stale" in event.last_error.lower()
|
||||
|
||||
# The reclaimed event is now a normal pending event, immediately claimable again.
|
||||
claimed = dispatcher._claim_due_events()
|
||||
assert event_id in claimed
|
||||
|
||||
|
||||
def test_run_dispatch_cycle_recovers_a_stale_lease_before_claiming(monkeypatch):
|
||||
event_id = _make_pending_event("MO-009")
|
||||
dispatcher._claim_due_events()
|
||||
db = SessionLocal()
|
||||
try:
|
||||
event = db.scalar(select(OutboxEvent).where(OutboxEvent.event_id == event_id))
|
||||
event.next_attempt_at = datetime.now(UTC) - timedelta(seconds=1)
|
||||
db.commit()
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
def fake_post(url, json, timeout):
|
||||
return SimpleNamespace(
|
||||
raise_for_status=lambda: None,
|
||||
json=lambda: {"ok": True, "event_id": str(event_id), "result": {}},
|
||||
)
|
||||
|
||||
monkeypatch.setattr(dispatcher.httpx, "post", fake_post)
|
||||
processed = dispatcher.run_dispatch_cycle()
|
||||
|
||||
assert processed >= 1
|
||||
assert _get_event(event_id).delivery_status == "succeeded"
|
||||
|
||||
|
||||
def test_run_dispatch_cycle_end_to_end(monkeypatch):
|
||||
event_id = _make_pending_event("MO-005")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user