@azure/storage-queue 12.20.0-alpha.20230615.1 → 12.20.0-alpha.20230616.2

package/dist-esm/storage-queue/src/QueueClient.js

@@ -2,13 +2,12 @@
 // Licensed under the MIT license.
 import { isTokenCredential } from "@azure/core-auth";
 import { isNode } from "@azure/core-util";
-import { SpanStatusCode } from "@azure/core-tracing";
 import { newPipeline, Pipeline } from "../../storage-blob/src/Pipeline";
 import { StorageClient, getStorageClientContext } from "./StorageClient";
 import { appendToURLPath, extractConnectionStringParts, isIpEndpointStyle, truncatedISO8061Date, appendToURLQuery, assertResponse, } from "./utils/utils.common";
 import { StorageSharedKeyCredential } from "../../storage-blob/src/credentials/StorageSharedKeyCredential";
 import { AnonymousCredential } from "../../storage-blob/src/credentials/AnonymousCredential";
-import { createSpan } from "./utils/tracing";
+import { tracingClient } from "./utils/tracing";
 import { generateQueueSASQueryParameters } from "./QueueSASSignatureValues";
 import { getDefaultProxySettings } from "@azure/core-rest-pipeline";
 /**
@@ -111,20 +110,9 @@ export class QueueClient extends StorageClient {
      * ```
      */
     async create(options = {}) {
-        const { span, updatedOptions } = createSpan("QueueClient-create", options);
-        try {
+        return tracingClient.withSpan("QueueClient-create", options, async (updatedOptions) => {
             return assertResponse(await this.queueContext.create(updatedOptions));
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Creates a new queue under the specified account if it doesn't already exist.
@@ -134,35 +122,25 @@ export class QueueClient extends StorageClient {
      * @param options -
      */
     async createIfNotExists(options = {}) {
-        var _a, _b;
-        const { span, updatedOptions } = createSpan("QueueClient-createIfNotExists", options);
-        try {
-            const response = await this.create(updatedOptions);
-            // When a queue with the specified name already exists, the Queue service checks the metadata associated with the existing queue.
-            // If the existing metadata is identical to the metadata specified on the Create Queue request, status code 204 (No Content) is returned.
-            // If the existing metadata does not match, the operation fails and status code 409 (Conflict) is returned.
-            if (response._response.status === 204) {
-                return Object.assign({ succeeded: false }, response);
+        return tracingClient.withSpan("QueueClient-createIfNotExists", options, async (updatedOptions) => {
+            var _a, _b;
+            try {
+                const response = await this.create(updatedOptions);
+                // When a queue with the specified name already exists, the Queue service checks the metadata associated with the existing queue.
+                // If the existing metadata is identical to the metadata specified on the Create Queue request, status code 204 (No Content) is returned.
+                // If the existing metadata does not match, the operation fails and status code 409 (Conflict) is returned.
+                if (response._response.status === 204) {
+                    return Object.assign({ succeeded: false }, response);
+                }
+                return Object.assign({ succeeded: true }, response);
             }
-            return Object.assign({ succeeded: true }, response);
-        }
-        catch (e) {
-            if (((_a = e.details) === null || _a === void 0 ? void 0 : _a.errorCode) === "QueueAlreadyExists") {
-                span.setStatus({
-                    code: SpanStatusCode.ERROR,
-                    message: "Expected exception when creating a queue only if it does not already exist.",
-                });
-                return Object.assign(Object.assign({ succeeded: false }, (_b = e.response) === null || _b === void 0 ? void 0 : _b.parsedHeaders), { _response: e.response });
+            catch (e) {
+                if (((_a = e.details) === null || _a === void 0 ? void 0 : _a.errorCode) === "QueueAlreadyExists") {
+                    return Object.assign(Object.assign({ succeeded: false }, (_b = e.response) === null || _b === void 0 ? void 0 : _b.parsedHeaders), { _response: e.response });
+                }
+                throw e;
             }
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Deletes the specified queue permanently if it exists.
@@ -171,29 +149,19 @@ export class QueueClient extends StorageClient {
      * @param options -
      */
     async deleteIfExists(options = {}) {
-        var _a, _b;
-        const { span, updatedOptions } = createSpan("QueueClient-deleteIfExists", options);
-        try {
-            const res = await this.delete(updatedOptions);
-            return Object.assign({ succeeded: true }, res);
-        }
-        catch (e) {
-            if (((_a = e.details) === null || _a === void 0 ? void 0 : _a.errorCode) === "QueueNotFound") {
-                span.setStatus({
-                    code: SpanStatusCode.ERROR,
-                    message: "Expected exception when deleting a queue only if it exists.",
-                });
-                return Object.assign(Object.assign({ succeeded: false }, (_b = e.response) === null || _b === void 0 ? void 0 : _b.parsedHeaders), { _response: e.response });
+        return tracingClient.withSpan("QueueClient-deleteIfExists", options, async (updatedOptions) => {
+            var _a, _b;
+            try {
+                const res = await this.delete(updatedOptions);
+                return Object.assign({ succeeded: true }, res);
             }
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+            catch (e) {
+                if (((_a = e.details) === null || _a === void 0 ? void 0 : _a.errorCode) === "QueueNotFound") {
+                    return Object.assign(Object.assign({ succeeded: false }, (_b = e.response) === null || _b === void 0 ? void 0 : _b.parsedHeaders), { _response: e.response });
+                }
+                throw e;
+            }
+        });
     }
     /**
      * Deletes the specified queue permanently.
@@ -212,23 +180,12 @@ export class QueueClient extends StorageClient {
      * ```
      */
     async delete(options = {}) {
-        const { span, updatedOptions } = createSpan("QueueClient-delete", options);
-        try {
+        return tracingClient.withSpan("QueueClient-delete", options, async (updatedOptions) => {
             return assertResponse(await this.queueContext.delete({
                 abortSignal: options.abortSignal,
                 tracingOptions: updatedOptions.tracingOptions,
             }));
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Returns true if the specified queue exists; false otherwise.
@@ -240,31 +197,18 @@ export class QueueClient extends StorageClient {
      * @param options - options to Exists operation.
      */
     async exists(options = {}) {
-        const { span, updatedOptions } = createSpan("QueueClient-exists", options);
-        try {
-            await this.getProperties({
-                abortSignal: options.abortSignal,
-                tracingOptions: updatedOptions.tracingOptions,
-            });
-            return true;
-        }
-        catch (e) {
-            if (e.statusCode === 404) {
-                span.setStatus({
-                    code: SpanStatusCode.ERROR,
-                    message: "Expected exception when checking queue existence",
-                });
-                return false;
+        return tracingClient.withSpan("QueueClient-exists", options, async (updatedOptions) => {
+            try {
+                await this.getProperties(updatedOptions);
+                return true;
             }
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+            catch (e) {
+                if (e.statusCode === 404) {
+                    return false;
+                }
+                throw e;
+            }
+        });
     }
     /**
      * Gets all user-defined metadata and system properties for the specified
@@ -280,23 +224,9 @@ export class QueueClient extends StorageClient {
      * @returns Response data for the Queue get properties operation.
      */
     async getProperties(options = {}) {
-        const { span, updatedOptions } = createSpan("QueueClient-getProperties", options);
-        try {
-            return assertResponse(await this.queueContext.getProperties({
-                abortSignal: options.abortSignal,
-                tracingOptions: updatedOptions.tracingOptions,
-            }));
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        return tracingClient.withSpan("QueueClient-getProperties", options, async (updatedOptions) => {
+            return assertResponse(await this.queueContext.getProperties(updatedOptions));
+        });
     }
     /**
      * Sets one or more user-defined name-value pairs for the specified queue.
@@ -310,24 +240,9 @@ export class QueueClient extends StorageClient {
      * @returns Response data for the Queue set metadata operation.
      */
     async setMetadata(metadata, options = {}) {
-        const { span, updatedOptions } = createSpan("QueueClient-setMetadata", options);
-        try {
-            return assertResponse(await this.queueContext.setMetadata({
-                abortSignal: options.abortSignal,
-                metadata,
-                tracingOptions: updatedOptions.tracingOptions,
-            }));
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        return tracingClient.withSpan("QueueClient-setMetadata", options, async (updatedOptions) => {
+            return assertResponse(await this.queueContext.setMetadata(Object.assign(Object.assign({}, updatedOptions), { metadata })));
+        });
     }
     /**
      * Gets details about any stored access policies specified on the queue that may be used with Shared Access Signatures.
@@ -341,8 +256,7 @@ export class QueueClient extends StorageClient {
      * @returns Response data for the Queue get access policy operation.
      */
     async getAccessPolicy(options = {}) {
-        const { span, updatedOptions } = createSpan("QueueClient-getAccessPolicy", options);
-        try {
+        return tracingClient.withSpan("QueueClient-getAccessPolicy", options, async (updatedOptions) => {
             const response = assertResponse(await this.queueContext.getAccessPolicy({
                 abortSignal: options.abortSignal,
                 tracingOptions: updatedOptions.tracingOptions,
@@ -375,17 +289,7 @@ export class QueueClient extends StorageClient {
                 });
             }
             return res;
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Sets stored access policies for the queue that may be used with Shared Access Signatures.
@@ -396,8 +300,7 @@ export class QueueClient extends StorageClient {
      * @returns Response data for the Queue set access policy operation.
      */
     async setAccessPolicy(queueAcl, options = {}) {
-        const { span, updatedOptions } = createSpan("QueueClient-setAccessPolicy", options);
-        try {
+        return tracingClient.withSpan("QueueClient-setAccessPolicy", options, async (updatedOptions) => {
             const acl = [];
             for (const identifier of queueAcl || []) {
                 acl.push({
@@ -413,22 +316,8 @@ export class QueueClient extends StorageClient {
                     id: identifier.id,
                 });
             }
-            return assertResponse(await this.queueContext.setAccessPolicy({
-                abortSignal: options.abortSignal,
-                queueAcl: acl,
-                tracingOptions: updatedOptions.tracingOptions,
-            }));
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+            return assertResponse(await this.queueContext.setAccessPolicy(Object.assign(Object.assign({}, updatedOptions), { queueAcl: acl })));
+        });
     }
     /**
      * Clear deletes all messages from a queue.
@@ -438,23 +327,9 @@ export class QueueClient extends StorageClient {
      * @returns Response data for the clear messages operation.
      */
     async clearMessages(options = {}) {
-        const { span, updatedOptions } = createSpan("QueueClient-clearMessages", options);
-        try {
-            return assertResponse(await this.messagesContext.clear({
-                abortSignal: options.abortSignal,
-                tracingOptions: updatedOptions.tracingOptions,
-            }));
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        return tracingClient.withSpan("QueueClient-clearMessages", options, async (updatedOptions) => {
+            return assertResponse(await this.messagesContext.clear(updatedOptions));
+        });
     }
     /**
      * sendMessage adds a new message to the back of a queue. The visibility timeout specifies how long
@@ -478,8 +353,7 @@ export class QueueClient extends StorageClient {
      * ```
      */
     async sendMessage(messageText, options = {}) {
-        const { span, updatedOptions } = createSpan("QueueClient-sendMessage", options);
-        try {
+        return tracingClient.withSpan("QueueClient-sendMessage", options, async (updatedOptions) => {
             const response = assertResponse(await this.messagesContext.enqueue({
                 messageText: messageText,
             }, updatedOptions));
@@ -497,17 +371,7 @@ export class QueueClient extends StorageClient {
                 insertedOn: item.insertedOn,
                 expiresOn: item.expiresOn,
             };
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * receiveMessages retrieves one or more messages from the front of the queue.
@@ -535,8 +399,7 @@ export class QueueClient extends StorageClient {
      * ```
      */
     async receiveMessages(options = {}) {
-        const { span, updatedOptions } = createSpan("QueueClient-receiveMessages", options);
-        try {
+        return tracingClient.withSpan("QueueClient-receiveMessages", options, async (updatedOptions) => {
             const response = assertResponse(await this.messagesContext.dequeue(updatedOptions));
             const res = {
                 _response: response._response,
@@ -551,17 +414,7 @@ export class QueueClient extends StorageClient {
                 res.receivedMessageItems.push(item);
             }
             return res;
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * peekMessages retrieves one or more messages from the front of the queue but does not alter the visibility of the message.
@@ -578,8 +431,7 @@ export class QueueClient extends StorageClient {
      * ```
      */
     async peekMessages(options = {}) {
-        const { span, updatedOptions } = createSpan("QueueClient-peekMessages", options);
-        try {
+        return tracingClient.withSpan("QueueClient-peekMessages", options, async (updatedOptions) => {
             const response = assertResponse(await this.messagesContext.peek(updatedOptions));
             const res = {
                 _response: response._response,
@@ -594,17 +446,7 @@ export class QueueClient extends StorageClient {
                 res.peekedMessageItems.push(item);
             }
             return res;
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * deleteMessage permanently removes the specified message from its queue.
@@ -616,23 +458,9 @@ export class QueueClient extends StorageClient {
      * @returns Response data for the delete message operation.
      */
     async deleteMessage(messageId, popReceipt, options = {}) {
-        const { span, updatedOptions } = createSpan("QueueClient-deleteMessage", options);
-        try {
-            return assertResponse(await this.getMessageIdContext(messageId).delete(popReceipt, {
-                abortSignal: options.abortSignal,
-                tracingOptions: updatedOptions.tracingOptions,
-            }));
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        return tracingClient.withSpan("QueueClient-deleteMessage", options, async (updatedOptions) => {
+            return assertResponse(await this.getMessageIdContext(messageId).delete(popReceipt, updatedOptions));
+        });
     }
     /**
      * Update changes a message's visibility timeout and contents.
@@ -652,28 +480,17 @@ export class QueueClient extends StorageClient {
      * @returns Response data for the update message operation.
      */
     async updateMessage(messageId, popReceipt, message, visibilityTimeout, options = {}) {
-        const { span, updatedOptions } = createSpan("QueueClient-updateMessage", options);
-        let queueMessage = undefined;
-        if (message !== undefined) {
-            queueMessage = { messageText: message };
-        }
-        try {
+        return tracingClient.withSpan("QueueClient-updateMessage", options, async (updatedOptions) => {
+            let queueMessage = undefined;
+            if (message !== undefined) {
+                queueMessage = { messageText: message };
+            }
             return assertResponse(await this.getMessageIdContext(messageId).update(popReceipt, visibilityTimeout || 0, {
                 abortSignal: options.abortSignal,
                 tracingOptions: updatedOptions.tracingOptions,
                 queueMessage,
             }));
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     getQueueNameFromUrl() {
         let queueName;