Minz/toho-jose
1
0
Fork 0
Code Issues 4 Pull Requests Actions Packages Projects Releases Wiki Activity

Handle node disconnection

Browse Source
This commit is contained in:
Minzkraut
2023-06-01 19:50:21 +02:00
parent eded4865b7
commit 39c052e6bd
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/index.js
View File

@@ -133,7 +133,6 @@ app.post("/jobs/:jobId/completed", async (req, res) => {
res.status(400).send({ 'message': `Job ${jobId} not found` }); res.status(400).send({ 'message': `Job ${jobId} not found` });
}); });
wss.on('connection', function connection(ws) { wss.on('connection', function connection(ws) {
var nodeID = uuid.v4(); var nodeID = uuid.v4();
ws.nodeID = nodeID; ws.nodeID = nodeID;
@@ -155,8 +154,12 @@ wss.on('connection', function connection(ws) {
jobs['queued'][jobResult["jobId"]]['promise'].resolve(jobResult); jobs['queued'][jobResult["jobId"]]['promise'].resolve(jobResult);
} }
}); });
ws.on('close', function(reasonCode, description) {
console.log(`Node ${ws.nodeID} disconnected.`);
delete nodes[ws.nodeID]
}); });
});
app.listen(PORT_WEB, () => { app.listen(PORT_WEB, () => {
console.log(`Job Server API running on ${PORT_WEB}`); console.log(`Job Server API running on ${PORT_WEB}`);