91 lines
2.3 KiB
TypeScript
91 lines
2.3 KiB
TypeScript
import { logger } from '../utils/logger';
|
|
|
|
type HealthResult = { ok: boolean; error?: string };
|
|
|
|
export class GatewayConnectionMonitor {
|
|
private pingInterval: NodeJS.Timeout | null = null;
|
|
private pongTimeout: NodeJS.Timeout | null = null;
|
|
private healthCheckInterval: NodeJS.Timeout | null = null;
|
|
|
|
startPing(
|
|
sendPing: () => void,
|
|
onPongTimeout?: () => void,
|
|
intervalMs = 30000,
|
|
timeoutMs = 15000,
|
|
): void {
|
|
if (this.pingInterval) {
|
|
clearInterval(this.pingInterval);
|
|
}
|
|
if (this.pongTimeout) {
|
|
clearTimeout(this.pongTimeout);
|
|
this.pongTimeout = null;
|
|
}
|
|
|
|
this.pingInterval = setInterval(() => {
|
|
sendPing();
|
|
|
|
if (onPongTimeout) {
|
|
if (this.pongTimeout) {
|
|
clearTimeout(this.pongTimeout);
|
|
}
|
|
this.pongTimeout = setTimeout(() => {
|
|
this.pongTimeout = null;
|
|
onPongTimeout();
|
|
}, timeoutMs);
|
|
}
|
|
}, intervalMs);
|
|
}
|
|
|
|
handlePong(): void {
|
|
if (this.pongTimeout) {
|
|
clearTimeout(this.pongTimeout);
|
|
this.pongTimeout = null;
|
|
}
|
|
}
|
|
|
|
startHealthCheck(options: {
|
|
shouldCheck: () => boolean;
|
|
checkHealth: () => Promise<HealthResult>;
|
|
onUnhealthy: (errorMessage: string) => void;
|
|
onError: (error: unknown) => void;
|
|
intervalMs?: number;
|
|
}): void {
|
|
if (this.healthCheckInterval) {
|
|
clearInterval(this.healthCheckInterval);
|
|
}
|
|
|
|
this.healthCheckInterval = setInterval(async () => {
|
|
if (!options.shouldCheck()) {
|
|
return;
|
|
}
|
|
|
|
try {
|
|
const health = await options.checkHealth();
|
|
if (!health.ok) {
|
|
const errorMessage = health.error ?? 'Health check failed';
|
|
logger.warn(`Gateway health check failed: ${errorMessage}`);
|
|
options.onUnhealthy(errorMessage);
|
|
}
|
|
} catch (error) {
|
|
logger.error('Gateway health check error:', error);
|
|
options.onError(error);
|
|
}
|
|
}, options.intervalMs ?? 30000);
|
|
}
|
|
|
|
clear(): void {
|
|
if (this.pingInterval) {
|
|
clearInterval(this.pingInterval);
|
|
this.pingInterval = null;
|
|
}
|
|
if (this.pongTimeout) {
|
|
clearTimeout(this.pongTimeout);
|
|
this.pongTimeout = null;
|
|
}
|
|
if (this.healthCheckInterval) {
|
|
clearInterval(this.healthCheckInterval);
|
|
this.healthCheckInterval = null;
|
|
}
|
|
}
|
|
}
|