Dokploy/dokploy
1
0
Fork 0
mirror of https://github.com/Dokploy/dokploy synced 2025-06-26 18:27:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor(dokploy): terminate connections

Browse Source
This commit is contained in:
Mauricio Siu 2024-10-27 15:09:27 -06:00
parent 9872c4fee3
commit ae5df82887
1 changed files with 23 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
apps/dokploy/server/wss/docker-container-logs.ts
View File

@ -5,6 +5,11 @@ import { Client } from "ssh2";
import { WebSocketServer } from "ws";
import { getShell } from "./utils";
function heartbeat() {
// @ts-ignore
this.isAlive = true;
}
export const setupDockerContainerLogsWebSocketServer = (
server: http.Server<typeof http.IncomingMessage, typeof http.ServerResponse>,
) => {
@ -44,14 +49,10 @@ export const setupDockerContainerLogsWebSocketServer = (
return;
}
const pingInterval = setInterval(() => {
if (ws.readyState === ws.OPEN) {
ws.ping();
}
}, 30000);
ws.on("close", () => clearInterval(pingInterval));
// @ts-ignore
ws.isAlive = true;
ws.on("error", console.error);
ws.on("pong", heartbeat);
try {
if (serverId) {
const server = await findServerById(serverId);
@ -147,4 +148,18 @@ export const setupDockerContainerLogsWebSocketServer = (
ws.send(errorMessage);
}
});
const interval = setInterval(function ping() {
for (const ws of wssTerm.clients) {
// @ts-ignore
if (ws.isAlive === false) return ws.terminate();
// @ts-ignore
ws.isAlive = false;
ws.ping();
}
}, 30000);
wssTerm.on("close", function close() {
clearInterval(interval);
});
};