@catbee/utils 2.1.1 → 2.2.1

package/server/index.mjs

@@ -26,17 +26,18 @@ import express from 'express';
 import http from 'http';
 import https from 'https';
 import { HttpStatusCodes } from '@catbee/utils/http-status-codes';
-import { createFinalErrorResponse, SuccessResponse } from '@catbee/utils/response';
+import { createFinalErrorResponse } from '@catbee/utils/response';
 import { requestId, setupRequestContext, timeout, responseTime, errorHandler } from '@catbee/utils/middleware';
 import { Env } from '@catbee/utils/env';
 import { getLogger } from '@catbee/utils/logger';
-import { ServiceUnavailableException, InternalServerErrorException, NotFoundException } from '@catbee/utils/exception';
+import { ServiceUnavailableException, NotFoundException } from '@catbee/utils/exception';
 import { getCatbeeServerGlobalConfig } from '@catbee/utils/config';
-import { deepObjMerge, isPlainObject, deepClone } from '@catbee/utils/object';
+import { isPlainObject, deepClone, deepObjMerge } from '@catbee/utils/object';
 import { fileExists, readFile, readFileSync } from '@catbee/utils/fs';
 import { isPort, isHostname } from '@catbee/utils/validation';
 import { optionalRequire } from '@catbee/utils/async';
 import { uuid } from '@catbee/utils/id';
+import { HealthzServer } from '@catbee/utils/healthz-server';
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -299,36 +300,53 @@ var ServerConfigBuilder = class {
     return this.setEnabled("requestLogging", false);
   }
   /**
-  * Configures server health check endpoint.
+  * Configures the dedicated Healthz probe HTTP server for Kubernetes.
   *
-  * @param opts - Health check configuration options
+  * @param opts - Healthz server configuration options or boolean toggle
   * @returns The builder instance for chaining
-  * @default - { path: '/healthz', detailed: true, withGlobalPrefix: false }
   *
   * @example
   * ```typescript
-  * builder.withHealthCheck({
-  *   path: '/health',
-  *   detailed: true
+  * builder.withHealthzServer({
+  *   port: 8282,
+  *   shutdownDelayMs: 5000,
+  *   readinessChecks: [
+  *     { name: 'db', check: () => checkDb() }
+  *   ]
   * })
   * ```
   */
-  withHealthCheck(opts) {
-    this.mergeConfig("healthCheck", opts);
+  withHealthzServer(opts) {
+    if (typeof opts === "boolean") {
+      this.config.healthzServer = opts;
+    } else {
+      const current = isPlainObject(this.config.healthzServer) ? deepClone(this.config.healthzServer) : {};
+      this.config.healthzServer = deepObjMerge({}, current, {
+        enable: true,
+        ...opts
+      });
+    }
     return this;
   }
   /**
-  * Enables health check endpoint with default or custom settings
-  * @param opts - Optional health check configuration
+  * Enables the dedicated Healthz probe HTTP server.
+  *
+  * @param opts - Optional Healthz server configuration options
   * @returns The builder instance for chaining
+  */
+  enableHealthzServer(opts = {}) {
+    return this.withHealthzServer({
+      ...opts,
+      enable: true
+    });
+  }
+  /**
+  * Disables the dedicated Healthz probe HTTP server.
   *
-  * @example
-  * ```typescript
-  * builder.disableHealthCheck()
-  * ```
+  * @returns The builder instance for chaining
   */
-  disableHealthCheck() {
-    return this.setEnabled("healthCheck", false);
+  disableHealthzServer() {
+    return this.withHealthzServer(false);
   }
   /**
   * Configures OpenAPI/Swagger documentation for the API.
@@ -718,11 +736,12 @@ var ExpressServer = class {
   gracefulShutdownRegistered = false;
   /** Map of registered signal listeners for clean teardown */
   signalListeners = /* @__PURE__ */ new Map();
-  /**
-  * Collection of registered health check functions.
-  * These are executed when the health check endpoint is accessed.
-  */
-  healthChecks = [];
+  /** Running address info for the Healthz probe server */
+  healthzAddress;
+  /** Named checks queued for Healthz liveness probe */
+  healthzChecks = [];
+  /** Named checks queued for Healthz readiness probe */
+  healthzReadinessChecks = [];
   /** Promise that resolves when initialization (middleware + routes) is complete */
   initPromise;
   /** In-flight start promise to protect against concurrent start() calls */
@@ -757,8 +776,19 @@ var ExpressServer = class {
       getLogger().error(msg);
       throw new Error(msg);
     }
-    if (config?.healthCheck?.checks) {
-      this.healthChecks.push(...config.healthCheck.checks);
+    if (typeof this.config.healthzServer === "boolean") {
+      this.config.healthzServer = {
+        ...HealthzServer.getDefaultConfig(),
+        enable: this.config.healthzServer
+      };
+    }
+    if (this.config.healthzServer && typeof this.config.healthzServer === "object") {
+      if (this.config.healthzServer.checks) {
+        this.healthzChecks.push(...this.config.healthzServer.checks);
+      }
+      if (this.config.healthzServer.readinessChecks) {
+        this.healthzReadinessChecks.push(...this.config.healthzServer.readinessChecks);
+      }
     }
     this.globalPrefix = this.normalizePath(this.config.globalPrefix ?? "", false);
     this.hooks = hooks;
@@ -1106,10 +1136,6 @@ var ExpressServer = class {
   * 4. Error handler
   */
   async setupRoutes() {
-    const healthCheckPath = this.normalizePath(this.config.healthCheck?.path || "/healthz", this.config.healthCheck?.withGlobalPrefix);
-    this.app.get(healthCheckPath, async (_req, res) => {
-      return this.handleHealthCheckRequest(res);
-    });
     this.app.use(this.globalPrefix, this.rootRouter);
     await this.runHook("afterRoutes", this.app);
     this.app.use((req, res) => {
@@ -1133,60 +1159,25 @@ var ExpressServer = class {
     });
   }
   /**
-  * Execute health check and return response.
+  * Whether the Healthz probe server is enabled.
   */
-  async handleHealthCheckRequest(res) {
-    try {
-      if (!this.healthChecks.length || getCatbeeServerGlobalConfig().skipHealthzChecksValidation) {
-        return res.status(HttpStatusCodes.OK).json(new SuccessResponse("OK"));
-      }
-      const results = await this.executeHealthChecks();
-      const allOk = results.every((r) => r.status);
-      const status = allOk ? HttpStatusCodes.OK : HttpStatusCodes.SERVICE_UNAVAILABLE;
-      const response = new SuccessResponse(allOk ? "OK" : "Service unavailable");
-      if (!allOk) response.error = true;
-      if (this.config.healthCheck?.detailed) response.data = {
-        checks: results
-      };
-      return res.status(status).json(response);
-    } catch {
-      return res.status(HttpStatusCodes.INTERNAL_SERVER_ERROR).json(new InternalServerErrorException("Health check failed"));
+  isHealthzServerEnabled() {
+    if (typeof this.config.healthzServer === "boolean") {
+      return this.config.healthzServer;
     }
+    return this.config.healthzServer?.enable === true;
   }
   /**
-  * Execute all registered health checks and return results.
+  * Get the graceful shutdown delay in milliseconds configured for HealthzServer.
   */
-  async executeHealthChecks() {
-    const checkResults = await Promise.allSettled(this.healthChecks.map(async ({ name, check }) => {
-      try {
-        const status = await Promise.resolve(check());
-        return {
-          name,
-          status,
-          error: null
-        };
-      } catch (error) {
-        return {
-          name,
-          status: false,
-          error: error.message
-        };
-      }
-    }));
-    return checkResults.map((result) => {
-      if (result.status === "fulfilled") return result.value;
-      return {
-        name: "unknown",
-        status: false,
-        error: result.reason
-      };
-    });
+  getHealthzShutdownDelay() {
+    return typeof this.config.healthzServer === "object" ? this.config.healthzServer.shutdownDelayMs ?? 0 : 0;
   }
   /**
-  * Register a new health check function for monitoring service dependencies.
+  * Register a new health check function for monitoring service dependencies on the Healthz probe server.
   *
-  * Health checks are executed when the health endpoint is accessed and
-  * help determine if the service is ready to handle requests.
+  * By default, external dependencies (DB, Redis, etc.) are registered as `readiness` checks.
+  * Can also be registered as `liveness` or `both`.
   *
   * Examples:
   * - Database connectivity
@@ -1195,33 +1186,76 @@ var ExpressServer = class {
   * - Memory/CPU usage checks
   *
   * @param name Unique identifier for the check (used in detailed responses)
-  * @param check Function returning boolean or Promise<boolean> indicating health
+  * @param check Function returning boolean or Promise<boolean> indicating health (supports optional AbortSignal)
+  * @param options Target probe type ('readiness' | 'liveness' | 'both') or options object
   * @returns This instance for method chaining
   */
-  registerHealthCheck(name, check) {
-    this.healthChecks.push({
+  registerHealthCheck(name, check, options) {
+    const probeType = typeof options === "string" ? options : options?.type ?? "readiness";
+    const namedCheck = {
       name,
       check
-    });
+    };
+    if (probeType === "liveness" || probeType === "both") {
+      this.healthzChecks.push(namedCheck);
+    }
+    if (probeType === "readiness" || probeType === "both") {
+      this.healthzReadinessChecks.push(namedCheck);
+    }
+    HealthzServer.registerCheck(namedCheck, probeType);
     return this;
   }
   /**
-  * Run registered health checks and return whether the service is ready.
-  * Useful for readiness probes in deployment tooling.
+  * Mark the service as ready / not-ready for traffic on the Healthz probe server.
   *
-  * @returns Promise resolving to `true` when all checks pass, otherwise `false`.
+  * @param ready Whether the service is ready to receive traffic
+  * @returns This instance for method chaining
   */
-  async ready() {
-    try {
-      if (!this.healthChecks.length || getCatbeeServerGlobalConfig().skipHealthzChecksValidation) return true;
-      const results = await this.executeHealthChecks();
-      return results.every((r) => r.status === true);
-    } catch (err) {
-      getLogger().error({
-        err
-      }, "Error while running readiness checks");
-      return false;
-    }
+  setReady(ready) {
+    HealthzServer.setReady(ready);
+    return this;
+  }
+  /**
+  * Whether the service is currently marked as ready for traffic on the Healthz probe server.
+  */
+  isReady() {
+    return HealthzServer.isReady();
+  }
+  /**
+  * Mark application startup as completed on the Healthz probe server (switches `/startupz` to 200).
+  *
+  * @returns This instance for method chaining
+  */
+  markStartupComplete() {
+    HealthzServer.markStartupComplete();
+    return this;
+  }
+  /**
+  * Whether application startup has completed on the Healthz probe server.
+  */
+  isStartupComplete() {
+    return HealthzServer.isStartupComplete();
+  }
+  /**
+  * Get the running HealthzServer instance (if started).
+  */
+  getHealthzServer() {
+    return HealthzServer.getInstance();
+  }
+  /**
+  * Get the address info of the running HealthzServer (if started).
+  */
+  getHealthzAddress() {
+    return this.healthzAddress;
+  }
+  /**
+  * Returns whether the service is currently marked as ready for traffic on the Healthz probe server.
+  * Useful for readiness checks in deployment tooling.
+  *
+  * @returns `true` when ready, otherwise `false`.
+  */
+  ready() {
+    return HealthzServer.isReady();
   }
   /**
   * Get the underlying Express application instance.
@@ -1278,48 +1312,105 @@ var ExpressServer = class {
   */
   async doStart() {
     await this.initPromise;
-    await this.runHook("beforeStart", this.app);
-    const server = this.createServerInstance();
-    this.server = server;
-    return new Promise((resolve, reject) => {
-      let isListening = false;
-      server.on("error", (err) => {
-        if (!isListening) {
-          getLogger().error({
-            err
-          }, "Server failed to start");
-          try {
-            server.removeAllListeners();
-            server.close();
-          } catch {
+    if (this.isHealthzServerEnabled()) {
+      const healthzConfig = {
+        ...typeof this.config.healthzServer === "object" ? this.config.healthzServer : {},
+        handleSignals: false,
+        checks: [
+          ...this.healthzChecks
+        ],
+        readinessChecks: [
+          ...this.healthzReadinessChecks
+        ]
+      };
+      const addr = await HealthzServer.start(healthzConfig);
+      if (!addr) {
+        throw new Error("Healthz probe server failed to start (already running in this process)");
+      }
+      this.healthzAddress = addr;
+    }
+    try {
+      await this.runHook("beforeStart", this.app);
+      const server = this.createServerInstance();
+      this.server = server;
+      return await new Promise((resolve, reject) => {
+        let isListening = false;
+        server.on("error", async (err) => {
+          if (!isListening) {
+            getLogger().error({
+              err
+            }, "Server failed to start");
+            try {
+              server.removeAllListeners();
+              server.close();
+              if (HealthzServer.isRunning()) {
+                await HealthzServer.stop().catch(() => {
+                });
+              }
+            } catch {
+            }
+            this.server = null;
+            this.healthzAddress = null;
+            this.connections.clear();
+            reject(err);
+          } else {
+            getLogger().error({
+              err
+            }, "Server runtime error");
           }
-          this.server = null;
-          this.connections.clear();
-          reject(err);
-        } else {
-          getLogger().error({
-            err
-          }, "Server runtime error");
-        }
-      });
-      this.setupConnectionTracking();
-      Promise.resolve(this.runHook("onServerCreated", server)).then(() => {
-        const onListening = /* @__PURE__ */ __name(async () => {
-          isListening = true;
-          this.logServerStartInfo();
-          await this.runHook("afterStart", server);
-          resolve(server);
-        }, "onListening");
-        const listenArgs = [
-          this.config.port,
-          this.config.host,
-          onListening
-        ];
-        server.listen(...listenArgs);
-      }).catch((err) => {
-        server.emit("error", err instanceof Error ? err : new Error(String(err)));
+        });
+        this.setupConnectionTracking();
+        Promise.resolve(this.runHook("onServerCreated", server)).then(() => {
+          const onListening = /* @__PURE__ */ __name(async () => {
+            try {
+              this.logServerStartInfo();
+              await this.runHook("afterStart", server);
+              if (this.isHealthzServerEnabled()) {
+                HealthzServer.markStartupComplete();
+                HealthzServer.setReady(true);
+              }
+              isListening = true;
+              resolve(server);
+            } catch (err) {
+              const error = err instanceof Error ? err : new Error(String(err));
+              getLogger().error({
+                err: error
+              }, "Server startup failed");
+              if (HealthzServer.isRunning()) {
+                await HealthzServer.stop().catch(() => {
+                });
+              }
+              try {
+                server.removeAllListeners();
+                server.close();
+              } catch {
+              }
+              this.server = null;
+              this.healthzAddress = null;
+              this.connections.clear();
+              reject(error);
+            }
+          }, "onListening");
+          const listenArgs = [
+            this.config.port,
+            this.config.host,
+            onListening
+          ];
+          server.listen(...listenArgs);
+        }).catch((err) => {
+          server.emit("error", err instanceof Error ? err : new Error(String(err)));
+        });
       });
-    });
+    } catch (err) {
+      if (HealthzServer.isRunning()) {
+        await HealthzServer.stop().catch(() => {
+        });
+      }
+      this.healthzAddress = null;
+      this.server = null;
+      this.connections.clear();
+      throw err;
+    }
   }
   /**
   * Create HTTP or HTTPS server instance (without listening).
@@ -1359,8 +1450,8 @@ var ExpressServer = class {
     const host = this.formatHostForUrl(this.config.host || "localhost");
     const url = `${protocol}://${host}:${port}`;
     getLogger().info(`Server running on ${url}`);
-    if (this.config.healthCheck && this.config.healthCheck?.path) {
-      getLogger().info(`Health check available at ${url}${this.normalizePath(this.config.healthCheck.path, this.config.healthCheck.withGlobalPrefix)}`);
+    if (this.healthzAddress) {
+      getLogger().info(`Healthz probe server running on http://${this.healthzAddress.address}:${this.healthzAddress.port}`);
     }
     if (this.config.openApi?.enable) {
       getLogger().info(`API docs available at ${url}${this.normalizePath(this.config.openApi.mountPath, this.config.openApi.withGlobalPrefix)}`);
@@ -1383,7 +1474,7 @@ var ExpressServer = class {
   * - Monitoring systems are notified
   */
   async stop(force = false) {
-    if (!this.server) {
+    if (!this.server && !HealthzServer.isRunning()) {
       getLogger().warn("Stop called but server is not running");
       return;
     }
@@ -1392,10 +1483,26 @@ var ExpressServer = class {
       return;
     }
     this.isShuttingDown = true;
-    await this.runHook("beforeStop", this.server);
+    if (HealthzServer.isRunning()) {
+      HealthzServer.setReady(false);
+    }
+    if (this.server) {
+      await this.runHook("beforeStop", this.server);
+    }
     try {
-      await this.gracefulShutdown(force);
+      const shutdownDelay = this.getHealthzShutdownDelay();
+      if (shutdownDelay > 0 && !force && HealthzServer.isRunning()) {
+        getLogger().info(`Waiting ${shutdownDelay}ms for load balancer to drain traffic...`);
+        await new Promise((resolve) => setTimeout(resolve, shutdownDelay));
+      }
+      if (this.server) {
+        await this.gracefulShutdown(force);
+      }
     } finally {
+      if (HealthzServer.isRunning()) {
+        await HealthzServer.stop();
+        this.healthzAddress = null;
+      }
       this.isShuttingDown = false;
     }
   }
@@ -1484,7 +1591,7 @@ var ExpressServer = class {
         getLogger().info(`Received ${signal}, initiating graceful shutdown...`);
         try {
           this.disableGracefulShutdown();
-          await this.stop(true);
+          await this.stop(false);
           process.exit(0);
         } catch (err) {
           getLogger().fatal({