add tcp timeout, fix worker stuck

components/DetailBar.tsx

@@ -1,6 +1,5 @@
 import { MonitorState, MonitorTarget } from "@/uptime.types";
 import { Box, Tooltip } from "@mantine/core";
-import { useEffect, useRef } from "react";
 
 
 export default function DetailBar({ monitor, state }: { monitor: MonitorTarget, state: MonitorState }) {  

components/MonitorDetail.tsx

@@ -22,8 +22,8 @@ export default function MonitorDetail({ monitor, state }: { monitor: MonitorTarg
 
   if (!state.latency[monitor.id]) return (
     <>
-      <Text>{monitor.name}</Text>
-      <Text>No data available, please make sure you have deployed your workers with latest config and check your worker status!</Text>
+      <Text mt='sm' fw={700}>{monitor.name}</Text>
+      <Text mt='sm' fw={700}>No data available, please make sure you have deployed your workers with latest config and check your worker status!</Text>
     </>
   )
 

worker/src/index.ts

@@ -1,6 +1,6 @@
 import { connect } from 'cloudflare:sockets'
 import config from '../../uptime.config'
-import { fetchTimeout, getWorkerLocation } from './util'
+import { fetchTimeout, getWorkerLocation, withTimeout } from './util'
 import { MonitorState, MonitorTarget } from "../../uptime.types"
 
 export interface Env {
@@ -30,7 +30,7 @@ async function getStatus(monitor: MonitorTarget): Promise<{ ping: number; up: bo
 			// Can't do this: await socket.close()
 
 			// https://github.com/cloudflare/workerd/issues/1305
-			await socket.closed
+			await withTimeout(monitor.timeout || 10000, socket.closed)
 			
 			console.log(`${monitor.name} connected to ${monitor.target}`)
 

worker/src/util.ts

@@ -14,4 +14,14 @@ const fetchTimeout = (url: string, ms: number, { signal, ...options }: RequestIn
   return promise.finally(() => clearTimeout(timeout))
 }
 
-export { getWorkerLocation, fetchTimeout }
+function withTimeout<T> (millis: number, promise: Promise<T>): Promise<T> {
+  const timeout = new Promise<T>((resolve, reject) =>
+  setTimeout(() => reject(new Error(`Promise timed out after ${millis}ms`)), millis))
+
+  return Promise.race([
+    promise,
+    timeout
+  ])
+}
+
+export { getWorkerLocation, fetchTimeout, withTimeout }