@nestjs/microservices
Version:
Nest - modern, fast, powerful node.js web framework (@microservices)
157 lines (156 loc) • 7.28 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.ClientRMQ = void 0;
const logger_service_1 = require("@nestjs/common/services/logger.service");
const load_package_util_1 = require("@nestjs/common/utils/load-package.util");
const random_string_generator_util_1 = require("@nestjs/common/utils/random-string-generator.util");
const events_1 = require("events");
const rxjs_1 = require("rxjs");
const operators_1 = require("rxjs/operators");
const constants_1 = require("../constants");
const rmq_record_serializer_1 = require("../serializers/rmq-record.serializer");
const client_proxy_1 = require("./client-proxy");
let rqmPackage = {};
const REPLY_QUEUE = 'amq.rabbitmq.reply-to';
class ClientRMQ extends client_proxy_1.ClientProxy {
constructor(options) {
super();
this.options = options;
this.logger = new logger_service_1.Logger(client_proxy_1.ClientProxy.name);
this.client = null;
this.channel = null;
this.urls = this.getOptionsProp(this.options, 'urls') || [constants_1.RQM_DEFAULT_URL];
this.queue =
this.getOptionsProp(this.options, 'queue') || constants_1.RQM_DEFAULT_QUEUE;
this.queueOptions =
this.getOptionsProp(this.options, 'queueOptions') ||
constants_1.RQM_DEFAULT_QUEUE_OPTIONS;
this.replyQueue =
this.getOptionsProp(this.options, 'replyQueue') || REPLY_QUEUE;
this.persistent =
this.getOptionsProp(this.options, 'persistent') || constants_1.RQM_DEFAULT_PERSISTENT;
load_package_util_1.loadPackage('amqplib', ClientRMQ.name, () => require('amqplib'));
rqmPackage = load_package_util_1.loadPackage('amqp-connection-manager', ClientRMQ.name, () => require('amqp-connection-manager'));
this.initializeSerializer(options);
this.initializeDeserializer(options);
}
close() {
this.channel && this.channel.close();
this.client && this.client.close();
this.channel = null;
this.client = null;
}
connect() {
if (this.client) {
return this.connection;
}
this.client = this.createClient();
this.handleError(this.client);
this.handleDisconnectError(this.client);
const connect$ = this.connect$(this.client);
this.connection = rxjs_1.lastValueFrom(this.mergeDisconnectEvent(this.client, connect$).pipe(operators_1.switchMap(() => this.createChannel()), operators_1.share())).catch(err => {
if (err instanceof rxjs_1.EmptyError) {
return;
}
throw err;
});
return this.connection;
}
createChannel() {
return new Promise(resolve => {
this.channel = this.client.createChannel({
json: false,
setup: (channel) => this.setupChannel(channel, resolve),
});
});
}
createClient() {
const socketOptions = this.getOptionsProp(this.options, 'socketOptions');
return rqmPackage.connect(this.urls, {
connectionOptions: socketOptions,
});
}
mergeDisconnectEvent(instance, source$) {
const close$ = rxjs_1.fromEvent(instance, constants_1.DISCONNECT_EVENT).pipe(operators_1.map((err) => {
throw err;
}));
return rxjs_1.merge(source$, close$).pipe(operators_1.first());
}
async setupChannel(channel, resolve) {
const prefetchCount = this.getOptionsProp(this.options, 'prefetchCount') ||
constants_1.RQM_DEFAULT_PREFETCH_COUNT;
const isGlobalPrefetchCount = this.getOptionsProp(this.options, 'isGlobalPrefetchCount') ||
constants_1.RQM_DEFAULT_IS_GLOBAL_PREFETCH_COUNT;
await channel.assertQueue(this.queue, this.queueOptions);
await channel.prefetch(prefetchCount, isGlobalPrefetchCount);
this.responseEmitter = new events_1.EventEmitter();
this.responseEmitter.setMaxListeners(0);
await this.consumeChannel(channel);
resolve();
}
async consumeChannel(channel) {
const noAck = this.getOptionsProp(this.options, 'noAck', constants_1.RQM_DEFAULT_NOACK);
await channel.consume(this.replyQueue, (msg) => this.responseEmitter.emit(msg.properties.correlationId, msg), {
noAck,
});
}
handleError(client) {
client.addListener(constants_1.ERROR_EVENT, (err) => this.logger.error(err));
}
handleDisconnectError(client) {
client.addListener(constants_1.DISCONNECT_EVENT, (err) => {
this.logger.error(constants_1.DISCONNECTED_RMQ_MESSAGE);
this.logger.error(err);
this.close();
});
}
async handleMessage(packet, callback) {
const { err, response, isDisposed } = await this.deserializer.deserialize(packet);
if (isDisposed || err) {
callback({
err,
response,
isDisposed: true,
});
}
callback({
err,
response,
});
}
publish(message, callback) {
try {
const correlationId = random_string_generator_util_1.randomStringGenerator();
const listener = ({ content }) => this.handleMessage(JSON.parse(content.toString()), callback);
Object.assign(message, { id: correlationId });
const serializedPacket = this.serializer.serialize(message);
const options = serializedPacket.options;
delete serializedPacket.options;
this.responseEmitter.on(correlationId, listener);
this.channel.sendToQueue(this.queue, Buffer.from(JSON.stringify(serializedPacket)), Object.assign(Object.assign({ replyTo: this.replyQueue, persistent: this.persistent }, options), { headers: this.mergeHeaders(options === null || options === void 0 ? void 0 : options.headers), correlationId }));
return () => this.responseEmitter.removeListener(correlationId, listener);
}
catch (err) {
callback({ err });
}
}
dispatchEvent(packet) {
const serializedPacket = this.serializer.serialize(packet);
return new Promise((resolve, reject) => {
var _a;
return this.channel.sendToQueue(this.queue, Buffer.from(JSON.stringify(serializedPacket)), Object.assign(Object.assign({ persistent: this.persistent }, serializedPacket.options), { headers: this.mergeHeaders((_a = serializedPacket.options) === null || _a === void 0 ? void 0 : _a.headers) }), (err) => (err ? reject(err) : resolve()));
});
}
initializeSerializer(options) {
var _a;
this.serializer = (_a = options === null || options === void 0 ? void 0 : options.serializer) !== null && _a !== void 0 ? _a : new rmq_record_serializer_1.RmqRecordSerializer();
}
mergeHeaders(requestHeaders) {
var _a, _b;
if (!requestHeaders && !((_a = this.options) === null || _a === void 0 ? void 0 : _a.headers)) {
return undefined;
}
return Object.assign(Object.assign({}, (_b = this.options) === null || _b === void 0 ? void 0 : _b.headers), requestHeaders);
}
}
exports.ClientRMQ = ClientRMQ;