Spaces:
Running
Running
| import net from "node:net"; | |
| export interface TcpProbeOptions { | |
| host: string; | |
| port: number; | |
| timeoutMs?: number; | |
| } | |
| export async function tcpProbe({ | |
| host, | |
| port, | |
| timeoutMs = 1500, | |
| }: TcpProbeOptions): Promise<boolean> { | |
| return new Promise<boolean>((resolve) => { | |
| const socket = new net.Socket(); | |
| let settled = false; | |
| const finish = (ok: boolean) => { | |
| if (settled) return; | |
| settled = true; | |
| socket.destroy(); | |
| resolve(ok); | |
| }; | |
| socket.setTimeout(timeoutMs); | |
| socket.once("connect", () => finish(true)); | |
| socket.once("timeout", () => finish(false)); | |
| socket.once("error", () => finish(false)); | |
| socket.connect(port, host); | |
| }); | |
| } | |
| export interface HttpProbeOptions { | |
| url: string; | |
| timeoutMs?: number; | |
| expectStatus?: number; | |
| } | |
| export async function httpProbe({ | |
| url, | |
| timeoutMs = 2000, | |
| expectStatus, | |
| }: HttpProbeOptions): Promise<{ ok: boolean; status?: number; body?: string }> { | |
| const controller = new AbortController(); | |
| const timer = setTimeout(() => controller.abort(), timeoutMs); | |
| try { | |
| const response = await fetch(url, { signal: controller.signal }); | |
| const body = await response.text(); | |
| const status = response.status; | |
| const matches = expectStatus !== undefined ? status === expectStatus : status < 500; | |
| return { ok: matches, status, body }; | |
| } catch { | |
| return { ok: false }; | |
| } finally { | |
| clearTimeout(timer); | |
| } | |
| } | |