@autofleet/nitur 2.4.1-beta.0 → 2.4.1

package/lib/health-manager/health-manager.cjs

@@ -1,2 +1,2 @@
-const e=require(`../_virtual/rolldown_runtime.cjs`),t=require(`./types.cjs`),n=require(`./client-defaults.cjs`);let r=require(`node:timers/promises`),i=require(`@autofleet/errors`),a=require(`http-terminator`);var o=class{constructor(e){this.DEFAULT_CONFIG={shutdownDelayMs:6e3,useExit0:!0,shouldSetupSignalHandlers:!0,enableLivenessOptimization:!0},this.isShuttingDown=!1,this.isReady=!1,this.onReadyTriggered=!1,this.clientsMap=new Map,this.livenessChecks=[],this.readinessChecks=[],this.canSafelySkipLivenessWhenReady=(e,t)=>e?!!t:!0,this.config={...this.DEFAULT_CONFIG,...e.config??{}},this.logger=e.logger,this.storeClientsWithDefaults(e),this.initializeHealthChecks(),this.config.shouldSetupSignalHandlers&&this.setupSignalHandlers()}storeClientsWithDefaults(e){let r=e.clients;if(r){for(let[e,i]of Object.entries(r))if(i)if(e===t.CLIENT_TYPES.SERVER)this.attachServer(i);else{let r=Array.isArray(i)?i:[i],a=n.CLIENT_DEFAULTS_MAP[e]??n.CLIENT_DEFAULTS_MAP[t.CLIENT_TYPES.CUSTOM],o=r.map(t=>({connection:t.connection,healthCheck:t.healthCheck??a?.healthCheck,shutdownHandlers:{...a?.shutdownHandlers,...t.shutdownHandlers},options:{...a?.options,...t.options,name:t.options?.name||a?.options?.name||e}}));this.clientsMap.set(e,o)}}}setupSignalHandlers(){this.logger.info(`Setting up global signal listeners for graceful shutdown`),process.on(`SIGTERM`,async()=>{await this.shutdown(`SIGTERM`)}),process.on(`SIGINT`,async()=>{await this.shutdown(`SIGINT`)})}initializeHealthChecks(){let e=!0;this.clientsMap.forEach(t=>{t.forEach(t=>{if(!t.healthCheck)return;let n=t.options?.includeInLivenessCheck,r=t.options?.includeInReadinessCheck;this.canSafelySkipLivenessWhenReady(n,r)||(e=!1),n&&this.livenessChecks.push({check:t.healthCheck,client:t.connection,name:t.options?.name||`unknown`,customHandlersPayload:t.options?.customHandlersPayload}),r&&this.readinessChecks.push({check:t.healthCheck,client:t.connection,name:t.options?.name||`unknown`,customHandlersPayload:t.options?.customHandlersPayload})})}),this.config.enableLivenessOptimization=this.config.enableLivenessOptimization&&e,this.config.enableLivenessOptimization?this.logger.info(`Liveness check optimization enabled - liveness checks will be skipped when pod is ready`):e||this.logger.warn(`Liveness check optimization disabled - some clients are in liveness checks but not in readiness checks`)}attachServer(e){let r=n.CLIENT_DEFAULTS_MAP[t.CLIENT_TYPES.SERVER],i=(Array.isArray(e)?e:[e]).map(e=>{let t=(0,a.createHttpTerminator)({server:e.connection});return{connection:e.connection,healthCheck:e.healthCheck??r?.healthCheck,shutdownHandlers:{...r?.shutdownHandlers,...e.shutdownHandlers},options:{...r?.options,...e.options,name:e.options?.name||r?.options?.name||`HTTP Server`,customHandlersPayload:{...e.options?.customHandlersPayload,httpTerminator:t}}}});this.clientsMap.set(t.CLIENT_TYPES.SERVER,i)}async aliveCheck(){if(this.isShuttingDown||this.config.enableLivenessOptimization&&this.isReady)return{statusCode:200,status:`ok`};try{return await Promise.all(this.livenessChecks.map(({check:e,client:t,name:n,customHandlersPayload:r})=>e({client:t,name:n,customHandlersPayload:r,logger:this.logger}))),{statusCode:200,status:`ok`}}catch(e){let t=new i.ServiceUnavailableError(`Liveness probe failed: ${e.message}`);throw t.stack=e.stack,t}}async readyCheck(){if(this.isShuttingDown)throw this.isReady=!1,this.logger.warn(`Application is shutting down: readiness probe failed`),new i.ServiceUnavailableError(`Readiness probe failed: Application is shutting down`);try{return await Promise.all(this.readinessChecks.map(({check:e,client:t,name:n,customHandlersPayload:r})=>e({client:t,name:n,customHandlersPayload:r,logger:this.logger}))),this.onReadyTriggered||(this.onReadyTriggered=!0,await this.config.onReady?.(),this.logger.info(`onReady hook executed successfully`)),this.isReady=!0,{statusCode:200,status:`ok`}}catch(e){this.isReady=!1;let t=new i.ServiceUnavailableError(`Readiness probe failed: ${e.message}`);throw t.stack=e.stack,t}}groupShutdownTasksByStage(e){let n={[t.SHUTDOWN_STAGES.STOP_ACCEPTING_WORK]:[],[t.SHUTDOWN_STAGES.DRAIN]:[],[t.SHUTDOWN_STAGES.CLOSE]:[]};return this.clientsMap.forEach(t=>{t.forEach(t=>{let r=t.options?.name||`unknown`,i=t.options?.enabledShutdownStages;if(t.shutdownHandlers)for(let[a,o]of Object.entries(t.shutdownHandlers))o&&i?.includes(a)&&n[a].push({fn:()=>o({signal:e,client:t.connection,name:r,customHandlersPayload:t.options?.customHandlersPayload,logger:this.logger}),name:r})})}),n}async executeShutdownStage(e,t){t.length!==0&&(this.logger.info(`[${e} stage] Start executing shutdown stage ${e}...`),(await Promise.allSettled(t.map(e=>e.fn()))).forEach((n,r)=>{if(n.status===`rejected`){let{name:i}=t[r];this.logger.error(`[${e} stage] Failed to shutdown '${i}'`,{error:n.reason})}}),this.logger.info(`[${e} stage] Shutdown stage ${e} completed`))}async shutdown(e=`MANUAL`){try{if(this.isShuttingDown){this.logger.warn(`Shutdown already in progress`);return}let{shutdownDelayMs:n,useExit0:i,onShutdown:a,beforeShutdown:o}=this.config;this.logger.info(`${e} received: setting readiness state to false`),this.isShuttingDown=!0,this.isReady=!1,await o?.(),this.logger.info(`Waiting ${n}ms before shutdown...`),await(0,r.setTimeout)(n);let s=this.groupShutdownTasksByStage(e);await this.executeShutdownStage(t.SHUTDOWN_STAGES.STOP_ACCEPTING_WORK,s[t.SHUTDOWN_STAGES.STOP_ACCEPTING_WORK]),this.logger.info(`Waiting 4000ms to allow in-flight work to complete...`),await(0,r.setTimeout)(4e3),await this.executeShutdownStage(t.SHUTDOWN_STAGES.DRAIN,s[t.SHUTDOWN_STAGES.DRAIN]),await this.executeShutdownStage(t.SHUTDOWN_STAGES.CLOSE,s[t.SHUTDOWN_STAGES.CLOSE]),await a?.(),this.logger.info(`Shutdown process completed`),i&&(this.logger.info(`Exiting process after ${e}`),process.exit(0))}catch(t){this.logger.error(`Error during shutdown process, forcing process exit despite shutdown error`,{error:t,signal:e}),process.exit(1)}}};exports.HealthManager=o;
+const e=require(`../_virtual/rolldown_runtime.cjs`),t=require(`./types.cjs`),n=require(`./client-defaults.cjs`);let r=require(`node:timers/promises`),i=require(`@autofleet/errors`),a=require(`http-terminator`);var o=class{constructor(e){this.DEFAULT_CONFIG={shutdownDelayMs:6e3,useExit0:!0,shouldSetupSignalHandlers:!0,enableLivenessOptimization:!0},this.isShuttingDown=!1,this.isReady=!1,this.clientsMap=new Map,this.livenessChecks=[],this.readinessChecks=[],this.canSafelySkipLivenessWhenReady=(e,t)=>e?!!t:!0,this.config={...this.DEFAULT_CONFIG,...e.config??{}},this.logger=e.logger,this.storeClientsWithDefaults(e),this.initializeHealthChecks(),this.config.shouldSetupSignalHandlers&&this.setupSignalHandlers()}storeClientsWithDefaults(e){let r=e.clients;if(r){for(let[e,i]of Object.entries(r))if(i)if(e===t.CLIENT_TYPES.SERVER)this.attachServer(i);else{let r=Array.isArray(i)?i:[i],a=n.CLIENT_DEFAULTS_MAP[e]??n.CLIENT_DEFAULTS_MAP[t.CLIENT_TYPES.CUSTOM],o=r.map(t=>({connection:t.connection,healthCheck:t.healthCheck??a?.healthCheck,shutdownHandlers:{...a?.shutdownHandlers,...t.shutdownHandlers},options:{...a?.options,...t.options,name:t.options?.name||a?.options?.name||e}}));this.clientsMap.set(e,o)}}}setupSignalHandlers(){this.logger.info(`Setting up global signal listeners for graceful shutdown`),process.on(`SIGTERM`,async()=>{await this.shutdown(`SIGTERM`)}),process.on(`SIGINT`,async()=>{await this.shutdown(`SIGINT`)})}initializeHealthChecks(){let e=!0;this.clientsMap.forEach(t=>{t.forEach(t=>{if(!t.healthCheck)return;let n=t.options?.includeInLivenessCheck,r=t.options?.includeInReadinessCheck;this.canSafelySkipLivenessWhenReady(n,r)||(e=!1),n&&this.livenessChecks.push({check:t.healthCheck,client:t.connection,name:t.options?.name||`unknown`,customHandlersPayload:t.options?.customHandlersPayload}),r&&this.readinessChecks.push({check:t.healthCheck,client:t.connection,name:t.options?.name||`unknown`,customHandlersPayload:t.options?.customHandlersPayload})})}),this.config.enableLivenessOptimization=this.config.enableLivenessOptimization&&e,this.config.enableLivenessOptimization?this.logger.info(`Liveness check optimization enabled - liveness checks will be skipped when pod is ready`):e||this.logger.warn(`Liveness check optimization disabled - some clients are in liveness checks but not in readiness checks`)}attachServer(e){let r=n.CLIENT_DEFAULTS_MAP[t.CLIENT_TYPES.SERVER],i=(Array.isArray(e)?e:[e]).map(e=>{let t=(0,a.createHttpTerminator)({server:e.connection});return{connection:e.connection,healthCheck:e.healthCheck??r?.healthCheck,shutdownHandlers:{...r?.shutdownHandlers,...e.shutdownHandlers},options:{...r?.options,...e.options,name:e.options?.name||r?.options?.name||`HTTP Server`,customHandlersPayload:{...e.options?.customHandlersPayload,httpTerminator:t}}}});this.clientsMap.set(t.CLIENT_TYPES.SERVER,i)}async aliveCheck(){if(this.isShuttingDown||this.config.enableLivenessOptimization&&this.isReady)return{statusCode:200,status:`ok`};try{return await Promise.all(this.livenessChecks.map(({check:e,client:t,name:n,customHandlersPayload:r})=>e({client:t,name:n,customHandlersPayload:r,logger:this.logger}))),{statusCode:200,status:`ok`}}catch(e){let t=new i.ServiceUnavailableError(`Liveness probe failed: ${e.message}`);throw t.stack=e.stack,t}}async readyCheck(){if(this.isShuttingDown)throw this.isReady=!1,this.logger.warn(`Application is shutting down: readiness probe failed`),new i.ServiceUnavailableError(`Readiness probe failed: Application is shutting down`);try{return await Promise.all(this.readinessChecks.map(({check:e,client:t,name:n,customHandlersPayload:r})=>e({client:t,name:n,customHandlersPayload:r,logger:this.logger}))),this.isReady=!0,{statusCode:200,status:`ok`}}catch(e){this.isReady=!1;let t=new i.ServiceUnavailableError(`Readiness probe failed: ${e.message}`);throw t.stack=e.stack,t}}groupShutdownTasksByStage(e){let n={[t.SHUTDOWN_STAGES.STOP_ACCEPTING_WORK]:[],[t.SHUTDOWN_STAGES.DRAIN]:[],[t.SHUTDOWN_STAGES.CLOSE]:[]};return this.clientsMap.forEach(t=>{t.forEach(t=>{let r=t.options?.name||`unknown`,i=t.options?.enabledShutdownStages;if(t.shutdownHandlers)for(let[a,o]of Object.entries(t.shutdownHandlers))o&&i?.includes(a)&&n[a].push({fn:()=>o({signal:e,client:t.connection,name:r,customHandlersPayload:t.options?.customHandlersPayload,logger:this.logger}),name:r})})}),n}async executeShutdownStage(e,t){t.length!==0&&(this.logger.info(`[${e} stage] Start executing shutdown stage ${e}...`),(await Promise.allSettled(t.map(e=>e.fn()))).forEach((n,r)=>{if(n.status===`rejected`){let{name:i}=t[r];this.logger.error(`[${e} stage] Failed to shutdown '${i}'`,{error:n.reason})}}),this.logger.info(`[${e} stage] Shutdown stage ${e} completed`))}async shutdown(e=`MANUAL`){try{if(this.isShuttingDown){this.logger.warn(`Shutdown already in progress`);return}let{shutdownDelayMs:n,useExit0:i,onShutdown:a,beforeShutdown:o}=this.config;this.logger.info(`${e} received: setting readiness state to false`),this.isShuttingDown=!0,this.isReady=!1,await o?.(),this.logger.info(`Waiting ${n}ms before shutdown...`),await(0,r.setTimeout)(n);let s=this.groupShutdownTasksByStage(e);await this.executeShutdownStage(t.SHUTDOWN_STAGES.STOP_ACCEPTING_WORK,s[t.SHUTDOWN_STAGES.STOP_ACCEPTING_WORK]),this.logger.info(`Waiting 4000ms to allow in-flight work to complete...`),await(0,r.setTimeout)(4e3),await this.executeShutdownStage(t.SHUTDOWN_STAGES.DRAIN,s[t.SHUTDOWN_STAGES.DRAIN]),await this.executeShutdownStage(t.SHUTDOWN_STAGES.CLOSE,s[t.SHUTDOWN_STAGES.CLOSE]),await a?.(),this.logger.info(`Shutdown process completed`),i&&(this.logger.info(`Exiting process after ${e}`),process.exit(0))}catch(t){this.logger.error(`Error during shutdown process, forcing process exit despite shutdown error`,{error:t,signal:e}),process.exit(1)}}};exports.HealthManager=o;
 //# sourceMappingURL=health-manager.cjs.map

package/lib/health-manager/health-manager.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"health-manager.cjs","names":["CLIENT_TYPES","CLIENT_DEFAULTS_MAP","client","error: any","ServiceUnavailableError","shutdownTasksByStage: Record<ShutdownStage, { fn: () => Promise<any>; name: string; }[]>","SHUTDOWN_STAGES"],"sources":["../../src/health-manager/health-manager.ts"],"sourcesContent":["import { setTimeout } from 'node:timers/promises';\nimport { ServiceUnavailableError } from '@autofleet/errors';\nimport type { LoggerInstanceManager } from '@autofleet/logger';\nimport { createHttpTerminator } from 'http-terminator';\nimport {\n  type ClientConfig,\n  type HealthManagerConfig,\n  type HealthManagerOptions,\n  type ClientType,\n  type HealthCheckHandler,\n  type ShutdownStage,\n  type ServerConfig,\n  CLIENT_TYPES,\n  SHUTDOWN_STAGES,\n  type ShutdownHandler,\n} from './types';\nimport { CLIENT_DEFAULTS_MAP } from './client-defaults';\n\n/**\n * Manages application health state and graceful shutdown for Kubernetes liveness/readiness probes\n *\n * Features:\n * - Provides aliveCheck() and readyCheck() methods for liveness/readiness probes\n * - Auto-registers SIGTERM/SIGINT handlers for graceful shutdown\n * - Manages ready state during shutdown to fail readiness probes\n */\nexport class HealthManager {\n  private readonly DEFAULT_CONFIG: HealthManagerConfig = {\n    shutdownDelayMs: 6_000,\n    useExit0: true,\n    shouldSetupSignalHandlers: true,\n    enableLivenessOptimization: true,\n  };\n\n  private isShuttingDown = false;\n  private isReady = false;\n  private onReadyTriggered = false;\n\n  private config: HealthManagerConfig;\n\n  private clientsMap = new Map<string, ClientConfig<any>[]>();\n\n  private livenessChecks: { check: HealthCheckHandler<any>; client: any; name: string; customHandlersPayload?: any; }[] = [];\n  private readinessChecks: { check: HealthCheckHandler<any>; client: any; name: string; customHandlersPayload?: any; }[] = [];\n\n  private logger: LoggerInstanceManager;\n\n  /**\n   * Creates a new HealthManager instance\n   *\n   * @param params Configuration options for health checks and shutdown behavior\n   */\n  constructor(params: HealthManagerOptions) {\n    this.config = {\n      ...this.DEFAULT_CONFIG,\n      ...(params.config ?? {}),\n    };\n\n    this.logger = params.logger;\n\n    this.storeClientsWithDefaults(params);\n\n    this.initializeHealthChecks();\n\n    if (this.config.shouldSetupSignalHandlers) {\n      this.setupSignalHandlers();\n    }\n  }\n\n  /**\n   * Store clients with default health check and shutdown functions\n   * @private\n   */\n  private storeClientsWithDefaults(params: HealthManagerOptions): void {\n    const clients = params.clients;\n    if (!clients) {\n      return;\n    }\n\n    for (const [type, client] of Object.entries(clients)) {\n      if (!client) {\n        continue;\n      }\n\n      if (type === CLIENT_TYPES.SERVER) {\n        this.attachServer(client as ServerConfig | ServerConfig[]);\n      } else {\n        const clientsArray = Array.isArray(client) ? client : [client];\n        const defaults = CLIENT_DEFAULTS_MAP[type as ClientType] ?? CLIENT_DEFAULTS_MAP[CLIENT_TYPES.CUSTOM];\n\n        const clientsWithDefaults = clientsArray.map(client => ({\n          connection: client.connection,\n          healthCheck: client.healthCheck ?? defaults?.healthCheck,\n          shutdownHandlers: {\n            ...defaults?.shutdownHandlers,\n            ...client.shutdownHandlers,\n          },\n          options: {\n            ...defaults?.options,\n            ...client.options,\n            name: client.options?.name || defaults?.options?.name || type,\n          },\n        }));\n\n        this.clientsMap.set(type, clientsWithDefaults);\n      }\n    }\n  }\n\n  /**\n   * Register SIGTERM and SIGINT signal listeners\n   * @private\n   */\n  private setupSignalHandlers(): void {\n    this.logger.info('Setting up global signal listeners for graceful shutdown');\n\n    process.on('SIGTERM', async () => {\n      await this.shutdown('SIGTERM');\n    });\n\n    process.on('SIGINT', async () => {\n      await this.shutdown('SIGINT');\n    });\n  }\n\n  /**\n   * Check if liveness optimization is safe to enable for a given client configuration\n   * If a client is in liveness but not readiness, we cannot skip liveness checks\n   * because the optimization relies on readiness passing, but this client isn't checked in readiness\n   * @private\n   */\n  private canSafelySkipLivenessWhenReady = (includeInLiveness: boolean | undefined, includeInReadiness: boolean | undefined): boolean => {\n    // If not in liveness, optimization is safe (skipping liveness doesn't affect this client)\n    if (!includeInLiveness) {\n      return true;\n    }\n\n    // If in liveness, must also be in readiness for liveness optimization to be safe\n    return !!includeInReadiness;\n  };\n\n  /**\n   * Initialize health check arrays based on configured clients\n   * @private\n   */\n  private initializeHealthChecks(): void {\n    let allClientsAllowLivenessOptimization = true;\n\n    this.clientsMap.forEach((clients) => {\n      clients.forEach((client) => {\n        if (!client.healthCheck) {\n          return;\n        }\n\n        const includeInLiveness = client.options?.includeInLivenessCheck;\n        const includeInReadiness = client.options?.includeInReadinessCheck;\n\n        // Check if this client configuration allows liveness optimization\n        if (!this.canSafelySkipLivenessWhenReady(includeInLiveness, includeInReadiness)) {\n          allClientsAllowLivenessOptimization = false;\n        }\n\n        if (includeInLiveness) {\n          this.livenessChecks.push({\n            check: client.healthCheck,\n            client: client.connection,\n            name: client.options?.name || 'unknown',\n            customHandlersPayload: client.options?.customHandlersPayload,\n          });\n        }\n\n        if (includeInReadiness) {\n          this.readinessChecks.push({\n            check: client.healthCheck,\n            client: client.connection,\n            name: client.options?.name || 'unknown',\n            customHandlersPayload: client.options?.customHandlersPayload,\n          });\n        }\n      });\n    });\n\n    // Apply optimization setting: only enable if user requested it AND all clients allow it\n    this.config.enableLivenessOptimization = this.config.enableLivenessOptimization && allClientsAllowLivenessOptimization;\n\n    // Log optimization status\n    if (this.config.enableLivenessOptimization) {\n      this.logger.info('Liveness check optimization enabled - liveness checks will be skipped when pod is ready');\n    } else if (!allClientsAllowLivenessOptimization) {\n      this.logger.warn('Liveness check optimization disabled - some clients are in liveness checks but not in readiness checks');\n    }\n  }\n\n  /**\n   * Attach an HTTP or HTTPS server to be managed during shutdown\n   * @param serverConfig Server configuration (ClientConfig with HTTP/HTTPS server)\n   */\n  attachServer(serverConfig: ServerConfig | ServerConfig[]): void {\n    const serverDefaults = CLIENT_DEFAULTS_MAP[CLIENT_TYPES.SERVER];\n    const serverConfigArray = Array.isArray(serverConfig) ? serverConfig : [serverConfig];\n\n    const serverClients = serverConfigArray.map((config) => {\n      // Create httpTerminator for this server instance\n      const httpTerminator = createHttpTerminator({\n        server: config.connection,\n      });\n\n      return {\n        connection: config.connection,\n        healthCheck: config.healthCheck ?? serverDefaults?.healthCheck,\n        shutdownHandlers: {\n          ...serverDefaults?.shutdownHandlers,\n          ...config.shutdownHandlers,\n        },\n        options: {\n          ...serverDefaults?.options,\n          ...config.options,\n          name: config.options?.name || serverDefaults?.options?.name || 'HTTP Server',\n          customHandlersPayload: {\n            ...config.options?.customHandlersPayload,\n            httpTerminator,\n          },\n        },\n      };\n    });\n\n    this.clientsMap.set(CLIENT_TYPES.SERVER, serverClients);\n  }\n\n  /**\n   * Performs liveness health checks for clients configured with includeInLivenessCheck\n   *\n   * @returns Promise that resolves to { status: 'ok', statusCode: 200 } on success\n   * @throws ServiceUnavailableError with statusCode 503 if any health check fails\n   */\n  async aliveCheck(): Promise<{ status: string; statusCode: number; }> {\n    if (this.isShuttingDown) {\n      return { statusCode: 200, status: 'ok' };\n    }\n\n    // Skip liveness check if optimization is enabled and the readiness check has already passed\n    if (this.config.enableLivenessOptimization && this.isReady) {\n      return { statusCode: 200, status: 'ok' };\n    }\n\n    try {\n      await Promise.all(\n        this.livenessChecks.map(({ check, client, name, customHandlersPayload }) =>\n          check({ client, name, customHandlersPayload, logger: this.logger }),\n        ),\n      );\n\n      return { statusCode: 200, status: 'ok' };\n    } catch (error: any) {\n      const err = new ServiceUnavailableError(`Liveness probe failed: ${error.message}`);\n      err.stack = error.stack;\n      throw err;\n    }\n  }\n\n  /**\n   * Performs readiness health checks for clients configured with includeInReadinessCheck\n   *\n   * @returns Promise that resolves to { status: 'ok', statusCode: 200 } if ready\n   * @throws ServiceUnavailableError with statusCode 503 if not ready or health check fails\n   */\n  async readyCheck(): Promise<{ status: string; statusCode: number; }> {\n    if (this.isShuttingDown) {\n      this.isReady = false;\n      this.logger.warn('Application is shutting down: readiness probe failed');\n      throw new ServiceUnavailableError('Readiness probe failed: Application is shutting down');\n    }\n\n    try {\n      await Promise.all(\n        this.readinessChecks.map(({ check, client, name, customHandlersPayload }) =>\n          check({ client, name, customHandlersPayload, logger: this.logger }),\n        ),\n      );\n\n      if (!this.onReadyTriggered) {\n        this.onReadyTriggered = true;\n        await this.config.onReady?.();\n        this.logger.info('onReady hook executed successfully');\n      }\n\n      this.isReady = true;\n\n      return { statusCode: 200, status: 'ok' };\n    } catch (error: any) {\n      this.isReady = false;\n      const err = new ServiceUnavailableError(`Readiness probe failed: ${error.message}`);\n      err.stack = error.stack;\n      throw err;\n    }\n  }\n\n  /**\n   * Groups client shutdown handlers by stage (stopAcceptingWork, drain, close).\n   *\n   * @private\n   * @param signal - Signal name to pass to shutdown handlers\n   * @returns Record of shutdown tasks organized by stage\n   */\n  private groupShutdownTasksByStage(signal: string): Record<ShutdownStage, { fn: () => Promise<any>; name: string; }[]> {\n    const shutdownTasksByStage: Record<ShutdownStage, { fn: () => Promise<any>; name: string; }[]> = {\n      [SHUTDOWN_STAGES.STOP_ACCEPTING_WORK]: [],\n      [SHUTDOWN_STAGES.DRAIN]: [],\n      [SHUTDOWN_STAGES.CLOSE]: [],\n    };\n\n    this.clientsMap.forEach((clients) => {\n      clients.forEach((client) => {\n        const clientName = client.options?.name || 'unknown';\n        const enabledShutdownStages = client.options?.enabledShutdownStages;\n\n        if (client.shutdownHandlers) {\n          for (const [stage, shutdownHandler] of Object.entries(client.shutdownHandlers) as [ShutdownStage, ShutdownHandler<any>][]) {\n            if (shutdownHandler && enabledShutdownStages?.includes(stage)) {\n              shutdownTasksByStage[stage].push({\n                fn: () => shutdownHandler({\n                  signal,\n                  client: client.connection,\n                  name: clientName,\n                  customHandlersPayload: client.options?.customHandlersPayload,\n                  logger: this.logger,\n                }),\n                name: clientName,\n              });\n            }\n          }\n        }\n      });\n    });\n\n    return shutdownTasksByStage;\n  }\n\n  /**\n   * Execute shutdown tasks for a specific stage\n   * @private\n   * @param stage - The shutdown stage to execute\n   * @param tasks - Array of shutdown tasks to execute\n   */\n  private async executeShutdownStage(stage: ShutdownStage, tasks: { fn: () => Promise<any>; name: string; }[]): Promise<void> {\n    if (tasks.length === 0) {\n      return;\n    }\n\n    this.logger.info(`[${stage} stage] Start executing shutdown stage ${stage}...`);\n    const results = await Promise.allSettled(tasks.map(t => t.fn()));\n    results.forEach((result, index) => {\n      if (result.status === 'rejected') {\n        const { name } = tasks[index];\n        this.logger.error(`[${stage} stage] Failed to shutdown '${name}'`, { error: result.reason });\n      }\n    });\n    this.logger.info(`[${stage} stage] Shutdown stage ${stage} completed`);\n  }\n\n  /**\n   * Triggers graceful shutdown process\n   * Can be called manually or automatically via SIGTERM/SIGINT signal handlers\n   *\n   * Shutdown happens in three stages:\n   * 1. stopAcceptingWork - Stop accepting new requests/work\n   * 2. drain - Allow in-flight work to complete\n   * 3. close - Close all connections and clean up resources\n   *\n   * @param signal - Signal name (e.g., 'SIGTERM', 'SIGINT', 'MANUAL')\n   */\n  async shutdown(signal = 'MANUAL'): Promise<void> {\n    try {\n      if (this.isShuttingDown) {\n        this.logger.warn('Shutdown already in progress');\n        return;\n      }\n\n      const { shutdownDelayMs, useExit0, onShutdown, beforeShutdown } = this.config;\n\n      this.logger.info(`${signal} received: setting readiness state to false`);\n\n      this.isShuttingDown = true;\n      this.isReady = false;\n\n      await beforeShutdown?.();\n\n      this.logger.info(`Waiting ${shutdownDelayMs}ms before shutdown...`);\n      await setTimeout(shutdownDelayMs);\n\n      const shutdownTasksByStage = this.groupShutdownTasksByStage(signal);\n\n      await this.executeShutdownStage(SHUTDOWN_STAGES.STOP_ACCEPTING_WORK, shutdownTasksByStage[SHUTDOWN_STAGES.STOP_ACCEPTING_WORK]);\n\n      // Wait between stages to allow in-flight work to complete (we are missing a drain stage in our rabbitmq package)\n      // TODO once we have proper drain handlers, we can remove this fixed wait\n      this.logger.info('Waiting 4000ms to allow in-flight work to complete...');\n      await setTimeout(4_000);\n\n      await this.executeShutdownStage(SHUTDOWN_STAGES.DRAIN, shutdownTasksByStage[SHUTDOWN_STAGES.DRAIN]);\n      await this.executeShutdownStage(SHUTDOWN_STAGES.CLOSE, shutdownTasksByStage[SHUTDOWN_STAGES.CLOSE]);\n\n      await onShutdown?.();\n\n      this.logger.info('Shutdown process completed');\n\n      if (useExit0) {\n        this.logger.info(`Exiting process after ${signal}`);\n        process.exit(0);\n      }\n    } catch (error: any) {\n      this.logger.error('Error during shutdown process, forcing process exit despite shutdown error', { error, signal });\n      process.exit(1);\n    }\n  }\n}\n"],"mappings":"kNA0BA,IAAa,EAAb,KAA2B,CA0BzB,YAAY,EAA8B,qBAzBa,CACrD,gBAAiB,IACjB,SAAU,GACV,0BAA2B,GAC3B,2BAA4B,GAC7B,qBAEwB,gBACP,yBACS,mBAIN,IAAI,wBAE+F,EAAE,sBACD,EAAE,sCAwFjF,EAAwC,IAE3E,EAKE,CAAC,CAAC,EAJA,GAjFT,KAAK,OAAS,CACZ,GAAG,KAAK,eACR,GAAI,EAAO,QAAU,EAAE,CACxB,CAED,KAAK,OAAS,EAAO,OAErB,KAAK,yBAAyB,EAAO,CAErC,KAAK,wBAAwB,CAEzB,KAAK,OAAO,2BACd,KAAK,qBAAqB,CAQ9B,yBAAiC,EAAoC,CACnE,IAAM,EAAU,EAAO,QAClB,KAIL,KAAK,GAAM,CAAC,EAAM,KAAW,OAAO,QAAQ,EAAQ,CAC7C,KAIL,GAAI,IAASA,EAAAA,aAAa,OACxB,KAAK,aAAa,EAAwC,KACrD,CACL,IAAM,EAAe,MAAM,QAAQ,EAAO,CAAG,EAAS,CAAC,EAAO,CACxD,EAAWC,EAAAA,oBAAoB,IAAuBA,EAAAA,oBAAoBD,EAAAA,aAAa,QAEvF,EAAsB,EAAa,IAAI,IAAW,CACtD,WAAYE,EAAO,WACnB,YAAaA,EAAO,aAAe,GAAU,YAC7C,iBAAkB,CAChB,GAAG,GAAU,iBACb,GAAGA,EAAO,iBACX,CACD,QAAS,CACP,GAAG,GAAU,QACb,GAAGA,EAAO,QACV,KAAMA,EAAO,SAAS,MAAQ,GAAU,SAAS,MAAQ,EAC1D,CACF,EAAE,CAEH,KAAK,WAAW,IAAI,EAAM,EAAoB,GASpD,qBAAoC,CAClC,KAAK,OAAO,KAAK,2DAA2D,CAE5E,QAAQ,GAAG,UAAW,SAAY,CAChC,MAAM,KAAK,SAAS,UAAU,EAC9B,CAEF,QAAQ,GAAG,SAAU,SAAY,CAC/B,MAAM,KAAK,SAAS,SAAS,EAC7B,CAuBJ,wBAAuC,CACrC,IAAI,EAAsC,GAE1C,KAAK,WAAW,QAAS,GAAY,CACnC,EAAQ,QAAS,GAAW,CAC1B,GAAI,CAAC,EAAO,YACV,OAGF,IAAM,EAAoB,EAAO,SAAS,uBACpC,EAAqB,EAAO,SAAS,wBAGtC,KAAK,+BAA+B,EAAmB,EAAmB,GAC7E,EAAsC,IAGpC,GACF,KAAK,eAAe,KAAK,CACvB,MAAO,EAAO,YACd,OAAQ,EAAO,WACf,KAAM,EAAO,SAAS,MAAQ,UAC9B,sBAAuB,EAAO,SAAS,sBACxC,CAAC,CAGA,GACF,KAAK,gBAAgB,KAAK,CACxB,MAAO,EAAO,YACd,OAAQ,EAAO,WACf,KAAM,EAAO,SAAS,MAAQ,UAC9B,sBAAuB,EAAO,SAAS,sBACxC,CAAC,EAEJ,EACF,CAGF,KAAK,OAAO,2BAA6B,KAAK,OAAO,4BAA8B,EAG/E,KAAK,OAAO,2BACd,KAAK,OAAO,KAAK,0FAA0F,CACjG,GACV,KAAK,OAAO,KAAK,yGAAyG,CAQ9H,aAAa,EAAmD,CAC9D,IAAM,EAAiBD,EAAAA,oBAAoBD,EAAAA,aAAa,QAGlD,GAFoB,MAAM,QAAQ,EAAa,CAAG,EAAe,CAAC,EAAa,EAE7C,IAAK,GAAW,CAEtD,IAAM,GAAA,EAAA,EAAA,sBAAsC,CAC1C,OAAQ,EAAO,WAChB,CAAC,CAEF,MAAO,CACL,WAAY,EAAO,WACnB,YAAa,EAAO,aAAe,GAAgB,YACnD,iBAAkB,CAChB,GAAG,GAAgB,iBACnB,GAAG,EAAO,iBACX,CACD,QAAS,CACP,GAAG,GAAgB,QACnB,GAAG,EAAO,QACV,KAAM,EAAO,SAAS,MAAQ,GAAgB,SAAS,MAAQ,cAC/D,sBAAuB,CACrB,GAAG,EAAO,SAAS,sBACnB,iBACD,CACF,CACF,EACD,CAEF,KAAK,WAAW,IAAIA,EAAAA,aAAa,OAAQ,EAAc,CASzD,MAAM,YAA+D,CAMnE,GALI,KAAK,gBAKL,KAAK,OAAO,4BAA8B,KAAK,QACjD,MAAO,CAAE,WAAY,IAAK,OAAQ,KAAM,CAG1C,GAAI,CAOF,OANA,MAAM,QAAQ,IACZ,KAAK,eAAe,KAAK,CAAE,QAAO,SAAQ,OAAM,2BAC9C,EAAM,CAAE,SAAQ,OAAM,wBAAuB,OAAQ,KAAK,OAAQ,CAAC,CACpE,CACF,CAEM,CAAE,WAAY,IAAK,OAAQ,KAAM,OACjCG,EAAY,CACnB,IAAM,EAAM,IAAIC,EAAAA,wBAAwB,0BAA0B,EAAM,UAAU,CAElF,KADA,GAAI,MAAQ,EAAM,MACZ,GAUV,MAAM,YAA+D,CACnE,GAAI,KAAK,eAGP,KAFA,MAAK,QAAU,GACf,KAAK,OAAO,KAAK,uDAAuD,CAClE,IAAIA,EAAAA,wBAAwB,uDAAuD,CAG3F,GAAI,CAeF,OAdA,MAAM,QAAQ,IACZ,KAAK,gBAAgB,KAAK,CAAE,QAAO,SAAQ,OAAM,2BAC/C,EAAM,CAAE,SAAQ,OAAM,wBAAuB,OAAQ,KAAK,OAAQ,CAAC,CACpE,CACF,CAEI,KAAK,mBACR,KAAK,iBAAmB,GACxB,MAAM,KAAK,OAAO,WAAW,CAC7B,KAAK,OAAO,KAAK,qCAAqC,EAGxD,KAAK,QAAU,GAER,CAAE,WAAY,IAAK,OAAQ,KAAM,OACjCD,EAAY,CACnB,KAAK,QAAU,GACf,IAAM,EAAM,IAAIC,EAAAA,wBAAwB,2BAA2B,EAAM,UAAU,CAEnF,KADA,GAAI,MAAQ,EAAM,MACZ,GAWV,0BAAkC,EAAoF,CACpH,IAAMC,EAA2F,EAC9FC,EAAAA,gBAAgB,qBAAsB,EAAE,EACxCA,EAAAA,gBAAgB,OAAQ,EAAE,EAC1BA,EAAAA,gBAAgB,OAAQ,EAAE,CAC5B,CA0BD,OAxBA,KAAK,WAAW,QAAS,GAAY,CACnC,EAAQ,QAAS,GAAW,CAC1B,IAAM,EAAa,EAAO,SAAS,MAAQ,UACrC,EAAwB,EAAO,SAAS,sBAE9C,GAAI,EAAO,qBACJ,GAAM,CAAC,EAAO,KAAoB,OAAO,QAAQ,EAAO,iBAAiB,CACxE,GAAmB,GAAuB,SAAS,EAAM,EAC3D,EAAqB,GAAO,KAAK,CAC/B,OAAU,EAAgB,CACxB,SACA,OAAQ,EAAO,WACf,KAAM,EACN,sBAAuB,EAAO,SAAS,sBACvC,OAAQ,KAAK,OACd,CAAC,CACF,KAAM,EACP,CAAC,EAIR,EACF,CAEK,EAST,MAAc,qBAAqB,EAAsB,EAAmE,CACtH,EAAM,SAAW,IAIrB,KAAK,OAAO,KAAK,IAAI,EAAM,yCAAyC,EAAM,KAAK,EAC/D,MAAM,QAAQ,WAAW,EAAM,IAAI,GAAK,EAAE,IAAI,CAAC,CAAC,EACxD,SAAS,EAAQ,IAAU,CACjC,GAAI,EAAO,SAAW,WAAY,CAChC,GAAM,CAAE,QAAS,EAAM,GACvB,KAAK,OAAO,MAAM,IAAI,EAAM,8BAA8B,EAAK,GAAI,CAAE,MAAO,EAAO,OAAQ,CAAC,GAE9F,CACF,KAAK,OAAO,KAAK,IAAI,EAAM,yBAAyB,EAAM,YAAY,EAcxE,MAAM,SAAS,EAAS,SAAyB,CAC/C,GAAI,CACF,GAAI,KAAK,eAAgB,CACvB,KAAK,OAAO,KAAK,+BAA+B,CAChD,OAGF,GAAM,CAAE,kBAAiB,WAAU,aAAY,kBAAmB,KAAK,OAEvE,KAAK,OAAO,KAAK,GAAG,EAAO,6CAA6C,CAExE,KAAK,eAAiB,GACtB,KAAK,QAAU,GAEf,MAAM,KAAkB,CAExB,KAAK,OAAO,KAAK,WAAW,EAAgB,uBAAuB,CACnE,MAAA,EAAA,EAAA,YAAiB,EAAgB,CAEjC,IAAM,EAAuB,KAAK,0BAA0B,EAAO,CAEnE,MAAM,KAAK,qBAAqBA,EAAAA,gBAAgB,oBAAqB,EAAqBA,EAAAA,gBAAgB,qBAAqB,CAI/H,KAAK,OAAO,KAAK,wDAAwD,CACzE,MAAA,EAAA,EAAA,YAAiB,IAAM,CAEvB,MAAM,KAAK,qBAAqBA,EAAAA,gBAAgB,MAAO,EAAqBA,EAAAA,gBAAgB,OAAO,CACnG,MAAM,KAAK,qBAAqBA,EAAAA,gBAAgB,MAAO,EAAqBA,EAAAA,gBAAgB,OAAO,CAEnG,MAAM,KAAc,CAEpB,KAAK,OAAO,KAAK,6BAA6B,CAE1C,IACF,KAAK,OAAO,KAAK,yBAAyB,IAAS,CACnD,QAAQ,KAAK,EAAE,QAEVH,EAAY,CACnB,KAAK,OAAO,MAAM,6EAA8E,CAAE,QAAO,SAAQ,CAAC,CAClH,QAAQ,KAAK,EAAE"}
+{"version":3,"file":"health-manager.cjs","names":["CLIENT_TYPES","CLIENT_DEFAULTS_MAP","client","error: any","ServiceUnavailableError","shutdownTasksByStage: Record<ShutdownStage, { fn: () => Promise<any>; name: string; }[]>","SHUTDOWN_STAGES"],"sources":["../../src/health-manager/health-manager.ts"],"sourcesContent":["import { setTimeout } from 'node:timers/promises';\nimport { ServiceUnavailableError } from '@autofleet/errors';\nimport type { LoggerInstanceManager } from '@autofleet/logger';\nimport { createHttpTerminator } from 'http-terminator';\nimport {\n  type ClientConfig,\n  type HealthManagerConfig,\n  type HealthManagerOptions,\n  type ClientType,\n  type HealthCheckHandler,\n  type ShutdownStage,\n  type ServerConfig,\n  CLIENT_TYPES,\n  SHUTDOWN_STAGES,\n  type ShutdownHandler,\n} from './types';\nimport { CLIENT_DEFAULTS_MAP } from './client-defaults';\n\n/**\n * Manages application health state and graceful shutdown for Kubernetes liveness/readiness probes\n *\n * Features:\n * - Provides aliveCheck() and readyCheck() methods for liveness/readiness probes\n * - Auto-registers SIGTERM/SIGINT handlers for graceful shutdown\n * - Manages ready state during shutdown to fail readiness probes\n */\nexport class HealthManager {\n  private readonly DEFAULT_CONFIG: HealthManagerConfig = {\n    shutdownDelayMs: 6_000,\n    useExit0: true,\n    shouldSetupSignalHandlers: true,\n    enableLivenessOptimization: true,\n  };\n\n  private isShuttingDown = false;\n  private isReady = false;\n\n  private config: HealthManagerConfig;\n\n  private clientsMap = new Map<string, ClientConfig<any>[]>();\n\n  private livenessChecks: { check: HealthCheckHandler<any>; client: any; name: string; customHandlersPayload?: any; }[] = [];\n  private readinessChecks: { check: HealthCheckHandler<any>; client: any; name: string; customHandlersPayload?: any; }[] = [];\n\n  private logger: LoggerInstanceManager;\n\n  /**\n   * Creates a new HealthManager instance\n   *\n   * @param params Configuration options for health checks and shutdown behavior\n   */\n  constructor(params: HealthManagerOptions) {\n    this.config = {\n      ...this.DEFAULT_CONFIG,\n      ...(params.config ?? {}),\n    };\n\n    this.logger = params.logger;\n\n    this.storeClientsWithDefaults(params);\n\n    this.initializeHealthChecks();\n\n    if (this.config.shouldSetupSignalHandlers) {\n      this.setupSignalHandlers();\n    }\n  }\n\n  /**\n   * Store clients with default health check and shutdown functions\n   * @private\n   */\n  private storeClientsWithDefaults(params: HealthManagerOptions): void {\n    const clients = params.clients;\n    if (!clients) {\n      return;\n    }\n\n    for (const [type, client] of Object.entries(clients)) {\n      if (!client) {\n        continue;\n      }\n\n      if (type === CLIENT_TYPES.SERVER) {\n        this.attachServer(client as ServerConfig | ServerConfig[]);\n      } else {\n        const clientsArray = Array.isArray(client) ? client : [client];\n        const defaults = CLIENT_DEFAULTS_MAP[type as ClientType] ?? CLIENT_DEFAULTS_MAP[CLIENT_TYPES.CUSTOM];\n\n        const clientsWithDefaults = clientsArray.map(client => ({\n          connection: client.connection,\n          healthCheck: client.healthCheck ?? defaults?.healthCheck,\n          shutdownHandlers: {\n            ...defaults?.shutdownHandlers,\n            ...client.shutdownHandlers,\n          },\n          options: {\n            ...defaults?.options,\n            ...client.options,\n            name: client.options?.name || defaults?.options?.name || type,\n          },\n        }));\n\n        this.clientsMap.set(type, clientsWithDefaults);\n      }\n    }\n  }\n\n  /**\n   * Register SIGTERM and SIGINT signal listeners\n   * @private\n   */\n  private setupSignalHandlers(): void {\n    this.logger.info('Setting up global signal listeners for graceful shutdown');\n\n    process.on('SIGTERM', async () => {\n      await this.shutdown('SIGTERM');\n    });\n\n    process.on('SIGINT', async () => {\n      await this.shutdown('SIGINT');\n    });\n  }\n\n  /**\n   * Check if liveness optimization is safe to enable for a given client configuration\n   * If a client is in liveness but not readiness, we cannot skip liveness checks\n   * because the optimization relies on readiness passing, but this client isn't checked in readiness\n   * @private\n   */\n  private canSafelySkipLivenessWhenReady = (includeInLiveness: boolean | undefined, includeInReadiness: boolean | undefined): boolean => {\n    // If not in liveness, optimization is safe (skipping liveness doesn't affect this client)\n    if (!includeInLiveness) {\n      return true;\n    }\n\n    // If in liveness, must also be in readiness for liveness optimization to be safe\n    return !!includeInReadiness;\n  };\n\n  /**\n   * Initialize health check arrays based on configured clients\n   * @private\n   */\n  private initializeHealthChecks(): void {\n    let allClientsAllowLivenessOptimization = true;\n\n    this.clientsMap.forEach((clients) => {\n      clients.forEach((client) => {\n        if (!client.healthCheck) {\n          return;\n        }\n\n        const includeInLiveness = client.options?.includeInLivenessCheck;\n        const includeInReadiness = client.options?.includeInReadinessCheck;\n\n        // Check if this client configuration allows liveness optimization\n        if (!this.canSafelySkipLivenessWhenReady(includeInLiveness, includeInReadiness)) {\n          allClientsAllowLivenessOptimization = false;\n        }\n\n        if (includeInLiveness) {\n          this.livenessChecks.push({\n            check: client.healthCheck,\n            client: client.connection,\n            name: client.options?.name || 'unknown',\n            customHandlersPayload: client.options?.customHandlersPayload,\n          });\n        }\n\n        if (includeInReadiness) {\n          this.readinessChecks.push({\n            check: client.healthCheck,\n            client: client.connection,\n            name: client.options?.name || 'unknown',\n            customHandlersPayload: client.options?.customHandlersPayload,\n          });\n        }\n      });\n    });\n\n    // Apply optimization setting: only enable if user requested it AND all clients allow it\n    this.config.enableLivenessOptimization = this.config.enableLivenessOptimization && allClientsAllowLivenessOptimization;\n\n    // Log optimization status\n    if (this.config.enableLivenessOptimization) {\n      this.logger.info('Liveness check optimization enabled - liveness checks will be skipped when pod is ready');\n    } else if (!allClientsAllowLivenessOptimization) {\n      this.logger.warn('Liveness check optimization disabled - some clients are in liveness checks but not in readiness checks');\n    }\n  }\n\n  /**\n   * Attach an HTTP or HTTPS server to be managed during shutdown\n   * @param serverConfig Server configuration (ClientConfig with HTTP/HTTPS server)\n   */\n  attachServer(serverConfig: ServerConfig | ServerConfig[]): void {\n    const serverDefaults = CLIENT_DEFAULTS_MAP[CLIENT_TYPES.SERVER];\n    const serverConfigArray = Array.isArray(serverConfig) ? serverConfig : [serverConfig];\n\n    const serverClients = serverConfigArray.map((config) => {\n      // Create httpTerminator for this server instance\n      const httpTerminator = createHttpTerminator({\n        server: config.connection,\n      });\n\n      return {\n        connection: config.connection,\n        healthCheck: config.healthCheck ?? serverDefaults?.healthCheck,\n        shutdownHandlers: {\n          ...serverDefaults?.shutdownHandlers,\n          ...config.shutdownHandlers,\n        },\n        options: {\n          ...serverDefaults?.options,\n          ...config.options,\n          name: config.options?.name || serverDefaults?.options?.name || 'HTTP Server',\n          customHandlersPayload: {\n            ...config.options?.customHandlersPayload,\n            httpTerminator,\n          },\n        },\n      };\n    });\n\n    this.clientsMap.set(CLIENT_TYPES.SERVER, serverClients);\n  }\n\n  /**\n   * Performs liveness health checks for clients configured with includeInLivenessCheck\n   *\n   * @returns Promise that resolves to { status: 'ok', statusCode: 200 } on success\n   * @throws ServiceUnavailableError with statusCode 503 if any health check fails\n   */\n  async aliveCheck(): Promise<{ status: string; statusCode: number; }> {\n    if (this.isShuttingDown) {\n      return { statusCode: 200, status: 'ok' };\n    }\n\n    // Skip liveness check if optimization is enabled and the readiness check has already passed\n    if (this.config.enableLivenessOptimization && this.isReady) {\n      return { statusCode: 200, status: 'ok' };\n    }\n\n    try {\n      await Promise.all(\n        this.livenessChecks.map(({ check, client, name, customHandlersPayload }) =>\n          check({ client, name, customHandlersPayload, logger: this.logger }),\n        ),\n      );\n\n      return { statusCode: 200, status: 'ok' };\n    } catch (error: any) {\n      const err = new ServiceUnavailableError(`Liveness probe failed: ${error.message}`);\n      err.stack = error.stack;\n      throw err;\n    }\n  }\n\n  /**\n   * Performs readiness health checks for clients configured with includeInReadinessCheck\n   *\n   * @returns Promise that resolves to { status: 'ok', statusCode: 200 } if ready\n   * @throws ServiceUnavailableError with statusCode 503 if not ready or health check fails\n   */\n  async readyCheck(): Promise<{ status: string; statusCode: number; }> {\n    if (this.isShuttingDown) {\n      this.isReady = false;\n      this.logger.warn('Application is shutting down: readiness probe failed');\n      throw new ServiceUnavailableError('Readiness probe failed: Application is shutting down');\n    }\n\n    try {\n      await Promise.all(\n        this.readinessChecks.map(({ check, client, name, customHandlersPayload }) =>\n          check({ client, name, customHandlersPayload, logger: this.logger }),\n        ),\n      );\n      this.isReady = true;\n\n      return { statusCode: 200, status: 'ok' };\n    } catch (error: any) {\n      this.isReady = false;\n      const err = new ServiceUnavailableError(`Readiness probe failed: ${error.message}`);\n      err.stack = error.stack;\n      throw err;\n    }\n  }\n\n  /**\n   * Groups client shutdown handlers by stage (stopAcceptingWork, drain, close).\n   *\n   * @private\n   * @param signal - Signal name to pass to shutdown handlers\n   * @returns Record of shutdown tasks organized by stage\n   */\n  private groupShutdownTasksByStage(signal: string): Record<ShutdownStage, { fn: () => Promise<any>; name: string; }[]> {\n    const shutdownTasksByStage: Record<ShutdownStage, { fn: () => Promise<any>; name: string; }[]> = {\n      [SHUTDOWN_STAGES.STOP_ACCEPTING_WORK]: [],\n      [SHUTDOWN_STAGES.DRAIN]: [],\n      [SHUTDOWN_STAGES.CLOSE]: [],\n    };\n\n    this.clientsMap.forEach((clients) => {\n      clients.forEach((client) => {\n        const clientName = client.options?.name || 'unknown';\n        const enabledShutdownStages = client.options?.enabledShutdownStages;\n\n        if (client.shutdownHandlers) {\n          for (const [stage, shutdownHandler] of Object.entries(client.shutdownHandlers) as [ShutdownStage, ShutdownHandler<any>][]) {\n            if (shutdownHandler && enabledShutdownStages?.includes(stage)) {\n              shutdownTasksByStage[stage].push({\n                fn: () => shutdownHandler({\n                  signal,\n                  client: client.connection,\n                  name: clientName,\n                  customHandlersPayload: client.options?.customHandlersPayload,\n                  logger: this.logger,\n                }),\n                name: clientName,\n              });\n            }\n          }\n        }\n      });\n    });\n\n    return shutdownTasksByStage;\n  }\n\n  /**\n   * Execute shutdown tasks for a specific stage\n   * @private\n   * @param stage - The shutdown stage to execute\n   * @param tasks - Array of shutdown tasks to execute\n   */\n  private async executeShutdownStage(stage: ShutdownStage, tasks: { fn: () => Promise<any>; name: string; }[]): Promise<void> {\n    if (tasks.length === 0) {\n      return;\n    }\n\n    this.logger.info(`[${stage} stage] Start executing shutdown stage ${stage}...`);\n    const results = await Promise.allSettled(tasks.map(t => t.fn()));\n    results.forEach((result, index) => {\n      if (result.status === 'rejected') {\n        const { name } = tasks[index];\n        this.logger.error(`[${stage} stage] Failed to shutdown '${name}'`, { error: result.reason });\n      }\n    });\n    this.logger.info(`[${stage} stage] Shutdown stage ${stage} completed`);\n  }\n\n  /**\n   * Triggers graceful shutdown process\n   * Can be called manually or automatically via SIGTERM/SIGINT signal handlers\n   *\n   * Shutdown happens in three stages:\n   * 1. stopAcceptingWork - Stop accepting new requests/work\n   * 2. drain - Allow in-flight work to complete\n   * 3. close - Close all connections and clean up resources\n   *\n   * @param signal - Signal name (e.g., 'SIGTERM', 'SIGINT', 'MANUAL')\n   */\n  async shutdown(signal = 'MANUAL'): Promise<void> {\n    try {\n      if (this.isShuttingDown) {\n        this.logger.warn('Shutdown already in progress');\n        return;\n      }\n\n      const { shutdownDelayMs, useExit0, onShutdown, beforeShutdown } = this.config;\n\n      this.logger.info(`${signal} received: setting readiness state to false`);\n\n      this.isShuttingDown = true;\n      this.isReady = false;\n\n      await beforeShutdown?.();\n\n      this.logger.info(`Waiting ${shutdownDelayMs}ms before shutdown...`);\n      await setTimeout(shutdownDelayMs);\n\n      const shutdownTasksByStage = this.groupShutdownTasksByStage(signal);\n\n      await this.executeShutdownStage(SHUTDOWN_STAGES.STOP_ACCEPTING_WORK, shutdownTasksByStage[SHUTDOWN_STAGES.STOP_ACCEPTING_WORK]);\n\n      // Wait between stages to allow in-flight work to complete (we are missing a drain stage in our rabbitmq package)\n      // TODO once we have proper drain handlers, we can remove this fixed wait\n      this.logger.info('Waiting 4000ms to allow in-flight work to complete...');\n      await setTimeout(4_000);\n\n      await this.executeShutdownStage(SHUTDOWN_STAGES.DRAIN, shutdownTasksByStage[SHUTDOWN_STAGES.DRAIN]);\n      await this.executeShutdownStage(SHUTDOWN_STAGES.CLOSE, shutdownTasksByStage[SHUTDOWN_STAGES.CLOSE]);\n\n      await onShutdown?.();\n\n      this.logger.info('Shutdown process completed');\n\n      if (useExit0) {\n        this.logger.info(`Exiting process after ${signal}`);\n        process.exit(0);\n      }\n    } catch (error: any) {\n      this.logger.error('Error during shutdown process, forcing process exit despite shutdown error', { error, signal });\n      process.exit(1);\n    }\n  }\n}\n"],"mappings":"kNA0BA,IAAa,EAAb,KAA2B,CAyBzB,YAAY,EAA8B,qBAxBa,CACrD,gBAAiB,IACjB,SAAU,GACV,0BAA2B,GAC3B,2BAA4B,GAC7B,qBAEwB,gBACP,mBAIG,IAAI,wBAE+F,EAAE,sBACD,EAAE,sCAwFjF,EAAwC,IAE3E,EAKE,CAAC,CAAC,EAJA,GAjFT,KAAK,OAAS,CACZ,GAAG,KAAK,eACR,GAAI,EAAO,QAAU,EAAE,CACxB,CAED,KAAK,OAAS,EAAO,OAErB,KAAK,yBAAyB,EAAO,CAErC,KAAK,wBAAwB,CAEzB,KAAK,OAAO,2BACd,KAAK,qBAAqB,CAQ9B,yBAAiC,EAAoC,CACnE,IAAM,EAAU,EAAO,QAClB,KAIL,KAAK,GAAM,CAAC,EAAM,KAAW,OAAO,QAAQ,EAAQ,CAC7C,KAIL,GAAI,IAASA,EAAAA,aAAa,OACxB,KAAK,aAAa,EAAwC,KACrD,CACL,IAAM,EAAe,MAAM,QAAQ,EAAO,CAAG,EAAS,CAAC,EAAO,CACxD,EAAWC,EAAAA,oBAAoB,IAAuBA,EAAAA,oBAAoBD,EAAAA,aAAa,QAEvF,EAAsB,EAAa,IAAI,IAAW,CACtD,WAAYE,EAAO,WACnB,YAAaA,EAAO,aAAe,GAAU,YAC7C,iBAAkB,CAChB,GAAG,GAAU,iBACb,GAAGA,EAAO,iBACX,CACD,QAAS,CACP,GAAG,GAAU,QACb,GAAGA,EAAO,QACV,KAAMA,EAAO,SAAS,MAAQ,GAAU,SAAS,MAAQ,EAC1D,CACF,EAAE,CAEH,KAAK,WAAW,IAAI,EAAM,EAAoB,GASpD,qBAAoC,CAClC,KAAK,OAAO,KAAK,2DAA2D,CAE5E,QAAQ,GAAG,UAAW,SAAY,CAChC,MAAM,KAAK,SAAS,UAAU,EAC9B,CAEF,QAAQ,GAAG,SAAU,SAAY,CAC/B,MAAM,KAAK,SAAS,SAAS,EAC7B,CAuBJ,wBAAuC,CACrC,IAAI,EAAsC,GAE1C,KAAK,WAAW,QAAS,GAAY,CACnC,EAAQ,QAAS,GAAW,CAC1B,GAAI,CAAC,EAAO,YACV,OAGF,IAAM,EAAoB,EAAO,SAAS,uBACpC,EAAqB,EAAO,SAAS,wBAGtC,KAAK,+BAA+B,EAAmB,EAAmB,GAC7E,EAAsC,IAGpC,GACF,KAAK,eAAe,KAAK,CACvB,MAAO,EAAO,YACd,OAAQ,EAAO,WACf,KAAM,EAAO,SAAS,MAAQ,UAC9B,sBAAuB,EAAO,SAAS,sBACxC,CAAC,CAGA,GACF,KAAK,gBAAgB,KAAK,CACxB,MAAO,EAAO,YACd,OAAQ,EAAO,WACf,KAAM,EAAO,SAAS,MAAQ,UAC9B,sBAAuB,EAAO,SAAS,sBACxC,CAAC,EAEJ,EACF,CAGF,KAAK,OAAO,2BAA6B,KAAK,OAAO,4BAA8B,EAG/E,KAAK,OAAO,2BACd,KAAK,OAAO,KAAK,0FAA0F,CACjG,GACV,KAAK,OAAO,KAAK,yGAAyG,CAQ9H,aAAa,EAAmD,CAC9D,IAAM,EAAiBD,EAAAA,oBAAoBD,EAAAA,aAAa,QAGlD,GAFoB,MAAM,QAAQ,EAAa,CAAG,EAAe,CAAC,EAAa,EAE7C,IAAK,GAAW,CAEtD,IAAM,GAAA,EAAA,EAAA,sBAAsC,CAC1C,OAAQ,EAAO,WAChB,CAAC,CAEF,MAAO,CACL,WAAY,EAAO,WACnB,YAAa,EAAO,aAAe,GAAgB,YACnD,iBAAkB,CAChB,GAAG,GAAgB,iBACnB,GAAG,EAAO,iBACX,CACD,QAAS,CACP,GAAG,GAAgB,QACnB,GAAG,EAAO,QACV,KAAM,EAAO,SAAS,MAAQ,GAAgB,SAAS,MAAQ,cAC/D,sBAAuB,CACrB,GAAG,EAAO,SAAS,sBACnB,iBACD,CACF,CACF,EACD,CAEF,KAAK,WAAW,IAAIA,EAAAA,aAAa,OAAQ,EAAc,CASzD,MAAM,YAA+D,CAMnE,GALI,KAAK,gBAKL,KAAK,OAAO,4BAA8B,KAAK,QACjD,MAAO,CAAE,WAAY,IAAK,OAAQ,KAAM,CAG1C,GAAI,CAOF,OANA,MAAM,QAAQ,IACZ,KAAK,eAAe,KAAK,CAAE,QAAO,SAAQ,OAAM,2BAC9C,EAAM,CAAE,SAAQ,OAAM,wBAAuB,OAAQ,KAAK,OAAQ,CAAC,CACpE,CACF,CAEM,CAAE,WAAY,IAAK,OAAQ,KAAM,OACjCG,EAAY,CACnB,IAAM,EAAM,IAAIC,EAAAA,wBAAwB,0BAA0B,EAAM,UAAU,CAElF,KADA,GAAI,MAAQ,EAAM,MACZ,GAUV,MAAM,YAA+D,CACnE,GAAI,KAAK,eAGP,KAFA,MAAK,QAAU,GACf,KAAK,OAAO,KAAK,uDAAuD,CAClE,IAAIA,EAAAA,wBAAwB,uDAAuD,CAG3F,GAAI,CAQF,OAPA,MAAM,QAAQ,IACZ,KAAK,gBAAgB,KAAK,CAAE,QAAO,SAAQ,OAAM,2BAC/C,EAAM,CAAE,SAAQ,OAAM,wBAAuB,OAAQ,KAAK,OAAQ,CAAC,CACpE,CACF,CACD,KAAK,QAAU,GAER,CAAE,WAAY,IAAK,OAAQ,KAAM,OACjCD,EAAY,CACnB,KAAK,QAAU,GACf,IAAM,EAAM,IAAIC,EAAAA,wBAAwB,2BAA2B,EAAM,UAAU,CAEnF,KADA,GAAI,MAAQ,EAAM,MACZ,GAWV,0BAAkC,EAAoF,CACpH,IAAMC,EAA2F,EAC9FC,EAAAA,gBAAgB,qBAAsB,EAAE,EACxCA,EAAAA,gBAAgB,OAAQ,EAAE,EAC1BA,EAAAA,gBAAgB,OAAQ,EAAE,CAC5B,CA0BD,OAxBA,KAAK,WAAW,QAAS,GAAY,CACnC,EAAQ,QAAS,GAAW,CAC1B,IAAM,EAAa,EAAO,SAAS,MAAQ,UACrC,EAAwB,EAAO,SAAS,sBAE9C,GAAI,EAAO,qBACJ,GAAM,CAAC,EAAO,KAAoB,OAAO,QAAQ,EAAO,iBAAiB,CACxE,GAAmB,GAAuB,SAAS,EAAM,EAC3D,EAAqB,GAAO,KAAK,CAC/B,OAAU,EAAgB,CACxB,SACA,OAAQ,EAAO,WACf,KAAM,EACN,sBAAuB,EAAO,SAAS,sBACvC,OAAQ,KAAK,OACd,CAAC,CACF,KAAM,EACP,CAAC,EAIR,EACF,CAEK,EAST,MAAc,qBAAqB,EAAsB,EAAmE,CACtH,EAAM,SAAW,IAIrB,KAAK,OAAO,KAAK,IAAI,EAAM,yCAAyC,EAAM,KAAK,EAC/D,MAAM,QAAQ,WAAW,EAAM,IAAI,GAAK,EAAE,IAAI,CAAC,CAAC,EACxD,SAAS,EAAQ,IAAU,CACjC,GAAI,EAAO,SAAW,WAAY,CAChC,GAAM,CAAE,QAAS,EAAM,GACvB,KAAK,OAAO,MAAM,IAAI,EAAM,8BAA8B,EAAK,GAAI,CAAE,MAAO,EAAO,OAAQ,CAAC,GAE9F,CACF,KAAK,OAAO,KAAK,IAAI,EAAM,yBAAyB,EAAM,YAAY,EAcxE,MAAM,SAAS,EAAS,SAAyB,CAC/C,GAAI,CACF,GAAI,KAAK,eAAgB,CACvB,KAAK,OAAO,KAAK,+BAA+B,CAChD,OAGF,GAAM,CAAE,kBAAiB,WAAU,aAAY,kBAAmB,KAAK,OAEvE,KAAK,OAAO,KAAK,GAAG,EAAO,6CAA6C,CAExE,KAAK,eAAiB,GACtB,KAAK,QAAU,GAEf,MAAM,KAAkB,CAExB,KAAK,OAAO,KAAK,WAAW,EAAgB,uBAAuB,CACnE,MAAA,EAAA,EAAA,YAAiB,EAAgB,CAEjC,IAAM,EAAuB,KAAK,0BAA0B,EAAO,CAEnE,MAAM,KAAK,qBAAqBA,EAAAA,gBAAgB,oBAAqB,EAAqBA,EAAAA,gBAAgB,qBAAqB,CAI/H,KAAK,OAAO,KAAK,wDAAwD,CACzE,MAAA,EAAA,EAAA,YAAiB,IAAM,CAEvB,MAAM,KAAK,qBAAqBA,EAAAA,gBAAgB,MAAO,EAAqBA,EAAAA,gBAAgB,OAAO,CACnG,MAAM,KAAK,qBAAqBA,EAAAA,gBAAgB,MAAO,EAAqBA,EAAAA,gBAAgB,OAAO,CAEnG,MAAM,KAAc,CAEpB,KAAK,OAAO,KAAK,6BAA6B,CAE1C,IACF,KAAK,OAAO,KAAK,yBAAyB,IAAS,CACnD,QAAQ,KAAK,EAAE,QAEVH,EAAY,CACnB,KAAK,OAAO,MAAM,6EAA8E,CAAE,QAAO,SAAQ,CAAC,CAClH,QAAQ,KAAK,EAAE"}

package/lib/health-manager/health-manager.d.cts

@@ -14,7 +14,6 @@ declare class HealthManager {
   private readonly DEFAULT_CONFIG;
   private isShuttingDown;
   private isReady;
-  private onReadyTriggered;
   private config;
   private clientsMap;
   private livenessChecks;

package/lib/health-manager/health-manager.d.ts

@@ -14,7 +14,6 @@ declare class HealthManager {
   private readonly DEFAULT_CONFIG;
   private isShuttingDown;
   private isReady;
-  private onReadyTriggered;
   private config;
   private clientsMap;
   private livenessChecks;

package/lib/health-manager/health-manager.js

@@ -1,2 +1,2 @@
-import{CLIENT_TYPES as e,SHUTDOWN_STAGES as t}from"./types.js";import{CLIENT_DEFAULTS_MAP as n}from"./client-defaults.js";import{setTimeout as r}from"node:timers/promises";import{ServiceUnavailableError as i}from"@autofleet/errors";import{createHttpTerminator as a}from"http-terminator";var o=class{constructor(e){this.DEFAULT_CONFIG={shutdownDelayMs:6e3,useExit0:!0,shouldSetupSignalHandlers:!0,enableLivenessOptimization:!0},this.isShuttingDown=!1,this.isReady=!1,this.onReadyTriggered=!1,this.clientsMap=new Map,this.livenessChecks=[],this.readinessChecks=[],this.canSafelySkipLivenessWhenReady=(e,t)=>e?!!t:!0,this.config={...this.DEFAULT_CONFIG,...e.config??{}},this.logger=e.logger,this.storeClientsWithDefaults(e),this.initializeHealthChecks(),this.config.shouldSetupSignalHandlers&&this.setupSignalHandlers()}storeClientsWithDefaults(t){let r=t.clients;if(r){for(let[t,i]of Object.entries(r))if(i)if(t===e.SERVER)this.attachServer(i);else{let r=Array.isArray(i)?i:[i],a=n[t]??n[e.CUSTOM],o=r.map(e=>({connection:e.connection,healthCheck:e.healthCheck??a?.healthCheck,shutdownHandlers:{...a?.shutdownHandlers,...e.shutdownHandlers},options:{...a?.options,...e.options,name:e.options?.name||a?.options?.name||t}}));this.clientsMap.set(t,o)}}}setupSignalHandlers(){this.logger.info(`Setting up global signal listeners for graceful shutdown`),process.on(`SIGTERM`,async()=>{await this.shutdown(`SIGTERM`)}),process.on(`SIGINT`,async()=>{await this.shutdown(`SIGINT`)})}initializeHealthChecks(){let e=!0;this.clientsMap.forEach(t=>{t.forEach(t=>{if(!t.healthCheck)return;let n=t.options?.includeInLivenessCheck,r=t.options?.includeInReadinessCheck;this.canSafelySkipLivenessWhenReady(n,r)||(e=!1),n&&this.livenessChecks.push({check:t.healthCheck,client:t.connection,name:t.options?.name||`unknown`,customHandlersPayload:t.options?.customHandlersPayload}),r&&this.readinessChecks.push({check:t.healthCheck,client:t.connection,name:t.options?.name||`unknown`,customHandlersPayload:t.options?.customHandlersPayload})})}),this.config.enableLivenessOptimization=this.config.enableLivenessOptimization&&e,this.config.enableLivenessOptimization?this.logger.info(`Liveness check optimization enabled - liveness checks will be skipped when pod is ready`):e||this.logger.warn(`Liveness check optimization disabled - some clients are in liveness checks but not in readiness checks`)}attachServer(t){let r=n[e.SERVER],i=(Array.isArray(t)?t:[t]).map(e=>{let t=a({server:e.connection});return{connection:e.connection,healthCheck:e.healthCheck??r?.healthCheck,shutdownHandlers:{...r?.shutdownHandlers,...e.shutdownHandlers},options:{...r?.options,...e.options,name:e.options?.name||r?.options?.name||`HTTP Server`,customHandlersPayload:{...e.options?.customHandlersPayload,httpTerminator:t}}}});this.clientsMap.set(e.SERVER,i)}async aliveCheck(){if(this.isShuttingDown||this.config.enableLivenessOptimization&&this.isReady)return{statusCode:200,status:`ok`};try{return await Promise.all(this.livenessChecks.map(({check:e,client:t,name:n,customHandlersPayload:r})=>e({client:t,name:n,customHandlersPayload:r,logger:this.logger}))),{statusCode:200,status:`ok`}}catch(e){let t=new i(`Liveness probe failed: ${e.message}`);throw t.stack=e.stack,t}}async readyCheck(){if(this.isShuttingDown)throw this.isReady=!1,this.logger.warn(`Application is shutting down: readiness probe failed`),new i(`Readiness probe failed: Application is shutting down`);try{return await Promise.all(this.readinessChecks.map(({check:e,client:t,name:n,customHandlersPayload:r})=>e({client:t,name:n,customHandlersPayload:r,logger:this.logger}))),this.onReadyTriggered||(this.onReadyTriggered=!0,await this.config.onReady?.(),this.logger.info(`onReady hook executed successfully`)),this.isReady=!0,{statusCode:200,status:`ok`}}catch(e){this.isReady=!1;let t=new i(`Readiness probe failed: ${e.message}`);throw t.stack=e.stack,t}}groupShutdownTasksByStage(e){let n={[t.STOP_ACCEPTING_WORK]:[],[t.DRAIN]:[],[t.CLOSE]:[]};return this.clientsMap.forEach(t=>{t.forEach(t=>{let r=t.options?.name||`unknown`,i=t.options?.enabledShutdownStages;if(t.shutdownHandlers)for(let[a,o]of Object.entries(t.shutdownHandlers))o&&i?.includes(a)&&n[a].push({fn:()=>o({signal:e,client:t.connection,name:r,customHandlersPayload:t.options?.customHandlersPayload,logger:this.logger}),name:r})})}),n}async executeShutdownStage(e,t){t.length!==0&&(this.logger.info(`[${e} stage] Start executing shutdown stage ${e}...`),(await Promise.allSettled(t.map(e=>e.fn()))).forEach((n,r)=>{if(n.status===`rejected`){let{name:i}=t[r];this.logger.error(`[${e} stage] Failed to shutdown '${i}'`,{error:n.reason})}}),this.logger.info(`[${e} stage] Shutdown stage ${e} completed`))}async shutdown(e=`MANUAL`){try{if(this.isShuttingDown){this.logger.warn(`Shutdown already in progress`);return}let{shutdownDelayMs:n,useExit0:i,onShutdown:a,beforeShutdown:o}=this.config;this.logger.info(`${e} received: setting readiness state to false`),this.isShuttingDown=!0,this.isReady=!1,await o?.(),this.logger.info(`Waiting ${n}ms before shutdown...`),await r(n);let s=this.groupShutdownTasksByStage(e);await this.executeShutdownStage(t.STOP_ACCEPTING_WORK,s[t.STOP_ACCEPTING_WORK]),this.logger.info(`Waiting 4000ms to allow in-flight work to complete...`),await r(4e3),await this.executeShutdownStage(t.DRAIN,s[t.DRAIN]),await this.executeShutdownStage(t.CLOSE,s[t.CLOSE]),await a?.(),this.logger.info(`Shutdown process completed`),i&&(this.logger.info(`Exiting process after ${e}`),process.exit(0))}catch(t){this.logger.error(`Error during shutdown process, forcing process exit despite shutdown error`,{error:t,signal:e}),process.exit(1)}}};export{o as HealthManager};
+import{CLIENT_TYPES as e,SHUTDOWN_STAGES as t}from"./types.js";import{CLIENT_DEFAULTS_MAP as n}from"./client-defaults.js";import{setTimeout as r}from"node:timers/promises";import{ServiceUnavailableError as i}from"@autofleet/errors";import{createHttpTerminator as a}from"http-terminator";var o=class{constructor(e){this.DEFAULT_CONFIG={shutdownDelayMs:6e3,useExit0:!0,shouldSetupSignalHandlers:!0,enableLivenessOptimization:!0},this.isShuttingDown=!1,this.isReady=!1,this.clientsMap=new Map,this.livenessChecks=[],this.readinessChecks=[],this.canSafelySkipLivenessWhenReady=(e,t)=>e?!!t:!0,this.config={...this.DEFAULT_CONFIG,...e.config??{}},this.logger=e.logger,this.storeClientsWithDefaults(e),this.initializeHealthChecks(),this.config.shouldSetupSignalHandlers&&this.setupSignalHandlers()}storeClientsWithDefaults(t){let r=t.clients;if(r){for(let[t,i]of Object.entries(r))if(i)if(t===e.SERVER)this.attachServer(i);else{let r=Array.isArray(i)?i:[i],a=n[t]??n[e.CUSTOM],o=r.map(e=>({connection:e.connection,healthCheck:e.healthCheck??a?.healthCheck,shutdownHandlers:{...a?.shutdownHandlers,...e.shutdownHandlers},options:{...a?.options,...e.options,name:e.options?.name||a?.options?.name||t}}));this.clientsMap.set(t,o)}}}setupSignalHandlers(){this.logger.info(`Setting up global signal listeners for graceful shutdown`),process.on(`SIGTERM`,async()=>{await this.shutdown(`SIGTERM`)}),process.on(`SIGINT`,async()=>{await this.shutdown(`SIGINT`)})}initializeHealthChecks(){let e=!0;this.clientsMap.forEach(t=>{t.forEach(t=>{if(!t.healthCheck)return;let n=t.options?.includeInLivenessCheck,r=t.options?.includeInReadinessCheck;this.canSafelySkipLivenessWhenReady(n,r)||(e=!1),n&&this.livenessChecks.push({check:t.healthCheck,client:t.connection,name:t.options?.name||`unknown`,customHandlersPayload:t.options?.customHandlersPayload}),r&&this.readinessChecks.push({check:t.healthCheck,client:t.connection,name:t.options?.name||`unknown`,customHandlersPayload:t.options?.customHandlersPayload})})}),this.config.enableLivenessOptimization=this.config.enableLivenessOptimization&&e,this.config.enableLivenessOptimization?this.logger.info(`Liveness check optimization enabled - liveness checks will be skipped when pod is ready`):e||this.logger.warn(`Liveness check optimization disabled - some clients are in liveness checks but not in readiness checks`)}attachServer(t){let r=n[e.SERVER],i=(Array.isArray(t)?t:[t]).map(e=>{let t=a({server:e.connection});return{connection:e.connection,healthCheck:e.healthCheck??r?.healthCheck,shutdownHandlers:{...r?.shutdownHandlers,...e.shutdownHandlers},options:{...r?.options,...e.options,name:e.options?.name||r?.options?.name||`HTTP Server`,customHandlersPayload:{...e.options?.customHandlersPayload,httpTerminator:t}}}});this.clientsMap.set(e.SERVER,i)}async aliveCheck(){if(this.isShuttingDown||this.config.enableLivenessOptimization&&this.isReady)return{statusCode:200,status:`ok`};try{return await Promise.all(this.livenessChecks.map(({check:e,client:t,name:n,customHandlersPayload:r})=>e({client:t,name:n,customHandlersPayload:r,logger:this.logger}))),{statusCode:200,status:`ok`}}catch(e){let t=new i(`Liveness probe failed: ${e.message}`);throw t.stack=e.stack,t}}async readyCheck(){if(this.isShuttingDown)throw this.isReady=!1,this.logger.warn(`Application is shutting down: readiness probe failed`),new i(`Readiness probe failed: Application is shutting down`);try{return await Promise.all(this.readinessChecks.map(({check:e,client:t,name:n,customHandlersPayload:r})=>e({client:t,name:n,customHandlersPayload:r,logger:this.logger}))),this.isReady=!0,{statusCode:200,status:`ok`}}catch(e){this.isReady=!1;let t=new i(`Readiness probe failed: ${e.message}`);throw t.stack=e.stack,t}}groupShutdownTasksByStage(e){let n={[t.STOP_ACCEPTING_WORK]:[],[t.DRAIN]:[],[t.CLOSE]:[]};return this.clientsMap.forEach(t=>{t.forEach(t=>{let r=t.options?.name||`unknown`,i=t.options?.enabledShutdownStages;if(t.shutdownHandlers)for(let[a,o]of Object.entries(t.shutdownHandlers))o&&i?.includes(a)&&n[a].push({fn:()=>o({signal:e,client:t.connection,name:r,customHandlersPayload:t.options?.customHandlersPayload,logger:this.logger}),name:r})})}),n}async executeShutdownStage(e,t){t.length!==0&&(this.logger.info(`[${e} stage] Start executing shutdown stage ${e}...`),(await Promise.allSettled(t.map(e=>e.fn()))).forEach((n,r)=>{if(n.status===`rejected`){let{name:i}=t[r];this.logger.error(`[${e} stage] Failed to shutdown '${i}'`,{error:n.reason})}}),this.logger.info(`[${e} stage] Shutdown stage ${e} completed`))}async shutdown(e=`MANUAL`){try{if(this.isShuttingDown){this.logger.warn(`Shutdown already in progress`);return}let{shutdownDelayMs:n,useExit0:i,onShutdown:a,beforeShutdown:o}=this.config;this.logger.info(`${e} received: setting readiness state to false`),this.isShuttingDown=!0,this.isReady=!1,await o?.(),this.logger.info(`Waiting ${n}ms before shutdown...`),await r(n);let s=this.groupShutdownTasksByStage(e);await this.executeShutdownStage(t.STOP_ACCEPTING_WORK,s[t.STOP_ACCEPTING_WORK]),this.logger.info(`Waiting 4000ms to allow in-flight work to complete...`),await r(4e3),await this.executeShutdownStage(t.DRAIN,s[t.DRAIN]),await this.executeShutdownStage(t.CLOSE,s[t.CLOSE]),await a?.(),this.logger.info(`Shutdown process completed`),i&&(this.logger.info(`Exiting process after ${e}`),process.exit(0))}catch(t){this.logger.error(`Error during shutdown process, forcing process exit despite shutdown error`,{error:t,signal:e}),process.exit(1)}}};export{o as HealthManager};
 //# sourceMappingURL=health-manager.js.map

package/lib/health-manager/health-manager.js.map

@@ -1 +1 @@
-{"version":3,"file":"health-manager.js","names":["client","error: any","shutdownTasksByStage: Record<ShutdownStage, { fn: () => Promise<any>; name: string; }[]>"],"sources":["../../src/health-manager/health-manager.ts"],"sourcesContent":["import { setTimeout } from 'node:timers/promises';\nimport { ServiceUnavailableError } from '@autofleet/errors';\nimport type { LoggerInstanceManager } from '@autofleet/logger';\nimport { createHttpTerminator } from 'http-terminator';\nimport {\n  type ClientConfig,\n  type HealthManagerConfig,\n  type HealthManagerOptions,\n  type ClientType,\n  type HealthCheckHandler,\n  type ShutdownStage,\n  type ServerConfig,\n  CLIENT_TYPES,\n  SHUTDOWN_STAGES,\n  type ShutdownHandler,\n} from './types';\nimport { CLIENT_DEFAULTS_MAP } from './client-defaults';\n\n/**\n * Manages application health state and graceful shutdown for Kubernetes liveness/readiness probes\n *\n * Features:\n * - Provides aliveCheck() and readyCheck() methods for liveness/readiness probes\n * - Auto-registers SIGTERM/SIGINT handlers for graceful shutdown\n * - Manages ready state during shutdown to fail readiness probes\n */\nexport class HealthManager {\n  private readonly DEFAULT_CONFIG: HealthManagerConfig = {\n    shutdownDelayMs: 6_000,\n    useExit0: true,\n    shouldSetupSignalHandlers: true,\n    enableLivenessOptimization: true,\n  };\n\n  private isShuttingDown = false;\n  private isReady = false;\n  private onReadyTriggered = false;\n\n  private config: HealthManagerConfig;\n\n  private clientsMap = new Map<string, ClientConfig<any>[]>();\n\n  private livenessChecks: { check: HealthCheckHandler<any>; client: any; name: string; customHandlersPayload?: any; }[] = [];\n  private readinessChecks: { check: HealthCheckHandler<any>; client: any; name: string; customHandlersPayload?: any; }[] = [];\n\n  private logger: LoggerInstanceManager;\n\n  /**\n   * Creates a new HealthManager instance\n   *\n   * @param params Configuration options for health checks and shutdown behavior\n   */\n  constructor(params: HealthManagerOptions) {\n    this.config = {\n      ...this.DEFAULT_CONFIG,\n      ...(params.config ?? {}),\n    };\n\n    this.logger = params.logger;\n\n    this.storeClientsWithDefaults(params);\n\n    this.initializeHealthChecks();\n\n    if (this.config.shouldSetupSignalHandlers) {\n      this.setupSignalHandlers();\n    }\n  }\n\n  /**\n   * Store clients with default health check and shutdown functions\n   * @private\n   */\n  private storeClientsWithDefaults(params: HealthManagerOptions): void {\n    const clients = params.clients;\n    if (!clients) {\n      return;\n    }\n\n    for (const [type, client] of Object.entries(clients)) {\n      if (!client) {\n        continue;\n      }\n\n      if (type === CLIENT_TYPES.SERVER) {\n        this.attachServer(client as ServerConfig | ServerConfig[]);\n      } else {\n        const clientsArray = Array.isArray(client) ? client : [client];\n        const defaults = CLIENT_DEFAULTS_MAP[type as ClientType] ?? CLIENT_DEFAULTS_MAP[CLIENT_TYPES.CUSTOM];\n\n        const clientsWithDefaults = clientsArray.map(client => ({\n          connection: client.connection,\n          healthCheck: client.healthCheck ?? defaults?.healthCheck,\n          shutdownHandlers: {\n            ...defaults?.shutdownHandlers,\n            ...client.shutdownHandlers,\n          },\n          options: {\n            ...defaults?.options,\n            ...client.options,\n            name: client.options?.name || defaults?.options?.name || type,\n          },\n        }));\n\n        this.clientsMap.set(type, clientsWithDefaults);\n      }\n    }\n  }\n\n  /**\n   * Register SIGTERM and SIGINT signal listeners\n   * @private\n   */\n  private setupSignalHandlers(): void {\n    this.logger.info('Setting up global signal listeners for graceful shutdown');\n\n    process.on('SIGTERM', async () => {\n      await this.shutdown('SIGTERM');\n    });\n\n    process.on('SIGINT', async () => {\n      await this.shutdown('SIGINT');\n    });\n  }\n\n  /**\n   * Check if liveness optimization is safe to enable for a given client configuration\n   * If a client is in liveness but not readiness, we cannot skip liveness checks\n   * because the optimization relies on readiness passing, but this client isn't checked in readiness\n   * @private\n   */\n  private canSafelySkipLivenessWhenReady = (includeInLiveness: boolean | undefined, includeInReadiness: boolean | undefined): boolean => {\n    // If not in liveness, optimization is safe (skipping liveness doesn't affect this client)\n    if (!includeInLiveness) {\n      return true;\n    }\n\n    // If in liveness, must also be in readiness for liveness optimization to be safe\n    return !!includeInReadiness;\n  };\n\n  /**\n   * Initialize health check arrays based on configured clients\n   * @private\n   */\n  private initializeHealthChecks(): void {\n    let allClientsAllowLivenessOptimization = true;\n\n    this.clientsMap.forEach((clients) => {\n      clients.forEach((client) => {\n        if (!client.healthCheck) {\n          return;\n        }\n\n        const includeInLiveness = client.options?.includeInLivenessCheck;\n        const includeInReadiness = client.options?.includeInReadinessCheck;\n\n        // Check if this client configuration allows liveness optimization\n        if (!this.canSafelySkipLivenessWhenReady(includeInLiveness, includeInReadiness)) {\n          allClientsAllowLivenessOptimization = false;\n        }\n\n        if (includeInLiveness) {\n          this.livenessChecks.push({\n            check: client.healthCheck,\n            client: client.connection,\n            name: client.options?.name || 'unknown',\n            customHandlersPayload: client.options?.customHandlersPayload,\n          });\n        }\n\n        if (includeInReadiness) {\n          this.readinessChecks.push({\n            check: client.healthCheck,\n            client: client.connection,\n            name: client.options?.name || 'unknown',\n            customHandlersPayload: client.options?.customHandlersPayload,\n          });\n        }\n      });\n    });\n\n    // Apply optimization setting: only enable if user requested it AND all clients allow it\n    this.config.enableLivenessOptimization = this.config.enableLivenessOptimization && allClientsAllowLivenessOptimization;\n\n    // Log optimization status\n    if (this.config.enableLivenessOptimization) {\n      this.logger.info('Liveness check optimization enabled - liveness checks will be skipped when pod is ready');\n    } else if (!allClientsAllowLivenessOptimization) {\n      this.logger.warn('Liveness check optimization disabled - some clients are in liveness checks but not in readiness checks');\n    }\n  }\n\n  /**\n   * Attach an HTTP or HTTPS server to be managed during shutdown\n   * @param serverConfig Server configuration (ClientConfig with HTTP/HTTPS server)\n   */\n  attachServer(serverConfig: ServerConfig | ServerConfig[]): void {\n    const serverDefaults = CLIENT_DEFAULTS_MAP[CLIENT_TYPES.SERVER];\n    const serverConfigArray = Array.isArray(serverConfig) ? serverConfig : [serverConfig];\n\n    const serverClients = serverConfigArray.map((config) => {\n      // Create httpTerminator for this server instance\n      const httpTerminator = createHttpTerminator({\n        server: config.connection,\n      });\n\n      return {\n        connection: config.connection,\n        healthCheck: config.healthCheck ?? serverDefaults?.healthCheck,\n        shutdownHandlers: {\n          ...serverDefaults?.shutdownHandlers,\n          ...config.shutdownHandlers,\n        },\n        options: {\n          ...serverDefaults?.options,\n          ...config.options,\n          name: config.options?.name || serverDefaults?.options?.name || 'HTTP Server',\n          customHandlersPayload: {\n            ...config.options?.customHandlersPayload,\n            httpTerminator,\n          },\n        },\n      };\n    });\n\n    this.clientsMap.set(CLIENT_TYPES.SERVER, serverClients);\n  }\n\n  /**\n   * Performs liveness health checks for clients configured with includeInLivenessCheck\n   *\n   * @returns Promise that resolves to { status: 'ok', statusCode: 200 } on success\n   * @throws ServiceUnavailableError with statusCode 503 if any health check fails\n   */\n  async aliveCheck(): Promise<{ status: string; statusCode: number; }> {\n    if (this.isShuttingDown) {\n      return { statusCode: 200, status: 'ok' };\n    }\n\n    // Skip liveness check if optimization is enabled and the readiness check has already passed\n    if (this.config.enableLivenessOptimization && this.isReady) {\n      return { statusCode: 200, status: 'ok' };\n    }\n\n    try {\n      await Promise.all(\n        this.livenessChecks.map(({ check, client, name, customHandlersPayload }) =>\n          check({ client, name, customHandlersPayload, logger: this.logger }),\n        ),\n      );\n\n      return { statusCode: 200, status: 'ok' };\n    } catch (error: any) {\n      const err = new ServiceUnavailableError(`Liveness probe failed: ${error.message}`);\n      err.stack = error.stack;\n      throw err;\n    }\n  }\n\n  /**\n   * Performs readiness health checks for clients configured with includeInReadinessCheck\n   *\n   * @returns Promise that resolves to { status: 'ok', statusCode: 200 } if ready\n   * @throws ServiceUnavailableError with statusCode 503 if not ready or health check fails\n   */\n  async readyCheck(): Promise<{ status: string; statusCode: number; }> {\n    if (this.isShuttingDown) {\n      this.isReady = false;\n      this.logger.warn('Application is shutting down: readiness probe failed');\n      throw new ServiceUnavailableError('Readiness probe failed: Application is shutting down');\n    }\n\n    try {\n      await Promise.all(\n        this.readinessChecks.map(({ check, client, name, customHandlersPayload }) =>\n          check({ client, name, customHandlersPayload, logger: this.logger }),\n        ),\n      );\n\n      if (!this.onReadyTriggered) {\n        this.onReadyTriggered = true;\n        await this.config.onReady?.();\n        this.logger.info('onReady hook executed successfully');\n      }\n\n      this.isReady = true;\n\n      return { statusCode: 200, status: 'ok' };\n    } catch (error: any) {\n      this.isReady = false;\n      const err = new ServiceUnavailableError(`Readiness probe failed: ${error.message}`);\n      err.stack = error.stack;\n      throw err;\n    }\n  }\n\n  /**\n   * Groups client shutdown handlers by stage (stopAcceptingWork, drain, close).\n   *\n   * @private\n   * @param signal - Signal name to pass to shutdown handlers\n   * @returns Record of shutdown tasks organized by stage\n   */\n  private groupShutdownTasksByStage(signal: string): Record<ShutdownStage, { fn: () => Promise<any>; name: string; }[]> {\n    const shutdownTasksByStage: Record<ShutdownStage, { fn: () => Promise<any>; name: string; }[]> = {\n      [SHUTDOWN_STAGES.STOP_ACCEPTING_WORK]: [],\n      [SHUTDOWN_STAGES.DRAIN]: [],\n      [SHUTDOWN_STAGES.CLOSE]: [],\n    };\n\n    this.clientsMap.forEach((clients) => {\n      clients.forEach((client) => {\n        const clientName = client.options?.name || 'unknown';\n        const enabledShutdownStages = client.options?.enabledShutdownStages;\n\n        if (client.shutdownHandlers) {\n          for (const [stage, shutdownHandler] of Object.entries(client.shutdownHandlers) as [ShutdownStage, ShutdownHandler<any>][]) {\n            if (shutdownHandler && enabledShutdownStages?.includes(stage)) {\n              shutdownTasksByStage[stage].push({\n                fn: () => shutdownHandler({\n                  signal,\n                  client: client.connection,\n                  name: clientName,\n                  customHandlersPayload: client.options?.customHandlersPayload,\n                  logger: this.logger,\n                }),\n                name: clientName,\n              });\n            }\n          }\n        }\n      });\n    });\n\n    return shutdownTasksByStage;\n  }\n\n  /**\n   * Execute shutdown tasks for a specific stage\n   * @private\n   * @param stage - The shutdown stage to execute\n   * @param tasks - Array of shutdown tasks to execute\n   */\n  private async executeShutdownStage(stage: ShutdownStage, tasks: { fn: () => Promise<any>; name: string; }[]): Promise<void> {\n    if (tasks.length === 0) {\n      return;\n    }\n\n    this.logger.info(`[${stage} stage] Start executing shutdown stage ${stage}...`);\n    const results = await Promise.allSettled(tasks.map(t => t.fn()));\n    results.forEach((result, index) => {\n      if (result.status === 'rejected') {\n        const { name } = tasks[index];\n        this.logger.error(`[${stage} stage] Failed to shutdown '${name}'`, { error: result.reason });\n      }\n    });\n    this.logger.info(`[${stage} stage] Shutdown stage ${stage} completed`);\n  }\n\n  /**\n   * Triggers graceful shutdown process\n   * Can be called manually or automatically via SIGTERM/SIGINT signal handlers\n   *\n   * Shutdown happens in three stages:\n   * 1. stopAcceptingWork - Stop accepting new requests/work\n   * 2. drain - Allow in-flight work to complete\n   * 3. close - Close all connections and clean up resources\n   *\n   * @param signal - Signal name (e.g., 'SIGTERM', 'SIGINT', 'MANUAL')\n   */\n  async shutdown(signal = 'MANUAL'): Promise<void> {\n    try {\n      if (this.isShuttingDown) {\n        this.logger.warn('Shutdown already in progress');\n        return;\n      }\n\n      const { shutdownDelayMs, useExit0, onShutdown, beforeShutdown } = this.config;\n\n      this.logger.info(`${signal} received: setting readiness state to false`);\n\n      this.isShuttingDown = true;\n      this.isReady = false;\n\n      await beforeShutdown?.();\n\n      this.logger.info(`Waiting ${shutdownDelayMs}ms before shutdown...`);\n      await setTimeout(shutdownDelayMs);\n\n      const shutdownTasksByStage = this.groupShutdownTasksByStage(signal);\n\n      await this.executeShutdownStage(SHUTDOWN_STAGES.STOP_ACCEPTING_WORK, shutdownTasksByStage[SHUTDOWN_STAGES.STOP_ACCEPTING_WORK]);\n\n      // Wait between stages to allow in-flight work to complete (we are missing a drain stage in our rabbitmq package)\n      // TODO once we have proper drain handlers, we can remove this fixed wait\n      this.logger.info('Waiting 4000ms to allow in-flight work to complete...');\n      await setTimeout(4_000);\n\n      await this.executeShutdownStage(SHUTDOWN_STAGES.DRAIN, shutdownTasksByStage[SHUTDOWN_STAGES.DRAIN]);\n      await this.executeShutdownStage(SHUTDOWN_STAGES.CLOSE, shutdownTasksByStage[SHUTDOWN_STAGES.CLOSE]);\n\n      await onShutdown?.();\n\n      this.logger.info('Shutdown process completed');\n\n      if (useExit0) {\n        this.logger.info(`Exiting process after ${signal}`);\n        process.exit(0);\n      }\n    } catch (error: any) {\n      this.logger.error('Error during shutdown process, forcing process exit despite shutdown error', { error, signal });\n      process.exit(1);\n    }\n  }\n}\n"],"mappings":"+RA0BA,IAAa,EAAb,KAA2B,CA0BzB,YAAY,EAA8B,qBAzBa,CACrD,gBAAiB,IACjB,SAAU,GACV,0BAA2B,GAC3B,2BAA4B,GAC7B,qBAEwB,gBACP,yBACS,mBAIN,IAAI,wBAE+F,EAAE,sBACD,EAAE,sCAwFjF,EAAwC,IAE3E,EAKE,CAAC,CAAC,EAJA,GAjFT,KAAK,OAAS,CACZ,GAAG,KAAK,eACR,GAAI,EAAO,QAAU,EAAE,CACxB,CAED,KAAK,OAAS,EAAO,OAErB,KAAK,yBAAyB,EAAO,CAErC,KAAK,wBAAwB,CAEzB,KAAK,OAAO,2BACd,KAAK,qBAAqB,CAQ9B,yBAAiC,EAAoC,CACnE,IAAM,EAAU,EAAO,QAClB,KAIL,KAAK,GAAM,CAAC,EAAM,KAAW,OAAO,QAAQ,EAAQ,CAC7C,KAIL,GAAI,IAAS,EAAa,OACxB,KAAK,aAAa,EAAwC,KACrD,CACL,IAAM,EAAe,MAAM,QAAQ,EAAO,CAAG,EAAS,CAAC,EAAO,CACxD,EAAW,EAAoB,IAAuB,EAAoB,EAAa,QAEvF,EAAsB,EAAa,IAAI,IAAW,CACtD,WAAYA,EAAO,WACnB,YAAaA,EAAO,aAAe,GAAU,YAC7C,iBAAkB,CAChB,GAAG,GAAU,iBACb,GAAGA,EAAO,iBACX,CACD,QAAS,CACP,GAAG,GAAU,QACb,GAAGA,EAAO,QACV,KAAMA,EAAO,SAAS,MAAQ,GAAU,SAAS,MAAQ,EAC1D,CACF,EAAE,CAEH,KAAK,WAAW,IAAI,EAAM,EAAoB,GASpD,qBAAoC,CAClC,KAAK,OAAO,KAAK,2DAA2D,CAE5E,QAAQ,GAAG,UAAW,SAAY,CAChC,MAAM,KAAK,SAAS,UAAU,EAC9B,CAEF,QAAQ,GAAG,SAAU,SAAY,CAC/B,MAAM,KAAK,SAAS,SAAS,EAC7B,CAuBJ,wBAAuC,CACrC,IAAI,EAAsC,GAE1C,KAAK,WAAW,QAAS,GAAY,CACnC,EAAQ,QAAS,GAAW,CAC1B,GAAI,CAAC,EAAO,YACV,OAGF,IAAM,EAAoB,EAAO,SAAS,uBACpC,EAAqB,EAAO,SAAS,wBAGtC,KAAK,+BAA+B,EAAmB,EAAmB,GAC7E,EAAsC,IAGpC,GACF,KAAK,eAAe,KAAK,CACvB,MAAO,EAAO,YACd,OAAQ,EAAO,WACf,KAAM,EAAO,SAAS,MAAQ,UAC9B,sBAAuB,EAAO,SAAS,sBACxC,CAAC,CAGA,GACF,KAAK,gBAAgB,KAAK,CACxB,MAAO,EAAO,YACd,OAAQ,EAAO,WACf,KAAM,EAAO,SAAS,MAAQ,UAC9B,sBAAuB,EAAO,SAAS,sBACxC,CAAC,EAEJ,EACF,CAGF,KAAK,OAAO,2BAA6B,KAAK,OAAO,4BAA8B,EAG/E,KAAK,OAAO,2BACd,KAAK,OAAO,KAAK,0FAA0F,CACjG,GACV,KAAK,OAAO,KAAK,yGAAyG,CAQ9H,aAAa,EAAmD,CAC9D,IAAM,EAAiB,EAAoB,EAAa,QAGlD,GAFoB,MAAM,QAAQ,EAAa,CAAG,EAAe,CAAC,EAAa,EAE7C,IAAK,GAAW,CAEtD,IAAM,EAAiB,EAAqB,CAC1C,OAAQ,EAAO,WAChB,CAAC,CAEF,MAAO,CACL,WAAY,EAAO,WACnB,YAAa,EAAO,aAAe,GAAgB,YACnD,iBAAkB,CAChB,GAAG,GAAgB,iBACnB,GAAG,EAAO,iBACX,CACD,QAAS,CACP,GAAG,GAAgB,QACnB,GAAG,EAAO,QACV,KAAM,EAAO,SAAS,MAAQ,GAAgB,SAAS,MAAQ,cAC/D,sBAAuB,CACrB,GAAG,EAAO,SAAS,sBACnB,iBACD,CACF,CACF,EACD,CAEF,KAAK,WAAW,IAAI,EAAa,OAAQ,EAAc,CASzD,MAAM,YAA+D,CAMnE,GALI,KAAK,gBAKL,KAAK,OAAO,4BAA8B,KAAK,QACjD,MAAO,CAAE,WAAY,IAAK,OAAQ,KAAM,CAG1C,GAAI,CAOF,OANA,MAAM,QAAQ,IACZ,KAAK,eAAe,KAAK,CAAE,QAAO,SAAQ,OAAM,2BAC9C,EAAM,CAAE,SAAQ,OAAM,wBAAuB,OAAQ,KAAK,OAAQ,CAAC,CACpE,CACF,CAEM,CAAE,WAAY,IAAK,OAAQ,KAAM,OACjCC,EAAY,CACnB,IAAM,EAAM,IAAI,EAAwB,0BAA0B,EAAM,UAAU,CAElF,KADA,GAAI,MAAQ,EAAM,MACZ,GAUV,MAAM,YAA+D,CACnE,GAAI,KAAK,eAGP,KAFA,MAAK,QAAU,GACf,KAAK,OAAO,KAAK,uDAAuD,CAClE,IAAI,EAAwB,uDAAuD,CAG3F,GAAI,CAeF,OAdA,MAAM,QAAQ,IACZ,KAAK,gBAAgB,KAAK,CAAE,QAAO,SAAQ,OAAM,2BAC/C,EAAM,CAAE,SAAQ,OAAM,wBAAuB,OAAQ,KAAK,OAAQ,CAAC,CACpE,CACF,CAEI,KAAK,mBACR,KAAK,iBAAmB,GACxB,MAAM,KAAK,OAAO,WAAW,CAC7B,KAAK,OAAO,KAAK,qCAAqC,EAGxD,KAAK,QAAU,GAER,CAAE,WAAY,IAAK,OAAQ,KAAM,OACjCA,EAAY,CACnB,KAAK,QAAU,GACf,IAAM,EAAM,IAAI,EAAwB,2BAA2B,EAAM,UAAU,CAEnF,KADA,GAAI,MAAQ,EAAM,MACZ,GAWV,0BAAkC,EAAoF,CACpH,IAAMC,EAA2F,EAC9F,EAAgB,qBAAsB,EAAE,EACxC,EAAgB,OAAQ,EAAE,EAC1B,EAAgB,OAAQ,EAAE,CAC5B,CA0BD,OAxBA,KAAK,WAAW,QAAS,GAAY,CACnC,EAAQ,QAAS,GAAW,CAC1B,IAAM,EAAa,EAAO,SAAS,MAAQ,UACrC,EAAwB,EAAO,SAAS,sBAE9C,GAAI,EAAO,qBACJ,GAAM,CAAC,EAAO,KAAoB,OAAO,QAAQ,EAAO,iBAAiB,CACxE,GAAmB,GAAuB,SAAS,EAAM,EAC3D,EAAqB,GAAO,KAAK,CAC/B,OAAU,EAAgB,CACxB,SACA,OAAQ,EAAO,WACf,KAAM,EACN,sBAAuB,EAAO,SAAS,sBACvC,OAAQ,KAAK,OACd,CAAC,CACF,KAAM,EACP,CAAC,EAIR,EACF,CAEK,EAST,MAAc,qBAAqB,EAAsB,EAAmE,CACtH,EAAM,SAAW,IAIrB,KAAK,OAAO,KAAK,IAAI,EAAM,yCAAyC,EAAM,KAAK,EAC/D,MAAM,QAAQ,WAAW,EAAM,IAAI,GAAK,EAAE,IAAI,CAAC,CAAC,EACxD,SAAS,EAAQ,IAAU,CACjC,GAAI,EAAO,SAAW,WAAY,CAChC,GAAM,CAAE,QAAS,EAAM,GACvB,KAAK,OAAO,MAAM,IAAI,EAAM,8BAA8B,EAAK,GAAI,CAAE,MAAO,EAAO,OAAQ,CAAC,GAE9F,CACF,KAAK,OAAO,KAAK,IAAI,EAAM,yBAAyB,EAAM,YAAY,EAcxE,MAAM,SAAS,EAAS,SAAyB,CAC/C,GAAI,CACF,GAAI,KAAK,eAAgB,CACvB,KAAK,OAAO,KAAK,+BAA+B,CAChD,OAGF,GAAM,CAAE,kBAAiB,WAAU,aAAY,kBAAmB,KAAK,OAEvE,KAAK,OAAO,KAAK,GAAG,EAAO,6CAA6C,CAExE,KAAK,eAAiB,GACtB,KAAK,QAAU,GAEf,MAAM,KAAkB,CAExB,KAAK,OAAO,KAAK,WAAW,EAAgB,uBAAuB,CACnE,MAAM,EAAW,EAAgB,CAEjC,IAAM,EAAuB,KAAK,0BAA0B,EAAO,CAEnE,MAAM,KAAK,qBAAqB,EAAgB,oBAAqB,EAAqB,EAAgB,qBAAqB,CAI/H,KAAK,OAAO,KAAK,wDAAwD,CACzE,MAAM,EAAW,IAAM,CAEvB,MAAM,KAAK,qBAAqB,EAAgB,MAAO,EAAqB,EAAgB,OAAO,CACnG,MAAM,KAAK,qBAAqB,EAAgB,MAAO,EAAqB,EAAgB,OAAO,CAEnG,MAAM,KAAc,CAEpB,KAAK,OAAO,KAAK,6BAA6B,CAE1C,IACF,KAAK,OAAO,KAAK,yBAAyB,IAAS,CACnD,QAAQ,KAAK,EAAE,QAEVD,EAAY,CACnB,KAAK,OAAO,MAAM,6EAA8E,CAAE,QAAO,SAAQ,CAAC,CAClH,QAAQ,KAAK,EAAE"}
+{"version":3,"file":"health-manager.js","names":["client","error: any","shutdownTasksByStage: Record<ShutdownStage, { fn: () => Promise<any>; name: string; }[]>"],"sources":["../../src/health-manager/health-manager.ts"],"sourcesContent":["import { setTimeout } from 'node:timers/promises';\nimport { ServiceUnavailableError } from '@autofleet/errors';\nimport type { LoggerInstanceManager } from '@autofleet/logger';\nimport { createHttpTerminator } from 'http-terminator';\nimport {\n  type ClientConfig,\n  type HealthManagerConfig,\n  type HealthManagerOptions,\n  type ClientType,\n  type HealthCheckHandler,\n  type ShutdownStage,\n  type ServerConfig,\n  CLIENT_TYPES,\n  SHUTDOWN_STAGES,\n  type ShutdownHandler,\n} from './types';\nimport { CLIENT_DEFAULTS_MAP } from './client-defaults';\n\n/**\n * Manages application health state and graceful shutdown for Kubernetes liveness/readiness probes\n *\n * Features:\n * - Provides aliveCheck() and readyCheck() methods for liveness/readiness probes\n * - Auto-registers SIGTERM/SIGINT handlers for graceful shutdown\n * - Manages ready state during shutdown to fail readiness probes\n */\nexport class HealthManager {\n  private readonly DEFAULT_CONFIG: HealthManagerConfig = {\n    shutdownDelayMs: 6_000,\n    useExit0: true,\n    shouldSetupSignalHandlers: true,\n    enableLivenessOptimization: true,\n  };\n\n  private isShuttingDown = false;\n  private isReady = false;\n\n  private config: HealthManagerConfig;\n\n  private clientsMap = new Map<string, ClientConfig<any>[]>();\n\n  private livenessChecks: { check: HealthCheckHandler<any>; client: any; name: string; customHandlersPayload?: any; }[] = [];\n  private readinessChecks: { check: HealthCheckHandler<any>; client: any; name: string; customHandlersPayload?: any; }[] = [];\n\n  private logger: LoggerInstanceManager;\n\n  /**\n   * Creates a new HealthManager instance\n   *\n   * @param params Configuration options for health checks and shutdown behavior\n   */\n  constructor(params: HealthManagerOptions) {\n    this.config = {\n      ...this.DEFAULT_CONFIG,\n      ...(params.config ?? {}),\n    };\n\n    this.logger = params.logger;\n\n    this.storeClientsWithDefaults(params);\n\n    this.initializeHealthChecks();\n\n    if (this.config.shouldSetupSignalHandlers) {\n      this.setupSignalHandlers();\n    }\n  }\n\n  /**\n   * Store clients with default health check and shutdown functions\n   * @private\n   */\n  private storeClientsWithDefaults(params: HealthManagerOptions): void {\n    const clients = params.clients;\n    if (!clients) {\n      return;\n    }\n\n    for (const [type, client] of Object.entries(clients)) {\n      if (!client) {\n        continue;\n      }\n\n      if (type === CLIENT_TYPES.SERVER) {\n        this.attachServer(client as ServerConfig | ServerConfig[]);\n      } else {\n        const clientsArray = Array.isArray(client) ? client : [client];\n        const defaults = CLIENT_DEFAULTS_MAP[type as ClientType] ?? CLIENT_DEFAULTS_MAP[CLIENT_TYPES.CUSTOM];\n\n        const clientsWithDefaults = clientsArray.map(client => ({\n          connection: client.connection,\n          healthCheck: client.healthCheck ?? defaults?.healthCheck,\n          shutdownHandlers: {\n            ...defaults?.shutdownHandlers,\n            ...client.shutdownHandlers,\n          },\n          options: {\n            ...defaults?.options,\n            ...client.options,\n            name: client.options?.name || defaults?.options?.name || type,\n          },\n        }));\n\n        this.clientsMap.set(type, clientsWithDefaults);\n      }\n    }\n  }\n\n  /**\n   * Register SIGTERM and SIGINT signal listeners\n   * @private\n   */\n  private setupSignalHandlers(): void {\n    this.logger.info('Setting up global signal listeners for graceful shutdown');\n\n    process.on('SIGTERM', async () => {\n      await this.shutdown('SIGTERM');\n    });\n\n    process.on('SIGINT', async () => {\n      await this.shutdown('SIGINT');\n    });\n  }\n\n  /**\n   * Check if liveness optimization is safe to enable for a given client configuration\n   * If a client is in liveness but not readiness, we cannot skip liveness checks\n   * because the optimization relies on readiness passing, but this client isn't checked in readiness\n   * @private\n   */\n  private canSafelySkipLivenessWhenReady = (includeInLiveness: boolean | undefined, includeInReadiness: boolean | undefined): boolean => {\n    // If not in liveness, optimization is safe (skipping liveness doesn't affect this client)\n    if (!includeInLiveness) {\n      return true;\n    }\n\n    // If in liveness, must also be in readiness for liveness optimization to be safe\n    return !!includeInReadiness;\n  };\n\n  /**\n   * Initialize health check arrays based on configured clients\n   * @private\n   */\n  private initializeHealthChecks(): void {\n    let allClientsAllowLivenessOptimization = true;\n\n    this.clientsMap.forEach((clients) => {\n      clients.forEach((client) => {\n        if (!client.healthCheck) {\n          return;\n        }\n\n        const includeInLiveness = client.options?.includeInLivenessCheck;\n        const includeInReadiness = client.options?.includeInReadinessCheck;\n\n        // Check if this client configuration allows liveness optimization\n        if (!this.canSafelySkipLivenessWhenReady(includeInLiveness, includeInReadiness)) {\n          allClientsAllowLivenessOptimization = false;\n        }\n\n        if (includeInLiveness) {\n          this.livenessChecks.push({\n            check: client.healthCheck,\n            client: client.connection,\n            name: client.options?.name || 'unknown',\n            customHandlersPayload: client.options?.customHandlersPayload,\n          });\n        }\n\n        if (includeInReadiness) {\n          this.readinessChecks.push({\n            check: client.healthCheck,\n            client: client.connection,\n            name: client.options?.name || 'unknown',\n            customHandlersPayload: client.options?.customHandlersPayload,\n          });\n        }\n      });\n    });\n\n    // Apply optimization setting: only enable if user requested it AND all clients allow it\n    this.config.enableLivenessOptimization = this.config.enableLivenessOptimization && allClientsAllowLivenessOptimization;\n\n    // Log optimization status\n    if (this.config.enableLivenessOptimization) {\n      this.logger.info('Liveness check optimization enabled - liveness checks will be skipped when pod is ready');\n    } else if (!allClientsAllowLivenessOptimization) {\n      this.logger.warn('Liveness check optimization disabled - some clients are in liveness checks but not in readiness checks');\n    }\n  }\n\n  /**\n   * Attach an HTTP or HTTPS server to be managed during shutdown\n   * @param serverConfig Server configuration (ClientConfig with HTTP/HTTPS server)\n   */\n  attachServer(serverConfig: ServerConfig | ServerConfig[]): void {\n    const serverDefaults = CLIENT_DEFAULTS_MAP[CLIENT_TYPES.SERVER];\n    const serverConfigArray = Array.isArray(serverConfig) ? serverConfig : [serverConfig];\n\n    const serverClients = serverConfigArray.map((config) => {\n      // Create httpTerminator for this server instance\n      const httpTerminator = createHttpTerminator({\n        server: config.connection,\n      });\n\n      return {\n        connection: config.connection,\n        healthCheck: config.healthCheck ?? serverDefaults?.healthCheck,\n        shutdownHandlers: {\n          ...serverDefaults?.shutdownHandlers,\n          ...config.shutdownHandlers,\n        },\n        options: {\n          ...serverDefaults?.options,\n          ...config.options,\n          name: config.options?.name || serverDefaults?.options?.name || 'HTTP Server',\n          customHandlersPayload: {\n            ...config.options?.customHandlersPayload,\n            httpTerminator,\n          },\n        },\n      };\n    });\n\n    this.clientsMap.set(CLIENT_TYPES.SERVER, serverClients);\n  }\n\n  /**\n   * Performs liveness health checks for clients configured with includeInLivenessCheck\n   *\n   * @returns Promise that resolves to { status: 'ok', statusCode: 200 } on success\n   * @throws ServiceUnavailableError with statusCode 503 if any health check fails\n   */\n  async aliveCheck(): Promise<{ status: string; statusCode: number; }> {\n    if (this.isShuttingDown) {\n      return { statusCode: 200, status: 'ok' };\n    }\n\n    // Skip liveness check if optimization is enabled and the readiness check has already passed\n    if (this.config.enableLivenessOptimization && this.isReady) {\n      return { statusCode: 200, status: 'ok' };\n    }\n\n    try {\n      await Promise.all(\n        this.livenessChecks.map(({ check, client, name, customHandlersPayload }) =>\n          check({ client, name, customHandlersPayload, logger: this.logger }),\n        ),\n      );\n\n      return { statusCode: 200, status: 'ok' };\n    } catch (error: any) {\n      const err = new ServiceUnavailableError(`Liveness probe failed: ${error.message}`);\n      err.stack = error.stack;\n      throw err;\n    }\n  }\n\n  /**\n   * Performs readiness health checks for clients configured with includeInReadinessCheck\n   *\n   * @returns Promise that resolves to { status: 'ok', statusCode: 200 } if ready\n   * @throws ServiceUnavailableError with statusCode 503 if not ready or health check fails\n   */\n  async readyCheck(): Promise<{ status: string; statusCode: number; }> {\n    if (this.isShuttingDown) {\n      this.isReady = false;\n      this.logger.warn('Application is shutting down: readiness probe failed');\n      throw new ServiceUnavailableError('Readiness probe failed: Application is shutting down');\n    }\n\n    try {\n      await Promise.all(\n        this.readinessChecks.map(({ check, client, name, customHandlersPayload }) =>\n          check({ client, name, customHandlersPayload, logger: this.logger }),\n        ),\n      );\n      this.isReady = true;\n\n      return { statusCode: 200, status: 'ok' };\n    } catch (error: any) {\n      this.isReady = false;\n      const err = new ServiceUnavailableError(`Readiness probe failed: ${error.message}`);\n      err.stack = error.stack;\n      throw err;\n    }\n  }\n\n  /**\n   * Groups client shutdown handlers by stage (stopAcceptingWork, drain, close).\n   *\n   * @private\n   * @param signal - Signal name to pass to shutdown handlers\n   * @returns Record of shutdown tasks organized by stage\n   */\n  private groupShutdownTasksByStage(signal: string): Record<ShutdownStage, { fn: () => Promise<any>; name: string; }[]> {\n    const shutdownTasksByStage: Record<ShutdownStage, { fn: () => Promise<any>; name: string; }[]> = {\n      [SHUTDOWN_STAGES.STOP_ACCEPTING_WORK]: [],\n      [SHUTDOWN_STAGES.DRAIN]: [],\n      [SHUTDOWN_STAGES.CLOSE]: [],\n    };\n\n    this.clientsMap.forEach((clients) => {\n      clients.forEach((client) => {\n        const clientName = client.options?.name || 'unknown';\n        const enabledShutdownStages = client.options?.enabledShutdownStages;\n\n        if (client.shutdownHandlers) {\n          for (const [stage, shutdownHandler] of Object.entries(client.shutdownHandlers) as [ShutdownStage, ShutdownHandler<any>][]) {\n            if (shutdownHandler && enabledShutdownStages?.includes(stage)) {\n              shutdownTasksByStage[stage].push({\n                fn: () => shutdownHandler({\n                  signal,\n                  client: client.connection,\n                  name: clientName,\n                  customHandlersPayload: client.options?.customHandlersPayload,\n                  logger: this.logger,\n                }),\n                name: clientName,\n              });\n            }\n          }\n        }\n      });\n    });\n\n    return shutdownTasksByStage;\n  }\n\n  /**\n   * Execute shutdown tasks for a specific stage\n   * @private\n   * @param stage - The shutdown stage to execute\n   * @param tasks - Array of shutdown tasks to execute\n   */\n  private async executeShutdownStage(stage: ShutdownStage, tasks: { fn: () => Promise<any>; name: string; }[]): Promise<void> {\n    if (tasks.length === 0) {\n      return;\n    }\n\n    this.logger.info(`[${stage} stage] Start executing shutdown stage ${stage}...`);\n    const results = await Promise.allSettled(tasks.map(t => t.fn()));\n    results.forEach((result, index) => {\n      if (result.status === 'rejected') {\n        const { name } = tasks[index];\n        this.logger.error(`[${stage} stage] Failed to shutdown '${name}'`, { error: result.reason });\n      }\n    });\n    this.logger.info(`[${stage} stage] Shutdown stage ${stage} completed`);\n  }\n\n  /**\n   * Triggers graceful shutdown process\n   * Can be called manually or automatically via SIGTERM/SIGINT signal handlers\n   *\n   * Shutdown happens in three stages:\n   * 1. stopAcceptingWork - Stop accepting new requests/work\n   * 2. drain - Allow in-flight work to complete\n   * 3. close - Close all connections and clean up resources\n   *\n   * @param signal - Signal name (e.g., 'SIGTERM', 'SIGINT', 'MANUAL')\n   */\n  async shutdown(signal = 'MANUAL'): Promise<void> {\n    try {\n      if (this.isShuttingDown) {\n        this.logger.warn('Shutdown already in progress');\n        return;\n      }\n\n      const { shutdownDelayMs, useExit0, onShutdown, beforeShutdown } = this.config;\n\n      this.logger.info(`${signal} received: setting readiness state to false`);\n\n      this.isShuttingDown = true;\n      this.isReady = false;\n\n      await beforeShutdown?.();\n\n      this.logger.info(`Waiting ${shutdownDelayMs}ms before shutdown...`);\n      await setTimeout(shutdownDelayMs);\n\n      const shutdownTasksByStage = this.groupShutdownTasksByStage(signal);\n\n      await this.executeShutdownStage(SHUTDOWN_STAGES.STOP_ACCEPTING_WORK, shutdownTasksByStage[SHUTDOWN_STAGES.STOP_ACCEPTING_WORK]);\n\n      // Wait between stages to allow in-flight work to complete (we are missing a drain stage in our rabbitmq package)\n      // TODO once we have proper drain handlers, we can remove this fixed wait\n      this.logger.info('Waiting 4000ms to allow in-flight work to complete...');\n      await setTimeout(4_000);\n\n      await this.executeShutdownStage(SHUTDOWN_STAGES.DRAIN, shutdownTasksByStage[SHUTDOWN_STAGES.DRAIN]);\n      await this.executeShutdownStage(SHUTDOWN_STAGES.CLOSE, shutdownTasksByStage[SHUTDOWN_STAGES.CLOSE]);\n\n      await onShutdown?.();\n\n      this.logger.info('Shutdown process completed');\n\n      if (useExit0) {\n        this.logger.info(`Exiting process after ${signal}`);\n        process.exit(0);\n      }\n    } catch (error: any) {\n      this.logger.error('Error during shutdown process, forcing process exit despite shutdown error', { error, signal });\n      process.exit(1);\n    }\n  }\n}\n"],"mappings":"+RA0BA,IAAa,EAAb,KAA2B,CAyBzB,YAAY,EAA8B,qBAxBa,CACrD,gBAAiB,IACjB,SAAU,GACV,0BAA2B,GAC3B,2BAA4B,GAC7B,qBAEwB,gBACP,mBAIG,IAAI,wBAE+F,EAAE,sBACD,EAAE,sCAwFjF,EAAwC,IAE3E,EAKE,CAAC,CAAC,EAJA,GAjFT,KAAK,OAAS,CACZ,GAAG,KAAK,eACR,GAAI,EAAO,QAAU,EAAE,CACxB,CAED,KAAK,OAAS,EAAO,OAErB,KAAK,yBAAyB,EAAO,CAErC,KAAK,wBAAwB,CAEzB,KAAK,OAAO,2BACd,KAAK,qBAAqB,CAQ9B,yBAAiC,EAAoC,CACnE,IAAM,EAAU,EAAO,QAClB,KAIL,KAAK,GAAM,CAAC,EAAM,KAAW,OAAO,QAAQ,EAAQ,CAC7C,KAIL,GAAI,IAAS,EAAa,OACxB,KAAK,aAAa,EAAwC,KACrD,CACL,IAAM,EAAe,MAAM,QAAQ,EAAO,CAAG,EAAS,CAAC,EAAO,CACxD,EAAW,EAAoB,IAAuB,EAAoB,EAAa,QAEvF,EAAsB,EAAa,IAAI,IAAW,CACtD,WAAYA,EAAO,WACnB,YAAaA,EAAO,aAAe,GAAU,YAC7C,iBAAkB,CAChB,GAAG,GAAU,iBACb,GAAGA,EAAO,iBACX,CACD,QAAS,CACP,GAAG,GAAU,QACb,GAAGA,EAAO,QACV,KAAMA,EAAO,SAAS,MAAQ,GAAU,SAAS,MAAQ,EAC1D,CACF,EAAE,CAEH,KAAK,WAAW,IAAI,EAAM,EAAoB,GASpD,qBAAoC,CAClC,KAAK,OAAO,KAAK,2DAA2D,CAE5E,QAAQ,GAAG,UAAW,SAAY,CAChC,MAAM,KAAK,SAAS,UAAU,EAC9B,CAEF,QAAQ,GAAG,SAAU,SAAY,CAC/B,MAAM,KAAK,SAAS,SAAS,EAC7B,CAuBJ,wBAAuC,CACrC,IAAI,EAAsC,GAE1C,KAAK,WAAW,QAAS,GAAY,CACnC,EAAQ,QAAS,GAAW,CAC1B,GAAI,CAAC,EAAO,YACV,OAGF,IAAM,EAAoB,EAAO,SAAS,uBACpC,EAAqB,EAAO,SAAS,wBAGtC,KAAK,+BAA+B,EAAmB,EAAmB,GAC7E,EAAsC,IAGpC,GACF,KAAK,eAAe,KAAK,CACvB,MAAO,EAAO,YACd,OAAQ,EAAO,WACf,KAAM,EAAO,SAAS,MAAQ,UAC9B,sBAAuB,EAAO,SAAS,sBACxC,CAAC,CAGA,GACF,KAAK,gBAAgB,KAAK,CACxB,MAAO,EAAO,YACd,OAAQ,EAAO,WACf,KAAM,EAAO,SAAS,MAAQ,UAC9B,sBAAuB,EAAO,SAAS,sBACxC,CAAC,EAEJ,EACF,CAGF,KAAK,OAAO,2BAA6B,KAAK,OAAO,4BAA8B,EAG/E,KAAK,OAAO,2BACd,KAAK,OAAO,KAAK,0FAA0F,CACjG,GACV,KAAK,OAAO,KAAK,yGAAyG,CAQ9H,aAAa,EAAmD,CAC9D,IAAM,EAAiB,EAAoB,EAAa,QAGlD,GAFoB,MAAM,QAAQ,EAAa,CAAG,EAAe,CAAC,EAAa,EAE7C,IAAK,GAAW,CAEtD,IAAM,EAAiB,EAAqB,CAC1C,OAAQ,EAAO,WAChB,CAAC,CAEF,MAAO,CACL,WAAY,EAAO,WACnB,YAAa,EAAO,aAAe,GAAgB,YACnD,iBAAkB,CAChB,GAAG,GAAgB,iBACnB,GAAG,EAAO,iBACX,CACD,QAAS,CACP,GAAG,GAAgB,QACnB,GAAG,EAAO,QACV,KAAM,EAAO,SAAS,MAAQ,GAAgB,SAAS,MAAQ,cAC/D,sBAAuB,CACrB,GAAG,EAAO,SAAS,sBACnB,iBACD,CACF,CACF,EACD,CAEF,KAAK,WAAW,IAAI,EAAa,OAAQ,EAAc,CASzD,MAAM,YAA+D,CAMnE,GALI,KAAK,gBAKL,KAAK,OAAO,4BAA8B,KAAK,QACjD,MAAO,CAAE,WAAY,IAAK,OAAQ,KAAM,CAG1C,GAAI,CAOF,OANA,MAAM,QAAQ,IACZ,KAAK,eAAe,KAAK,CAAE,QAAO,SAAQ,OAAM,2BAC9C,EAAM,CAAE,SAAQ,OAAM,wBAAuB,OAAQ,KAAK,OAAQ,CAAC,CACpE,CACF,CAEM,CAAE,WAAY,IAAK,OAAQ,KAAM,OACjCC,EAAY,CACnB,IAAM,EAAM,IAAI,EAAwB,0BAA0B,EAAM,UAAU,CAElF,KADA,GAAI,MAAQ,EAAM,MACZ,GAUV,MAAM,YAA+D,CACnE,GAAI,KAAK,eAGP,KAFA,MAAK,QAAU,GACf,KAAK,OAAO,KAAK,uDAAuD,CAClE,IAAI,EAAwB,uDAAuD,CAG3F,GAAI,CAQF,OAPA,MAAM,QAAQ,IACZ,KAAK,gBAAgB,KAAK,CAAE,QAAO,SAAQ,OAAM,2BAC/C,EAAM,CAAE,SAAQ,OAAM,wBAAuB,OAAQ,KAAK,OAAQ,CAAC,CACpE,CACF,CACD,KAAK,QAAU,GAER,CAAE,WAAY,IAAK,OAAQ,KAAM,OACjCA,EAAY,CACnB,KAAK,QAAU,GACf,IAAM,EAAM,IAAI,EAAwB,2BAA2B,EAAM,UAAU,CAEnF,KADA,GAAI,MAAQ,EAAM,MACZ,GAWV,0BAAkC,EAAoF,CACpH,IAAMC,EAA2F,EAC9F,EAAgB,qBAAsB,EAAE,EACxC,EAAgB,OAAQ,EAAE,EAC1B,EAAgB,OAAQ,EAAE,CAC5B,CA0BD,OAxBA,KAAK,WAAW,QAAS,GAAY,CACnC,EAAQ,QAAS,GAAW,CAC1B,IAAM,EAAa,EAAO,SAAS,MAAQ,UACrC,EAAwB,EAAO,SAAS,sBAE9C,GAAI,EAAO,qBACJ,GAAM,CAAC,EAAO,KAAoB,OAAO,QAAQ,EAAO,iBAAiB,CACxE,GAAmB,GAAuB,SAAS,EAAM,EAC3D,EAAqB,GAAO,KAAK,CAC/B,OAAU,EAAgB,CACxB,SACA,OAAQ,EAAO,WACf,KAAM,EACN,sBAAuB,EAAO,SAAS,sBACvC,OAAQ,KAAK,OACd,CAAC,CACF,KAAM,EACP,CAAC,EAIR,EACF,CAEK,EAST,MAAc,qBAAqB,EAAsB,EAAmE,CACtH,EAAM,SAAW,IAIrB,KAAK,OAAO,KAAK,IAAI,EAAM,yCAAyC,EAAM,KAAK,EAC/D,MAAM,QAAQ,WAAW,EAAM,IAAI,GAAK,EAAE,IAAI,CAAC,CAAC,EACxD,SAAS,EAAQ,IAAU,CACjC,GAAI,EAAO,SAAW,WAAY,CAChC,GAAM,CAAE,QAAS,EAAM,GACvB,KAAK,OAAO,MAAM,IAAI,EAAM,8BAA8B,EAAK,GAAI,CAAE,MAAO,EAAO,OAAQ,CAAC,GAE9F,CACF,KAAK,OAAO,KAAK,IAAI,EAAM,yBAAyB,EAAM,YAAY,EAcxE,MAAM,SAAS,EAAS,SAAyB,CAC/C,GAAI,CACF,GAAI,KAAK,eAAgB,CACvB,KAAK,OAAO,KAAK,+BAA+B,CAChD,OAGF,GAAM,CAAE,kBAAiB,WAAU,aAAY,kBAAmB,KAAK,OAEvE,KAAK,OAAO,KAAK,GAAG,EAAO,6CAA6C,CAExE,KAAK,eAAiB,GACtB,KAAK,QAAU,GAEf,MAAM,KAAkB,CAExB,KAAK,OAAO,KAAK,WAAW,EAAgB,uBAAuB,CACnE,MAAM,EAAW,EAAgB,CAEjC,IAAM,EAAuB,KAAK,0BAA0B,EAAO,CAEnE,MAAM,KAAK,qBAAqB,EAAgB,oBAAqB,EAAqB,EAAgB,qBAAqB,CAI/H,KAAK,OAAO,KAAK,wDAAwD,CACzE,MAAM,EAAW,IAAM,CAEvB,MAAM,KAAK,qBAAqB,EAAgB,MAAO,EAAqB,EAAgB,OAAO,CACnG,MAAM,KAAK,qBAAqB,EAAgB,MAAO,EAAqB,EAAgB,OAAO,CAEnG,MAAM,KAAc,CAEpB,KAAK,OAAO,KAAK,6BAA6B,CAE1C,IACF,KAAK,OAAO,KAAK,yBAAyB,IAAS,CACnD,QAAQ,KAAK,EAAE,QAEVD,EAAY,CACnB,KAAK,OAAO,MAAM,6EAA8E,CAAE,QAAO,SAAQ,CAAC,CAClH,QAAQ,KAAK,EAAE"}

package/lib/health-manager/types.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"types.cjs","names":[],"sources":["../../src/health-manager/types.ts"],"sourcesContent":["import type { LoggerInstanceManager } from '@autofleet/logger';\nimport type { Server as HttpServer } from 'node:http';\nimport type { Server as HttpsServer } from 'node:https';\n\nexport type HealthCheckHandler<T> = (options: {\n  client: T;\n  name: string;\n  customHandlersPayload?: any;\n  logger: LoggerInstanceManager;\n}) => Promise<void>;\n\nexport type ShutdownHandler<T> = (options: {\n  client: T;\n  name: string;\n  signal?: string;\n  customHandlersPayload?: any;\n  logger: LoggerInstanceManager;\n}) => Promise<void>;\n\nexport type ShutdownHandlersByStage<T> = Partial<Record<ShutdownStage, ShutdownHandler<T>>>;\n\nexport const SHUTDOWN_STAGES = {\n  STOP_ACCEPTING_WORK: 'stopAcceptingWork',\n  DRAIN: 'drain',\n  CLOSE: 'close',\n} as const;\n\nexport type ShutdownStage = typeof SHUTDOWN_STAGES[keyof typeof SHUTDOWN_STAGES];\n\n/**\n * Configuration for a single client (e.g., rabbit, sequelize, redis, elasticsearch)\n * Allows custom health checks and shutdown logic\n */\nexport interface ClientConfig<T = any, TPayload = any> {\n  /** The actual connection instance */\n  connection: T;\n\n  /** Custom health check function (optional) - will be called during alive/ready checks with client connection and name */\n  healthCheck?: HealthCheckHandler<T>;\n\n  /**\n   * Custom shutdown handlers by stage (optional).\n   * Handlers are called during graceful shutdown based on the stages configured in options.enabledShutdownStages.\n   * Available stages: stopAcceptingWork (stop new requests), drain (complete in-flight work), close (cleanup resources)\n   */\n  shutdownHandlers?: ShutdownHandlersByStage<T>;\n  /**\n   * Additional options specific to the client (optional).\n   * If not provided, default values will be used based on the client type.\n   */\n  options?: {\n    /** Whether to include this client in liveness checks (default: true) */\n    includeInLivenessCheck?: boolean;\n\n    /** Whether to include this client in readiness checks (default: true) */\n    includeInReadinessCheck?: boolean;\n\n    /** Custom name for this client (used in logging) */\n    name?: string;\n\n    /** Custom payload to pass to health check and shutdown handlers - use this to pass any additional data your handlers need */\n    customHandlersPayload?: TPayload;\n\n    /** Which shutdown stages are enabled for this client */\n    enabledShutdownStages?: ShutdownStage[];\n  };\n}\n\n/** Type alias for server configuration - uses ClientConfig with server instance */\nexport type ServerConfig = ClientConfig<HttpServer | HttpsServer>;\n\n/**\n * General configuration options for the HealthManager\n */\nexport interface HealthManagerConfig {\n  /**\n   * Delay in milliseconds before shutting down after SIGTERM/SIGINT\n   * This gives time for load balancers to stop routing traffic. Should be equal to or greater than readiness probe interval.\n   * @default 6000\n   */\n  shutdownDelayMs?: number;\n\n  /**\n   * Whether to call process.exit(0) after shutdown sequence completes\n   * @default true\n   */\n  useExit0?: boolean;\n\n  /** Whether to auto-setup SIGTERM/SIGINT handlers for graceful shutdown\n   * @default true\n   */\n  shouldSetupSignalHandlers?: boolean;\n\n  /**\n   * Global shutdown hook called before all clients are shut down\n   * @default undefined\n   */\n  beforeShutdown?: (() => Promise<void>) | undefined;\n\n  /**\n   * Global shutdown hook called after all clients are shut down\n   * @default undefined\n   */\n  onShutdown?: (() => Promise<void>) | undefined;\n\n  /**\n   * Hook called once when the readiness check passes for the first time, before the app is marked as ready.\n   * Use this to set up work that should only start once all dependencies are healthy (e.g. starting RabbitMQ consumers).\n   * @default undefined\n   */\n  onReady?: (() => Promise<void> | void) | undefined;\n\n  /**\n   * Enable liveness check optimization - if true and all clients are in both liveness and readiness checks,\n   * liveness checks will be skipped when pod is already ready (readiness check passed)\n   * @default true\n   */\n  enableLivenessOptimization?: boolean;\n}\n\n/**\n * List of clients and general configuration for HealthManager\n */\nexport interface HealthManagerOptions {\n  /** Client configurations */\n  clients?: {\n    /** RabbitMQ client configuration - can be a single config or array of configs */\n    rabbit?: ClientConfig<RabbitMQClient> | ClientConfig<RabbitMQClient>[];\n\n    /** Sequelize instance configuration - can be a single config or array of configs */\n    sequelize?: ClientConfig<SequelizeClient> | ClientConfig<SequelizeClient>[];\n\n    /** Redis client configuration - can be a single config or array of configs */\n    redis?: ClientConfig<RedisClient> | ClientConfig<RedisClient>[];\n\n    /** Elasticsearch client configuration - can be a single config or array of configs */\n    elasticsearch?: ClientConfig<ElasticsearchClient> | ClientConfig<ElasticsearchClient>[];\n\n    /** Events client configuration - can be a single config or array of configs */\n    events?: ClientConfig<EventsClient> | ClientConfig<EventsClient>[];\n\n    /** HTTP/HTTPS server configuration - can be a single config or array of configs */\n    server?: ServerConfig | ServerConfig[];\n\n    /** Sadot initialization state configuration - can be a single config or array of configs */\n    sadot?: ClientConfig<SadotClient> | ClientConfig<SadotClient>[];\n\n    /** Custom client configurations - for any client type not explicitly listed above */\n    [key: string]: ClientConfig<any> | ClientConfig<any>[] | undefined;\n  };\n\n  /** General configuration options */\n  config?: HealthManagerConfig;\n\n  /** Logger instance for logging messages */\n  logger: LoggerInstanceManager;\n}\n\nexport interface RedisClient {\n  ping?: () => Promise<string | void>;\n  quit?: () => Promise<unknown>;\n}\n\n/**\n * Sequelize database interface\n */\nexport interface SequelizeClient {\n  query?(sql: string | { query: string; values: unknown[]; }): Promise<any>;\n  close?: () => Promise<void>;\n}\n\n/**\n * Elasticsearch client interface\n */\nexport interface ElasticsearchClient {\n  ping?: () => Promise<boolean>;\n  close?: () => Promise<void>;\n}\n\nexport interface EventsClient {\n  exitHandler?: (signal: string) => Promise<void>;\n}\n\n/**\n * RabbitMQ client interface\n */\nexport interface RabbitMQClient {\n  isConnected?: () => Promise<boolean> | boolean;\n  gracefulShutdown?: (signal: string) => Promise<void>;\n}\n\n/** Sadot initialization state and database connection client interface */\nexport interface SadotClient {\n  waitForInitialization?: () => Promise<SequelizeClient | undefined>;\n  isReady?: boolean;\n  error?: Error | null;\n  sequelize?: SequelizeClient | null;\n}\n\nexport const CLIENT_TYPES = {\n  RABBIT: 'rabbit',\n  SEQUELIZE: 'sequelize',\n  REDIS: 'redis',\n  ELASTIC_SEARCH: 'elasticsearch',\n  EVENTS: 'events',\n  SERVER: 'server',\n  SADOT: 'sadot',\n  CUSTOM: 'custom',\n} as const;\n\nexport type ClientType = typeof CLIENT_TYPES[keyof typeof CLIENT_TYPES];\n"],"mappings":"AAqBA,MAAa,EAAkB,CAC7B,oBAAqB,oBACrB,MAAO,QACP,MAAO,QACR,CA8KY,EAAe,CAC1B,OAAQ,SACR,UAAW,YACX,MAAO,QACP,eAAgB,gBAChB,OAAQ,SACR,OAAQ,SACR,MAAO,QACP,OAAQ,SACT"}
+{"version":3,"file":"types.cjs","names":[],"sources":["../../src/health-manager/types.ts"],"sourcesContent":["import type { LoggerInstanceManager } from '@autofleet/logger';\nimport type { Server as HttpServer } from 'node:http';\nimport type { Server as HttpsServer } from 'node:https';\n\nexport type HealthCheckHandler<T> = (options: {\n  client: T;\n  name: string;\n  customHandlersPayload?: any;\n  logger: LoggerInstanceManager;\n}) => Promise<void>;\n\nexport type ShutdownHandler<T> = (options: {\n  client: T;\n  name: string;\n  signal?: string;\n  customHandlersPayload?: any;\n  logger: LoggerInstanceManager;\n}) => Promise<void>;\n\nexport type ShutdownHandlersByStage<T> = Partial<Record<ShutdownStage, ShutdownHandler<T>>>;\n\nexport const SHUTDOWN_STAGES = {\n  STOP_ACCEPTING_WORK: 'stopAcceptingWork',\n  DRAIN: 'drain',\n  CLOSE: 'close',\n} as const;\n\nexport type ShutdownStage = typeof SHUTDOWN_STAGES[keyof typeof SHUTDOWN_STAGES];\n\n/**\n * Configuration for a single client (e.g., rabbit, sequelize, redis, elasticsearch)\n * Allows custom health checks and shutdown logic\n */\nexport interface ClientConfig<T = any, TPayload = any> {\n  /** The actual connection instance */\n  connection: T;\n\n  /** Custom health check function (optional) - will be called during alive/ready checks with client connection and name */\n  healthCheck?: HealthCheckHandler<T>;\n\n  /**\n   * Custom shutdown handlers by stage (optional).\n   * Handlers are called during graceful shutdown based on the stages configured in options.enabledShutdownStages.\n   * Available stages: stopAcceptingWork (stop new requests), drain (complete in-flight work), close (cleanup resources)\n   */\n  shutdownHandlers?: ShutdownHandlersByStage<T>;\n  /**\n   * Additional options specific to the client (optional).\n   * If not provided, default values will be used based on the client type.\n   */\n  options?: {\n    /** Whether to include this client in liveness checks (default: true) */\n    includeInLivenessCheck?: boolean;\n\n    /** Whether to include this client in readiness checks (default: true) */\n    includeInReadinessCheck?: boolean;\n\n    /** Custom name for this client (used in logging) */\n    name?: string;\n\n    /** Custom payload to pass to health check and shutdown handlers - use this to pass any additional data your handlers need */\n    customHandlersPayload?: TPayload;\n\n    /** Which shutdown stages are enabled for this client */\n    enabledShutdownStages?: ShutdownStage[];\n  };\n}\n\n/** Type alias for server configuration - uses ClientConfig with server instance */\nexport type ServerConfig = ClientConfig<HttpServer | HttpsServer>;\n\n/**\n * General configuration options for the HealthManager\n */\nexport interface HealthManagerConfig {\n  /**\n   * Delay in milliseconds before shutting down after SIGTERM/SIGINT\n   * This gives time for load balancers to stop routing traffic. Should be equal to or greater than readiness probe interval.\n   * @default 6000\n   */\n  shutdownDelayMs?: number;\n\n  /**\n   * Whether to call process.exit(0) after shutdown sequence completes\n   * @default true\n   */\n  useExit0?: boolean;\n\n  /** Whether to auto-setup SIGTERM/SIGINT handlers for graceful shutdown\n   * @default true\n   */\n  shouldSetupSignalHandlers?: boolean;\n\n  /**\n   * Global shutdown hook called before all clients are shut down\n   * @default undefined\n   */\n  beforeShutdown?: (() => Promise<void>) | undefined;\n\n  /**\n   * Global shutdown hook called after all clients are shut down\n   * @default undefined\n   */\n  onShutdown?: (() => Promise<void>) | undefined;\n\n  /**\n   * Enable liveness check optimization - if true and all clients are in both liveness and readiness checks,\n   * liveness checks will be skipped when pod is already ready (readiness check passed)\n   * @default true\n   */\n  enableLivenessOptimization?: boolean;\n}\n\n/**\n * List of clients and general configuration for HealthManager\n */\nexport interface HealthManagerOptions {\n  /** Client configurations */\n  clients?: {\n    /** RabbitMQ client configuration - can be a single config or array of configs */\n    rabbit?: ClientConfig<RabbitMQClient> | ClientConfig<RabbitMQClient>[];\n\n    /** Sequelize instance configuration - can be a single config or array of configs */\n    sequelize?: ClientConfig<SequelizeClient> | ClientConfig<SequelizeClient>[];\n\n    /** Redis client configuration - can be a single config or array of configs */\n    redis?: ClientConfig<RedisClient> | ClientConfig<RedisClient>[];\n\n    /** Elasticsearch client configuration - can be a single config or array of configs */\n    elasticsearch?: ClientConfig<ElasticsearchClient> | ClientConfig<ElasticsearchClient>[];\n\n    /** Events client configuration - can be a single config or array of configs */\n    events?: ClientConfig<EventsClient> | ClientConfig<EventsClient>[];\n\n    /** HTTP/HTTPS server configuration - can be a single config or array of configs */\n    server?: ServerConfig | ServerConfig[];\n\n    /** Sadot initialization state configuration - can be a single config or array of configs */\n    sadot?: ClientConfig<SadotClient> | ClientConfig<SadotClient>[];\n\n    /** Custom client configurations - for any client type not explicitly listed above */\n    [key: string]: ClientConfig<any> | ClientConfig<any>[] | undefined;\n  };\n\n  /** General configuration options */\n  config?: HealthManagerConfig;\n\n  /** Logger instance for logging messages */\n  logger: LoggerInstanceManager;\n}\n\nexport interface RedisClient {\n  ping?: () => Promise<string | void>;\n  quit?: () => Promise<unknown>;\n}\n\n/**\n * Sequelize database interface\n */\nexport interface SequelizeClient {\n  query?(sql: string | { query: string; values: unknown[]; }): Promise<any>;\n  close?: () => Promise<void>;\n}\n\n/**\n * Elasticsearch client interface\n */\nexport interface ElasticsearchClient {\n  ping?: () => Promise<boolean>;\n  close?: () => Promise<void>;\n}\n\nexport interface EventsClient {\n  exitHandler?: (signal: string) => Promise<void>;\n}\n\n/**\n * RabbitMQ client interface\n */\nexport interface RabbitMQClient {\n  isConnected?: () => Promise<boolean> | boolean;\n  gracefulShutdown?: (signal: string) => Promise<void>;\n}\n\n/** Sadot initialization state and database connection client interface */\nexport interface SadotClient {\n  waitForInitialization?: () => Promise<SequelizeClient | undefined>;\n  isReady?: boolean;\n  error?: Error | null;\n  sequelize?: SequelizeClient | null;\n}\n\nexport const CLIENT_TYPES = {\n  RABBIT: 'rabbit',\n  SEQUELIZE: 'sequelize',\n  REDIS: 'redis',\n  ELASTIC_SEARCH: 'elasticsearch',\n  EVENTS: 'events',\n  SERVER: 'server',\n  SADOT: 'sadot',\n  CUSTOM: 'custom',\n} as const;\n\nexport type ClientType = typeof CLIENT_TYPES[keyof typeof CLIENT_TYPES];\n"],"mappings":"AAqBA,MAAa,EAAkB,CAC7B,oBAAqB,oBACrB,MAAO,QACP,MAAO,QACR,CAuKY,EAAe,CAC1B,OAAQ,SACR,UAAW,YACX,MAAO,QACP,eAAgB,gBAChB,OAAQ,SACR,OAAQ,SACR,MAAO,QACP,OAAQ,SACT"}

package/lib/health-manager/types.d.cts

@@ -87,12 +87,6 @@ interface HealthManagerConfig {
   */
   onShutdown?: (() => Promise<void>) | undefined;
   /**
-  * Hook called once when the readiness check passes for the first time, before the app is marked as ready.
-  * Use this to set up work that should only start once all dependencies are healthy (e.g. starting RabbitMQ consumers).
-  * @default undefined
-  */
-  onReady?: (() => Promise<void> | void) | undefined;
-  /**
   * Enable liveness check optimization - if true and all clients are in both liveness and readiness checks,
   * liveness checks will be skipped when pod is already ready (readiness check passed)
   * @default true

package/lib/health-manager/types.d.ts

@@ -87,12 +87,6 @@ interface HealthManagerConfig {
   */
   onShutdown?: (() => Promise<void>) | undefined;
   /**
-  * Hook called once when the readiness check passes for the first time, before the app is marked as ready.
-  * Use this to set up work that should only start once all dependencies are healthy (e.g. starting RabbitMQ consumers).
-  * @default undefined
-  */
-  onReady?: (() => Promise<void> | void) | undefined;
-  /**
   * Enable liveness check optimization - if true and all clients are in both liveness and readiness checks,
   * liveness checks will be skipped when pod is already ready (readiness check passed)
   * @default true

package/lib/health-manager/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","names":[],"sources":["../../src/health-manager/types.ts"],"sourcesContent":["import type { LoggerInstanceManager } from '@autofleet/logger';\nimport type { Server as HttpServer } from 'node:http';\nimport type { Server as HttpsServer } from 'node:https';\n\nexport type HealthCheckHandler<T> = (options: {\n  client: T;\n  name: string;\n  customHandlersPayload?: any;\n  logger: LoggerInstanceManager;\n}) => Promise<void>;\n\nexport type ShutdownHandler<T> = (options: {\n  client: T;\n  name: string;\n  signal?: string;\n  customHandlersPayload?: any;\n  logger: LoggerInstanceManager;\n}) => Promise<void>;\n\nexport type ShutdownHandlersByStage<T> = Partial<Record<ShutdownStage, ShutdownHandler<T>>>;\n\nexport const SHUTDOWN_STAGES = {\n  STOP_ACCEPTING_WORK: 'stopAcceptingWork',\n  DRAIN: 'drain',\n  CLOSE: 'close',\n} as const;\n\nexport type ShutdownStage = typeof SHUTDOWN_STAGES[keyof typeof SHUTDOWN_STAGES];\n\n/**\n * Configuration for a single client (e.g., rabbit, sequelize, redis, elasticsearch)\n * Allows custom health checks and shutdown logic\n */\nexport interface ClientConfig<T = any, TPayload = any> {\n  /** The actual connection instance */\n  connection: T;\n\n  /** Custom health check function (optional) - will be called during alive/ready checks with client connection and name */\n  healthCheck?: HealthCheckHandler<T>;\n\n  /**\n   * Custom shutdown handlers by stage (optional).\n   * Handlers are called during graceful shutdown based on the stages configured in options.enabledShutdownStages.\n   * Available stages: stopAcceptingWork (stop new requests), drain (complete in-flight work), close (cleanup resources)\n   */\n  shutdownHandlers?: ShutdownHandlersByStage<T>;\n  /**\n   * Additional options specific to the client (optional).\n   * If not provided, default values will be used based on the client type.\n   */\n  options?: {\n    /** Whether to include this client in liveness checks (default: true) */\n    includeInLivenessCheck?: boolean;\n\n    /** Whether to include this client in readiness checks (default: true) */\n    includeInReadinessCheck?: boolean;\n\n    /** Custom name for this client (used in logging) */\n    name?: string;\n\n    /** Custom payload to pass to health check and shutdown handlers - use this to pass any additional data your handlers need */\n    customHandlersPayload?: TPayload;\n\n    /** Which shutdown stages are enabled for this client */\n    enabledShutdownStages?: ShutdownStage[];\n  };\n}\n\n/** Type alias for server configuration - uses ClientConfig with server instance */\nexport type ServerConfig = ClientConfig<HttpServer | HttpsServer>;\n\n/**\n * General configuration options for the HealthManager\n */\nexport interface HealthManagerConfig {\n  /**\n   * Delay in milliseconds before shutting down after SIGTERM/SIGINT\n   * This gives time for load balancers to stop routing traffic. Should be equal to or greater than readiness probe interval.\n   * @default 6000\n   */\n  shutdownDelayMs?: number;\n\n  /**\n   * Whether to call process.exit(0) after shutdown sequence completes\n   * @default true\n   */\n  useExit0?: boolean;\n\n  /** Whether to auto-setup SIGTERM/SIGINT handlers for graceful shutdown\n   * @default true\n   */\n  shouldSetupSignalHandlers?: boolean;\n\n  /**\n   * Global shutdown hook called before all clients are shut down\n   * @default undefined\n   */\n  beforeShutdown?: (() => Promise<void>) | undefined;\n\n  /**\n   * Global shutdown hook called after all clients are shut down\n   * @default undefined\n   */\n  onShutdown?: (() => Promise<void>) | undefined;\n\n  /**\n   * Hook called once when the readiness check passes for the first time, before the app is marked as ready.\n   * Use this to set up work that should only start once all dependencies are healthy (e.g. starting RabbitMQ consumers).\n   * @default undefined\n   */\n  onReady?: (() => Promise<void> | void) | undefined;\n\n  /**\n   * Enable liveness check optimization - if true and all clients are in both liveness and readiness checks,\n   * liveness checks will be skipped when pod is already ready (readiness check passed)\n   * @default true\n   */\n  enableLivenessOptimization?: boolean;\n}\n\n/**\n * List of clients and general configuration for HealthManager\n */\nexport interface HealthManagerOptions {\n  /** Client configurations */\n  clients?: {\n    /** RabbitMQ client configuration - can be a single config or array of configs */\n    rabbit?: ClientConfig<RabbitMQClient> | ClientConfig<RabbitMQClient>[];\n\n    /** Sequelize instance configuration - can be a single config or array of configs */\n    sequelize?: ClientConfig<SequelizeClient> | ClientConfig<SequelizeClient>[];\n\n    /** Redis client configuration - can be a single config or array of configs */\n    redis?: ClientConfig<RedisClient> | ClientConfig<RedisClient>[];\n\n    /** Elasticsearch client configuration - can be a single config or array of configs */\n    elasticsearch?: ClientConfig<ElasticsearchClient> | ClientConfig<ElasticsearchClient>[];\n\n    /** Events client configuration - can be a single config or array of configs */\n    events?: ClientConfig<EventsClient> | ClientConfig<EventsClient>[];\n\n    /** HTTP/HTTPS server configuration - can be a single config or array of configs */\n    server?: ServerConfig | ServerConfig[];\n\n    /** Sadot initialization state configuration - can be a single config or array of configs */\n    sadot?: ClientConfig<SadotClient> | ClientConfig<SadotClient>[];\n\n    /** Custom client configurations - for any client type not explicitly listed above */\n    [key: string]: ClientConfig<any> | ClientConfig<any>[] | undefined;\n  };\n\n  /** General configuration options */\n  config?: HealthManagerConfig;\n\n  /** Logger instance for logging messages */\n  logger: LoggerInstanceManager;\n}\n\nexport interface RedisClient {\n  ping?: () => Promise<string | void>;\n  quit?: () => Promise<unknown>;\n}\n\n/**\n * Sequelize database interface\n */\nexport interface SequelizeClient {\n  query?(sql: string | { query: string; values: unknown[]; }): Promise<any>;\n  close?: () => Promise<void>;\n}\n\n/**\n * Elasticsearch client interface\n */\nexport interface ElasticsearchClient {\n  ping?: () => Promise<boolean>;\n  close?: () => Promise<void>;\n}\n\nexport interface EventsClient {\n  exitHandler?: (signal: string) => Promise<void>;\n}\n\n/**\n * RabbitMQ client interface\n */\nexport interface RabbitMQClient {\n  isConnected?: () => Promise<boolean> | boolean;\n  gracefulShutdown?: (signal: string) => Promise<void>;\n}\n\n/** Sadot initialization state and database connection client interface */\nexport interface SadotClient {\n  waitForInitialization?: () => Promise<SequelizeClient | undefined>;\n  isReady?: boolean;\n  error?: Error | null;\n  sequelize?: SequelizeClient | null;\n}\n\nexport const CLIENT_TYPES = {\n  RABBIT: 'rabbit',\n  SEQUELIZE: 'sequelize',\n  REDIS: 'redis',\n  ELASTIC_SEARCH: 'elasticsearch',\n  EVENTS: 'events',\n  SERVER: 'server',\n  SADOT: 'sadot',\n  CUSTOM: 'custom',\n} as const;\n\nexport type ClientType = typeof CLIENT_TYPES[keyof typeof CLIENT_TYPES];\n"],"mappings":"AAqBA,MAAa,EAAkB,CAC7B,oBAAqB,oBACrB,MAAO,QACP,MAAO,QACR,CA8KY,EAAe,CAC1B,OAAQ,SACR,UAAW,YACX,MAAO,QACP,eAAgB,gBAChB,OAAQ,SACR,OAAQ,SACR,MAAO,QACP,OAAQ,SACT"}
+{"version":3,"file":"types.js","names":[],"sources":["../../src/health-manager/types.ts"],"sourcesContent":["import type { LoggerInstanceManager } from '@autofleet/logger';\nimport type { Server as HttpServer } from 'node:http';\nimport type { Server as HttpsServer } from 'node:https';\n\nexport type HealthCheckHandler<T> = (options: {\n  client: T;\n  name: string;\n  customHandlersPayload?: any;\n  logger: LoggerInstanceManager;\n}) => Promise<void>;\n\nexport type ShutdownHandler<T> = (options: {\n  client: T;\n  name: string;\n  signal?: string;\n  customHandlersPayload?: any;\n  logger: LoggerInstanceManager;\n}) => Promise<void>;\n\nexport type ShutdownHandlersByStage<T> = Partial<Record<ShutdownStage, ShutdownHandler<T>>>;\n\nexport const SHUTDOWN_STAGES = {\n  STOP_ACCEPTING_WORK: 'stopAcceptingWork',\n  DRAIN: 'drain',\n  CLOSE: 'close',\n} as const;\n\nexport type ShutdownStage = typeof SHUTDOWN_STAGES[keyof typeof SHUTDOWN_STAGES];\n\n/**\n * Configuration for a single client (e.g., rabbit, sequelize, redis, elasticsearch)\n * Allows custom health checks and shutdown logic\n */\nexport interface ClientConfig<T = any, TPayload = any> {\n  /** The actual connection instance */\n  connection: T;\n\n  /** Custom health check function (optional) - will be called during alive/ready checks with client connection and name */\n  healthCheck?: HealthCheckHandler<T>;\n\n  /**\n   * Custom shutdown handlers by stage (optional).\n   * Handlers are called during graceful shutdown based on the stages configured in options.enabledShutdownStages.\n   * Available stages: stopAcceptingWork (stop new requests), drain (complete in-flight work), close (cleanup resources)\n   */\n  shutdownHandlers?: ShutdownHandlersByStage<T>;\n  /**\n   * Additional options specific to the client (optional).\n   * If not provided, default values will be used based on the client type.\n   */\n  options?: {\n    /** Whether to include this client in liveness checks (default: true) */\n    includeInLivenessCheck?: boolean;\n\n    /** Whether to include this client in readiness checks (default: true) */\n    includeInReadinessCheck?: boolean;\n\n    /** Custom name for this client (used in logging) */\n    name?: string;\n\n    /** Custom payload to pass to health check and shutdown handlers - use this to pass any additional data your handlers need */\n    customHandlersPayload?: TPayload;\n\n    /** Which shutdown stages are enabled for this client */\n    enabledShutdownStages?: ShutdownStage[];\n  };\n}\n\n/** Type alias for server configuration - uses ClientConfig with server instance */\nexport type ServerConfig = ClientConfig<HttpServer | HttpsServer>;\n\n/**\n * General configuration options for the HealthManager\n */\nexport interface HealthManagerConfig {\n  /**\n   * Delay in milliseconds before shutting down after SIGTERM/SIGINT\n   * This gives time for load balancers to stop routing traffic. Should be equal to or greater than readiness probe interval.\n   * @default 6000\n   */\n  shutdownDelayMs?: number;\n\n  /**\n   * Whether to call process.exit(0) after shutdown sequence completes\n   * @default true\n   */\n  useExit0?: boolean;\n\n  /** Whether to auto-setup SIGTERM/SIGINT handlers for graceful shutdown\n   * @default true\n   */\n  shouldSetupSignalHandlers?: boolean;\n\n  /**\n   * Global shutdown hook called before all clients are shut down\n   * @default undefined\n   */\n  beforeShutdown?: (() => Promise<void>) | undefined;\n\n  /**\n   * Global shutdown hook called after all clients are shut down\n   * @default undefined\n   */\n  onShutdown?: (() => Promise<void>) | undefined;\n\n  /**\n   * Enable liveness check optimization - if true and all clients are in both liveness and readiness checks,\n   * liveness checks will be skipped when pod is already ready (readiness check passed)\n   * @default true\n   */\n  enableLivenessOptimization?: boolean;\n}\n\n/**\n * List of clients and general configuration for HealthManager\n */\nexport interface HealthManagerOptions {\n  /** Client configurations */\n  clients?: {\n    /** RabbitMQ client configuration - can be a single config or array of configs */\n    rabbit?: ClientConfig<RabbitMQClient> | ClientConfig<RabbitMQClient>[];\n\n    /** Sequelize instance configuration - can be a single config or array of configs */\n    sequelize?: ClientConfig<SequelizeClient> | ClientConfig<SequelizeClient>[];\n\n    /** Redis client configuration - can be a single config or array of configs */\n    redis?: ClientConfig<RedisClient> | ClientConfig<RedisClient>[];\n\n    /** Elasticsearch client configuration - can be a single config or array of configs */\n    elasticsearch?: ClientConfig<ElasticsearchClient> | ClientConfig<ElasticsearchClient>[];\n\n    /** Events client configuration - can be a single config or array of configs */\n    events?: ClientConfig<EventsClient> | ClientConfig<EventsClient>[];\n\n    /** HTTP/HTTPS server configuration - can be a single config or array of configs */\n    server?: ServerConfig | ServerConfig[];\n\n    /** Sadot initialization state configuration - can be a single config or array of configs */\n    sadot?: ClientConfig<SadotClient> | ClientConfig<SadotClient>[];\n\n    /** Custom client configurations - for any client type not explicitly listed above */\n    [key: string]: ClientConfig<any> | ClientConfig<any>[] | undefined;\n  };\n\n  /** General configuration options */\n  config?: HealthManagerConfig;\n\n  /** Logger instance for logging messages */\n  logger: LoggerInstanceManager;\n}\n\nexport interface RedisClient {\n  ping?: () => Promise<string | void>;\n  quit?: () => Promise<unknown>;\n}\n\n/**\n * Sequelize database interface\n */\nexport interface SequelizeClient {\n  query?(sql: string | { query: string; values: unknown[]; }): Promise<any>;\n  close?: () => Promise<void>;\n}\n\n/**\n * Elasticsearch client interface\n */\nexport interface ElasticsearchClient {\n  ping?: () => Promise<boolean>;\n  close?: () => Promise<void>;\n}\n\nexport interface EventsClient {\n  exitHandler?: (signal: string) => Promise<void>;\n}\n\n/**\n * RabbitMQ client interface\n */\nexport interface RabbitMQClient {\n  isConnected?: () => Promise<boolean> | boolean;\n  gracefulShutdown?: (signal: string) => Promise<void>;\n}\n\n/** Sadot initialization state and database connection client interface */\nexport interface SadotClient {\n  waitForInitialization?: () => Promise<SequelizeClient | undefined>;\n  isReady?: boolean;\n  error?: Error | null;\n  sequelize?: SequelizeClient | null;\n}\n\nexport const CLIENT_TYPES = {\n  RABBIT: 'rabbit',\n  SEQUELIZE: 'sequelize',\n  REDIS: 'redis',\n  ELASTIC_SEARCH: 'elasticsearch',\n  EVENTS: 'events',\n  SERVER: 'server',\n  SADOT: 'sadot',\n  CUSTOM: 'custom',\n} as const;\n\nexport type ClientType = typeof CLIENT_TYPES[keyof typeof CLIENT_TYPES];\n"],"mappings":"AAqBA,MAAa,EAAkB,CAC7B,oBAAqB,oBACrB,MAAO,QACP,MAAO,QACR,CAuKY,EAAe,CAC1B,OAAQ,SACR,UAAW,YACX,MAAO,QACP,eAAgB,gBAChB,OAAQ,SACR,OAAQ,SACR,MAAO,QACP,OAAQ,SACT"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@autofleet/nitur",
-  "version": "2.4.1-beta.0",
+  "version": "2.4.1",
   "description": "A package for service monitoring",
   "type": "module",
   "main": "./lib/index.js",
@@ -47,8 +47,8 @@
   ],
   "dependencies": {
     "http-terminator": "3.2.0",
-    "@autofleet/logger": "^4.2.46",
-    "@autofleet/errors": "^3.1.42"
+    "@autofleet/errors": "^3.1.42",
+    "@autofleet/logger": "^4.2.46"
   },
   "engines": {
     "node": "^18 || ^20 || >=22.0.0"