UNPKG

@rxap/nest-amqp

Version:
654 lines (551 loc) 21.7 kB
<!doctype html> <html class="no-js" lang=""> <head> <meta charset="utf-8"> <meta http-equiv="x-ua-compatible" content="ie=edge"> <title>nest-rabbitmq</title> <meta name="description" content=""> <meta name="viewport" content="width=device-width, initial-scale=1"> <link rel="icon" type="image/x-icon" href="../images/favicon.ico"> <link rel="stylesheet" href="../styles/style.css"> <link rel="stylesheet" href="../styles/dark.css"> </head> <body> <script> // Blocking script to avoid flickering dark mode // Dark mode toggle button var useDark = window.matchMedia('(prefers-color-scheme: dark)'); var darkModeState = useDark.matches; var $darkModeToggleSwitchers = document.querySelectorAll('.dark-mode-switch input'); var $darkModeToggles = document.querySelectorAll('.dark-mode-switch'); var darkModeStateLocal = localStorage.getItem('compodoc_darkmode-state'); function checkToggle(check) { for (var i = 0; i < $darkModeToggleSwitchers.length; i++) { $darkModeToggleSwitchers[i].checked = check; } } function toggleDarkMode(state) { if (window.localStorage) { localStorage.setItem('compodoc_darkmode-state', state); } checkToggle(state); const hasClass = document.body.classList.contains('dark'); if (state) { for (var i = 0; i < $darkModeToggles.length; i++) { $darkModeToggles[i].classList.add('dark'); } if (!hasClass) { document.body.classList.add('dark'); } } else { for (var i = 0; i < $darkModeToggles.length; i++) { $darkModeToggles[i].classList.remove('dark'); } if (hasClass) { document.body.classList.remove('dark'); } } } useDark.addEventListener('change', function (evt) { toggleDarkMode(evt.matches); }); if (darkModeStateLocal) { darkModeState = darkModeStateLocal === 'true'; } toggleDarkMode(darkModeState); </script> <div class="navbar navbar-default navbar-fixed-top d-md-none p-0"> <div class="d-flex"> <a href="../" class="navbar-brand">nest-rabbitmq</a> <button type="button" class="btn btn-default btn-menu ion-ios-menu" id="btn-menu"></button> </div> </div> <div class="xs-menu menu" id="mobile-menu"> <div id="book-search-input" role="search"><input type="text" placeholder="Type to search"></div> <compodoc-menu></compodoc-menu> </div> <div class="container-fluid main"> <div class="row main"> <div class="d-none d-md-block menu"> <compodoc-menu mode="normal"></compodoc-menu> </div> <!-- START CONTENT --> <div class="content interface"> <div class="content-data"> <ol class="breadcrumb"> <li class="breadcrumb-item">Interfaces</li> <li class="breadcrumb-item" > RmqExchangeOptions</li> </ol> <ul class="nav nav-tabs" role="tablist"> <li class="nav-item"> <a href="#info" class="nav-link" class="nav-link active" role="tab" id="info-tab" data-bs-toggle="tab" data-link="info">Info</a> </li> <li class="nav-item"> <a href="#source" class="nav-link" role="tab" id="source-tab" data-bs-toggle="tab" data-link="source">Source</a> </li> </ul> <div class="tab-content"> <div class="tab-pane fade active in" id="info"> <p class="comment"> <h3>File</h3> </p> <p class="comment"> <code>src/lib/client-rmq-exchange.ts</code> </p> <section data-compodoc="block-index"> <h3 id="index">Index</h3> <table class="table table-sm table-bordered index-table"> <tbody> <tr> <td class="col-md-4"> <h6><b>Properties</b></h6> </td> </tr> <tr> <td class="col-md-4"> <ul class="index-list"> <li> <span class="modifier">Optional</span> <a href="#options" > options </a> </li> </ul> </td> </tr> </tbody> </table> </section> <section data-compodoc="block-properties"> <h3 id="inputs">Properties</h3> <table class="table table-sm table-bordered"> <tbody> <tr> <td class="col-md-4"> <a name="options"></a> <span class="name "><b>options</b> <a href="#options"> <span class="icon ion-ios-link"></span> </a> </span> </td> </tr> <tr> <td class="col-md-4"> <code>options: <code><a href="../interfaces/ExchangeRmqOptions.html" target="_self" >ExchangeRmqOptions</a></code> </code> </td> </tr> <tr> <td class="col-md-4"> <i>Type : </i> <code><a href="../interfaces/ExchangeRmqOptions.html" target="_self" >ExchangeRmqOptions</a></code> </td> </tr> <tr> <td class="col-md-4"> <i>Optional</i> </td> </tr> </tbody> </table> </section> </div> <div class="tab-pane fade tab-source-code" id="source"> <pre class="line-numbers compodoc-sourcecode"><code class="language-typescript">import { Logger } from &#x27;@nestjs/common/services/logger.service&#x27;; import { randomStringGenerator } from &#x27;@nestjs/common/utils/random-string-generator.util&#x27;; import { isFunction } from &#x27;@nestjs/common/utils/shared.utils&#x27;; import { ClientProxy, ReadPacket, RmqRecord, WritePacket, } from &#x27;@nestjs/microservices&#x27;; import { CONNECT_EVENT, CONNECT_FAILED_EVENT, DISCONNECT_EVENT, DISCONNECTED_RMQ_MESSAGE, ERROR_EVENT, RQM_DEFAULT_NOACK, } from &#x27;@nestjs/microservices/constants&#x27;; import { RmqRecordSerializer } from &#x27;@nestjs/microservices/serializers&#x27;; import { ExchangeRmqOptions } from &#x27;./options&#x27;; import { AmqpConnectionManager, ChannelWrapper, connect, } from &#x27;amqp-connection-manager&#x27;; import { Channel, Connection, ConsumeMessage, Options, } from &#x27;amqplib&#x27;; import { EventEmitter } from &#x27;events&#x27;; import { firstValueFrom, fromEvent, merge, Observable, ReplaySubject, tap, } from &#x27;rxjs&#x27;; import { first, map, retryWhen, scan, skip, } from &#x27;rxjs/operators&#x27;; import { IncomingResponseDeserializer } from &#x27;./incoming-response.deserializer&#x27;; export const REPLY_QUEUE &#x3D; &#x27;amq.rabbitmq.reply-to&#x27;; export interface RmqExchangeOptions { options?: ExchangeRmqOptions; } export class ClientRMQExchange extends ClientProxy { protected connection$!: ReplaySubject&lt;Connection&gt;; protected client: AmqpConnectionManager | null &#x3D; null; protected channel: ChannelWrapper | null &#x3D; null; protected responseEmitter!: EventEmitter; constructor( protected readonly options: ExchangeRmqOptions, protected readonly logger &#x3D; new Logger(ClientProxy.name), ) { super(); this.initializeSerializer(options); this.initializeDeserializer(options); } public isConnected() { return this.client &amp;&amp; this.client.isConnected(); } public close(): void { this.channel &amp;&amp; this.channel.close(); this.client &amp;&amp; this.client.close(); this.channel &#x3D; null; this.client &#x3D; null; } public connect(): Promise&lt;Connection&gt; { if (this.client) { return this.convertConnectionToPromise(); } this.client &#x3D; this.createClient(); this.handleError(this.client); this.handleDisconnectError(this.client); this.responseEmitter &#x3D; new EventEmitter(); this.responseEmitter.setMaxListeners(0); const connect$ &#x3D; this.connect$(this.client); const withDisconnect$ &#x3D; this.mergeDisconnectEvent( this.client, connect$, ).pipe( tap(() &#x3D;&gt; this.createChannel()), ); const withReconnect$ &#x3D; fromEvent(this.client, CONNECT_EVENT).pipe( skip(1), ); const source$: Observable&lt;{ url: string, connection: Connection }&gt; &#x3D; merge(withDisconnect$, withReconnect$); this.connection$ &#x3D; new ReplaySubject(1); source$.subscribe({ next: data &#x3D;&gt; { this.connection$.next(data.connection); }, }); return this.convertConnectionToPromise(); } public createChannel(): Promise&lt;void&gt; { return new Promise(resolve &#x3D;&gt; { this.channel &#x3D; this.client!.createChannel({ json: false, setup: (channel: Channel) &#x3D;&gt; this.setupExchange(channel, resolve), }); }); } public createClient(): AmqpConnectionManager { const socketOptions &#x3D; this.getOptionsProp(this.options, &#x27;socketOptions&#x27;); return connect(this.options.urls, socketOptions); } public mergeDisconnectEvent&lt;T &#x3D; any&gt;( instance: any, source$: Observable&lt;T&gt;, ): Observable&lt;T&gt; { const eventToError &#x3D; (eventType: string) &#x3D;&gt; fromEvent(instance, eventType).pipe( map((err: unknown) &#x3D;&gt; { throw err; }), ); const disconnect$ &#x3D; eventToError(DISCONNECT_EVENT); const urls &#x3D; this.getOptionsProp(this.options, &#x27;urls&#x27;, []) ?? []; const connectFailed$ &#x3D; eventToError(CONNECT_FAILED_EVENT).pipe( retryWhen(e &#x3D;&gt; e.pipe( scan((errorCount, error: any) &#x3D;&gt; { if (urls.indexOf(error.url) &gt;&#x3D; urls.length - 1) { throw error; } return errorCount + 1; }, 0), ), ), ); // If we ever decide to propagate all disconnect errors &amp; re-emit them through // the &quot;connection&quot; stream then comment out &quot;first()&quot; operator. return merge(source$, disconnect$, connectFailed$).pipe(first()); } public async convertConnectionToPromise() { // try { return await firstValueFrom(this.connection$); // } catch (err) { // if (err instanceof EmptyError) { // return; // } // throw err; // } } public async setupExchange(channel: Channel, resolve: () &#x3D;&gt; unknown) { this.logger.verbose(&#x60;Setting up exchange &#x27;${ this.options.exchange.name }&#x27;&#x60;, &#x27;ClientRMQExchange&#x27;); if (!this.options.noAssert) { await channel.assertExchange( this.options.exchange.name, this.options.exchange.type, this.options.exchange.options); } await this.consumeChannel(channel); resolve(); } public async consumeChannel(channel: Channel) { const noAck &#x3D; this.getOptionsProp(this.options, &#x27;noAck&#x27;, RQM_DEFAULT_NOACK); await channel.consume( this.options.replyQueue ?? REPLY_QUEUE, (msg: ConsumeMessage | null) &#x3D;&gt; { if (msg) { this.responseEmitter.emit(msg.properties.correlationId, msg); } else { this.logger.warn(&#x60;Message is empty from RMQ queue ${ this.options.replyQueue ?? REPLY_QUEUE }&#x60;); } }, { noAck, }, ); } public handleError(client: AmqpConnectionManager): void { client.addListener(ERROR_EVENT, (err: any) &#x3D;&gt; this.logger.error(err)); } public handleDisconnectError(client: AmqpConnectionManager): void { client.addListener(DISCONNECT_EVENT, (err: any) &#x3D;&gt; { this.logger.error(DISCONNECTED_RMQ_MESSAGE); this.logger.error(err); }); } public async handleMessage( packet: unknown, callback: (packet: WritePacket) &#x3D;&gt; any, ): Promise&lt;void&gt;; public async handleMessage( packet: unknown, options: Record&lt;string, unknown&gt;, callback: (packet: WritePacket) &#x3D;&gt; any, ): Promise&lt;void&gt;; public async handleMessage( packet: unknown, optionsOrCallback: Record&lt;string, unknown&gt; | ((packet: WritePacket) &#x3D;&gt; any), callback?: (packet: WritePacket) &#x3D;&gt; any, ): Promise&lt;void&gt; { this.logger.verbose(&#x27;Received message: %JSON&#x27;, packet, &#x27;ClientRMQExchange&#x27;); let options: Record&lt;string, unknown&gt; | undefined &#x3D; undefined; if (isFunction(options)) { callback &#x3D; options as (packet: WritePacket) &#x3D;&gt; any; } else { options &#x3D; optionsOrCallback as Record&lt;string, unknown&gt;; } if (!callback) { throw new Error(&#x27;No callback provided&#x27;); } const { err, response, isDisposed, } &#x3D; await this.deserializer.deserialize( packet, options, ); this.logger.verbose(&#x27;Deserialized response: %JSON&#x27;, response, &#x27;ClientRMQExchange&#x27;); if (err) { this.logger.verbose(&#x27;Deserialized error: %JSON&#x27;, err, &#x27;ClientRMQExchange&#x27;); } this.logger.verbose(&#x27;Deserialized isDisposed: %JSON&#x27;, isDisposed, &#x27;ClientRMQExchange&#x27;); if (isDisposed || err) { callback({ err, response, isDisposed: true, }); } callback({ err, response, }); } protected publish( message: ReadPacket, callback: (packet: WritePacket) &#x3D;&gt; any, ): () &#x3D;&gt; void { this.logger.verbose(&#x27;Publishing message: %JSON&#x27;, message, &#x27;ClientRMQExchange&#x27;); try { const correlationId &#x3D; randomStringGenerator(); const listener &#x3D; ({ content, fields, properties, }: ConsumeMessage) &#x3D;&gt; this.handleMessage( this.parseMessageContent(content), { fields, properties, }, callback, ); Object.assign(message, { id: correlationId }); const serializedPacket: ReadPacket &amp; Partial&lt;RmqRecord&gt; &#x3D; this.serializer.serialize(message); const options &#x3D; serializedPacket.options; delete serializedPacket.options; this.responseEmitter.on(correlationId, listener); this.channel! .publish( this.options.exchange.name, serializedPacket.pattern, Buffer.from(JSON.stringify(serializedPacket.data)), { replyTo: this.options.replyQueue ?? REPLY_QUEUE, persistent: this.options.persistent, ...options, headers: this.mergeHeaders(options?.headers), correlationId, } as Options.Publish, ) .catch(err &#x3D;&gt; callback({ err })); return () &#x3D;&gt; this.responseEmitter.removeListener(correlationId, listener); } catch (err) { callback({ err }); } return () &#x3D;&gt; void 0; } protected dispatchEvent(packet: ReadPacket): Promise&lt;any&gt; { this.logger.verbose(&#x27;Dispatching event: %JSON&#x27;, packet, &#x27;ClientRMQExchange&#x27;); const serializedPacket: ReadPacket &amp; Partial&lt;RmqRecord&gt; &#x3D; this.serializer.serialize(packet); const options &#x3D; serializedPacket.options; delete serializedPacket.options; return new Promise&lt;void&gt;((resolve, reject) &#x3D;&gt; this.channel!.publish( this.options.exchange.name, serializedPacket.pattern, Buffer.from(JSON.stringify(serializedPacket.data)), { persistent: this.options.persistent, ...options, headers: this.mergeHeaders(options?.headers), } as Options.Publish, (err: unknown) &#x3D;&gt; ( err ? reject(err) : resolve() ), ), ); } protected override initializeSerializer(options: ExchangeRmqOptions) { this.serializer &#x3D; options.serializer ?? new RmqRecordSerializer(); } protected override initializeDeserializer(options: ExchangeRmqOptions) { this.deserializer &#x3D; options.deserializer ?? new IncomingResponseDeserializer(); } protected mergeHeaders( requestHeaders?: Record&lt;string, string&gt;, ): Record&lt;string, string&gt; | undefined { if (!requestHeaders &amp;&amp; !this.options?.headers) { return undefined; } return { ...this.options?.headers, ...requestHeaders, }; } protected parseMessageContent(content: Buffer) { const rawContent &#x3D; content.toString(); try { return JSON.parse(rawContent); } catch { return rawContent; } } } </code></pre> </div> </div> </div><div class="search-results"> <div class="has-results"> <h1 class="search-results-title"><span class='search-results-count'></span> results matching "<span class='search-query'></span>"</h1> <ul class="search-results-list"></ul> </div> <div class="no-results"> <h1 class="search-results-title">No results matching "<span class='search-query'></span>"</h1> </div> </div> </div> <!-- END CONTENT --> </div> </div> <label class="dark-mode-switch"> <input type="checkbox"> <span class="slider"> <svg class="slider-icon" viewBox="0 0 24 24" fill="none" height="20" stroke="#000" stroke-linecap="round" stroke-linejoin="round" stroke-width="2" width="20" xmlns="http://www.w3.org/2000/svg"> <path d="M21 12.79A9 9 0 1111.21 3 7 7 0 0021 12.79z"></path> </svg> </span> </label> <script> var COMPODOC_CURRENT_PAGE_DEPTH = 1; var COMPODOC_CURRENT_PAGE_CONTEXT = 'interface'; var COMPODOC_CURRENT_PAGE_URL = 'RmqExchangeOptions.html'; var MAX_SEARCH_RESULTS = 15; </script> <script> $darkModeToggleSwitchers = document.querySelectorAll('.dark-mode-switch input'); checkToggle(darkModeState); if ($darkModeToggleSwitchers.length > 0) { for (var i = 0; i < $darkModeToggleSwitchers.length; i++) { $darkModeToggleSwitchers[i].addEventListener('change', function (event) { darkModeState = !darkModeState; toggleDarkMode(darkModeState); }); } } </script> <script src="../js/libs/custom-elements.min.js"></script> <script src="../js/libs/lit-html.js"></script> <script src="../js/menu-wc.js" defer></script> <script nomodule src="../js/menu-wc_es5.js" defer></script> <script src="../js/libs/bootstrap-native.js"></script> <script src="../js/libs/es6-shim.min.js"></script> <script src="../js/libs/EventDispatcher.js"></script> <script src="../js/libs/promise.min.js"></script> <script src="../js/libs/zepto.min.js"></script> <script src="../js/compodoc.js"></script> <script src="../js/tabs.js"></script> <script src="../js/menu.js"></script> <script src="../js/libs/clipboard.min.js"></script> <script src="../js/libs/prism.js"></script> <script src="../js/sourceCode.js"></script> <script src="../js/search/search.js"></script> <script src="../js/search/lunr.min.js"></script> <script src="../js/search/search-lunr.js"></script> <script src="../js/search/search_index.js"></script> <script src="../js/lazy-load-graphs.js"></script> </body> </html>