Set the checkpointEventId in marqs when the checkpoint is created for batchTriggerAndWait

This should fix a horrible race condition when at max concurrency
2024-09-12 13:29:17 +01:00
parent c183416ae0
commit 7850e442c8
1 changed files with 10 additions and 0 deletions
@@ -258,6 +258,16 @@ export class CreateCheckpointService extends BaseService {
            };
          }

+          //if there's a message in the queue, we make sure the checkpoint event is on it
+          await marqs?.replaceMessage(
+            attempt.taskRun.id,
+            {
+              checkpointEventId: checkpointEvent.id,
+            },
+            undefined,
+            true
+          );
+
          await ResumeBatchRunService.enqueue(batchRun.id, this._prisma);

          return {