@harmonyjs/controller-cloud-health/dist-src/index.js

REST controller for Harmony

import Fastify from 'fastify';
import * as health from '@cloudnative/health-connect';
function wrap(path) {
    if (path.startsWith('/')) {
        return path;
    }
    return `/${path}`;
}
const ControllerCloudHealth = function ControllerCloudHealth(config) {
    return ({
        name: 'ControllerCloudHealth',
        async initialize({ logger, server }) {
            const readinessPath = config.readinessPath || '/ready';
            const livenessPath = config.livenessPath || '/live';
            const healthPath = config.healthPath || '/health';
            const healthcheck = new health.HealthChecker();
            const instance = config.standalone ? Fastify() : server;
            instance.register((fastify, opts, done) => {
                logger.info(`Registering liveness probe on path ${wrap(livenessPath)}`);
                fastify.use(wrap(livenessPath), health.LivenessEndpoint(healthcheck));
                logger.info(`Registering readiness probe on path ${wrap(readinessPath)}`);
                fastify.use(wrap(readinessPath), health.ReadinessEndpoint(healthcheck));
                logger.info(`Registering health probe on path ${wrap(healthPath)}`);
                fastify.use(wrap(healthPath), health.HealthEndpoint(healthcheck));
                logger.info('All probes registered');
                done();
            }, {
                prefix: config.prefix,
            });
            config.checks.startup.forEach((check, i) => {
                healthcheck.registerStartupCheck(new health.StartupCheck(`startup_${i}`, check));
            });
            config.checks.liveness.forEach((check, i) => {
                healthcheck.registerLivenessCheck(new health.LivenessCheck(`liveness_${i}`, check));
            });
            config.checks.readiness.forEach((check, i) => {
                healthcheck.registerReadinessCheck(new health.ReadinessCheck(`readiness_${i}`, check));
            });
            config.checks.shutdown.forEach((check, i) => {
                healthcheck.registerShutdownCheck(new health.ShutdownCheck(`shutdown_${i}`, check));
            });
            if (config.standalone) {
                instance.listen(config.standalone.port, config.standalone.host);
            }
        },
    });
};
ControllerCloudHealth.startupLock = () => {
    let release = () => { };
    const lock = new Promise((resolve) => {
        release = resolve;
    });
    return {
        lock,
        release,
    };
};
export default ControllerCloudHealth;