exit process after checkpointing

This commit is contained in:
nicktrn
2024-03-20 14:38:21 +00:00
parent 84250a6f9e
commit 18bd7f4623
+21
View File
@@ -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,