@nestjs/microservices
Version:
Nest - modern, fast, powerful node.js web framework (@microservices)
77 lines (76 loc) • 2.74 kB
JavaScript
;
var ClientsModule_1;
Object.defineProperty(exports, "__esModule", { value: true });
exports.ClientsModule = void 0;
const tslib_1 = require("tslib");
const common_1 = require("@nestjs/common");
const client_1 = require("../client");
let ClientsModule = ClientsModule_1 = class ClientsModule {
static register(options) {
const clients = (options || []).map(item => ({
provide: item.name,
useValue: this.assignOnAppShutdownHook(client_1.ClientProxyFactory.create(item)),
}));
return {
module: ClientsModule_1,
providers: clients,
exports: clients,
};
}
static registerAsync(options) {
const providers = options.reduce((accProviders, item) => accProviders
.concat(this.createAsyncProviders(item))
.concat(item.extraProviders || []), []);
const imports = options.reduce((accImports, option) => option.imports && !accImports.includes(option.imports)
? accImports.concat(option.imports)
: accImports, []);
return {
module: ClientsModule_1,
imports,
providers: providers,
exports: providers,
};
}
static createAsyncProviders(options) {
if (options.useExisting || options.useFactory) {
return [this.createAsyncOptionsProvider(options)];
}
return [
this.createAsyncOptionsProvider(options),
{
provide: options.useClass,
useClass: options.useClass,
},
];
}
static createAsyncOptionsProvider(options) {
if (options.useFactory) {
return {
provide: options.name,
useFactory: this.createFactoryWrapper(options.useFactory),
inject: options.inject || [],
};
}
return {
provide: options.name,
useFactory: this.createFactoryWrapper((optionsFactory) => optionsFactory.createClientOptions()),
inject: [options.useExisting || options.useClass],
};
}
static createFactoryWrapper(useFactory) {
return async (...args) => {
const clientOptions = await useFactory(...args);
const clientProxyRef = client_1.ClientProxyFactory.create(clientOptions);
return this.assignOnAppShutdownHook(clientProxyRef);
};
}
static assignOnAppShutdownHook(client) {
client.onApplicationShutdown =
client.close;
return client;
}
};
ClientsModule = ClientsModule_1 = tslib_1.__decorate([
(0, common_1.Module)({})
], ClientsModule);
exports.ClientsModule = ClientsModule;