add timeouts

Author: jgoux

apps/browser-proxy/src/tcp-server.ts

@@ -26,6 +26,13 @@ tcpServer.on('connection', async (socket) => {
     connectionId: string
   } | null = null
 
+  // 5 minutes idle timeout for the tcp connection
+  socket.setTimeout(1000 * 60 * 5)
+  socket.on('timeout', () => {
+    debug('tcp connection timeout')
+    socket.end()
+  })
+
   debug('new tcp connection')
 
   const connection = await fromNodeSocket(socket, {

apps/browser-proxy/src/websocket-server.ts

@@ -48,6 +48,15 @@ websocketServer.on('error', (error) => {
 websocketServer.on('connection', async (websocket, request) => {
   debug('websocket connection')
 
+  // 1 hour lifetime for the websocket connection
+  const websocketConnectionTimeout = setTimeout(
+    () => {
+      debug('websocket connection timed out')
+      websocket.close()
+    },
+    1000 * 60 * 60 * 1
+  )
+
   const host = request.headers.host
 
   if (!host) {
@@ -101,6 +110,7 @@ websocketServer.on('connection', async (websocket, request) => {
   })
 
   websocket.on('close', () => {
+    clearTimeout(websocketConnectionTimeout)
     connectionManager.deleteWebsocket(databaseId)
     // TODO: have a way of ending a PostgresConnection
     logEvent(new DatabaseUnshared({ databaseId, userId: user.id }))