@banzaicloud/service-tools/dist/graceful-shutdown.js

"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
Object.defineProperty(exports, "__esModule", { value: true });
const logger_1 = require("./logger");
function registerGracefulShutdown(closeHandlers = [], { logger = {
    info: logger_1.default.info.bind(logger_1.default),
    error: logger_1.default.error.bind(logger_1.default),
}, timeout = 30, } = {}) {
    // gracefully shutdown on SIGTERM or SIGINT signal
    process.once('SIGTERM', gracefulShutDown.bind(this, 'SIGTERM'));
    process.once('SIGINT', gracefulShutDown.bind(this, 'SIGINT'));
    function gracefulShutDown(signal = 'SIGTERM') {
        return __awaiter(this, void 0, void 0, function* () {
            logger.info(`got kill signal (${signal}), starting graceful shut down`);
            // shut down anyway after `timeout` seconds
            if (timeout) {
                setTimeout(() => {
                    logger.error('could not finish in time, forcefully exiting');
                    process.exit(1);
                }, timeout * 1000).unref();
            }
            // release resources
            let isError = false;
            for (const handler of closeHandlers) {
                try {
                    yield Promise.resolve(handler());
                }
                catch (err) {
                    logger.error(err, 'error happened during graceful shut down');
                    isError = true;
                }
            }
            if (isError) {
                process.exit(1);
            }
            logger.info('graceful shut down finished');
            process.kill(process.pid, signal);
        });
    }
}
exports.default = registerGracefulShutdown;
//# sourceMappingURL=graceful-shutdown.js.map