Function decorateWithHealthCheck

lib/terminus.js:73–128

(server, state, options)

function noop () {}

function decorateWithHealthCheck (server, state, options) {
  const { healthChecks, logger, headers, onSendFailureDuringShutdown, sendFailuresDuringShutdown, caseInsensitive, statusOk, statusOkResponse, statusError, statusErrorResponse } = options

  let hasSetHandler = false
  const createHandler = (listener) => {
    const check = hasSetHandler
      ? () => {}
      : async (healthCheck, res) => {
        if (state.isShuttingDown && sendFailuresDuringShutdown) {
          return sendFailure(res, { onSendFailureDuringShutdown, statusError, statusErrorResponse })
        }
        let info
        try {
          info = await healthCheck({ state })
        } catch (error) {
          logger('healthcheck failed', error)
          const statusCode = error.statusCode
          const statusResponse = error.statusResponse
          return sendFailure(
            res,
            {
              error: error.causes,
              headers,
              exposeStackTraces: healthChecks.__unsafeExposeStackTraces,
              statusCode,
              statusResponse,
              statusError,
              statusErrorResponse
            }
          )
        }
        return sendSuccess(
          res,
          { info, verbatim: healthChecks.verbatim, statusOk, statusOkResponse, headers }
        )
      }

    hasSetHandler = true

    return async (req, res) => {
      const url = caseInsensitive ? req.url.toLowerCase() : req.url
      const healthCheck = healthChecks[url]

      if (healthCheck) {
        return check(healthCheck, res)
      } else {
        listener(req, res)
      }
    }
  }

  server.listeners('request').forEach((listener) => {
    server.removeListener('request', listener)
    server.on('request', createHandler(listener))
  })
}

function decorateWithSignalHandler (server, state, options) {