UNPKG

@nestjs/core

Version:

Nest - modern, fast, powerful node.js web framework (@core)

46 lines (45 loc) 1.92 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); const shared_utils_1 = require("@nestjs/common/utils/shared.utils"); const iterare_1 = require("iterare"); const transient_instances_1 = require("../injector/transient-instances"); /** * Checks if the given instance has the `onApplicationShutdown` function * * @param instance The instance which should be checked */ function hasOnAppShutdownHook(instance) { return !shared_utils_1.isNil(instance.onApplicationShutdown); } /** * Calls the given instances */ function callOperator(instances, signal) { return iterare_1.default(instances) .filter(instance => !shared_utils_1.isNil(instance)) .filter(hasOnAppShutdownHook) .map(async (instance) => instance.onApplicationShutdown(signal)) .toArray(); } /** * Calls the `onApplicationShutdown` function on the module and its children * (providers / controllers). * * @param module The module which will be initialized */ async function callAppShutdownHook(module, signal) { const providers = [...module.providers]; // Module (class) instance is the first element of the providers array // Lifecycle hook has to be called once all classes are properly initialized const [_, { instance: moduleClassInstance }] = providers.shift(); const instances = [...module.controllers, ...providers]; const nonTransientInstances = transient_instances_1.getNonTransientInstances(instances); await Promise.all(callOperator(nonTransientInstances, signal)); const transientInstances = transient_instances_1.getTransientInstances(instances); await Promise.all(callOperator(transientInstances, signal)); // Call the instance itself if (moduleClassInstance && hasOnAppShutdownHook(moduleClassInstance)) { await moduleClassInstance.onApplicationShutdown(signal); } } exports.callAppShutdownHook = callAppShutdownHook;