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: {
color: 16711680,
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: {
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: {
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", () => {
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...`);
this.connections.delete(server);