@azure/service-bus 7.10.0-beta.1 → 7.10.0-beta.2

package/dist/react-native/core/messageSender.d.ts

@@ -0,0 +1,93 @@
+import type { AwaitableSender, AwaitableSenderOptions } from "rhea-promise";
+import type { RetryOptions, AmqpAnnotatedMessage } from "@azure/core-amqp";
+import type { ServiceBusMessage } from "../serviceBusMessage.js";
+import type { ConnectionContext } from "../connectionContext.js";
+import { LinkEntity } from "./linkEntity.js";
+import type { ServiceBusMessageBatch } from "../serviceBusMessageBatch.js";
+import type { CreateMessageBatchOptions } from "../models.js";
+import type { OperationOptionsBase } from "../modelsToBeSharedWithEventHubs.js";
+import type { AbortSignalLike } from "@azure/abort-controller";
+/**
+ * @internal
+ * Describes the MessageSender that will send messages to ServiceBus.
+ */
+export declare class MessageSender extends LinkEntity<AwaitableSender> {
+    private identifier;
+    /**
+     * The handler function to handle errors that happen on the
+     * underlying sender.
+     * @readonly
+     */
+    private readonly _onAmqpError;
+    /**
+     * The handler function to handle "sender_close" event
+     * that happens on the underlying sender.
+     * @readonly
+     */
+    private readonly _onAmqpClose;
+    /**
+     * The message handler that will be set as the handler on
+     * the underlying rhea sender's session for the "session_error" event.
+     */
+    private _onSessionError;
+    /**
+     * The message handler that will be set as the handler on
+     * the underlying rhea sender's session for the "session_close" event.
+     */
+    private _onSessionClose;
+    private _retryOptions;
+    constructor(identifier: string, connectionContext: ConnectionContext, entityPath: string, retryOptions: RetryOptions);
+    private _createSenderOptions;
+    /**
+     * Tries to send the message to ServiceBus if there is enough credit to send them
+     * and the circular buffer has available space to settle the message after sending them.
+     *
+     * We have implemented a synchronous send over here in the sense that we shall be waiting
+     * for the message to be accepted or rejected and accordingly resolve or reject the promise.
+     *
+     * @param encodedMessage - The encoded message to be sent to ServiceBus.
+     * @param sendBatch - Boolean indicating whether the encoded message represents a batch of messages or not
+     */
+    private _trySend;
+    protected createRheaLink(options: AwaitableSenderOptions): Promise<AwaitableSender>;
+    /**
+     * Initializes the sender session on the connection.
+     */
+    open(options?: AwaitableSenderOptions, abortSignal?: AbortSignalLike): Promise<void>;
+    /**
+     * Closes the rhea link.
+     * To be called when connection is disconnected, onAmqpClose and onSessionClose events.
+     */
+    onDetached(): Promise<void>;
+    /**
+     * Determines whether the AMQP sender link is open. If open then returns true else returns false.
+     */
+    isOpen(): boolean;
+    /**
+     * Sends the given message, with the given options on this link
+     *
+     * @param data - Message to send. Will be sent as UTF8-encoded JSON string.
+     */
+    send(data: ServiceBusMessage | AmqpAnnotatedMessage, options?: OperationOptionsBase): Promise<void>;
+    /**
+     * Returns maximum message size on the AMQP sender link.
+     *
+     * Options to configure the `createBatch` method on the `Sender`.
+     * - `maxSizeInBytes`: The upper limit for the size of batch.
+     *
+     * Example usage:
+     * ```ts snippet:ignore
+     * {
+     *     retryOptions: { maxRetries: 5; timeoutInMs: 10 }
+     * }
+     * ```
+     */
+    getMaxMessageSize(options?: {
+        retryOptions?: RetryOptions;
+    } & Pick<OperationOptionsBase, "abortSignal">): Promise<number>;
+    createBatch(options?: CreateMessageBatchOptions): Promise<ServiceBusMessageBatch>;
+    sendBatch(batchMessage: ServiceBusMessageBatch, options?: OperationOptionsBase): Promise<void>;
+    static create(identifier: string, context: ConnectionContext, entityPath: string, retryOptions: RetryOptions): MessageSender;
+    protected removeLinkFromContext(): void;
+}
+//# sourceMappingURL=messageSender.d.ts.map

package/dist/react-native/core/messageSender.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"messageSender.d.ts","sourceRoot":"","sources":["../../../src/core/messageSender.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAEV,eAAe,EACf,sBAAsB,EAGvB,MAAM,cAAc,CAAC;AAEtB,OAAO,KAAK,EAGV,YAAY,EACZ,oBAAoB,EACrB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAEjE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAG7C,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AAE3E,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,cAAc,CAAC;AAC9D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,qCAAqC,CAAC;AAChF,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAK/D;;;GAGG;AACH,qBAAa,aAAc,SAAQ,UAAU,CAAC,eAAe,CAAC;IA0B1D,OAAO,CAAC,UAAU;IAzBpB;;;;OAIG;IACH,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAc;IAC3C;;;;OAIG;IACH,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAc;IAC3C;;;OAGG;IACH,OAAO,CAAC,eAAe,CAAc;IACrC;;;OAGG;IACH,OAAO,CAAC,eAAe,CAAc;IACrC,OAAO,CAAC,aAAa,CAAe;gBAG1B,UAAU,EAAE,MAAM,EAC1B,iBAAiB,EAAE,iBAAiB,EACpC,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,YAAY;IAkE5B,OAAO,CAAC,oBAAoB;IAiB5B;;;;;;;;;OASG;IACH,OAAO,CAAC,QAAQ;IA4GhB,SAAS,CAAC,cAAc,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,eAAe,CAAC;IAInF;;OAEG;IACU,IAAI,CACf,OAAO,CAAC,EAAE,sBAAsB,EAChC,WAAW,CAAC,EAAE,eAAe,GAC5B,OAAO,CAAC,IAAI,CAAC;IAqBhB;;;OAGG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAMjC;;OAEG;IACH,MAAM,IAAI,OAAO;IAYjB;;;;OAIG;IACG,IAAI,CACR,IAAI,EAAE,iBAAiB,GAAG,oBAAoB,EAC9C,OAAO,CAAC,EAAE,oBAAoB,GAC7B,OAAO,CAAC,IAAI,CAAC;IAoBhB;;;;;;;;;;;;OAYG;IACG,iBAAiB,CACrB,OAAO,GAAE;QACP,YAAY,CAAC,EAAE,YAAY,CAAC;KAC7B,GAAG,IAAI,CAAC,oBAAoB,EAAE,aAAa,CAAM,GACjD,OAAO,CAAC,MAAM,CAAC;IA4BZ,WAAW,CAAC,OAAO,CAAC,EAAE,yBAAyB,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAkBjF,SAAS,CACb,YAAY,EAAE,sBAAsB,EACpC,OAAO,CAAC,EAAE,oBAAoB,GAC7B,OAAO,CAAC,IAAI,CAAC;IAsBhB,MAAM,CAAC,MAAM,CACX,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,iBAAiB,EAC1B,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,YAAY,GACzB,aAAa;IAQhB,SAAS,CAAC,qBAAqB,IAAI,IAAI;CAGxC"}

package/dist/react-native/core/messageSender.js

@@ -0,0 +1,280 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { senderLogger as logger } from "../log.js";
+import { message as RheaMessageUtil } from "rhea-promise";
+import { Constants, ErrorNameConditionMapper, RetryOperationType, retry } from "@azure/core-amqp";
+import { toRheaMessage } from "../serviceBusMessage.js";
+import { LinkEntity } from "./linkEntity.js";
+import { getUniqueName, waitForSendable, waitForTimeoutOrAbortOrResolve } from "../util/utils.js";
+import { throwErrorIfConnectionClosed } from "../util/errors.js";
+import { ServiceBusMessageBatchImpl } from "../serviceBusMessageBatch.js";
+import { ServiceBusError, translateServiceBusError } from "../serviceBusError.js";
+import { isDefined } from "@azure/core-util";
+import { defaultDataTransformer } from "../dataTransformer.js";
+/**
+ * @internal
+ * Describes the MessageSender that will send messages to ServiceBus.
+ */
+export class MessageSender extends LinkEntity {
+    constructor(identifier, connectionContext, entityPath, retryOptions) {
+        super(entityPath, entityPath, connectionContext, "sender", logger, {
+            address: entityPath,
+            audience: `${connectionContext.config.endpoint}${entityPath}`,
+        });
+        this.identifier = identifier;
+        this._retryOptions = retryOptions;
+        this._onAmqpError = (context) => {
+            const senderError = context.sender && context.sender.error;
+            logger.logError(senderError, "%s 'sender_error' event occurred on the sender '%s' with address '%s'. " +
+                "The associated error", this.logPrefix, this.name, this.address);
+            // TODO: Consider rejecting promise in trySendBatch() or createBatch()
+        };
+        this._onSessionError = (context) => {
+            const sessionError = context.session && context.session.error;
+            logger.logError(sessionError, "%s 'session_error' event occurred on the session of sender '%s' with address '%s'. " +
+                "The associated error", this.logPrefix, this.name, this.address);
+            // TODO: Consider rejecting promise in trySendBatch() or createBatch()
+        };
+        this._onAmqpClose = async (context) => {
+            const senderError = context.sender && context.sender.error;
+            logger.logError(senderError, `${this.logPrefix} 'sender_close' event occurred. The associated error is`);
+            await this.onDetached().catch((err) => {
+                logger.logError(err, `${this.logPrefix} error when closing sender after 'sender_close' event`);
+            });
+        };
+        this._onSessionClose = async (context) => {
+            const sessionError = context.session && context.session.error;
+            logger.logError(sessionError, `${this.logPrefix} 'session_close' event occurred. The associated error is`);
+            await this.onDetached().catch((err) => {
+                logger.logError(err, `${this.logPrefix} error when closing sender after 'session_close' event`);
+            });
+        };
+    }
+    _createSenderOptions(newName) {
+        if (newName)
+            this.name = getUniqueName(this.baseName);
+        const srOptions = {
+            name: this.name,
+            target: {
+                address: this.address,
+            },
+            source: this.identifier,
+            onError: this._onAmqpError,
+            onClose: this._onAmqpClose,
+            onSessionError: this._onSessionError,
+            onSessionClose: this._onSessionClose,
+        };
+        logger.verbose(`${this.logPrefix} Creating sender with options: %O`, srOptions);
+        return srOptions;
+    }
+    /**
+     * Tries to send the message to ServiceBus if there is enough credit to send them
+     * and the circular buffer has available space to settle the message after sending them.
+     *
+     * We have implemented a synchronous send over here in the sense that we shall be waiting
+     * for the message to be accepted or rejected and accordingly resolve or reject the promise.
+     *
+     * @param encodedMessage - The encoded message to be sent to ServiceBus.
+     * @param sendBatch - Boolean indicating whether the encoded message represents a batch of messages or not
+     */
+    _trySend(encodedMessage, sendBatch, options) {
+        const abortSignal = options === null || options === void 0 ? void 0 : options.abortSignal;
+        const timeoutInMs = !isDefined(this._retryOptions.timeoutInMs)
+            ? Constants.defaultOperationTimeoutInMs
+            : this._retryOptions.timeoutInMs;
+        const sendEventPromise = async () => {
+            var _a, _b, _c, _d, _e, _f, _g;
+            const initStartTime = Date.now();
+            if (!this.isOpen()) {
+                try {
+                    await waitForTimeoutOrAbortOrResolve({
+                        actionFn: () => this.open(undefined, options === null || options === void 0 ? void 0 : options.abortSignal),
+                        abortSignal: options === null || options === void 0 ? void 0 : options.abortSignal,
+                        timeoutMs: timeoutInMs,
+                        timeoutMessage: `[${this._context.connectionId}] Sender "${this.name}" ` +
+                            `with address "${this.address}", was not able to send the message right now, due ` +
+                            `to operation timeout.`,
+                    });
+                }
+                catch (err) {
+                    const translatedError = translateServiceBusError(err);
+                    logger.logError(translatedError, "%s An error occurred while creating the sender", this.logPrefix, this.name);
+                    throw translatedError;
+                }
+            }
+            const timeTakenByInit = Date.now() - initStartTime;
+            logger.verbose("%s Sender '%s', credit: %d available: %d", this.logPrefix, this.name, (_a = this.link) === null || _a === void 0 ? void 0 : _a.credit, (_d = (_c = (_b = this.link) === null || _b === void 0 ? void 0 : _b.session) === null || _c === void 0 ? void 0 : _c.outgoing) === null || _d === void 0 ? void 0 : _d.available());
+            const waitingTime = await waitForSendable(logger, this.logPrefix, this.name, timeoutInMs - timeTakenByInit, this.link, (_g = (_f = (_e = this.link) === null || _e === void 0 ? void 0 : _e.session) === null || _f === void 0 ? void 0 : _f.outgoing) === null || _g === void 0 ? void 0 : _g.available());
+            if (timeoutInMs <= timeTakenByInit + waitingTime) {
+                const desc = `${this.logPrefix} Sender "${this.name}" ` +
+                    `with address "${this.address}", was not able to send the message right now, due ` +
+                    `to operation timeout.`;
+                logger.warning(desc);
+                const e = {
+                    condition: ErrorNameConditionMapper.ServiceUnavailableError,
+                    description: desc,
+                };
+                throw translateServiceBusError(e);
+            }
+            if (!this.link) {
+                const msg = `[${this.logPrefix}] Cannot send the message. Link is not ready.`;
+                logger.warning(msg);
+                const amqpError = {
+                    condition: ErrorNameConditionMapper.SenderNotReadyError,
+                    description: msg,
+                };
+                throw translateServiceBusError(amqpError);
+            }
+            try {
+                const delivery = await this.link.send(encodedMessage, {
+                    format: sendBatch ? 0x80013700 : 0,
+                    timeoutInSeconds: (timeoutInMs - timeTakenByInit - waitingTime) / 1000,
+                    abortSignal,
+                });
+                logger.verbose("%s Sender '%s', sent message with delivery id: %d", this.logPrefix, this.name, delivery.id);
+            }
+            catch (error) {
+                const translatedError = translateServiceBusError(error.innerError || error);
+                logger.logError(translatedError, `${this.logPrefix} An error occurred while sending the message`);
+                throw translatedError;
+            }
+        };
+        const config = {
+            operation: sendEventPromise,
+            connectionId: this._context.connectionId,
+            operationType: RetryOperationType.sendMessage,
+            retryOptions: this._retryOptions,
+            abortSignal: abortSignal,
+        };
+        return retry(config);
+    }
+    createRheaLink(options) {
+        return this._context.connection.createAwaitableSender(options);
+    }
+    /**
+     * Initializes the sender session on the connection.
+     */
+    async open(options, abortSignal) {
+        try {
+            if (!options) {
+                options = this._createSenderOptions();
+            }
+            await this.initLink(options, abortSignal);
+        }
+        catch (err) {
+            const translatedError = translateServiceBusError(err);
+            logger.logError(translatedError, `${this.logPrefix} An error occurred while creating the sender`);
+            // Fix the unhelpful error messages for the OperationTimeoutError that comes from `rhea-promise`.
+            if (translatedError.code === "OperationTimeoutError") {
+                translatedError.message =
+                    "Failed to create a sender within allocated time and retry attempts.";
+            }
+            throw translatedError;
+        }
+    }
+    /**
+     * Closes the rhea link.
+     * To be called when connection is disconnected, onAmqpClose and onSessionClose events.
+     */
+    async onDetached() {
+        // Clears the token renewal timer. Closes the link and its session if they are open.
+        // Removes the link and its session if they are present in rhea's cache.
+        await this.closeLink();
+    }
+    /**
+     * Determines whether the AMQP sender link is open. If open then returns true else returns false.
+     */
+    isOpen() {
+        const result = this.link == null ? false : this.link.isOpen();
+        logger.verbose("%s Sender '%s' with address '%s' is open? -> %s", this.logPrefix, this.name, this.address, result);
+        return result;
+    }
+    /**
+     * Sends the given message, with the given options on this link
+     *
+     * @param data - Message to send. Will be sent as UTF8-encoded JSON string.
+     */
+    async send(data, options) {
+        throwErrorIfConnectionClosed(this._context);
+        try {
+            const amqpMessage = toRheaMessage(data, defaultDataTransformer);
+            // TODO: this body of logic is really similar to what's in sendMessages. Unify what we can.
+            const encodedMessage = RheaMessageUtil.encode(amqpMessage);
+            logger.verbose("%s Sender '%s', trying to send message: %O", this.logPrefix, this.name, data);
+            return await this._trySend(encodedMessage, false, options);
+        }
+        catch (err) {
+            logger.logError(err, "%s An error occurred while sending the message: %O\nError", this.logPrefix, data);
+            throw err;
+        }
+    }
+    /**
+     * Returns maximum message size on the AMQP sender link.
+     *
+     * Options to configure the `createBatch` method on the `Sender`.
+     * - `maxSizeInBytes`: The upper limit for the size of batch.
+     *
+     * Example usage:
+     * ```ts snippet:ignore
+     * {
+     *     retryOptions: { maxRetries: 5; timeoutInMs: 10 }
+     * }
+     * ```
+     */
+    async getMaxMessageSize(options = {}) {
+        const retryOptions = options.retryOptions || {};
+        if (this.isOpen()) {
+            return this.link.maxMessageSize;
+        }
+        const config = {
+            operation: async () => {
+                await this.open(undefined, options === null || options === void 0 ? void 0 : options.abortSignal);
+                if (this.link) {
+                    return this.link.maxMessageSize;
+                }
+                throw new ServiceBusError("Link failed to initialize, cannot get max message size.", "GeneralError");
+            },
+            connectionId: this._context.connectionId,
+            operationType: RetryOperationType.senderLink,
+            retryOptions: retryOptions,
+            abortSignal: options === null || options === void 0 ? void 0 : options.abortSignal,
+        };
+        return retry(config);
+    }
+    async createBatch(options) {
+        throwErrorIfConnectionClosed(this._context);
+        let maxMessageSize = await this.getMaxMessageSize({
+            retryOptions: this._retryOptions,
+            abortSignal: options === null || options === void 0 ? void 0 : options.abortSignal,
+        });
+        if (options === null || options === void 0 ? void 0 : options.maxSizeInBytes) {
+            if (options.maxSizeInBytes > maxMessageSize) {
+                const error = new Error(`Max message size (${options.maxSizeInBytes} bytes) is greater than maximum message size (${maxMessageSize} bytes) on the AMQP sender link.`);
+                throw error;
+            }
+            maxMessageSize = options.maxSizeInBytes;
+        }
+        return new ServiceBusMessageBatchImpl(this._context, maxMessageSize);
+    }
+    async sendBatch(batchMessage, options) {
+        throwErrorIfConnectionClosed(this._context);
+        try {
+            logger.verbose("%s Sender '%s', sending encoded batch message.", this.logPrefix, this.name, batchMessage);
+            return await this._trySend(batchMessage._generateMessage(), true, options);
+        }
+        catch (err) {
+            logger.logError(err, "%s Sender '%s': An error occurred while sending the messages: %O\nError", this.logPrefix, this.name, batchMessage);
+            throw err;
+        }
+    }
+    static create(identifier, context, entityPath, retryOptions) {
+        throwErrorIfConnectionClosed(context);
+        const sbSender = new MessageSender(identifier, context, entityPath, retryOptions);
+        context.senders[sbSender.name] = sbSender;
+        return sbSender;
+    }
+    removeLinkFromContext() {
+        delete this._context.senders[this.name];
+    }
+}
+//# sourceMappingURL=messageSender.js.map

package/dist/react-native/core/messageSender.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"messageSender.js","sourceRoot":"","sources":["../../../src/core/messageSender.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,YAAY,IAAI,MAAM,EAAE,MAAM,WAAW,CAAC;AAQnD,OAAO,EAAE,OAAO,IAAI,eAAe,EAAE,MAAM,cAAc,CAAC;AAO1D,OAAO,EAAE,SAAS,EAAE,wBAAwB,EAAE,kBAAkB,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AAElG,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAExD,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,8BAA8B,EAAE,MAAM,kBAAkB,CAAC;AAClG,OAAO,EAAE,4BAA4B,EAAE,MAAM,mBAAmB,CAAC;AAEjE,OAAO,EAAE,0BAA0B,EAAE,MAAM,8BAA8B,CAAC;AAI1E,OAAO,EAAE,eAAe,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AAClF,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAE/D;;;GAGG;AACH,MAAM,OAAO,aAAc,SAAQ,UAA2B;IAyB5D,YACU,UAAkB,EAC1B,iBAAoC,EACpC,UAAkB,EAClB,YAA0B;QAE1B,KAAK,CAAC,UAAU,EAAE,UAAU,EAAE,iBAAiB,EAAE,QAAQ,EAAE,MAAM,EAAE;YACjE,OAAO,EAAE,UAAU;YACnB,QAAQ,EAAE,GAAG,iBAAiB,CAAC,MAAM,CAAC,QAAQ,GAAG,UAAU,EAAE;SAC9D,CAAC,CAAC;QARK,eAAU,GAAV,UAAU,CAAQ;QAS1B,IAAI,CAAC,aAAa,GAAG,YAAY,CAAC;QAClC,IAAI,CAAC,YAAY,GAAG,CAAC,OAAqB,EAAE,EAAE;YAC5C,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;YAC3D,MAAM,CAAC,QAAQ,CACb,WAAW,EACX,yEAAyE;gBACvE,sBAAsB,EACxB,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,IAAI,EACT,IAAI,CAAC,OAAO,CACb,CAAC;YACF,sEAAsE;QACxE,CAAC,CAAC;QAEF,IAAI,CAAC,eAAe,GAAG,CAAC,OAAqB,EAAE,EAAE;YAC/C,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC;YAC9D,MAAM,CAAC,QAAQ,CACb,YAAY,EACZ,qFAAqF;gBACnF,sBAAsB,EACxB,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,IAAI,EACT,IAAI,CAAC,OAAO,CACb,CAAC;YACF,sEAAsE;QACxE,CAAC,CAAC;QAEF,IAAI,CAAC,YAAY,GAAG,KAAK,EAAE,OAAqB,EAAE,EAAE;YAClD,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;YAE3D,MAAM,CAAC,QAAQ,CACb,WAAW,EACX,GAAG,IAAI,CAAC,SAAS,yDAAyD,CAC3E,CAAC;YAEF,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACpC,MAAM,CAAC,QAAQ,CACb,GAAG,EACH,GAAG,IAAI,CAAC,SAAS,uDAAuD,CACzE,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC;QAEF,IAAI,CAAC,eAAe,GAAG,KAAK,EAAE,OAAqB,EAAE,EAAE;YACrD,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC;YAE9D,MAAM,CAAC,QAAQ,CACb,YAAY,EACZ,GAAG,IAAI,CAAC,SAAS,0DAA0D,CAC5E,CAAC;YAEF,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACpC,MAAM,CAAC,QAAQ,CACb,GAAG,EACH,GAAG,IAAI,CAAC,SAAS,wDAAwD,CAC1E,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC;IACJ,CAAC;IAEO,oBAAoB,CAAC,OAAiB;QAC5C,IAAI,OAAO;YAAE,IAAI,CAAC,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtD,MAAM,SAAS,GAA2B;YACxC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,MAAM,EAAE;gBACN,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB;YACD,MAAM,EAAE,IAAI,CAAC,UAAU;YACvB,OAAO,EAAE,IAAI,CAAC,YAAY;YAC1B,OAAO,EAAE,IAAI,CAAC,YAAY;YAC1B,cAAc,EAAE,IAAI,CAAC,eAAe;YACpC,cAAc,EAAE,IAAI,CAAC,eAAe;SACrC,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,SAAS,mCAAmC,EAAE,SAAS,CAAC,CAAC;QAChF,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;;;;;;;OASG;IACK,QAAQ,CACd,cAAsB,EACtB,SAAkB,EAClB,OAAyC;QAEzC,MAAM,WAAW,GAAG,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW,CAAC;QACzC,MAAM,WAAW,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC;YAC5D,CAAC,CAAC,SAAS,CAAC,2BAA2B;YACvC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC;QAEnC,MAAM,gBAAgB,GAAG,KAAK,IAAmB,EAAE;;YACjD,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACjC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC;gBACnB,IAAI,CAAC;oBACH,MAAM,8BAA8B,CAAC;wBACnC,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW,CAAC;wBAC1D,WAAW,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW;wBACjC,SAAS,EAAE,WAAW;wBACtB,cAAc,EACZ,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,aAAa,IAAI,CAAC,IAAI,IAAI;4BACxD,iBAAiB,IAAI,CAAC,OAAO,qDAAqD;4BAClF,uBAAuB;qBAC1B,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,GAAQ,EAAE,CAAC;oBAClB,MAAM,eAAe,GAAG,wBAAwB,CAAC,GAAG,CAAC,CAAC;oBACtD,MAAM,CAAC,QAAQ,CACb,eAAe,EACf,gDAAgD,EAChD,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,IAAI,CACV,CAAC;oBACF,MAAM,eAAe,CAAC;gBACxB,CAAC;YACH,CAAC;YAED,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,aAAa,CAAC;YAEnD,MAAM,CAAC,OAAO,CACZ,0CAA0C,EAC1C,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,IAAI,EACT,MAAA,IAAI,CAAC,IAAI,0CAAE,MAAM,EACjB,MAAA,MAAA,MAAA,IAAI,CAAC,IAAI,0CAAE,OAAO,0CAAE,QAAQ,0CAAE,SAAS,EAAE,CAC1C,CAAC;YAEF,MAAM,WAAW,GAAG,MAAM,eAAe,CACvC,MAAM,EACN,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,IAAI,EACT,WAAW,GAAG,eAAe,EAC7B,IAAI,CAAC,IAAI,EACT,MAAA,MAAA,MAAA,IAAI,CAAC,IAAI,0CAAE,OAAO,0CAAE,QAAQ,0CAAE,SAAS,EAAE,CAC1C,CAAC;YAEF,IAAI,WAAW,IAAI,eAAe,GAAG,WAAW,EAAE,CAAC;gBACjD,MAAM,IAAI,GACR,GAAG,IAAI,CAAC,SAAS,YAAY,IAAI,CAAC,IAAI,IAAI;oBAC1C,iBAAiB,IAAI,CAAC,OAAO,qDAAqD;oBAClF,uBAAuB,CAAC;gBAC1B,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBACrB,MAAM,CAAC,GAAc;oBACnB,SAAS,EAAE,wBAAwB,CAAC,uBAAuB;oBAC3D,WAAW,EAAE,IAAI;iBAClB,CAAC;gBACF,MAAM,wBAAwB,CAAC,CAAC,CAAC,CAAC;YACpC,CAAC;YAED,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACf,MAAM,GAAG,GAAG,IAAI,IAAI,CAAC,SAAS,+CAA+C,CAAC;gBAC9E,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;gBACpB,MAAM,SAAS,GAAc;oBAC3B,SAAS,EAAE,wBAAwB,CAAC,mBAAmB;oBACvD,WAAW,EAAE,GAAG;iBACjB,CAAC;gBACF,MAAM,wBAAwB,CAAC,SAAS,CAAC,CAAC;YAC5C,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE;oBACpD,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;oBAClC,gBAAgB,EAAE,CAAC,WAAW,GAAG,eAAe,GAAG,WAAW,CAAC,GAAG,IAAI;oBACtE,WAAW;iBACZ,CAAC,CAAC;gBACH,MAAM,CAAC,OAAO,CACZ,mDAAmD,EACnD,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,IAAI,EACT,QAAQ,CAAC,EAAE,CACZ,CAAC;YACJ,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,MAAM,eAAe,GAAG,wBAAwB,CAAC,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,CAAC;gBAC5E,MAAM,CAAC,QAAQ,CACb,eAAe,EACf,GAAG,IAAI,CAAC,SAAS,8CAA8C,CAChE,CAAC;gBACF,MAAM,eAAe,CAAC;YACxB,CAAC;QACH,CAAC,CAAC;QACF,MAAM,MAAM,GAAsB;YAChC,SAAS,EAAE,gBAAgB;YAC3B,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAa;YACzC,aAAa,EAAE,kBAAkB,CAAC,WAAW;YAC7C,YAAY,EAAE,IAAI,CAAC,aAAa;YAChC,WAAW,EAAE,WAAW;SACzB,CAAC;QAEF,OAAO,KAAK,CAAO,MAAM,CAAC,CAAC;IAC7B,CAAC;IAES,cAAc,CAAC,OAA+B;QACtD,OAAO,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;IACjE,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,IAAI,CACf,OAAgC,EAChC,WAA6B;QAE7B,IAAI,CAAC;YACH,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAC;YACxC,CAAC;YACD,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAC5C,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,eAAe,GAAG,wBAAwB,CAAC,GAAG,CAAC,CAAC;YACtD,MAAM,CAAC,QAAQ,CACb,eAAe,EACf,GAAG,IAAI,CAAC,SAAS,8CAA8C,CAChE,CAAC;YACF,iGAAiG;YACjG,IAAK,eAAkC,CAAC,IAAI,KAAK,uBAAuB,EAAE,CAAC;gBACzE,eAAe,CAAC,OAAO;oBACrB,qEAAqE,CAAC;YAC1E,CAAC;YACD,MAAM,eAAe,CAAC;QACxB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,UAAU;QACd,oFAAoF;QACpF,wEAAwE;QACxE,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,MAAM,MAAM,GAAY,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QACvE,MAAM,CAAC,OAAO,CACZ,iDAAiD,EACjD,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,IAAI,EACT,IAAI,CAAC,OAAO,EACZ,MAAM,CACP,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,IAAI,CACR,IAA8C,EAC9C,OAA8B;QAE9B,4BAA4B,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,aAAa,CAAC,IAAI,EAAE,sBAAsB,CAAC,CAAC;YAEhE,2FAA2F;YAC3F,MAAM,cAAc,GAAG,eAAe,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAC3D,MAAM,CAAC,OAAO,CAAC,4CAA4C,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC9F,OAAO,MAAM,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC7D,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,CAAC,QAAQ,CACb,GAAG,EACH,2DAA2D,EAC3D,IAAI,CAAC,SAAS,EACd,IAAI,CACL,CAAC;YACF,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;;;;;;;;;;;OAYG;IACH,KAAK,CAAC,iBAAiB,CACrB,UAEgD,EAAE;QAElD,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,EAAE,CAAC;QAChD,IAAI,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC,IAAK,CAAC,cAAc,CAAC;QACnC,CAAC;QAED,MAAM,MAAM,GAAwB;YAClC,SAAS,EAAE,KAAK,IAAI,EAAE;gBACpB,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW,CAAC,CAAC;gBAEjD,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;oBACd,OAAO,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC;gBAClC,CAAC;gBAED,MAAM,IAAI,eAAe,CACvB,yDAAyD,EACzD,cAAc,CACf,CAAC;YACJ,CAAC;YACD,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY;YACxC,aAAa,EAAE,kBAAkB,CAAC,UAAU;YAC5C,YAAY,EAAE,YAAY;YAC1B,WAAW,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW;SAClC,CAAC;QAEF,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAAmC;QACnD,4BAA4B,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,cAAc,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC;YAChD,YAAY,EAAE,IAAI,CAAC,aAAa;YAChC,WAAW,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW;SAClC,CAAC,CAAC;QACH,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc,EAAE,CAAC;YAC5B,IAAI,OAAO,CAAC,cAAc,GAAG,cAAe,EAAE,CAAC;gBAC7C,MAAM,KAAK,GAAG,IAAI,KAAK,CACrB,qBAAqB,OAAO,CAAC,cAAc,iDAAiD,cAAc,kCAAkC,CAC7I,CAAC;gBACF,MAAM,KAAK,CAAC;YACd,CAAC;YACD,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;QAC1C,CAAC;QACD,OAAO,IAAI,0BAA0B,CAAC,IAAI,CAAC,QAAQ,EAAE,cAAe,CAAC,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,SAAS,CACb,YAAoC,EACpC,OAA8B;QAE9B,4BAA4B,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,CAAC;YACH,MAAM,CAAC,OAAO,CACZ,gDAAgD,EAChD,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,IAAI,EACT,YAAY,CACb,CAAC;YACF,OAAO,MAAM,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,gBAAgB,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QAC7E,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,CAAC,QAAQ,CACb,GAAG,EACH,yEAAyE,EACzE,IAAI,CAAC,SAAS,EACd,IAAI,CAAC,IAAI,EACT,YAAY,CACb,CAAC;YACF,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED,MAAM,CAAC,MAAM,CACX,UAAkB,EAClB,OAA0B,EAC1B,UAAkB,EAClB,YAA0B;QAE1B,4BAA4B,CAAC,OAAO,CAAC,CAAC;QAEtC,MAAM,QAAQ,GAAG,IAAI,aAAa,CAAC,UAAU,EAAE,OAAO,EAAE,UAAU,EAAE,YAAY,CAAC,CAAC;QAClF,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC;QAC1C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAES,qBAAqB;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1C,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { senderLogger as logger } from \"../log.js\";\nimport type {\n  AmqpError,\n  AwaitableSender,\n  AwaitableSenderOptions,\n  EventContext,\n  OnAmqpEvent,\n} from \"rhea-promise\";\nimport { message as RheaMessageUtil } from \"rhea-promise\";\nimport type {\n  MessagingError,\n  RetryConfig,\n  RetryOptions,\n  AmqpAnnotatedMessage,\n} from \"@azure/core-amqp\";\nimport { Constants, ErrorNameConditionMapper, RetryOperationType, retry } from \"@azure/core-amqp\";\nimport type { ServiceBusMessage } from \"../serviceBusMessage.js\";\nimport { toRheaMessage } from \"../serviceBusMessage.js\";\nimport type { ConnectionContext } from \"../connectionContext.js\";\nimport { LinkEntity } from \"./linkEntity.js\";\nimport { getUniqueName, waitForSendable, waitForTimeoutOrAbortOrResolve } from \"../util/utils.js\";\nimport { throwErrorIfConnectionClosed } from \"../util/errors.js\";\nimport type { ServiceBusMessageBatch } from \"../serviceBusMessageBatch.js\";\nimport { ServiceBusMessageBatchImpl } from \"../serviceBusMessageBatch.js\";\nimport type { CreateMessageBatchOptions } from \"../models.js\";\nimport type { OperationOptionsBase } from \"../modelsToBeSharedWithEventHubs.js\";\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport { ServiceBusError, translateServiceBusError } from \"../serviceBusError.js\";\nimport { isDefined } from \"@azure/core-util\";\nimport { defaultDataTransformer } from \"../dataTransformer.js\";\n\n/**\n * @internal\n * Describes the MessageSender that will send messages to ServiceBus.\n */\nexport class MessageSender extends LinkEntity<AwaitableSender> {\n  /**\n   * The handler function to handle errors that happen on the\n   * underlying sender.\n   * @readonly\n   */\n  private readonly _onAmqpError: OnAmqpEvent;\n  /**\n   * The handler function to handle \"sender_close\" event\n   * that happens on the underlying sender.\n   * @readonly\n   */\n  private readonly _onAmqpClose: OnAmqpEvent;\n  /**\n   * The message handler that will be set as the handler on\n   * the underlying rhea sender's session for the \"session_error\" event.\n   */\n  private _onSessionError: OnAmqpEvent;\n  /**\n   * The message handler that will be set as the handler on\n   * the underlying rhea sender's session for the \"session_close\" event.\n   */\n  private _onSessionClose: OnAmqpEvent;\n  private _retryOptions: RetryOptions;\n\n  constructor(\n    private identifier: string,\n    connectionContext: ConnectionContext,\n    entityPath: string,\n    retryOptions: RetryOptions,\n  ) {\n    super(entityPath, entityPath, connectionContext, \"sender\", logger, {\n      address: entityPath,\n      audience: `${connectionContext.config.endpoint}${entityPath}`,\n    });\n    this._retryOptions = retryOptions;\n    this._onAmqpError = (context: EventContext) => {\n      const senderError = context.sender && context.sender.error;\n      logger.logError(\n        senderError,\n        \"%s 'sender_error' event occurred on the sender '%s' with address '%s'. \" +\n          \"The associated error\",\n        this.logPrefix,\n        this.name,\n        this.address,\n      );\n      // TODO: Consider rejecting promise in trySendBatch() or createBatch()\n    };\n\n    this._onSessionError = (context: EventContext) => {\n      const sessionError = context.session && context.session.error;\n      logger.logError(\n        sessionError,\n        \"%s 'session_error' event occurred on the session of sender '%s' with address '%s'. \" +\n          \"The associated error\",\n        this.logPrefix,\n        this.name,\n        this.address,\n      );\n      // TODO: Consider rejecting promise in trySendBatch() or createBatch()\n    };\n\n    this._onAmqpClose = async (context: EventContext) => {\n      const senderError = context.sender && context.sender.error;\n\n      logger.logError(\n        senderError,\n        `${this.logPrefix} 'sender_close' event occurred. The associated error is`,\n      );\n\n      await this.onDetached().catch((err) => {\n        logger.logError(\n          err,\n          `${this.logPrefix} error when closing sender after 'sender_close' event`,\n        );\n      });\n    };\n\n    this._onSessionClose = async (context: EventContext) => {\n      const sessionError = context.session && context.session.error;\n\n      logger.logError(\n        sessionError,\n        `${this.logPrefix} 'session_close' event occurred. The associated error is`,\n      );\n\n      await this.onDetached().catch((err) => {\n        logger.logError(\n          err,\n          `${this.logPrefix} error when closing sender after 'session_close' event`,\n        );\n      });\n    };\n  }\n\n  private _createSenderOptions(newName?: boolean): AwaitableSenderOptions {\n    if (newName) this.name = getUniqueName(this.baseName);\n    const srOptions: AwaitableSenderOptions = {\n      name: this.name,\n      target: {\n        address: this.address,\n      },\n      source: this.identifier,\n      onError: this._onAmqpError,\n      onClose: this._onAmqpClose,\n      onSessionError: this._onSessionError,\n      onSessionClose: this._onSessionClose,\n    };\n    logger.verbose(`${this.logPrefix} Creating sender with options: %O`, srOptions);\n    return srOptions;\n  }\n\n  /**\n   * Tries to send the message to ServiceBus if there is enough credit to send them\n   * and the circular buffer has available space to settle the message after sending them.\n   *\n   * We have implemented a synchronous send over here in the sense that we shall be waiting\n   * for the message to be accepted or rejected and accordingly resolve or reject the promise.\n   *\n   * @param encodedMessage - The encoded message to be sent to ServiceBus.\n   * @param sendBatch - Boolean indicating whether the encoded message represents a batch of messages or not\n   */\n  private _trySend(\n    encodedMessage: Buffer,\n    sendBatch: boolean,\n    options: OperationOptionsBase | undefined,\n  ): Promise<void> {\n    const abortSignal = options?.abortSignal;\n    const timeoutInMs = !isDefined(this._retryOptions.timeoutInMs)\n      ? Constants.defaultOperationTimeoutInMs\n      : this._retryOptions.timeoutInMs;\n\n    const sendEventPromise = async (): Promise<void> => {\n      const initStartTime = Date.now();\n      if (!this.isOpen()) {\n        try {\n          await waitForTimeoutOrAbortOrResolve({\n            actionFn: () => this.open(undefined, options?.abortSignal),\n            abortSignal: options?.abortSignal,\n            timeoutMs: timeoutInMs,\n            timeoutMessage:\n              `[${this._context.connectionId}] Sender \"${this.name}\" ` +\n              `with address \"${this.address}\", was not able to send the message right now, due ` +\n              `to operation timeout.`,\n          });\n        } catch (err: any) {\n          const translatedError = translateServiceBusError(err);\n          logger.logError(\n            translatedError,\n            \"%s An error occurred while creating the sender\",\n            this.logPrefix,\n            this.name,\n          );\n          throw translatedError;\n        }\n      }\n\n      const timeTakenByInit = Date.now() - initStartTime;\n\n      logger.verbose(\n        \"%s Sender '%s', credit: %d available: %d\",\n        this.logPrefix,\n        this.name,\n        this.link?.credit,\n        this.link?.session?.outgoing?.available(),\n      );\n\n      const waitingTime = await waitForSendable(\n        logger,\n        this.logPrefix,\n        this.name,\n        timeoutInMs - timeTakenByInit,\n        this.link,\n        this.link?.session?.outgoing?.available(),\n      );\n\n      if (timeoutInMs <= timeTakenByInit + waitingTime) {\n        const desc: string =\n          `${this.logPrefix} Sender \"${this.name}\" ` +\n          `with address \"${this.address}\", was not able to send the message right now, due ` +\n          `to operation timeout.`;\n        logger.warning(desc);\n        const e: AmqpError = {\n          condition: ErrorNameConditionMapper.ServiceUnavailableError,\n          description: desc,\n        };\n        throw translateServiceBusError(e);\n      }\n\n      if (!this.link) {\n        const msg = `[${this.logPrefix}] Cannot send the message. Link is not ready.`;\n        logger.warning(msg);\n        const amqpError: AmqpError = {\n          condition: ErrorNameConditionMapper.SenderNotReadyError,\n          description: msg,\n        };\n        throw translateServiceBusError(amqpError);\n      }\n      try {\n        const delivery = await this.link.send(encodedMessage, {\n          format: sendBatch ? 0x80013700 : 0,\n          timeoutInSeconds: (timeoutInMs - timeTakenByInit - waitingTime) / 1000,\n          abortSignal,\n        });\n        logger.verbose(\n          \"%s Sender '%s', sent message with delivery id: %d\",\n          this.logPrefix,\n          this.name,\n          delivery.id,\n        );\n      } catch (error: any) {\n        const translatedError = translateServiceBusError(error.innerError || error);\n        logger.logError(\n          translatedError,\n          `${this.logPrefix} An error occurred while sending the message`,\n        );\n        throw translatedError;\n      }\n    };\n    const config: RetryConfig<void> = {\n      operation: sendEventPromise,\n      connectionId: this._context.connectionId!,\n      operationType: RetryOperationType.sendMessage,\n      retryOptions: this._retryOptions,\n      abortSignal: abortSignal,\n    };\n\n    return retry<void>(config);\n  }\n\n  protected createRheaLink(options: AwaitableSenderOptions): Promise<AwaitableSender> {\n    return this._context.connection.createAwaitableSender(options);\n  }\n\n  /**\n   * Initializes the sender session on the connection.\n   */\n  public async open(\n    options?: AwaitableSenderOptions,\n    abortSignal?: AbortSignalLike,\n  ): Promise<void> {\n    try {\n      if (!options) {\n        options = this._createSenderOptions();\n      }\n      await this.initLink(options, abortSignal);\n    } catch (err: any) {\n      const translatedError = translateServiceBusError(err);\n      logger.logError(\n        translatedError,\n        `${this.logPrefix} An error occurred while creating the sender`,\n      );\n      // Fix the unhelpful error messages for the OperationTimeoutError that comes from `rhea-promise`.\n      if ((translatedError as MessagingError).code === \"OperationTimeoutError\") {\n        translatedError.message =\n          \"Failed to create a sender within allocated time and retry attempts.\";\n      }\n      throw translatedError;\n    }\n  }\n\n  /**\n   * Closes the rhea link.\n   * To be called when connection is disconnected, onAmqpClose and onSessionClose events.\n   */\n  async onDetached(): Promise<void> {\n    // Clears the token renewal timer. Closes the link and its session if they are open.\n    // Removes the link and its session if they are present in rhea's cache.\n    await this.closeLink();\n  }\n\n  /**\n   * Determines whether the AMQP sender link is open. If open then returns true else returns false.\n   */\n  isOpen(): boolean {\n    const result: boolean = this.link == null ? false : this.link.isOpen();\n    logger.verbose(\n      \"%s Sender '%s' with address '%s' is open? -> %s\",\n      this.logPrefix,\n      this.name,\n      this.address,\n      result,\n    );\n    return result;\n  }\n\n  /**\n   * Sends the given message, with the given options on this link\n   *\n   * @param data - Message to send. Will be sent as UTF8-encoded JSON string.\n   */\n  async send(\n    data: ServiceBusMessage | AmqpAnnotatedMessage,\n    options?: OperationOptionsBase,\n  ): Promise<void> {\n    throwErrorIfConnectionClosed(this._context);\n    try {\n      const amqpMessage = toRheaMessage(data, defaultDataTransformer);\n\n      // TODO: this body of logic is really similar to what's in sendMessages. Unify what we can.\n      const encodedMessage = RheaMessageUtil.encode(amqpMessage);\n      logger.verbose(\"%s Sender '%s', trying to send message: %O\", this.logPrefix, this.name, data);\n      return await this._trySend(encodedMessage, false, options);\n    } catch (err: any) {\n      logger.logError(\n        err,\n        \"%s An error occurred while sending the message: %O\\nError\",\n        this.logPrefix,\n        data,\n      );\n      throw err;\n    }\n  }\n\n  /**\n   * Returns maximum message size on the AMQP sender link.\n   *\n   * Options to configure the `createBatch` method on the `Sender`.\n   * - `maxSizeInBytes`: The upper limit for the size of batch.\n   *\n   * Example usage:\n   * ```ts snippet:ignore\n   * {\n   *     retryOptions: { maxRetries: 5; timeoutInMs: 10 }\n   * }\n   * ```\n   */\n  async getMaxMessageSize(\n    options: {\n      retryOptions?: RetryOptions;\n    } & Pick<OperationOptionsBase, \"abortSignal\"> = {},\n  ): Promise<number> {\n    const retryOptions = options.retryOptions || {};\n    if (this.isOpen()) {\n      return this.link!.maxMessageSize;\n    }\n\n    const config: RetryConfig<number> = {\n      operation: async () => {\n        await this.open(undefined, options?.abortSignal);\n\n        if (this.link) {\n          return this.link.maxMessageSize;\n        }\n\n        throw new ServiceBusError(\n          \"Link failed to initialize, cannot get max message size.\",\n          \"GeneralError\",\n        );\n      },\n      connectionId: this._context.connectionId,\n      operationType: RetryOperationType.senderLink,\n      retryOptions: retryOptions,\n      abortSignal: options?.abortSignal,\n    };\n\n    return retry(config);\n  }\n\n  async createBatch(options?: CreateMessageBatchOptions): Promise<ServiceBusMessageBatch> {\n    throwErrorIfConnectionClosed(this._context);\n    let maxMessageSize = await this.getMaxMessageSize({\n      retryOptions: this._retryOptions,\n      abortSignal: options?.abortSignal,\n    });\n    if (options?.maxSizeInBytes) {\n      if (options.maxSizeInBytes > maxMessageSize!) {\n        const error = new Error(\n          `Max message size (${options.maxSizeInBytes} bytes) is greater than maximum message size (${maxMessageSize} bytes) on the AMQP sender link.`,\n        );\n        throw error;\n      }\n      maxMessageSize = options.maxSizeInBytes;\n    }\n    return new ServiceBusMessageBatchImpl(this._context, maxMessageSize!);\n  }\n\n  async sendBatch(\n    batchMessage: ServiceBusMessageBatch,\n    options?: OperationOptionsBase,\n  ): Promise<void> {\n    throwErrorIfConnectionClosed(this._context);\n    try {\n      logger.verbose(\n        \"%s Sender '%s', sending encoded batch message.\",\n        this.logPrefix,\n        this.name,\n        batchMessage,\n      );\n      return await this._trySend(batchMessage._generateMessage(), true, options);\n    } catch (err: any) {\n      logger.logError(\n        err,\n        \"%s Sender '%s': An error occurred while sending the messages: %O\\nError\",\n        this.logPrefix,\n        this.name,\n        batchMessage,\n      );\n      throw err;\n    }\n  }\n\n  static create(\n    identifier: string,\n    context: ConnectionContext,\n    entityPath: string,\n    retryOptions: RetryOptions,\n  ): MessageSender {\n    throwErrorIfConnectionClosed(context);\n\n    const sbSender = new MessageSender(identifier, context, entityPath, retryOptions);\n    context.senders[sbSender.name] = sbSender;\n    return sbSender;\n  }\n\n  protected removeLinkFromContext(): void {\n    delete this._context.senders[this.name];\n  }\n}\n"]}

package/dist/react-native/core/receiverHelper.d.ts

@@ -0,0 +1,45 @@
+import type { Receiver } from "rhea-promise";
+/**
+ * Wraps the receiver with some higher level operations for managing state
+ * like credits, draining, etc...
+ *
+ * @internal
+ */
+export declare class ReceiverHelper {
+    private _getCurrentReceiver;
+    private _isSuspended;
+    constructor(_getCurrentReceiver: () => {
+        receiver: Receiver | undefined;
+        logPrefix: string;
+    });
+    private _getCurrentReceiverOrError;
+    /**
+     * Adds credits to the receiver, respecting any state that
+     * indicates the receiver is closed or should not continue
+     * to receive more messages.
+     *
+     * @param credits - Number of credits to add.
+     * or `stopReceivingMessages` has been called.
+     */
+    addCredit(credits: number): void;
+    /**
+     * Drains the credits for the receiver and prevents the `receiverHelper.addCredit()` method from adding credits.
+     * Call `resume()` to enable the `addCredit()` method.
+     */
+    suspend(): Promise<void>;
+    /**
+     * Resets tracking so `addCredit` works again by toggling the `_isSuspended` flag.
+     */
+    resume(): void;
+    isSuspended(): boolean;
+    /**
+     * Initiates a drain for the current receiver and resolves when
+     * the drain has completed.
+     *
+     * NOTE: This method returns immediately if the receiver is not valid or if there
+     * are no pending credits on the receiver (ie: `receiver.credit === 0`).
+     */
+    drain(): Promise<void>;
+    private _isValidReceiver;
+}
+//# sourceMappingURL=receiverHelper.d.ts.map

package/dist/react-native/core/receiverHelper.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"receiverHelper.d.ts","sourceRoot":"","sources":["../../../src/core/receiverHelper.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAM7C;;;;;GAKG;AACH,qBAAa,cAAc;IAIvB,OAAO,CAAC,mBAAmB;IAH7B,OAAO,CAAC,YAAY,CAAiB;gBAG3B,mBAAmB,EAAE,MAAM;QAAE,QAAQ,EAAE,QAAQ,GAAG,SAAS,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE;IAG1F,OAAO,CAAC,0BAA0B;IAsBlC;;;;;;;OAOG;IACH,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAqBhC;;;OAGG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAgB9B;;OAEG;IACH,MAAM,IAAI,IAAI;IAId,WAAW,IAAI,OAAO;IAItB;;;;;;OAMG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAqC5B,OAAO,CAAC,gBAAgB;CAGzB"}

package/dist/react-native/core/receiverHelper.js

@@ -0,0 +1,118 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { AbortError } from "@azure/abort-controller";
+import { ReceiverEvents } from "rhea-promise";
+import { receiverLogger as logger } from "../log.js";
+import { ServiceBusError } from "../serviceBusError.js";
+import { receiveDrainTimeoutInMs } from "../util/constants.js";
+/**
+ * Wraps the receiver with some higher level operations for managing state
+ * like credits, draining, etc...
+ *
+ * @internal
+ */
+export class ReceiverHelper {
+    constructor(_getCurrentReceiver) {
+        this._getCurrentReceiver = _getCurrentReceiver;
+        this._isSuspended = true;
+    }
+    _getCurrentReceiverOrError() {
+        const currentReceiverData = this._getCurrentReceiver();
+        if (currentReceiverData.receiver == null) {
+            return "is undefined";
+        }
+        if (!currentReceiverData.receiver.isOpen()) {
+            return "is not open";
+        }
+        if (this._isSuspended) {
+            return "is suspended";
+        }
+        return currentReceiverData;
+    }
+    /**
+     * Adds credits to the receiver, respecting any state that
+     * indicates the receiver is closed or should not continue
+     * to receive more messages.
+     *
+     * @param credits - Number of credits to add.
+     * or `stopReceivingMessages` has been called.
+     */
+    addCredit(credits) {
+        const currentReceiverOrError = this._getCurrentReceiverOrError();
+        if (typeof currentReceiverOrError === "string") {
+            const errorMessage = `Cannot request messages on the receiver since it ${currentReceiverOrError}.`;
+            if (currentReceiverOrError === "is suspended") {
+                // if a user has suspended the receiver we should consider this a non-retryable
+                // error since it absolutely requires user intervention.
+                throw new AbortError(errorMessage);
+            }
+            throw new ServiceBusError(errorMessage, "GeneralError");
+        }
+        if (currentReceiverOrError.receiver != null) {
+            logger.verbose(`${currentReceiverOrError.logPrefix} Adding ${credits} credits`);
+            currentReceiverOrError.receiver.addCredit(credits);
+        }
+    }
+    /**
+     * Drains the credits for the receiver and prevents the `receiverHelper.addCredit()` method from adding credits.
+     * Call `resume()` to enable the `addCredit()` method.
+     */
+    async suspend() {
+        const { receiver, logPrefix } = this._getCurrentReceiver();
+        this._isSuspended = true;
+        if (!this._isValidReceiver(receiver)) {
+            return;
+        }
+        logger.verbose(`${logPrefix} User has requested to stop receiving new messages, attempting to drain.`);
+        return this.drain();
+    }
+    /**
+     * Resets tracking so `addCredit` works again by toggling the `_isSuspended` flag.
+     */
+    resume() {
+        this._isSuspended = false;
+    }
+    isSuspended() {
+        return this._isSuspended;
+    }
+    /**
+     * Initiates a drain for the current receiver and resolves when
+     * the drain has completed.
+     *
+     * NOTE: This method returns immediately if the receiver is not valid or if there
+     * are no pending credits on the receiver (ie: `receiver.credit === 0`).
+     */
+    async drain() {
+        const { receiver, logPrefix } = this._getCurrentReceiver();
+        if (!this._isValidReceiver(receiver)) {
+            // TODO: should we throw?
+            return;
+        }
+        if (receiver.credit === 0) {
+            // nothing to drain
+            return;
+        }
+        logger.verbose(`${logPrefix} Receiver is starting drain. Remaining credits; ${receiver.credit}`);
+        const drainPromise = new Promise((resolve) => {
+            const timer = setTimeout(async () => {
+                logger.warning(`${logPrefix} Time out when draining credits in suspend().`);
+                // Close the receiver link since we have not received the receiver_drained event
+                // to prevent out-of-sync link state between local and remote
+                await (receiver === null || receiver === void 0 ? void 0 : receiver.close());
+                resolve();
+            }, receiveDrainTimeoutInMs);
+            receiver.once(ReceiverEvents.receiverDrained, () => {
+                logger.verbose(`${logPrefix} Receiver has been drained.`);
+                receiver.drain = false;
+                clearTimeout(timer);
+                resolve();
+            });
+            receiver.drainCredit();
+        });
+        return drainPromise;
+    }
+    _isValidReceiver(receiver) {
+        return receiver != null && receiver.isOpen();
+    }
+}
+//# sourceMappingURL=receiverHelper.js.map

package/dist/react-native/core/receiverHelper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"receiverHelper.js","sourceRoot":"","sources":["../../../src/core/receiverHelper.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAErD,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,cAAc,IAAI,MAAM,EAAE,MAAM,WAAW,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAE/D;;;;;GAKG;AACH,MAAM,OAAO,cAAc;IAGzB,YACU,mBAAgF;QAAhF,wBAAmB,GAAnB,mBAAmB,CAA6D;QAHlF,iBAAY,GAAY,IAAI,CAAC;IAIlC,CAAC;IAEI,0BAA0B;QAKhC,MAAM,mBAAmB,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAEvD,IAAI,mBAAmB,CAAC,QAAQ,IAAI,IAAI,EAAE,CAAC;YACzC,OAAO,cAAc,CAAC;QACxB,CAAC;QAED,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,OAAO,aAAa,CAAC;QACvB,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,OAAO,cAAc,CAAC;QACxB,CAAC;QAED,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED;;;;;;;OAOG;IACH,SAAS,CAAC,OAAe;QACvB,MAAM,sBAAsB,GAAG,IAAI,CAAC,0BAA0B,EAAE,CAAC;QAEjE,IAAI,OAAO,sBAAsB,KAAK,QAAQ,EAAE,CAAC;YAC/C,MAAM,YAAY,GAAG,oDAAoD,sBAAsB,GAAG,CAAC;YAEnG,IAAI,sBAAsB,KAAK,cAAc,EAAE,CAAC;gBAC9C,+EAA+E;gBAC/E,wDAAwD;gBACxD,MAAM,IAAI,UAAU,CAAC,YAAY,CAAC,CAAC;YACrC,CAAC;YAED,MAAM,IAAI,eAAe,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC;QAC1D,CAAC;QAED,IAAI,sBAAsB,CAAC,QAAQ,IAAI,IAAI,EAAE,CAAC;YAC5C,MAAM,CAAC,OAAO,CAAC,GAAG,sBAAsB,CAAC,SAAS,WAAW,OAAO,UAAU,CAAC,CAAC;YAChF,sBAAsB,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,OAAO;QACX,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAE3D,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAEzB,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,CAAC;YACrC,OAAO;QACT,CAAC;QAED,MAAM,CAAC,OAAO,CACZ,GAAG,SAAS,0EAA0E,CACvF,CAAC;QAEF,OAAO,IAAI,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,MAAM;QACJ,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;IAC5B,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,KAAK;QACT,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAE3D,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,CAAC;YACrC,yBAAyB;YACzB,OAAO;QACT,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,mBAAmB;YACnB,OAAO;QACT,CAAC;QAED,MAAM,CAAC,OAAO,CACZ,GAAG,SAAS,mDAAmD,QAAQ,CAAC,MAAM,EAAE,CACjF,CAAC;QAEF,MAAM,YAAY,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YACjD,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,IAAI,EAAE;gBAClC,MAAM,CAAC,OAAO,CAAC,GAAG,SAAS,+CAA+C,CAAC,CAAC;gBAC5E,gFAAgF;gBAChF,6DAA6D;gBAC7D,MAAM,CAAA,QAAQ,aAAR,QAAQ,uBAAR,QAAQ,CAAE,KAAK,EAAE,CAAA,CAAC;gBACxB,OAAO,EAAE,CAAC;YACZ,CAAC,EAAE,uBAAuB,CAAC,CAAC;YAC5B,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,eAAe,EAAE,GAAG,EAAE;gBACjD,MAAM,CAAC,OAAO,CAAC,GAAG,SAAS,6BAA6B,CAAC,CAAC;gBAC1D,QAAQ,CAAC,KAAK,GAAG,KAAK,CAAC;gBACvB,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpB,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;YACH,QAAQ,CAAC,WAAW,EAAE,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,OAAO,YAAY,CAAC;IACtB,CAAC;IAEO,gBAAgB,CAAC,QAA8B;QACrD,OAAO,QAAQ,IAAI,IAAI,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;IAC/C,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { AbortError } from \"@azure/abort-controller\";\nimport type { Receiver } from \"rhea-promise\";\nimport { ReceiverEvents } from \"rhea-promise\";\nimport { receiverLogger as logger } from \"../log.js\";\nimport { ServiceBusError } from \"../serviceBusError.js\";\nimport { receiveDrainTimeoutInMs } from \"../util/constants.js\";\n\n/**\n * Wraps the receiver with some higher level operations for managing state\n * like credits, draining, etc...\n *\n * @internal\n */\nexport class ReceiverHelper {\n  private _isSuspended: boolean = true;\n\n  constructor(\n    private _getCurrentReceiver: () => { receiver: Receiver | undefined; logPrefix: string },\n  ) {}\n\n  private _getCurrentReceiverOrError():\n    | \"is undefined\"\n    | \"is not open\"\n    | \"is suspended\"\n    | { receiver: Receiver | undefined; logPrefix: string } {\n    const currentReceiverData = this._getCurrentReceiver();\n\n    if (currentReceiverData.receiver == null) {\n      return \"is undefined\";\n    }\n\n    if (!currentReceiverData.receiver.isOpen()) {\n      return \"is not open\";\n    }\n\n    if (this._isSuspended) {\n      return \"is suspended\";\n    }\n\n    return currentReceiverData;\n  }\n\n  /**\n   * Adds credits to the receiver, respecting any state that\n   * indicates the receiver is closed or should not continue\n   * to receive more messages.\n   *\n   * @param credits - Number of credits to add.\n   * or `stopReceivingMessages` has been called.\n   */\n  addCredit(credits: number): void {\n    const currentReceiverOrError = this._getCurrentReceiverOrError();\n\n    if (typeof currentReceiverOrError === \"string\") {\n      const errorMessage = `Cannot request messages on the receiver since it ${currentReceiverOrError}.`;\n\n      if (currentReceiverOrError === \"is suspended\") {\n        // if a user has suspended the receiver we should consider this a non-retryable\n        // error since it absolutely requires user intervention.\n        throw new AbortError(errorMessage);\n      }\n\n      throw new ServiceBusError(errorMessage, \"GeneralError\");\n    }\n\n    if (currentReceiverOrError.receiver != null) {\n      logger.verbose(`${currentReceiverOrError.logPrefix} Adding ${credits} credits`);\n      currentReceiverOrError.receiver.addCredit(credits);\n    }\n  }\n\n  /**\n   * Drains the credits for the receiver and prevents the `receiverHelper.addCredit()` method from adding credits.\n   * Call `resume()` to enable the `addCredit()` method.\n   */\n  async suspend(): Promise<void> {\n    const { receiver, logPrefix } = this._getCurrentReceiver();\n\n    this._isSuspended = true;\n\n    if (!this._isValidReceiver(receiver)) {\n      return;\n    }\n\n    logger.verbose(\n      `${logPrefix} User has requested to stop receiving new messages, attempting to drain.`,\n    );\n\n    return this.drain();\n  }\n\n  /**\n   * Resets tracking so `addCredit` works again by toggling the `_isSuspended` flag.\n   */\n  resume(): void {\n    this._isSuspended = false;\n  }\n\n  isSuspended(): boolean {\n    return this._isSuspended;\n  }\n\n  /**\n   * Initiates a drain for the current receiver and resolves when\n   * the drain has completed.\n   *\n   * NOTE: This method returns immediately if the receiver is not valid or if there\n   * are no pending credits on the receiver (ie: `receiver.credit === 0`).\n   */\n  async drain(): Promise<void> {\n    const { receiver, logPrefix } = this._getCurrentReceiver();\n\n    if (!this._isValidReceiver(receiver)) {\n      // TODO: should we throw?\n      return;\n    }\n\n    if (receiver.credit === 0) {\n      // nothing to drain\n      return;\n    }\n\n    logger.verbose(\n      `${logPrefix} Receiver is starting drain. Remaining credits; ${receiver.credit}`,\n    );\n\n    const drainPromise = new Promise<void>((resolve) => {\n      const timer = setTimeout(async () => {\n        logger.warning(`${logPrefix} Time out when draining credits in suspend().`);\n        // Close the receiver link since we have not received the receiver_drained event\n        // to prevent out-of-sync link state between local and remote\n        await receiver?.close();\n        resolve();\n      }, receiveDrainTimeoutInMs);\n      receiver.once(ReceiverEvents.receiverDrained, () => {\n        logger.verbose(`${logPrefix} Receiver has been drained.`);\n        receiver.drain = false;\n        clearTimeout(timer);\n        resolve();\n      });\n      receiver.drainCredit();\n    });\n\n    return drainPromise;\n  }\n\n  private _isValidReceiver(receiver: Receiver | undefined): receiver is Receiver {\n    return receiver != null && receiver.isOpen();\n  }\n}\n"]}