diff --git a/apps/coordinator/src/index.ts b/apps/coordinator/src/index.ts index 385f0b2a8..45d72023c 100644 --- a/apps/coordinator/src/index.ts +++ b/apps/coordinator/src/index.ts @@ -502,9 +502,12 @@ class TaskCoordinator { reason: { type: "RETRYING_AFTER_FAILURE", attemptNumber: execution.attempt.number, + // TODO: attach completion data here }, }); + // TODO: replace this with + // callback({ didCheckpoint: true, shouldExit: false }); confirmCompletion({ didCheckpoint: true, shouldExit: false }); }); @@ -537,6 +540,12 @@ class TaskCoordinator { return; } + if (!checkpoint.docker) { + socket.emit("REQUEST_EXIT", { + version: "v1", + }); + } + this.#platformSocket?.send("CHECKPOINT_CREATED", { version: "v1", attemptId: socket.data.attemptId, @@ -573,6 +582,12 @@ class TaskCoordinator { return; } + if (!checkpoint.docker) { + socket.emit("REQUEST_EXIT", { + version: "v1", + }); + } + this.#platformSocket?.send("CHECKPOINT_CREATED", { version: "v1", attemptId: socket.data.attemptId, @@ -609,6 +624,12 @@ class TaskCoordinator { return; } + if (!checkpoint.docker) { + socket.emit("REQUEST_EXIT", { + version: "v1", + }); + } + this.#platformSocket?.send("CHECKPOINT_CREATED", { version: "v1", attemptId: socket.data.attemptId,