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

package/dist/commonjs/serviceBusMessageBatch.js.map

@@ -1 +1 @@
-{"version":3,"file":"serviceBusMessageBatch.js","sourceRoot":"","sources":["../../src/serviceBusMessageBatch.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAGlC,iEAAuD;AACvD,gDAI0B;AAG1B,+CAGsB;AAItB,6DAA8D;AAC9D,iGAAiF;AAEjF;;;GAGG;AACH,MAAM,oBAAoB,GAAG,CAAC,CAAC;AAC/B;;;GAGG;AACH,MAAM,oBAAoB,GAAG,CAAC,CAAC;AAC/B;;;GAGG;AACH,MAAM,oBAAoB,GAAG,GAAG,CAAC;AA6DjC;;;;GAIG;AACH,MAAa,0BAA0B;IAarC;;;;;OAKG;IACH,YACU,QAA2B,EAC3B,eAAuB;QADvB,aAAQ,GAAR,QAAQ,CAAmB;QAC3B,oBAAe,GAAf,eAAe,CAAQ;QAhBjC;;WAEG;QACK,qBAAgB,GAAa,EAAE,CAAC;QACxC;;WAEG;QACK,kBAAa,GAAqB,EAAE,CAAC;QAW3C,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;QACtB,IAAI,CAAC,uBAAuB,GAAG,EAAE,CAAC;IACpC,CAAC;IAED;;;OAGG;IACH,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED;;;;OAIG;IACH,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;;OAGG;IACH,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC;IACtC,CAAC;IAED;;;;OAIG;IACH,IAAI,oBAAoB;QACtB,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED;;;;;;;OAOG;IACK,cAAc,CACpB,eAAyB,EACzB,WAAgC,EAChC,qBAA8C,EAC9C,iBAA6C;QAE7C,MAAM,aAAa,GAAgB;YACjC,IAAI,EAAE,sBAAe,CAAC,aAAa,CAAC,eAAe,CAAC;YACpD,mBAAmB,EAAE,WAAW;YAChC,sBAAsB,EAAE,qBAAqB;SAC9C,CAAC;QACF,IAAI,iBAAiB,EAAE,CAAC;YACtB,KAAK,MAAM,IAAI,IAAI,gCAAyB,EAAE,CAAC;gBAC7C,IAAK,iBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;oBACpC,aAAqB,CAAC,IAAI,CAAC,GAAI,iBAAyB,CAAC,IAAI,CAAC,CAAC;gBAClE,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,sBAAe,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IAC/C,CAAC;IAED;;;;;;;;;OASG;IACH,gBAAgB;QACd,OAAO,IAAI,CAAC,cAAc,CACxB,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,iBAAiB,EACtB,IAAI,CAAC,2BAA2B,EAChC,IAAI,CAAC,uBAAuB,CAC7B,CAAC;IACJ,CAAC;IAqBD;;;;;;;OAOG;IACI,aAAa,CAClB,eAAyD,EACzD,UAAyB,EAAE;QAE3B,IAAA,4CAAgC,EAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,SAAS,EAAE,eAAe,CAAC,CAAC;QACzF,IAAA,4CAAgC,EAAC,eAAe,EAAE,yCAA6B,CAAC,CAAC;QAEjF,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,IAAA,kDAAiB,EAChD,eAAe,EACf,OAAO,EACP,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,UAAW,EAChC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,EACzB,SAAS,CACV,CAAC;QAEF,4CAA4C;QAC5C,MAAM,WAAW,GAAG,IAAA,oCAAa,EAAC,OAAO,EAAE,2CAAsB,CAAC,CAAC;QAEnE,MAAM,cAAc,GAAG,sBAAe,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAC3D,IAAI,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC;QAEpC,yDAAyD;QACzD,wDAAwD;QACxD,qEAAqE;QACrE,uCAAuC;QACvC,IAAI,IAAI,CAAC,KAAK,KAAK,CAAC,EAAE,CAAC;YACrB,IAAI,WAAW,CAAC,mBAAmB,EAAE,CAAC;gBACpC,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,mBAAmB,CAAC;YAC3D,CAAC;YACD,IAAI,WAAW,CAAC,sBAAsB,EAAE,CAAC;gBACvC,IAAI,CAAC,2BAA2B,GAAG,WAAW,CAAC,sBAAsB,CAAC;YACxE,CAAC;YACD,KAAK,MAAM,IAAI,IAAI,gCAAyB,EAAE,CAAC;gBAC7C,IAAK,WAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC9B,IAAI,CAAC,uBAA+B,CAAC,IAAI,CAAC,GAAI,WAAmB,CAAC,IAAI,CAAC,CAAC;gBAC3E,CAAC;YACH,CAAC;YACD,2EAA2E;YAC3E,uCAAuC;YACvC,WAAW,IAAI,IAAI,CAAC,cAAc,CAChC,EAAE,EACF,IAAI,CAAC,iBAAiB,EACtB,IAAI,CAAC,2BAA2B,EAChC,IAAI,CAAC,uBAAuB,CAC7B,CAAC,MAAM,CAAC;QACX,CAAC;QAED,MAAM,WAAW,GAAG,cAAc,CAAC,MAAM,CAAC;QAC1C,MAAM,eAAe,GACnB,WAAW,IAAI,oBAAoB,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,oBAAoB,CAAC;QACpF,WAAW,IAAI,WAAW,GAAG,eAAe,CAAC;QAC7C,kEAAkE;QAClE,mCAAmC;QACnC,IAAI,WAAW,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;YACvC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,oEAAoE;QACpE,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC3C,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;QAChC,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AA1MD,gEA0MC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { ServiceBusMessage } from \"./serviceBusMessage.js\";\nimport { toRheaMessage } from \"./serviceBusMessage.js\";\nimport {\n  errorInvalidMessageTypeSingle,\n  throwIfNotValidServiceBusMessage,\n  throwTypeErrorIfParameterMissing,\n} from \"./util/errors.js\";\nimport type { ConnectionContext } from \"./connectionContext.js\";\nimport type { MessageAnnotations, Message as RheaMessage } from \"rhea-promise\";\nimport {\n  messageProperties as RheaMessagePropertiesList,\n  message as RheaMessageUtil,\n} from \"rhea-promise\";\nimport type { TracingContext } from \"@azure/core-tracing\";\nimport type { TryAddOptions } from \"./modelsToBeSharedWithEventHubs.js\";\nimport type { AmqpAnnotatedMessage } from \"@azure/core-amqp\";\nimport { defaultDataTransformer } from \"./dataTransformer.js\";\nimport { instrumentMessage } from \"./diagnostics/instrumentServiceBusMessage.js\";\n\n/**\n * @internal\n * The amount of bytes to reserve as overhead for a small message.\n */\nconst smallMessageOverhead = 5;\n/**\n * @internal\n * The amount of bytes to reserve as overhead for a large message.\n */\nconst largeMessageOverhead = 8;\n/**\n * @internal\n * The maximum number of bytes that a message may be to be considered small.\n */\nconst smallMessageMaxBytes = 255;\n\n/**\n * A batch of messages that you can create using the {@link createBatch} method.\n *\n */\nexport interface ServiceBusMessageBatch {\n  /**\n   * Size of the batch in bytes after the events added to it have been encoded into a single AMQP\n   * message.\n   * @readonly\n   */\n  readonly sizeInBytes: number;\n\n  /**\n   * Number of messages added to the batch.\n   * @readonly\n   */\n  readonly count: number;\n\n  /**\n   * The maximum size of the batch, in bytes. The `tryAddMessage` function on the batch will return `false`\n   * if the message being added causes the size of the batch to exceed this limit. Use the `createMessageBatch()` method on\n   * the `Sender` to set the maxSizeInBytes.\n   * @readonly\n   */\n  readonly maxSizeInBytes: number;\n\n  /**\n   * Adds a message to the batch if permitted by the batch's size limit.\n   * **NOTE**: Always remember to check the return value of this method, before calling it again\n   * for the next event.\n   *\n   * @param message - The message to add to the batch.\n   * @returns A boolean value indicating if the message has been added to the batch or not.\n   */\n  tryAddMessage(\n    message: ServiceBusMessage | AmqpAnnotatedMessage,\n    options?: TryAddOptions,\n  ): boolean;\n\n  /**\n   * The AMQP message containing encoded events that were added to the batch.\n   * Used internally by the `sendBatch()` method on the `Sender`.\n   * This is not meant for the user to use directly.\n   *\n   * @readonly\n   * @internal\n   * @hidden\n   */\n  _generateMessage(): Buffer;\n\n  /**\n   * Gets the \"message\" span contexts that were created when adding events to the batch.\n   * Used internally by the `sendBatch()` method to set up the right spans in traces if tracing is enabled.\n   * @internal\n   * @hidden\n   */\n  readonly _messageSpanContexts: TracingContext[];\n}\n\n/**\n * An internal class representing a batch of messages which can be used to send messages to Service Bus.\n *\n * @internal\n */\nexport class ServiceBusMessageBatchImpl implements ServiceBusMessageBatch {\n  /**\n   * Current size of the batch in bytes.\n   */\n  private _sizeInBytes: number;\n  /**\n   * Encoded amqp messages.\n   */\n  private _encodedMessages: Buffer[] = [];\n  /**\n   * List of 'message' span contexts.\n   */\n  private _spanContexts: TracingContext[] = [];\n  /**\n   * ServiceBusMessageBatch should not be constructed using `new ServiceBusMessageBatch()`\n   * Use the `createBatch()` method on your `Sender` instead.\n   * @internal\n   * @hidden\n   */\n  constructor(\n    private _context: ConnectionContext,\n    private _maxSizeInBytes: number,\n  ) {\n    this._sizeInBytes = 0;\n    this._batchMessageProperties = {};\n  }\n\n  /**\n   * The maximum size of the batch, in bytes.\n   * @readonly\n   */\n  get maxSizeInBytes(): number {\n    return this._maxSizeInBytes;\n  }\n\n  /**\n   * Size of the `ServiceBusMessageBatch` instance after the messages added to it have been\n   * encoded into a single AMQP message.\n   * @readonly\n   */\n  get sizeInBytes(): number {\n    return this._sizeInBytes;\n  }\n\n  /**\n   * Number of messages in the `ServiceBusMessageBatch` instance.\n   * @readonly\n   */\n  get count(): number {\n    return this._encodedMessages.length;\n  }\n\n  /**\n   * Gets the \"message\" span contexts that were created when adding messages to the batch.\n   * @internal\n   * @hidden\n   */\n  get _messageSpanContexts(): TracingContext[] {\n    return this._spanContexts;\n  }\n\n  /**\n   * Generates an AMQP message that contains the provided encoded messages and annotations.\n   *\n   * @param encodedMessages - The already encoded messages to include in the AMQP batch.\n   * @param annotations - The message annotations to set on the batch.\n   * @param applicationProperties - The application properties to set on the batch.\n   * @param messageProperties - The message properties to set on the batch.\n   */\n  private _generateBatch(\n    encodedMessages: Buffer[],\n    annotations?: MessageAnnotations,\n    applicationProperties?: { [key: string]: any },\n    messageProperties?: { [key: string]: string },\n  ): Buffer {\n    const batchEnvelope: RheaMessage = {\n      body: RheaMessageUtil.data_sections(encodedMessages),\n      message_annotations: annotations,\n      application_properties: applicationProperties,\n    };\n    if (messageProperties) {\n      for (const prop of RheaMessagePropertiesList) {\n        if ((messageProperties as any)[prop]) {\n          (batchEnvelope as any)[prop] = (messageProperties as any)[prop];\n        }\n      }\n    }\n    return RheaMessageUtil.encode(batchEnvelope);\n  }\n\n  /**\n   * Represents the single AMQP message which is the result of encoding all the events\n   * added into the `ServiceBusMessageBatch` instance.\n   *\n   * This is not meant for the user to use directly.\n   *\n   * When the `ServiceBusMessageBatch` instance is passed to the `sendBatch()` method on the `Sender`,\n   * this single batched AMQP message is what gets sent over the wire to the service.\n   * @readonly\n   */\n  _generateMessage(): Buffer {\n    return this._generateBatch(\n      this._encodedMessages,\n      this._batchAnnotations,\n      this._batchApplicationProperties,\n      this._batchMessageProperties,\n    );\n  }\n\n  /**\n   * The message annotations to apply on the batch envelope.\n   * This will reflect the message annotations on the first message\n   * that was added to the batch.\n   */\n  private _batchAnnotations?: MessageAnnotations;\n  /**\n   * The message properties to apply on the batch envelope.\n   * This will reflect the message properties on the first message\n   * that was added to the batch.\n   */\n  private _batchMessageProperties?: { [key: string]: string };\n  /**\n   * The application properties to apply on the batch envelope.\n   * This will reflect the application properties on the first message\n   * that was added to the batch.\n   */\n  private _batchApplicationProperties?: { [key: string]: any };\n\n  /**\n   * Tries to add a message to the batch if permitted by the batch's size limit.\n   * **NOTE**: Always remember to check the return value of this method, before calling it again\n   * for the next message.\n   *\n   * @param originalMessage - An individual service bus message.\n   * @returns A boolean value indicating if the message has been added to the batch or not.\n   */\n  public tryAddMessage(\n    originalMessage: ServiceBusMessage | AmqpAnnotatedMessage,\n    options: TryAddOptions = {},\n  ): boolean {\n    throwTypeErrorIfParameterMissing(this._context.connectionId, \"message\", originalMessage);\n    throwIfNotValidServiceBusMessage(originalMessage, errorInvalidMessageTypeSingle);\n\n    const { message, spanContext } = instrumentMessage(\n      originalMessage,\n      options,\n      this._context.config.entityPath!,\n      this._context.config.host,\n      \"publish\",\n    );\n\n    // Convert ServiceBusMessage to AmqpMessage.\n    const amqpMessage = toRheaMessage(message, defaultDataTransformer);\n\n    const encodedMessage = RheaMessageUtil.encode(amqpMessage);\n    let currentSize = this._sizeInBytes;\n\n    // The first time an event is added, we need to calculate\n    // the overhead of creating an AMQP batch, including the\n    // message_annotations, application_properties and message_properties\n    // that are taken from the 1st message.\n    if (this.count === 0) {\n      if (amqpMessage.message_annotations) {\n        this._batchAnnotations = amqpMessage.message_annotations;\n      }\n      if (amqpMessage.application_properties) {\n        this._batchApplicationProperties = amqpMessage.application_properties;\n      }\n      for (const prop of RheaMessagePropertiesList) {\n        if ((amqpMessage as any)[prop]) {\n          (this._batchMessageProperties as any)[prop] = (amqpMessage as any)[prop];\n        }\n      }\n      // Figure out the overhead of creating a batch by generating an empty batch\n      // with the expected batch annotations.\n      currentSize += this._generateBatch(\n        [],\n        this._batchAnnotations,\n        this._batchApplicationProperties,\n        this._batchMessageProperties,\n      ).length;\n    }\n\n    const messageSize = encodedMessage.length;\n    const messageOverhead =\n      messageSize <= smallMessageMaxBytes ? smallMessageOverhead : largeMessageOverhead;\n    currentSize += messageSize + messageOverhead;\n    // Check if the size of the batch exceeds the maximum allowed size\n    // once we add the new event to it.\n    if (currentSize > this._maxSizeInBytes) {\n      return false;\n    }\n\n    // The message will fit in the batch, so it is now safe to store it.\n    this._encodedMessages.push(encodedMessage);\n    if (spanContext) {\n      this._spanContexts.push(spanContext);\n    }\n\n    this._sizeInBytes = currentSize;\n    return true;\n  }\n}\n"]}
+{"version":3,"file":"serviceBusMessageBatch.js","sourceRoot":"","sources":["../../src/serviceBusMessageBatch.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAGlC,iEAAuD;AACvD,gDAI0B;AAG1B,+CAGsB;AAItB,6DAA8D;AAC9D,iGAAiF;AAEjF;;;GAGG;AACH,MAAM,oBAAoB,GAAG,CAAC,CAAC;AAC/B;;;GAGG;AACH,MAAM,oBAAoB,GAAG,CAAC,CAAC;AAC/B;;;GAGG;AACH,MAAM,oBAAoB,GAAG,GAAG,CAAC;AA6DjC;;;;GAIG;AACH,MAAa,0BAA0B;IAoB3B;IACA;IApBV;;OAEG;IACK,YAAY,CAAS;IAC7B;;OAEG;IACK,gBAAgB,GAAa,EAAE,CAAC;IACxC;;OAEG;IACK,aAAa,GAAqB,EAAE,CAAC;IAC7C;;;;;OAKG;IACH,YACU,QAA2B,EAC3B,eAAuB;QADvB,aAAQ,GAAR,QAAQ,CAAmB;QAC3B,oBAAe,GAAf,eAAe,CAAQ;QAE/B,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;QACtB,IAAI,CAAC,uBAAuB,GAAG,EAAE,CAAC;IACpC,CAAC;IAED;;;OAGG;IACH,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED;;;;OAIG;IACH,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;;OAGG;IACH,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC;IACtC,CAAC;IAED;;;;OAIG;IACH,IAAI,oBAAoB;QACtB,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED;;;;;;;OAOG;IACK,cAAc,CACpB,eAAyB,EACzB,WAAgC,EAChC,qBAA8C,EAC9C,iBAA6C;QAE7C,MAAM,aAAa,GAAgB;YACjC,IAAI,EAAE,sBAAe,CAAC,aAAa,CAAC,eAAe,CAAC;YACpD,mBAAmB,EAAE,WAAW;YAChC,sBAAsB,EAAE,qBAAqB;SAC9C,CAAC;QACF,IAAI,iBAAiB,EAAE,CAAC;YACtB,KAAK,MAAM,IAAI,IAAI,gCAAyB,EAAE,CAAC;gBAC7C,IAAK,iBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC;oBACpC,aAAqB,CAAC,IAAI,CAAC,GAAI,iBAAyB,CAAC,IAAI,CAAC,CAAC;gBAClE,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,sBAAe,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IAC/C,CAAC;IAED;;;;;;;;;OASG;IACH,gBAAgB;QACd,OAAO,IAAI,CAAC,cAAc,CACxB,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,iBAAiB,EACtB,IAAI,CAAC,2BAA2B,EAChC,IAAI,CAAC,uBAAuB,CAC7B,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACK,iBAAiB,CAAsB;IAC/C;;;;OAIG;IACK,uBAAuB,CAA6B;IAC5D;;;;OAIG;IACK,2BAA2B,CAA0B;IAE7D;;;;;;;OAOG;IACI,aAAa,CAClB,eAAyD,EACzD,UAAyB,EAAE;QAE3B,IAAA,4CAAgC,EAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,SAAS,EAAE,eAAe,CAAC,CAAC;QACzF,IAAA,4CAAgC,EAAC,eAAe,EAAE,yCAA6B,CAAC,CAAC;QAEjF,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,IAAA,kDAAiB,EAChD,eAAe,EACf,OAAO,EACP,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,UAAW,EAChC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,EACzB,SAAS,CACV,CAAC;QAEF,4CAA4C;QAC5C,MAAM,WAAW,GAAG,IAAA,oCAAa,EAAC,OAAO,EAAE,2CAAsB,CAAC,CAAC;QAEnE,MAAM,cAAc,GAAG,sBAAe,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAC3D,IAAI,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC;QAEpC,yDAAyD;QACzD,wDAAwD;QACxD,qEAAqE;QACrE,uCAAuC;QACvC,IAAI,IAAI,CAAC,KAAK,KAAK,CAAC,EAAE,CAAC;YACrB,IAAI,WAAW,CAAC,mBAAmB,EAAE,CAAC;gBACpC,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,mBAAmB,CAAC;YAC3D,CAAC;YACD,IAAI,WAAW,CAAC,sBAAsB,EAAE,CAAC;gBACvC,IAAI,CAAC,2BAA2B,GAAG,WAAW,CAAC,sBAAsB,CAAC;YACxE,CAAC;YACD,KAAK,MAAM,IAAI,IAAI,gCAAyB,EAAE,CAAC;gBAC7C,IAAK,WAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC9B,IAAI,CAAC,uBAA+B,CAAC,IAAI,CAAC,GAAI,WAAmB,CAAC,IAAI,CAAC,CAAC;gBAC3E,CAAC;YACH,CAAC;YACD,2EAA2E;YAC3E,uCAAuC;YACvC,WAAW,IAAI,IAAI,CAAC,cAAc,CAChC,EAAE,EACF,IAAI,CAAC,iBAAiB,EACtB,IAAI,CAAC,2BAA2B,EAChC,IAAI,CAAC,uBAAuB,CAC7B,CAAC,MAAM,CAAC;QACX,CAAC;QAED,MAAM,WAAW,GAAG,cAAc,CAAC,MAAM,CAAC;QAC1C,MAAM,eAAe,GACnB,WAAW,IAAI,oBAAoB,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,oBAAoB,CAAC;QACpF,WAAW,IAAI,WAAW,GAAG,eAAe,CAAC;QAC7C,kEAAkE;QAClE,mCAAmC;QACnC,IAAI,WAAW,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;YACvC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,oEAAoE;QACpE,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC3C,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;QAChC,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AA1MD,gEA0MC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { ServiceBusMessage } from \"./serviceBusMessage.js\";\nimport { toRheaMessage } from \"./serviceBusMessage.js\";\nimport {\n  errorInvalidMessageTypeSingle,\n  throwIfNotValidServiceBusMessage,\n  throwTypeErrorIfParameterMissing,\n} from \"./util/errors.js\";\nimport type { ConnectionContext } from \"./connectionContext.js\";\nimport type { MessageAnnotations, Message as RheaMessage } from \"rhea-promise\";\nimport {\n  messageProperties as RheaMessagePropertiesList,\n  message as RheaMessageUtil,\n} from \"rhea-promise\";\nimport type { TracingContext } from \"@azure/core-tracing\";\nimport type { TryAddOptions } from \"./modelsToBeSharedWithEventHubs.js\";\nimport type { AmqpAnnotatedMessage } from \"@azure/core-amqp\";\nimport { defaultDataTransformer } from \"./dataTransformer.js\";\nimport { instrumentMessage } from \"./diagnostics/instrumentServiceBusMessage.js\";\n\n/**\n * @internal\n * The amount of bytes to reserve as overhead for a small message.\n */\nconst smallMessageOverhead = 5;\n/**\n * @internal\n * The amount of bytes to reserve as overhead for a large message.\n */\nconst largeMessageOverhead = 8;\n/**\n * @internal\n * The maximum number of bytes that a message may be to be considered small.\n */\nconst smallMessageMaxBytes = 255;\n\n/**\n * A batch of messages that you can create using the {@link createBatch} method.\n *\n */\nexport interface ServiceBusMessageBatch {\n  /**\n   * Size of the batch in bytes after the events added to it have been encoded into a single AMQP\n   * message.\n   * @readonly\n   */\n  readonly sizeInBytes: number;\n\n  /**\n   * Number of messages added to the batch.\n   * @readonly\n   */\n  readonly count: number;\n\n  /**\n   * The maximum size of the batch, in bytes. The `tryAddMessage` function on the batch will return `false`\n   * if the message being added causes the size of the batch to exceed this limit. Use the `createMessageBatch()` method on\n   * the `Sender` to set the maxSizeInBytes.\n   * @readonly\n   */\n  readonly maxSizeInBytes: number;\n\n  /**\n   * Adds a message to the batch if permitted by the batch's size limit.\n   * **NOTE**: Always remember to check the return value of this method, before calling it again\n   * for the next event.\n   *\n   * @param message - The message to add to the batch.\n   * @returns A boolean value indicating if the message has been added to the batch or not.\n   */\n  tryAddMessage(\n    message: ServiceBusMessage | AmqpAnnotatedMessage,\n    options?: TryAddOptions,\n  ): boolean;\n\n  /**\n   * The AMQP message containing encoded events that were added to the batch.\n   * Used internally by the `sendBatch()` method on the `Sender`.\n   * This is not meant for the user to use directly.\n   *\n   * @readonly\n   * @internal\n   * @hidden\n   */\n  _generateMessage(): Buffer;\n\n  /**\n   * Gets the \"message\" span contexts that were created when adding events to the batch.\n   * Used internally by the `sendBatch()` method to set up the right spans in traces if tracing is enabled.\n   * @internal\n   * @hidden\n   */\n  readonly _messageSpanContexts: TracingContext[];\n}\n\n/**\n * An internal class representing a batch of messages which can be used to send messages to Service Bus.\n *\n * @internal\n */\nexport class ServiceBusMessageBatchImpl implements ServiceBusMessageBatch {\n  /**\n   * Current size of the batch in bytes.\n   */\n  private _sizeInBytes: number;\n  /**\n   * Encoded amqp messages.\n   */\n  private _encodedMessages: Buffer[] = [];\n  /**\n   * List of 'message' span contexts.\n   */\n  private _spanContexts: TracingContext[] = [];\n  /**\n   * ServiceBusMessageBatch should not be constructed using `new ServiceBusMessageBatch()`\n   * Use the `createBatch()` method on your `Sender` instead.\n   * @internal\n   * @hidden\n   */\n  constructor(\n    private _context: ConnectionContext,\n    private _maxSizeInBytes: number,\n  ) {\n    this._sizeInBytes = 0;\n    this._batchMessageProperties = {};\n  }\n\n  /**\n   * The maximum size of the batch, in bytes.\n   * @readonly\n   */\n  get maxSizeInBytes(): number {\n    return this._maxSizeInBytes;\n  }\n\n  /**\n   * Size of the `ServiceBusMessageBatch` instance after the messages added to it have been\n   * encoded into a single AMQP message.\n   * @readonly\n   */\n  get sizeInBytes(): number {\n    return this._sizeInBytes;\n  }\n\n  /**\n   * Number of messages in the `ServiceBusMessageBatch` instance.\n   * @readonly\n   */\n  get count(): number {\n    return this._encodedMessages.length;\n  }\n\n  /**\n   * Gets the \"message\" span contexts that were created when adding messages to the batch.\n   * @internal\n   * @hidden\n   */\n  get _messageSpanContexts(): TracingContext[] {\n    return this._spanContexts;\n  }\n\n  /**\n   * Generates an AMQP message that contains the provided encoded messages and annotations.\n   *\n   * @param encodedMessages - The already encoded messages to include in the AMQP batch.\n   * @param annotations - The message annotations to set on the batch.\n   * @param applicationProperties - The application properties to set on the batch.\n   * @param messageProperties - The message properties to set on the batch.\n   */\n  private _generateBatch(\n    encodedMessages: Buffer[],\n    annotations?: MessageAnnotations,\n    applicationProperties?: { [key: string]: any },\n    messageProperties?: { [key: string]: string },\n  ): Buffer {\n    const batchEnvelope: RheaMessage = {\n      body: RheaMessageUtil.data_sections(encodedMessages),\n      message_annotations: annotations,\n      application_properties: applicationProperties,\n    };\n    if (messageProperties) {\n      for (const prop of RheaMessagePropertiesList) {\n        if ((messageProperties as any)[prop]) {\n          (batchEnvelope as any)[prop] = (messageProperties as any)[prop];\n        }\n      }\n    }\n    return RheaMessageUtil.encode(batchEnvelope);\n  }\n\n  /**\n   * Represents the single AMQP message which is the result of encoding all the events\n   * added into the `ServiceBusMessageBatch` instance.\n   *\n   * This is not meant for the user to use directly.\n   *\n   * When the `ServiceBusMessageBatch` instance is passed to the `sendBatch()` method on the `Sender`,\n   * this single batched AMQP message is what gets sent over the wire to the service.\n   * @readonly\n   */\n  _generateMessage(): Buffer {\n    return this._generateBatch(\n      this._encodedMessages,\n      this._batchAnnotations,\n      this._batchApplicationProperties,\n      this._batchMessageProperties,\n    );\n  }\n\n  /**\n   * The message annotations to apply on the batch envelope.\n   * This will reflect the message annotations on the first message\n   * that was added to the batch.\n   */\n  private _batchAnnotations?: MessageAnnotations;\n  /**\n   * The message properties to apply on the batch envelope.\n   * This will reflect the message properties on the first message\n   * that was added to the batch.\n   */\n  private _batchMessageProperties?: { [key: string]: string };\n  /**\n   * The application properties to apply on the batch envelope.\n   * This will reflect the application properties on the first message\n   * that was added to the batch.\n   */\n  private _batchApplicationProperties?: { [key: string]: any };\n\n  /**\n   * Tries to add a message to the batch if permitted by the batch's size limit.\n   * **NOTE**: Always remember to check the return value of this method, before calling it again\n   * for the next message.\n   *\n   * @param originalMessage - An individual service bus message.\n   * @returns A boolean value indicating if the message has been added to the batch or not.\n   */\n  public tryAddMessage(\n    originalMessage: ServiceBusMessage | AmqpAnnotatedMessage,\n    options: TryAddOptions = {},\n  ): boolean {\n    throwTypeErrorIfParameterMissing(this._context.connectionId, \"message\", originalMessage);\n    throwIfNotValidServiceBusMessage(originalMessage, errorInvalidMessageTypeSingle);\n\n    const { message, spanContext } = instrumentMessage(\n      originalMessage,\n      options,\n      this._context.config.entityPath!,\n      this._context.config.host,\n      \"publish\",\n    );\n\n    // Convert ServiceBusMessage to AmqpMessage.\n    const amqpMessage = toRheaMessage(message, defaultDataTransformer);\n\n    const encodedMessage = RheaMessageUtil.encode(amqpMessage);\n    let currentSize = this._sizeInBytes;\n\n    // The first time an event is added, we need to calculate\n    // the overhead of creating an AMQP batch, including the\n    // message_annotations, application_properties and message_properties\n    // that are taken from the 1st message.\n    if (this.count === 0) {\n      if (amqpMessage.message_annotations) {\n        this._batchAnnotations = amqpMessage.message_annotations;\n      }\n      if (amqpMessage.application_properties) {\n        this._batchApplicationProperties = amqpMessage.application_properties;\n      }\n      for (const prop of RheaMessagePropertiesList) {\n        if ((amqpMessage as any)[prop]) {\n          (this._batchMessageProperties as any)[prop] = (amqpMessage as any)[prop];\n        }\n      }\n      // Figure out the overhead of creating a batch by generating an empty batch\n      // with the expected batch annotations.\n      currentSize += this._generateBatch(\n        [],\n        this._batchAnnotations,\n        this._batchApplicationProperties,\n        this._batchMessageProperties,\n      ).length;\n    }\n\n    const messageSize = encodedMessage.length;\n    const messageOverhead =\n      messageSize <= smallMessageMaxBytes ? smallMessageOverhead : largeMessageOverhead;\n    currentSize += messageSize + messageOverhead;\n    // Check if the size of the batch exceeds the maximum allowed size\n    // once we add the new event to it.\n    if (currentSize > this._maxSizeInBytes) {\n      return false;\n    }\n\n    // The message will fit in the batch, so it is now safe to store it.\n    this._encodedMessages.push(encodedMessage);\n    if (spanContext) {\n      this._spanContexts.push(spanContext);\n    }\n\n    this._sizeInBytes = currentSize;\n    return true;\n  }\n}\n"]}

package/dist/commonjs/serviceBusRuleManager.js

@@ -14,6 +14,15 @@ const core_paging_1 = require("@azure/core-paging");
  * @internal
  */
 class ServiceBusRuleManagerImpl {
+    _context;
+    _entityPath;
+    _retryOptions;
+    /**
+     * Denotes if close() was called on this sender
+     */
+    _isClosed = false;
+    name;
+    entityPath;
     /**
      * @internal
      * @throws Error if the underlying connection is closed.
@@ -22,10 +31,6 @@ class ServiceBusRuleManagerImpl {
         this._context = _context;
         this._entityPath = _entityPath;
         this._retryOptions = _retryOptions;
-        /**
-         * Denotes if close() was called on this sender
-         */
-        this._isClosed = false;
         (0, errors_js_1.throwErrorIfConnectionClosed)(_context);
         this.entityPath = _entityPath;
         this.name = (0, utils_js_1.getUniqueName)("ruleManager");
@@ -44,21 +49,26 @@ class ServiceBusRuleManagerImpl {
             }
             else {
                 // Overload#1 - where the sqlExpression in the ruleAction is undefined
-                operOptions = Object.assign(Object.assign({}, ruleActionOrOperationOptions), options);
+                operOptions = { ...ruleActionOrOperationOptions, ...options };
             }
         }
-        return tracing_js_1.tracingClient.withSpan("ServiceBusRuleManager.createRule", operOptions !== null && operOptions !== void 0 ? operOptions : {}, async (updatedOptions) => {
+        return tracing_js_1.tracingClient.withSpan("ServiceBusRuleManager.createRule", operOptions ?? {}, async (updatedOptions) => {
             const addRuleOperationPromise = async () => {
                 return this._context
                     .getManagementClient(this._entityPath)
-                    .addRule(ruleName, filter, sqlRuleAction === null || sqlRuleAction === void 0 ? void 0 : sqlRuleAction.sqlExpression, Object.assign(Object.assign({}, updatedOptions), { associatedLinkName: this.name, requestName: "addRule", timeoutInMs: this._retryOptions.timeoutInMs }));
+                    .addRule(ruleName, filter, sqlRuleAction?.sqlExpression, {
+                    ...updatedOptions,
+                    associatedLinkName: this.name,
+                    requestName: "addRule",
+                    timeoutInMs: this._retryOptions.timeoutInMs,
+                });
             };
             const config = {
                 operation: addRuleOperationPromise,
                 connectionId: this._context.connectionId,
                 operationType: core_amqp_1.RetryOperationType.management,
                 retryOptions: this._retryOptions,
-                abortSignal: updatedOptions === null || updatedOptions === void 0 ? void 0 : updatedOptions.abortSignal,
+                abortSignal: updatedOptions?.abortSignal,
             };
             return (0, core_amqp_1.retry)(config);
         });
@@ -67,16 +77,21 @@ class ServiceBusRuleManagerImpl {
      * Get all rules associated with the subscription.
      */
     async getRules(options) {
-        return tracing_js_1.tracingClient.withSpan("ServiceBusRuleManager.getRules", options !== null && options !== void 0 ? options : {}, async (updatedOptions) => {
+        return tracing_js_1.tracingClient.withSpan("ServiceBusRuleManager.getRules", options ?? {}, async (updatedOptions) => {
             const getRulesOperationPromise = async () => {
-                return this._context.getManagementClient(this._entityPath).getRules(Object.assign(Object.assign({}, updatedOptions), { associatedLinkName: this.name, requestName: "getRules", timeoutInMs: this._retryOptions.timeoutInMs }));
+                return this._context.getManagementClient(this._entityPath).getRules({
+                    ...updatedOptions,
+                    associatedLinkName: this.name,
+                    requestName: "getRules",
+                    timeoutInMs: this._retryOptions.timeoutInMs,
+                });
             };
             const config = {
                 operation: getRulesOperationPromise,
                 connectionId: this._context.connectionId,
                 operationType: core_amqp_1.RetryOperationType.management,
                 retryOptions: this._retryOptions,
-                abortSignal: updatedOptions === null || updatedOptions === void 0 ? void 0 : updatedOptions.abortSignal,
+                abortSignal: updatedOptions?.abortSignal,
             };
             return (0, core_amqp_1.retry)(config);
         });
@@ -94,8 +109,12 @@ class ServiceBusRuleManagerImpl {
         const pagedResult = {
             firstPageLink: 0,
             getPage: async (pageLink, maxPageSize) => {
-                const top = maxPageSize !== null && maxPageSize !== void 0 ? maxPageSize : 100;
-                const rules = await this.getRules(Object.assign({ skip: pageLink, maxCount: top }, options));
+                const top = maxPageSize ?? 100;
+                const rules = await this.getRules({
+                    skip: pageLink,
+                    maxCount: top,
+                    ...options,
+                });
                 return rules.length
                     ? {
                         page: rules,
@@ -112,14 +131,19 @@ class ServiceBusRuleManagerImpl {
     async deleteRule(ruleName, options = {}) {
         return tracing_js_1.tracingClient.withSpan("ServiceBusRuleManager.deleteRule", options, async (updatedOptions) => {
             const removeRuleOperationPromise = async () => {
-                return this._context.getManagementClient(this._entityPath).removeRule(ruleName, Object.assign(Object.assign({}, updatedOptions), { associatedLinkName: this.name, requestName: "removeRule", timeoutInMs: this._retryOptions.timeoutInMs }));
+                return this._context.getManagementClient(this._entityPath).removeRule(ruleName, {
+                    ...updatedOptions,
+                    associatedLinkName: this.name,
+                    requestName: "removeRule",
+                    timeoutInMs: this._retryOptions.timeoutInMs,
+                });
             };
             const config = {
                 operation: removeRuleOperationPromise,
                 connectionId: this._context.connectionId,
                 operationType: core_amqp_1.RetryOperationType.management,
                 retryOptions: this._retryOptions,
-                abortSignal: updatedOptions === null || updatedOptions === void 0 ? void 0 : updatedOptions.abortSignal,
+                abortSignal: updatedOptions?.abortSignal,
             };
             return (0, core_amqp_1.retry)(config);
         });

package/dist/commonjs/serviceBusRuleManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"serviceBusRuleManager.js","sourceRoot":"","sources":["../../src/serviceBusRuleManager.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAKlC,gDAA6D;AAE7D,qCAAuD;AAEvD,uFAA0E;AAC1E,8CAAgD;AAChD,gDAAgE;AAEhE,yDAAyD;AAEzD,oDAA2D;AAoD3D;;GAEG;AACH,MAAa,yBAAyB;IAQpC;;;OAGG;IACH,YACU,QAA2B,EAC3B,WAAmB,EACnB,gBAA8B,EAAE;QAFhC,aAAQ,GAAR,QAAQ,CAAmB;QAC3B,gBAAW,GAAX,WAAW,CAAQ;QACnB,kBAAa,GAAb,aAAa,CAAmB;QAd1C;;WAEG;QACK,cAAS,GAAY,KAAK,CAAC;QAajC,IAAA,wCAA4B,EAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,CAAC,UAAU,GAAG,WAAW,CAAC;QAC9B,IAAI,CAAC,IAAI,GAAG,IAAA,wBAAa,EAAC,aAAa,CAAC,CAAC;IAC3C,CAAC;IAED,IAAW,QAAQ;QACjB,OAAO,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,QAAQ,CAAC,wBAAwB,CAAC;IAClE,CAAC;IA4BD,KAAK,CAAC,UAAU,CACd,QAAgB,EAChB,MAA6C,EAC7C,4BAAmE,EACnE,UAA4B,EAAE;QAE9B,IAAI,aAAa,GAA8B,SAAS,CAAC;QACzD,IAAI,WAAyC,CAAC;QAC9C,IAAI,4BAA4B,EAAE,CAAC;YACjC,IAAI,IAAA,2CAAe,EAAC,4BAA4B,CAAC,EAAE,CAAC;gBAClD,oEAAoE;gBACpE,aAAa,GAAG,4BAA4B,CAAC;gBAC7C,WAAW,GAAG,OAAO,CAAC;YACxB,CAAC;iBAAM,CAAC;gBACN,sEAAsE;gBACtE,WAAW,mCAAQ,4BAA4B,GAAK,OAAO,CAAE,CAAC;YAChE,CAAC;QACH,CAAC;QAED,OAAO,0BAAa,CAAC,QAAQ,CAC3B,kCAAkC,EAClC,WAAW,aAAX,WAAW,cAAX,WAAW,GAAI,EAAE,EACjB,KAAK,EAAE,cAAc,EAAE,EAAE;YACvB,MAAM,uBAAuB,GAAG,KAAK,IAAmB,EAAE;gBACxD,OAAO,IAAI,CAAC,QAAQ;qBACjB,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC;qBACrC,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,aAAa,aAAb,aAAa,uBAAb,aAAa,CAAE,aAAa,kCAClD,cAAc,KACjB,kBAAkB,EAAE,IAAI,CAAC,IAAI,EAC7B,WAAW,EAAE,SAAS,EACtB,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,WAAW,IAC3C,CAAC;YACP,CAAC,CAAC;YACF,MAAM,MAAM,GAAsB;gBAChC,SAAS,EAAE,uBAAuB;gBAClC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY;gBACxC,aAAa,EAAE,8BAAkB,CAAC,UAAU;gBAC5C,YAAY,EAAE,IAAI,CAAC,aAAa;gBAChC,WAAW,EAAE,cAAc,aAAd,cAAc,uBAAd,cAAc,CAAE,WAAW;aACzC,CAAC;YACF,OAAO,IAAA,iBAAK,EAAO,MAAM,CAAC,CAAC;QAC7B,CAAC,CACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,QAAQ,CACpB,OAA+C;QAE/C,OAAO,0BAAa,CAAC,QAAQ,CAC3B,gCAAgC,EAChC,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,EAAE,EACb,KAAK,EAAE,cAAc,EAAE,EAAE;YACvB,MAAM,wBAAwB,GAAG,KAAK,IAA+B,EAAE;gBACrE,OAAO,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,QAAQ,iCAC9D,cAAc,KACjB,kBAAkB,EAAE,IAAI,CAAC,IAAI,EAC7B,WAAW,EAAE,UAAU,EACvB,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,WAAW,IAC3C,CAAC;YACL,CAAC,CAAC;YACF,MAAM,MAAM,GAAkC;gBAC5C,SAAS,EAAE,wBAAwB;gBACnC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY;gBACxC,aAAa,EAAE,8BAAkB,CAAC,UAAU;gBAC5C,YAAY,EAAE,IAAI,CAAC,aAAa;gBAChC,WAAW,EAAE,cAAc,aAAd,cAAc,uBAAd,cAAc,CAAE,WAAW;aACzC,CAAC;YACF,OAAO,IAAA,iBAAK,EAAmB,MAAM,CAAC,CAAC;QACzC,CAAC,CACF,CAAC;IACJ,CAAC;IAED;;;;;;;OAOG;IACI,SAAS,CACd,OAA0B;QAE1B,0BAAM,CAAC,OAAO,CAAC,qDAAqD,EAAE,OAAO,CAAC,CAAC;QAC/E,MAAM,WAAW,GAAoE;YACnF,aAAa,EAAE,CAAC;YAChB,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,EAAE;gBACvC,MAAM,GAAG,GAAG,WAAW,aAAX,WAAW,cAAX,WAAW,GAAI,GAAG,CAAC;gBAC/B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,iBAC/B,IAAI,EAAE,QAAQ,EACd,QAAQ,EAAE,GAAG,IACV,OAAO,EACV,CAAC;gBACH,OAAO,KAAK,CAAC,MAAM;oBACjB,CAAC,CAAC;wBACE,IAAI,EAAE,KAAK;wBACX,YAAY,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;qBACrE;oBACH,CAAC,CAAC,SAAS,CAAC;YAChB,CAAC;SACF,CAAC;QAEF,OAAO,IAAA,mCAAqB,EAAC,WAAW,CAAC,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,QAAgB,EAAE,UAA4B,EAAE;QAC/D,OAAO,0BAAa,CAAC,QAAQ,CAC3B,kCAAkC,EAClC,OAAO,EACP,KAAK,EAAE,cAAc,EAAE,EAAE;YACvB,MAAM,0BAA0B,GAAG,KAAK,IAAmB,EAAE;gBAC3D,OAAO,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,UAAU,CAAC,QAAQ,kCACzE,cAAc,KACjB,kBAAkB,EAAE,IAAI,CAAC,IAAI,EAC7B,WAAW,EAAE,YAAY,EACzB,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,WAAW,IAC3C,CAAC;YACL,CAAC,CAAC;YACF,MAAM,MAAM,GAAsB;gBAChC,SAAS,EAAE,0BAA0B;gBACrC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY;gBACxC,aAAa,EAAE,8BAAkB,CAAC,UAAU;gBAC5C,YAAY,EAAE,IAAI,CAAC,aAAa;gBAChC,WAAW,EAAE,cAAc,aAAd,cAAc,uBAAd,cAAc,CAAE,WAAW;aACzC,CAAC;YACF,OAAO,IAAA,iBAAK,EAAO,MAAM,CAAC,CAAC;QAC7B,CAAC,CACF,CAAC;IACJ,CAAC;CACF;AA3LD,8DA2LC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { OperationOptionsBase } from \"./modelsToBeSharedWithEventHubs.js\";\nimport type { ConnectionContext } from \"./connectionContext.js\";\nimport type { RetryConfig, RetryOptions } from \"@azure/core-amqp\";\nimport { RetryOperationType, retry } from \"@azure/core-amqp\";\nimport type { CorrelationRuleFilter } from \"./core/managementClient.js\";\nimport { ruleManagerLogger as logger } from \"./log.js\";\nimport type { RuleProperties, SqlRuleAction } from \"./serializers/ruleResourceSerializer.js\";\nimport { isSqlRuleAction } from \"./serializers/ruleResourceSerializer.js\";\nimport { getUniqueName } from \"./util/utils.js\";\nimport { throwErrorIfConnectionClosed } from \"./util/errors.js\";\nimport type { SqlRuleFilter } from \"./serializers/ruleResourceSerializer.js\";\nimport { tracingClient } from \"./diagnostics/tracing.js\";\nimport type { PagedAsyncIterableIterator, PagedResult } from \"@azure/core-paging\";\nimport { getPagedAsyncIterator } from \"@azure/core-paging\";\nimport type { OperationOptions } from \"@azure/core-client\";\nimport type { ListRequestOptions } from \"./serviceBusAtomManagementClient.js\";\n\n/**\n * Allows rules for a subscription to be managed. This rule manager requires only Listen claims, whereas the\n * {@link ServiceBusAdministrationClient} requires Manage claims.\n */\nexport interface ServiceBusRuleManager {\n  /**\n   * Adds a rule to the current subscription to filter the messages reaching from topic to the subscription.\n   *\n   * @param ruleName - the name of the rule\n   * @param filter - the filter expression that the rule evaluates.\n   * @param options - The options that can be used to abort, trace and control other configurations on the request.\n   */\n  createRule(\n    ruleName: string,\n    filter: SqlRuleFilter | CorrelationRuleFilter,\n    options?: OperationOptionsBase,\n  ): Promise<void>;\n  /**\n   * Adds a rule to the current subscription to filter the messages reaching from topic to the subscription.\n   *\n   * @param ruleName - the name of the rule\n   * @param filter - the filter expression that the rule evaluates.\n   * @param ruleAction - The SQL like expression that can be executed on the message should the associated filter apply.\n   * @param options - The options that can be used to abort, trace and control other configurations on the request.\n   */\n  createRule(\n    ruleName: string,\n    filter: SqlRuleFilter | CorrelationRuleFilter,\n    ruleAction?: SqlRuleAction,\n    options?: OperationOptionsBase,\n  ): Promise<void>;\n  /**\n   * Deletes a rule.\n   *\n   * @param ruleName - the name of the rule\n   */\n  deleteRule(ruleName: string, options?: OperationOptionsBase): Promise<void>;\n  /**\n   * Returns an async iterable iterator to list all the rules\n   * under the specified subscription.\n   *\n   * .byPage() returns an async iterable iterator to list the rules in pages.\n   *\n   * @returns An asyncIterableIterator that supports paging.\n   */\n  listRules(options?: OperationOptions): PagedAsyncIterableIterator<RuleProperties>;\n}\n\n/**\n * @internal\n */\nexport class ServiceBusRuleManagerImpl implements ServiceBusRuleManager {\n  /**\n   * Denotes if close() was called on this sender\n   */\n  private _isClosed: boolean = false;\n  private name: string;\n  public entityPath: string;\n\n  /**\n   * @internal\n   * @throws Error if the underlying connection is closed.\n   */\n  constructor(\n    private _context: ConnectionContext,\n    private _entityPath: string,\n    private _retryOptions: RetryOptions = {},\n  ) {\n    throwErrorIfConnectionClosed(_context);\n    this.entityPath = _entityPath;\n    this.name = getUniqueName(\"ruleManager\");\n  }\n\n  public get isClosed(): boolean {\n    return this._isClosed || this._context.wasConnectionCloseCalled;\n  }\n\n  /**\n   * Adds a rule to the current subscription to filter the messages reaching from topic to the subscription.\n   *\n   * @param ruleName - the name of the rule\n   * @param filter - the filter expression that the rule evaluates.\n   * @param options - The options that can be used to abort, trace and control other configurations on the request.\n   */\n  createRule(\n    ruleName: string,\n    filter: SqlRuleFilter | CorrelationRuleFilter,\n    options?: OperationOptions,\n  ): Promise<void>;\n  /**\n   * Adds a rule to the current subscription to filter the messages reaching from topic to the subscription.\n   *\n   * @param ruleName - the name of the rule\n   * @param filter - the filter expression that the rule evaluates.\n   * @param ruleAction - The SQL like expression that can be executed on the message should the associated filter apply.\n   * @param options - The options that can be used to abort, trace and control other configurations on the request.\n   */\n  createRule(\n    ruleName: string,\n    filter: SqlRuleFilter | CorrelationRuleFilter,\n    ruleAction?: SqlRuleAction,\n    options?: OperationOptions,\n  ): Promise<void>;\n  async createRule(\n    ruleName: string,\n    filter: SqlRuleFilter | CorrelationRuleFilter,\n    ruleActionOrOperationOptions?: SqlRuleAction | OperationOptionsBase,\n    options: OperationOptions = {},\n  ): Promise<void> {\n    let sqlRuleAction: SqlRuleAction | undefined = undefined;\n    let operOptions: OperationOptions | undefined;\n    if (ruleActionOrOperationOptions) {\n      if (isSqlRuleAction(ruleActionOrOperationOptions)) {\n        // Overload#2 - where the sqlExpression in the ruleAction is defined\n        sqlRuleAction = ruleActionOrOperationOptions;\n        operOptions = options;\n      } else {\n        // Overload#1 - where the sqlExpression in the ruleAction is undefined\n        operOptions = { ...ruleActionOrOperationOptions, ...options };\n      }\n    }\n\n    return tracingClient.withSpan(\n      \"ServiceBusRuleManager.createRule\",\n      operOptions ?? {},\n      async (updatedOptions) => {\n        const addRuleOperationPromise = async (): Promise<void> => {\n          return this._context\n            .getManagementClient(this._entityPath)\n            .addRule(ruleName, filter, sqlRuleAction?.sqlExpression, {\n              ...updatedOptions,\n              associatedLinkName: this.name,\n              requestName: \"addRule\",\n              timeoutInMs: this._retryOptions.timeoutInMs,\n            });\n        };\n        const config: RetryConfig<void> = {\n          operation: addRuleOperationPromise,\n          connectionId: this._context.connectionId,\n          operationType: RetryOperationType.management,\n          retryOptions: this._retryOptions,\n          abortSignal: updatedOptions?.abortSignal,\n        };\n        return retry<void>(config);\n      },\n    );\n  }\n\n  /**\n   * Get all rules associated with the subscription.\n   */\n  private async getRules(\n    options?: ListRequestOptions & OperationOptions,\n  ): Promise<RuleProperties[]> {\n    return tracingClient.withSpan(\n      \"ServiceBusRuleManager.getRules\",\n      options ?? {},\n      async (updatedOptions) => {\n        const getRulesOperationPromise = async (): Promise<RuleProperties[]> => {\n          return this._context.getManagementClient(this._entityPath).getRules({\n            ...updatedOptions,\n            associatedLinkName: this.name,\n            requestName: \"getRules\",\n            timeoutInMs: this._retryOptions.timeoutInMs,\n          });\n        };\n        const config: RetryConfig<RuleProperties[]> = {\n          operation: getRulesOperationPromise,\n          connectionId: this._context.connectionId,\n          operationType: RetryOperationType.management,\n          retryOptions: this._retryOptions,\n          abortSignal: updatedOptions?.abortSignal,\n        };\n        return retry<RuleProperties[]>(config);\n      },\n    );\n  }\n\n  /**\n   * Returns an async iterable iterator to list all the rules\n   * under the specified subscription.\n   *\n   * .byPage() returns an async iterable iterator to list the rules in pages.\n   *\n   * @returns An asyncIterableIterator that supports paging.\n   */\n  public listRules(\n    options?: OperationOptions,\n  ): PagedAsyncIterableIterator<RuleProperties, RuleProperties[], { maxPageSize?: number }> {\n    logger.verbose(`Performing operation - listRules() with options: %j`, options);\n    const pagedResult: PagedResult<RuleProperties[], { maxPageSize?: number }, number> = {\n      firstPageLink: 0,\n      getPage: async (pageLink, maxPageSize) => {\n        const top = maxPageSize ?? 100;\n        const rules = await this.getRules({\n          skip: pageLink,\n          maxCount: top,\n          ...options,\n        });\n        return rules.length\n          ? {\n              page: rules,\n              nextPageLink: rules.length > 0 ? pageLink + rules.length : undefined,\n            }\n          : undefined;\n      },\n    };\n\n    return getPagedAsyncIterator(pagedResult);\n  }\n\n  /**\n   * Deletes a rule.\n   */\n  async deleteRule(ruleName: string, options: OperationOptions = {}): Promise<void> {\n    return tracingClient.withSpan(\n      \"ServiceBusRuleManager.deleteRule\",\n      options,\n      async (updatedOptions) => {\n        const removeRuleOperationPromise = async (): Promise<void> => {\n          return this._context.getManagementClient(this._entityPath).removeRule(ruleName, {\n            ...updatedOptions,\n            associatedLinkName: this.name,\n            requestName: \"removeRule\",\n            timeoutInMs: this._retryOptions.timeoutInMs,\n          });\n        };\n        const config: RetryConfig<void> = {\n          operation: removeRuleOperationPromise,\n          connectionId: this._context.connectionId,\n          operationType: RetryOperationType.management,\n          retryOptions: this._retryOptions,\n          abortSignal: updatedOptions?.abortSignal,\n        };\n        return retry<void>(config);\n      },\n    );\n  }\n}\n"]}
+{"version":3,"file":"serviceBusRuleManager.js","sourceRoot":"","sources":["../../src/serviceBusRuleManager.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAKlC,gDAA6D;AAE7D,qCAAuD;AAEvD,uFAA0E;AAC1E,8CAAgD;AAChD,gDAAgE;AAEhE,yDAAyD;AAEzD,oDAA2D;AAoD3D;;GAEG;AACH,MAAa,yBAAyB;IAa1B;IACA;IACA;IAdV;;OAEG;IACK,SAAS,GAAY,KAAK,CAAC;IAC3B,IAAI,CAAS;IACd,UAAU,CAAS;IAE1B;;;OAGG;IACH,YACU,QAA2B,EAC3B,WAAmB,EACnB,gBAA8B,EAAE;QAFhC,aAAQ,GAAR,QAAQ,CAAmB;QAC3B,gBAAW,GAAX,WAAW,CAAQ;QACnB,kBAAa,GAAb,aAAa,CAAmB;QAExC,IAAA,wCAA4B,EAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,CAAC,UAAU,GAAG,WAAW,CAAC;QAC9B,IAAI,CAAC,IAAI,GAAG,IAAA,wBAAa,EAAC,aAAa,CAAC,CAAC;IAC3C,CAAC;IAED,IAAW,QAAQ;QACjB,OAAO,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,QAAQ,CAAC,wBAAwB,CAAC;IAClE,CAAC;IA4BD,KAAK,CAAC,UAAU,CACd,QAAgB,EAChB,MAA6C,EAC7C,4BAAmE,EACnE,UAA4B,EAAE;QAE9B,IAAI,aAAa,GAA8B,SAAS,CAAC;QACzD,IAAI,WAAyC,CAAC;QAC9C,IAAI,4BAA4B,EAAE,CAAC;YACjC,IAAI,IAAA,2CAAe,EAAC,4BAA4B,CAAC,EAAE,CAAC;gBAClD,oEAAoE;gBACpE,aAAa,GAAG,4BAA4B,CAAC;gBAC7C,WAAW,GAAG,OAAO,CAAC;YACxB,CAAC;iBAAM,CAAC;gBACN,sEAAsE;gBACtE,WAAW,GAAG,EAAE,GAAG,4BAA4B,EAAE,GAAG,OAAO,EAAE,CAAC;YAChE,CAAC;QACH,CAAC;QAED,OAAO,0BAAa,CAAC,QAAQ,CAC3B,kCAAkC,EAClC,WAAW,IAAI,EAAE,EACjB,KAAK,EAAE,cAAc,EAAE,EAAE;YACvB,MAAM,uBAAuB,GAAG,KAAK,IAAmB,EAAE;gBACxD,OAAO,IAAI,CAAC,QAAQ;qBACjB,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC;qBACrC,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,aAAa,EAAE,aAAa,EAAE;oBACvD,GAAG,cAAc;oBACjB,kBAAkB,EAAE,IAAI,CAAC,IAAI;oBAC7B,WAAW,EAAE,SAAS;oBACtB,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,WAAW;iBAC5C,CAAC,CAAC;YACP,CAAC,CAAC;YACF,MAAM,MAAM,GAAsB;gBAChC,SAAS,EAAE,uBAAuB;gBAClC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY;gBACxC,aAAa,EAAE,8BAAkB,CAAC,UAAU;gBAC5C,YAAY,EAAE,IAAI,CAAC,aAAa;gBAChC,WAAW,EAAE,cAAc,EAAE,WAAW;aACzC,CAAC;YACF,OAAO,IAAA,iBAAK,EAAO,MAAM,CAAC,CAAC;QAC7B,CAAC,CACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,QAAQ,CACpB,OAA+C;QAE/C,OAAO,0BAAa,CAAC,QAAQ,CAC3B,gCAAgC,EAChC,OAAO,IAAI,EAAE,EACb,KAAK,EAAE,cAAc,EAAE,EAAE;YACvB,MAAM,wBAAwB,GAAG,KAAK,IAA+B,EAAE;gBACrE,OAAO,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,QAAQ,CAAC;oBAClE,GAAG,cAAc;oBACjB,kBAAkB,EAAE,IAAI,CAAC,IAAI;oBAC7B,WAAW,EAAE,UAAU;oBACvB,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,WAAW;iBAC5C,CAAC,CAAC;YACL,CAAC,CAAC;YACF,MAAM,MAAM,GAAkC;gBAC5C,SAAS,EAAE,wBAAwB;gBACnC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY;gBACxC,aAAa,EAAE,8BAAkB,CAAC,UAAU;gBAC5C,YAAY,EAAE,IAAI,CAAC,aAAa;gBAChC,WAAW,EAAE,cAAc,EAAE,WAAW;aACzC,CAAC;YACF,OAAO,IAAA,iBAAK,EAAmB,MAAM,CAAC,CAAC;QACzC,CAAC,CACF,CAAC;IACJ,CAAC;IAED;;;;;;;OAOG;IACI,SAAS,CACd,OAA0B;QAE1B,0BAAM,CAAC,OAAO,CAAC,qDAAqD,EAAE,OAAO,CAAC,CAAC;QAC/E,MAAM,WAAW,GAAoE;YACnF,aAAa,EAAE,CAAC;YAChB,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,EAAE;gBACvC,MAAM,GAAG,GAAG,WAAW,IAAI,GAAG,CAAC;gBAC/B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC;oBAChC,IAAI,EAAE,QAAQ;oBACd,QAAQ,EAAE,GAAG;oBACb,GAAG,OAAO;iBACX,CAAC,CAAC;gBACH,OAAO,KAAK,CAAC,MAAM;oBACjB,CAAC,CAAC;wBACE,IAAI,EAAE,KAAK;wBACX,YAAY,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;qBACrE;oBACH,CAAC,CAAC,SAAS,CAAC;YAChB,CAAC;SACF,CAAC;QAEF,OAAO,IAAA,mCAAqB,EAAC,WAAW,CAAC,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,QAAgB,EAAE,UAA4B,EAAE;QAC/D,OAAO,0BAAa,CAAC,QAAQ,CAC3B,kCAAkC,EAClC,OAAO,EACP,KAAK,EAAE,cAAc,EAAE,EAAE;YACvB,MAAM,0BAA0B,GAAG,KAAK,IAAmB,EAAE;gBAC3D,OAAO,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,UAAU,CAAC,QAAQ,EAAE;oBAC9E,GAAG,cAAc;oBACjB,kBAAkB,EAAE,IAAI,CAAC,IAAI;oBAC7B,WAAW,EAAE,YAAY;oBACzB,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,WAAW;iBAC5C,CAAC,CAAC;YACL,CAAC,CAAC;YACF,MAAM,MAAM,GAAsB;gBAChC,SAAS,EAAE,0BAA0B;gBACrC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY;gBACxC,aAAa,EAAE,8BAAkB,CAAC,UAAU;gBAC5C,YAAY,EAAE,IAAI,CAAC,aAAa;gBAChC,WAAW,EAAE,cAAc,EAAE,WAAW;aACzC,CAAC;YACF,OAAO,IAAA,iBAAK,EAAO,MAAM,CAAC,CAAC;QAC7B,CAAC,CACF,CAAC;IACJ,CAAC;CACF;AA3LD,8DA2LC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { OperationOptionsBase } from \"./modelsToBeSharedWithEventHubs.js\";\nimport type { ConnectionContext } from \"./connectionContext.js\";\nimport type { RetryConfig, RetryOptions } from \"@azure/core-amqp\";\nimport { RetryOperationType, retry } from \"@azure/core-amqp\";\nimport type { CorrelationRuleFilter } from \"./core/managementClient.js\";\nimport { ruleManagerLogger as logger } from \"./log.js\";\nimport type { RuleProperties, SqlRuleAction } from \"./serializers/ruleResourceSerializer.js\";\nimport { isSqlRuleAction } from \"./serializers/ruleResourceSerializer.js\";\nimport { getUniqueName } from \"./util/utils.js\";\nimport { throwErrorIfConnectionClosed } from \"./util/errors.js\";\nimport type { SqlRuleFilter } from \"./serializers/ruleResourceSerializer.js\";\nimport { tracingClient } from \"./diagnostics/tracing.js\";\nimport type { PagedAsyncIterableIterator, PagedResult } from \"@azure/core-paging\";\nimport { getPagedAsyncIterator } from \"@azure/core-paging\";\nimport type { OperationOptions } from \"@azure/core-client\";\nimport type { ListRequestOptions } from \"./serviceBusAtomManagementClient.js\";\n\n/**\n * Allows rules for a subscription to be managed. This rule manager requires only Listen claims, whereas the\n * {@link ServiceBusAdministrationClient} requires Manage claims.\n */\nexport interface ServiceBusRuleManager {\n  /**\n   * Adds a rule to the current subscription to filter the messages reaching from topic to the subscription.\n   *\n   * @param ruleName - the name of the rule\n   * @param filter - the filter expression that the rule evaluates.\n   * @param options - The options that can be used to abort, trace and control other configurations on the request.\n   */\n  createRule(\n    ruleName: string,\n    filter: SqlRuleFilter | CorrelationRuleFilter,\n    options?: OperationOptionsBase,\n  ): Promise<void>;\n  /**\n   * Adds a rule to the current subscription to filter the messages reaching from topic to the subscription.\n   *\n   * @param ruleName - the name of the rule\n   * @param filter - the filter expression that the rule evaluates.\n   * @param ruleAction - The SQL like expression that can be executed on the message should the associated filter apply.\n   * @param options - The options that can be used to abort, trace and control other configurations on the request.\n   */\n  createRule(\n    ruleName: string,\n    filter: SqlRuleFilter | CorrelationRuleFilter,\n    ruleAction?: SqlRuleAction,\n    options?: OperationOptionsBase,\n  ): Promise<void>;\n  /**\n   * Deletes a rule.\n   *\n   * @param ruleName - the name of the rule\n   */\n  deleteRule(ruleName: string, options?: OperationOptionsBase): Promise<void>;\n  /**\n   * Returns an async iterable iterator to list all the rules\n   * under the specified subscription.\n   *\n   * .byPage() returns an async iterable iterator to list the rules in pages.\n   *\n   * @returns An asyncIterableIterator that supports paging.\n   */\n  listRules(options?: OperationOptions): PagedAsyncIterableIterator<RuleProperties>;\n}\n\n/**\n * @internal\n */\nexport class ServiceBusRuleManagerImpl implements ServiceBusRuleManager {\n  /**\n   * Denotes if close() was called on this sender\n   */\n  private _isClosed: boolean = false;\n  private name: string;\n  public entityPath: string;\n\n  /**\n   * @internal\n   * @throws Error if the underlying connection is closed.\n   */\n  constructor(\n    private _context: ConnectionContext,\n    private _entityPath: string,\n    private _retryOptions: RetryOptions = {},\n  ) {\n    throwErrorIfConnectionClosed(_context);\n    this.entityPath = _entityPath;\n    this.name = getUniqueName(\"ruleManager\");\n  }\n\n  public get isClosed(): boolean {\n    return this._isClosed || this._context.wasConnectionCloseCalled;\n  }\n\n  /**\n   * Adds a rule to the current subscription to filter the messages reaching from topic to the subscription.\n   *\n   * @param ruleName - the name of the rule\n   * @param filter - the filter expression that the rule evaluates.\n   * @param options - The options that can be used to abort, trace and control other configurations on the request.\n   */\n  createRule(\n    ruleName: string,\n    filter: SqlRuleFilter | CorrelationRuleFilter,\n    options?: OperationOptions,\n  ): Promise<void>;\n  /**\n   * Adds a rule to the current subscription to filter the messages reaching from topic to the subscription.\n   *\n   * @param ruleName - the name of the rule\n   * @param filter - the filter expression that the rule evaluates.\n   * @param ruleAction - The SQL like expression that can be executed on the message should the associated filter apply.\n   * @param options - The options that can be used to abort, trace and control other configurations on the request.\n   */\n  createRule(\n    ruleName: string,\n    filter: SqlRuleFilter | CorrelationRuleFilter,\n    ruleAction?: SqlRuleAction,\n    options?: OperationOptions,\n  ): Promise<void>;\n  async createRule(\n    ruleName: string,\n    filter: SqlRuleFilter | CorrelationRuleFilter,\n    ruleActionOrOperationOptions?: SqlRuleAction | OperationOptionsBase,\n    options: OperationOptions = {},\n  ): Promise<void> {\n    let sqlRuleAction: SqlRuleAction | undefined = undefined;\n    let operOptions: OperationOptions | undefined;\n    if (ruleActionOrOperationOptions) {\n      if (isSqlRuleAction(ruleActionOrOperationOptions)) {\n        // Overload#2 - where the sqlExpression in the ruleAction is defined\n        sqlRuleAction = ruleActionOrOperationOptions;\n        operOptions = options;\n      } else {\n        // Overload#1 - where the sqlExpression in the ruleAction is undefined\n        operOptions = { ...ruleActionOrOperationOptions, ...options };\n      }\n    }\n\n    return tracingClient.withSpan(\n      \"ServiceBusRuleManager.createRule\",\n      operOptions ?? {},\n      async (updatedOptions) => {\n        const addRuleOperationPromise = async (): Promise<void> => {\n          return this._context\n            .getManagementClient(this._entityPath)\n            .addRule(ruleName, filter, sqlRuleAction?.sqlExpression, {\n              ...updatedOptions,\n              associatedLinkName: this.name,\n              requestName: \"addRule\",\n              timeoutInMs: this._retryOptions.timeoutInMs,\n            });\n        };\n        const config: RetryConfig<void> = {\n          operation: addRuleOperationPromise,\n          connectionId: this._context.connectionId,\n          operationType: RetryOperationType.management,\n          retryOptions: this._retryOptions,\n          abortSignal: updatedOptions?.abortSignal,\n        };\n        return retry<void>(config);\n      },\n    );\n  }\n\n  /**\n   * Get all rules associated with the subscription.\n   */\n  private async getRules(\n    options?: ListRequestOptions & OperationOptions,\n  ): Promise<RuleProperties[]> {\n    return tracingClient.withSpan(\n      \"ServiceBusRuleManager.getRules\",\n      options ?? {},\n      async (updatedOptions) => {\n        const getRulesOperationPromise = async (): Promise<RuleProperties[]> => {\n          return this._context.getManagementClient(this._entityPath).getRules({\n            ...updatedOptions,\n            associatedLinkName: this.name,\n            requestName: \"getRules\",\n            timeoutInMs: this._retryOptions.timeoutInMs,\n          });\n        };\n        const config: RetryConfig<RuleProperties[]> = {\n          operation: getRulesOperationPromise,\n          connectionId: this._context.connectionId,\n          operationType: RetryOperationType.management,\n          retryOptions: this._retryOptions,\n          abortSignal: updatedOptions?.abortSignal,\n        };\n        return retry<RuleProperties[]>(config);\n      },\n    );\n  }\n\n  /**\n   * Returns an async iterable iterator to list all the rules\n   * under the specified subscription.\n   *\n   * .byPage() returns an async iterable iterator to list the rules in pages.\n   *\n   * @returns An asyncIterableIterator that supports paging.\n   */\n  public listRules(\n    options?: OperationOptions,\n  ): PagedAsyncIterableIterator<RuleProperties, RuleProperties[], { maxPageSize?: number }> {\n    logger.verbose(`Performing operation - listRules() with options: %j`, options);\n    const pagedResult: PagedResult<RuleProperties[], { maxPageSize?: number }, number> = {\n      firstPageLink: 0,\n      getPage: async (pageLink, maxPageSize) => {\n        const top = maxPageSize ?? 100;\n        const rules = await this.getRules({\n          skip: pageLink,\n          maxCount: top,\n          ...options,\n        });\n        return rules.length\n          ? {\n              page: rules,\n              nextPageLink: rules.length > 0 ? pageLink + rules.length : undefined,\n            }\n          : undefined;\n      },\n    };\n\n    return getPagedAsyncIterator(pagedResult);\n  }\n\n  /**\n   * Deletes a rule.\n   */\n  async deleteRule(ruleName: string, options: OperationOptions = {}): Promise<void> {\n    return tracingClient.withSpan(\n      \"ServiceBusRuleManager.deleteRule\",\n      options,\n      async (updatedOptions) => {\n        const removeRuleOperationPromise = async (): Promise<void> => {\n          return this._context.getManagementClient(this._entityPath).removeRule(ruleName, {\n            ...updatedOptions,\n            associatedLinkName: this.name,\n            requestName: \"removeRule\",\n            timeoutInMs: this._retryOptions.timeoutInMs,\n          });\n        };\n        const config: RetryConfig<void> = {\n          operation: removeRuleOperationPromise,\n          connectionId: this._context.connectionId,\n          operationType: RetryOperationType.management,\n          retryOptions: this._retryOptions,\n          abortSignal: updatedOptions?.abortSignal,\n        };\n        return retry<void>(config);\n      },\n    );\n  }\n}\n"]}

package/dist/commonjs/session/messageSession.js

@@ -22,15 +22,127 @@ const core_util_1 = require("@azure/core-util");
  * Describes the receiver for a Message Session.
  */
 class MessageSession extends linkEntity_js_1.LinkEntity {
+    identifier;
+    _providedSessionId;
+    /**
+     * Provides the duration until which the session is locked.
+     */
+    sessionLockedUntilUtc;
+    /**
+     * The sessionId for the message session. Empty string is valid sessionId.
+     */
+    sessionId;
+    /**
+     * The maximum number of concurrent sessions that the
+     * client should initiate.
+     * - **Default**: `1`.
+     */
+    maxConcurrentSessions;
+    /**
+     * The maximum number of messages that should be
+     * processed concurrently in a session while in streaming mode. Once this limit has been reached,
+     * more messages will not be received until the user's message handler has completed processing current message.
+     * - **Default**: `1` (message in a session at a time).
+     */
+    maxConcurrentCalls = 1;
+    /**
+     * The mode in which messages should be received.
+     * Default: ReceiveMode.peekLock
+     */
+    receiveMode;
+    /**
+     * Indicates whether `Message.complete()` should be called
+     * automatically after the message processing is complete while receiving messages with handlers.
+     * Default: false.
+     */
+    autoComplete;
+    /**
+     * The maximum duration within which the
+     * lock will be renewed automatically. This value should be greater than the longest message
+     * lock duration; for example, the `lockDuration` property on the received message.
+     *
+     * Default: `300 * 1000` (5 minutes);
+     */
+    maxAutoRenewDurationInMs;
+    /**
+     * Should lock renewal happen automatically.
+     */
+    autoRenewLock;
     /**
      * Denotes if we are currently receiving messages
      */
     get isReceivingMessages() {
         return this._batchingReceiverLite.isReceivingMessages || this._isReceivingMessagesForSubscriber;
     }
+    _batchingReceiverLite;
+    _isReceivingMessagesForSubscriber;
+    /**
+     * Maintains a map of deliveries that
+     * are being actively disposed. It acts as a store for correlating the responses received for
+     * active dispositions.
+     */
+    _deliveryDispositionMap = new Map();
+    /**
+     * The message handler provided by the user that will
+     * be wrapped inside _onAmqpMessage.
+     */
+    _onMessage;
+    /**
+     * The error handler provided by the user that will be wrapped
+     * inside _onAmqpError.
+     */
+    _onError;
+    /**
+     * If the user provided error handler is present then it will
+     * notify the user's error handler about the error.
+     */
+    _notifyError;
+    /**
+     * The message handler that will be set as the handler on the
+     * underlying rhea receiver for the "receiver_close" event.
+     */
+    _onAmqpClose;
+    /**
+     * The message handler that will be set as the handler on
+     * the underlying rhea receiver's session for the "session_error" event.
+     */
+    _onSessionError;
+    /**
+     * The message handler that will be set as the handler on
+     * the underlying rhea receiver's session for the "session_close" event.
+     */
+    _onSessionClose;
+    /**
+     * The message handler that will be set as the handler on the
+     * underlying rhea receiver for the "receiver_error" event.
+     */
+    _onAmqpError;
+    /**
+     * The message handler that will be set as the handler on the
+     * underlying rhea receiver for the "settled" event.
+     */
+    _onSettled;
+    /**
+     * The session lock renewal timer that keeps
+     * track of when the MessageSession is due for session lock renewal.
+     */
+    _sessionLockRenewalTimer;
+    _totalAutoLockRenewDuration;
+    /**
+     * Whether to prevent the client from running JSON.parse() on the message body when receiving the message.
+     */
+    skipParsingBodyAsJson;
+    /**
+     * Whether to skip converting Date type on properties of message annotations
+     * or application properties into numbers when receiving the message. By
+     * default, properties of Date type is converted into UNIX epoch number for
+     * compatibility.
+     */
+    skipConvertingDate;
     get receiverHelper() {
         return this._receiverHelper;
     }
+    _receiverHelper;
     /**
      * Ensures that the session lock is renewed before it expires. The lock will not be renewed for
      * more than the configured totalAutoLockRenewDuration.
@@ -62,12 +174,11 @@ class MessageSession extends linkEntity_js_1.LinkEntity {
         }
     }
     async createRheaLink(options, _abortSignal) {
-        var _a, _b;
         this._lastSBError = undefined;
         let errorMessage = "";
         const link = await this._context.connection.createReceiver(options);
         this._intermediateLink = link;
-        const receivedSessionId = (_b = (_a = link.source) === null || _a === void 0 ? void 0 : _a.filter) === null || _b === void 0 ? void 0 : _b[core_amqp_1.Constants.sessionFilterName];
+        const receivedSessionId = link.source?.filter?.[core_amqp_1.Constants.sessionFilterName];
         if (this._providedSessionId === undefined && receivedSessionId === undefined) {
             // When we ask for any sessions (passing option of session-filter: undefined),
             // but don't receive one back, check whether service has sent any error.
@@ -111,14 +222,13 @@ class MessageSession extends linkEntity_js_1.LinkEntity {
      * Creates a new AMQP receiver under a new AMQP session.
      */
     async _init(opts = {}) {
-        var _a, _b;
         try {
             const sessionOptions = this._createMessageSessionOptions(this.identifier, opts.timeoutInMs);
             await this.initLink(sessionOptions, opts.abortSignal);
             if (!this.link) {
                 throw new Error("INTERNAL ERROR: failed to create receiver but without an error.");
             }
-            const receivedSessionId = (_b = (_a = this.link.source) === null || _a === void 0 ? void 0 : _a.filter) === null || _b === void 0 ? void 0 : _b[core_amqp_1.Constants.sessionFilterName];
+            const receivedSessionId = this.link.source?.filter?.[core_amqp_1.Constants.sessionFilterName];
             if (this._providedSessionId === undefined)
                 this.sessionId = receivedSessionId;
             this.sessionLockedUntilUtc = (0, utils_js_1.convertTicksToDate)(this.link.properties["com.microsoft:locked-until-utc"]);
@@ -170,6 +280,9 @@ class MessageSession extends linkEntity_js_1.LinkEntity {
         }, timeoutInMs);
         return rcvrOptions;
     }
+    _retryOptions;
+    _lastSBError;
+    _intermediateLink;
     /**
      * Constructs a MessageSession instance which lets you receive messages as batches
      * or via callbacks using subscribe.
@@ -185,19 +298,6 @@ class MessageSession extends linkEntity_js_1.LinkEntity {
         });
         this.identifier = identifier;
         this._providedSessionId = _providedSessionId;
-        /**
-         * The maximum number of messages that should be
-         * processed concurrently in a session while in streaming mode. Once this limit has been reached,
-         * more messages will not be received until the user's message handler has completed processing current message.
-         * - **Default**: `1` (message in a session at a time).
-         */
-        this.maxConcurrentCalls = 1;
-        /**
-         * Maintains a map of deliveries that
-         * are being actively disposed. It acts as a store for correlating the responses received for
-         * active dispositions.
-         */
-        this._deliveryDispositionMap = new Map();
         this._receiverHelper = new receiverHelper_js_1.ReceiverHelper(() => ({
             receiver: this.link,
             logPrefix: this.logPrefix,
@@ -357,10 +457,9 @@ class MessageSession extends linkEntity_js_1.LinkEntity {
         this._subscribeImpl(onMessage, onError, options);
     }
     _subscribeImpl(onMessage, onError, options) {
-        var _a;
         if (!options)
             options = {};
-        if ((_a = options.abortSignal) === null || _a === void 0 ? void 0 : _a.aborted) {
+        if (options.abortSignal?.aborted) {
             throw new abort_controller_1.AbortError(core_amqp_1.StandardAbortMessage);
         }
         this._isReceivingMessagesForSubscriber = true;
@@ -514,9 +613,12 @@ class MessageSession extends linkEntity_js_1.LinkEntity {
      */
     async receiveMessages(maxMessageCount, maxWaitTimeInMs, maxTimeAfterFirstMessageInMs, options) {
         try {
-            return await this._batchingReceiverLite.receiveMessages(Object.assign({ maxMessageCount,
+            return await this._batchingReceiverLite.receiveMessages({
+                maxMessageCount,
                 maxWaitTimeInMs,
-                maxTimeAfterFirstMessageInMs }, options));
+                maxTimeAfterFirstMessageInMs,
+                ...options,
+            });
         }
         catch (error) {
             log_js_1.receiverLogger.logError(error, `${this.logPrefix} Rejecting receiveMessages() with error`);
@@ -594,7 +696,11 @@ class MessageSession extends linkEntity_js_1.LinkEntity {
             else if (operation === serviceBusMessage_js_1.DispositionType.deadletter) {
                 const error = {
                     condition: core_amqp_1.Constants.deadLetterName,
-                    info: Object.assign(Object.assign({}, options.propertiesToModify), { DeadLetterReason: options.deadLetterReason, DeadLetterErrorDescription: options.deadLetterDescription }),
+                    info: {
+                        ...options.propertiesToModify,
+                        DeadLetterReason: options.deadLetterReason,
+                        DeadLetterErrorDescription: options.deadLetterDescription,
+                    },
                 };
                 delivery.reject(error);
             }
@@ -607,14 +713,13 @@ class MessageSession extends linkEntity_js_1.LinkEntity {
      * @param options - Options that can be provided while creating the MessageSession.
      */
     static async create(identifier, context, entityPath, sessionId, options) {
-        var _a, _b;
         (0, errors_js_1.throwErrorIfConnectionClosed)(context);
         const messageSession = new MessageSession(identifier, context, entityPath, sessionId, options);
         let timeoutInMs;
         // Only passing client timeout in link properties for accepting next available
         // session as this is the only long-polling scenario.
         if (sessionId === undefined) {
-            timeoutInMs = (_b = (_a = options.retryOptions) === null || _a === void 0 ? void 0 : _a.timeoutInMs) !== null && _b !== void 0 ? _b : core_amqp_1.Constants.defaultOperationTimeoutInMs;
+            timeoutInMs = options.retryOptions?.timeoutInMs ?? core_amqp_1.Constants.defaultOperationTimeoutInMs;
             // The number of milliseconds to use as the basis for calculating a random jitter amount
             // opening receiver links. This is intended to ensure that multiple
             // session operations don't timeout at the same exact moment.
@@ -641,7 +746,7 @@ class MessageSession extends linkEntity_js_1.LinkEntity {
             }
         }
         await messageSession._init({
-            abortSignal: options === null || options === void 0 ? void 0 : options.abortSignal,
+            abortSignal: options?.abortSignal,
             timeoutInMs,
         });
         return messageSession;