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
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 { WebSocketServer } from "ws";
import { getShell } from "./utils"; import { getShell } from "./utils";
function heartbeat() {
// @ts-ignore
this.isAlive = true;
}
export const setupDockerContainerLogsWebSocketServer = ( export const setupDockerContainerLogsWebSocketServer = (
server: http.Server<typeof http.IncomingMessage, typeof http.ServerResponse>, server: http.Server<typeof http.IncomingMessage, typeof http.ServerResponse>,
) => { ) => {
@@ -44,14 +49,10 @@ export const setupDockerContainerLogsWebSocketServer = (
return; return;
} }
const pingInterval = setInterval(() => { // @ts-ignore
if (ws.readyState === ws.OPEN) { ws.isAlive = true;
ws.ping(); ws.on("error", console.error);
} ws.on("pong", heartbeat);
}, 30000);
ws.on("close", () => clearInterval(pingInterval));
try { try {
if (serverId) { if (serverId) {
const server = await findServerById(serverId); const server = await findServerById(serverId);
@@ -147,4 +148,18 @@ export const setupDockerContainerLogsWebSocketServer = (
ws.send(errorMessage); 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);
});
}; };