Delete unfinished jobs on connection close

This commit is contained in:
TheEssem 2021-06-27 17:06:52 -05:00
parent 137120b4b7
commit 0fbc165154
No known key found for this signature in database
GPG key ID: A3F9F02129092FCA
2 changed files with 6 additions and 3 deletions

View file

@ -105,9 +105,9 @@ module.exports = async (client, cluster, ipc, message) => {
embed: { embed: {
color: 16711680, color: 16711680,
title: "Here's your image!", title: "Here's your image!",
url: `${process.env.TMP_DOMAIN == "" ? "https://projectlounge.pw/tmp" : process.env.TMP_DOMAIN}/${filename}`, url: `${process.env.TMP_DOMAIN == "" ? "https://tmp.projectlounge.pw" : process.env.TMP_DOMAIN}/${filename}`,
image: { image: {
url: `${process.env.TMP_DOMAIN == "" ? "https://projectlounge.pw/tmp" : process.env.TMP_DOMAIN}/${filename}` url: `${process.env.TMP_DOMAIN == "" ? "https://tmp.projectlounge.pw" : process.env.TMP_DOMAIN}/${filename}`
}, },
footer: { footer: {
text: "The result image was more than 8MB in size, so it was uploaded to an external site instead." text: "The result image was more than 8MB in size, so it was uploaded to an external site instead."

View file

@ -98,7 +98,10 @@ exports.connect = async (server) => {
}); });
connection.once("close", () => { connection.once("close", () => {
for (const uuid of Object.keys(this.jobs)) { for (const uuid of Object.keys(this.jobs)) {
if (this.jobs[uuid].addr === server) this.jobs[uuid].event.emit("error", "Job ended prematurely due to a closed connection; please run your image job again"); if (this.jobs[uuid].addr === server) {
this.jobs[uuid].event.emit("error", "Job ended prematurely due to a closed connection; please run your image job again");
delete this.jobs[uuid];
}
} }
//logger.log(`Lost connection to ${server}, attempting to reconnect...`); //logger.log(`Lost connection to ${server}, attempting to reconnect...`);
this.connections.delete(server); this.connections.delete(server);