wehub-resource-sync
71 lines
1.6 KiB
TypeScript
71 lines
1.6 KiB
TypeScript
import net from "node:net";
|
|
|
|
export function sleep(ms: number) {
|
|
return new Promise((resolve) => setTimeout(resolve, ms));
|
|
}
|
|
|
|
export async function waitForTcpPort(
|
|
host: string,
|
|
port: number,
|
|
timeoutMs = 60_000
|
|
): Promise<void> {
|
|
const deadline = Date.now() + timeoutMs;
|
|
let lastError: unknown;
|
|
|
|
while (Date.now() < deadline) {
|
|
try {
|
|
await new Promise<void>((resolve, reject) => {
|
|
const socket = net.createConnection({ host, port });
|
|
socket.setTimeout(2000);
|
|
socket.once("connect", () => {
|
|
socket.end();
|
|
resolve();
|
|
});
|
|
socket.once("timeout", () => {
|
|
socket.destroy();
|
|
reject(new Error(`timeout waiting for ${host}:${port}`));
|
|
});
|
|
socket.once("error", reject);
|
|
});
|
|
return;
|
|
} catch (error) {
|
|
lastError = error;
|
|
await sleep(1000);
|
|
}
|
|
}
|
|
|
|
throw new Error(
|
|
`timed out waiting for ${host}:${port}: ${
|
|
lastError instanceof Error ? lastError.message : String(lastError)
|
|
}`
|
|
);
|
|
}
|
|
|
|
export async function waitForHttpStatus(
|
|
url: string,
|
|
expectedStatus = 200,
|
|
timeoutMs = 60_000
|
|
): Promise<void> {
|
|
const deadline = Date.now() + timeoutMs;
|
|
let lastStatus = 0;
|
|
let lastError = "";
|
|
|
|
while (Date.now() < deadline) {
|
|
try {
|
|
const response = await fetch(url);
|
|
lastStatus = response.status;
|
|
if (response.status === expectedStatus) {
|
|
return;
|
|
}
|
|
} catch (error) {
|
|
lastError = error instanceof Error ? error.message : String(error);
|
|
}
|
|
await sleep(1000);
|
|
}
|
|
|
|
throw new Error(
|
|
`timed out waiting for ${url}; last status=${lastStatus}; last error=${lastError}`
|
|
);
|
|
}
|
|
|