feat(worker): add PlanningMergeOrchestrator.AbortAsync

This commit is contained in:
mika kuns
2026-04-24 18:18:49 +02:00
parent 7d87c03cfa
commit b9896399fa
2 changed files with 39 additions and 0 deletions

View File

@@ -82,6 +82,16 @@ public sealed class PlanningMergeOrchestrator
await DrainAsync(planningTaskId, ct);
}
public async Task AbortAsync(string planningTaskId, CancellationToken ct)
{
if (!_states.TryGetValue(planningTaskId, out var state) || state.CurrentSubtaskId is null)
throw new InvalidOperationException("no in-progress merge to abort");
await _merge.AbortMergeAsync(state.CurrentSubtaskId, ct);
_states.TryRemove(planningTaskId, out _);
await _broadcaster.PlanningMergeAborted(planningTaskId);
}
private async Task DrainAsync(string planningTaskId, CancellationToken ct)
{
if (!_states.TryGetValue(planningTaskId, out var state)) return;