@azure/storage-blob 12.13.0 → 12.20.0-alpha.20230306.3

package/dist-esm/storage-blob/src/BlobServiceClient.js

@@ -1,17 +1,16 @@
 import { __asyncDelegator, __asyncGenerator, __asyncValues, __await } from "tslib";
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT license.
-import { isTokenCredential, isNode, getDefaultProxySettings, } from "@azure/core-http";
-import { SpanStatusCode } from "@azure/core-tracing";
-import { Container, Service } from "./generated/src/operations";
+import { isTokenCredential } from "@azure/core-auth";
+import { getDefaultProxySettings } from "@azure/core-rest-pipeline";
+import { isNode } from "@azure/core-util";
 import { newPipeline, isPipelineLike } from "./Pipeline";
 import { ContainerClient, } from "./ContainerClient";
 import { appendToURLPath, appendToURLQuery, extractConnectionStringParts, toTags, } from "./utils/utils.common";
 import { StorageSharedKeyCredential } from "./credentials/StorageSharedKeyCredential";
 import { AnonymousCredential } from "./credentials/AnonymousCredential";
-import "@azure/core-paging";
-import { truncatedISO8061Date } from "./utils/utils.common";
-import { convertTracingToRequestOptionsBase, createSpan } from "./utils/tracing";
+import { truncatedISO8061Date, assertResponse } from "./utils/utils.common";
+import { tracingClient } from "./utils/tracing";
 import { BlobBatchClient } from "./BlobBatchClient";
 import { StorageClient } from "./StorageClient";
 import { AccountSASPermissions } from "./sas/AccountSASPermissions";
@@ -40,7 +39,7 @@ export class BlobServiceClient extends StorageClient {
             pipeline = newPipeline(new AnonymousCredential(), options);
         }
         super(url, pipeline);
-        this.serviceContext = new Service(this.storageClientContext);
+        this.serviceContext = this.storageClientContext.service;
     }
     /**
      *
@@ -104,25 +103,14 @@ export class BlobServiceClient extends StorageClient {
      * @returns Container creation response and the corresponding container client.
      */
     async createContainer(containerName, options = {}) {
-        const { span, updatedOptions } = createSpan("BlobServiceClient-createContainer", options);
-        try {
+        return tracingClient.withSpan("BlobServiceClient-createContainer", options, async (updatedOptions) => {
             const containerClient = this.getContainerClient(containerName);
             const containerCreateResponse = await containerClient.create(updatedOptions);
             return {
                 containerClient,
                 containerCreateResponse,
             };
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Deletes a Blob container.
@@ -132,21 +120,10 @@ export class BlobServiceClient extends StorageClient {
      * @returns Container deletion response.
      */
     async deleteContainer(containerName, options = {}) {
-        const { span, updatedOptions } = createSpan("BlobServiceClient-deleteContainer", options);
-        try {
+        return tracingClient.withSpan("BlobServiceClient-deleteContainer", options, async (updatedOptions) => {
             const containerClient = this.getContainerClient(containerName);
-            return await containerClient.delete(updatedOptions);
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+            return containerClient.delete(updatedOptions);
+        });
     }
     /**
      * Restore a previously deleted Blob container.
@@ -158,25 +135,17 @@ export class BlobServiceClient extends StorageClient {
      * @returns Container deletion response.
      */
     async undeleteContainer(deletedContainerName, deletedContainerVersion, options = {}) {
-        const { span, updatedOptions } = createSpan("BlobServiceClient-undeleteContainer", options);
-        try {
+        return tracingClient.withSpan("BlobServiceClient-undeleteContainer", options, async (updatedOptions) => {
             const containerClient = this.getContainerClient(options.destinationContainerName || deletedContainerName);
             // Hack to access a protected member.
-            const containerContext = new Container(containerClient["storageClientContext"]);
-            const containerUndeleteResponse = await containerContext.restore(Object.assign({ deletedContainerName,
-                deletedContainerVersion }, updatedOptions));
+            const containerContext = containerClient["storageClientContext"].container;
+            const containerUndeleteResponse = assertResponse(await containerContext.restore({
+                deletedContainerName,
+                deletedContainerVersion,
+                tracingOptions: updatedOptions.tracingOptions,
+            }));
             return { containerClient, containerUndeleteResponse };
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Rename an existing Blob Container.
@@ -188,25 +157,14 @@ export class BlobServiceClient extends StorageClient {
     /* eslint-disable-next-line @typescript-eslint/ban-ts-comment */
     // @ts-ignore Need to hide this interface for now. Make it public and turn on the live tests for it when the service is ready.
     async renameContainer(sourceContainerName, destinationContainerName, options = {}) {
-        var _a;
-        const { span, updatedOptions } = createSpan("BlobServiceClient-renameContainer", options);
-        try {
+        return tracingClient.withSpan("BlobServiceClient-renameContainer", options, async (updatedOptions) => {
+            var _a;
             const containerClient = this.getContainerClient(destinationContainerName);
             // Hack to access a protected member.
-            const containerContext = new Container(containerClient["storageClientContext"]);
-            const containerRenameResponse = await containerContext.rename(sourceContainerName, Object.assign(Object.assign({}, updatedOptions), { sourceLeaseId: (_a = options.sourceCondition) === null || _a === void 0 ? void 0 : _a.leaseId }));
+            const containerContext = containerClient["storageClientContext"].container;
+            const containerRenameResponse = assertResponse(await containerContext.rename(sourceContainerName, Object.assign(Object.assign({}, updatedOptions), { sourceLeaseId: (_a = options.sourceCondition) === null || _a === void 0 ? void 0 : _a.leaseId })));
             return { containerClient, containerRenameResponse };
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Gets the properties of a storage account’s Blob service, including properties
@@ -217,20 +175,12 @@ export class BlobServiceClient extends StorageClient {
      * @returns Response data for the Service Get Properties operation.
      */
     async getProperties(options = {}) {
-        const { span, updatedOptions } = createSpan("BlobServiceClient-getProperties", options);
-        try {
-            return await this.serviceContext.getProperties(Object.assign({ abortSignal: options.abortSignal }, convertTracingToRequestOptionsBase(updatedOptions)));
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        return tracingClient.withSpan("BlobServiceClient-getProperties", options, async (updatedOptions) => {
+            return assertResponse(await this.serviceContext.getProperties({
+                abortSignal: options.abortSignal,
+                tracingOptions: updatedOptions.tracingOptions,
+            }));
+        });
     }
     /**
      * Sets properties for a storage account’s Blob service endpoint, including properties
@@ -242,20 +192,12 @@ export class BlobServiceClient extends StorageClient {
      * @returns Response data for the Service Set Properties operation.
      */
     async setProperties(properties, options = {}) {
-        const { span, updatedOptions } = createSpan("BlobServiceClient-setProperties", options);
-        try {
-            return await this.serviceContext.setProperties(properties, Object.assign({ abortSignal: options.abortSignal }, convertTracingToRequestOptionsBase(updatedOptions)));
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        return tracingClient.withSpan("BlobServiceClient-setProperties", options, async (updatedOptions) => {
+            return assertResponse(await this.serviceContext.setProperties(properties, {
+                abortSignal: options.abortSignal,
+                tracingOptions: updatedOptions.tracingOptions,
+            }));
+        });
     }
     /**
      * Retrieves statistics related to replication for the Blob service. It is only
@@ -267,20 +209,12 @@ export class BlobServiceClient extends StorageClient {
      * @returns Response data for the Service Get Statistics operation.
      */
     async getStatistics(options = {}) {
-        const { span, updatedOptions } = createSpan("BlobServiceClient-getStatistics", options);
-        try {
-            return await this.serviceContext.getStatistics(Object.assign({ abortSignal: options.abortSignal }, convertTracingToRequestOptionsBase(updatedOptions)));
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        return tracingClient.withSpan("BlobServiceClient-getStatistics", options, async (updatedOptions) => {
+            return assertResponse(await this.serviceContext.getStatistics({
+                abortSignal: options.abortSignal,
+                tracingOptions: updatedOptions.tracingOptions,
+            }));
+        });
     }
     /**
      * The Get Account Information operation returns the sku name and account kind
@@ -293,20 +227,12 @@ export class BlobServiceClient extends StorageClient {
      * @returns Response data for the Service Get Account Info operation.
      */
     async getAccountInfo(options = {}) {
-        const { span, updatedOptions } = createSpan("BlobServiceClient-getAccountInfo", options);
-        try {
-            return await this.serviceContext.getAccountInfo(Object.assign({ abortSignal: options.abortSignal }, convertTracingToRequestOptionsBase(updatedOptions)));
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        return tracingClient.withSpan("BlobServiceClient-getAccountInfo", options, async (updatedOptions) => {
+            return assertResponse(await this.serviceContext.getAccountInfo({
+                abortSignal: options.abortSignal,
+                tracingOptions: updatedOptions.tracingOptions,
+            }));
+        });
     }
     /**
      * Returns a list of the containers under the specified account.
@@ -323,20 +249,9 @@ export class BlobServiceClient extends StorageClient {
      * @returns Response data for the Service List Container Segment operation.
      */
     async listContainersSegment(marker, options = {}) {
-        const { span, updatedOptions } = createSpan("BlobServiceClient-listContainersSegment", options);
-        try {
-            return await this.serviceContext.listContainersSegment(Object.assign(Object.assign(Object.assign({ abortSignal: options.abortSignal, marker }, options), { include: typeof options.include === "string" ? [options.include] : options.include }), convertTracingToRequestOptionsBase(updatedOptions)));
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        return tracingClient.withSpan("BlobServiceClient-listContainersSegment", options, async (updatedOptions) => {
+            return assertResponse(await this.serviceContext.listContainersSegment(Object.assign(Object.assign({ abortSignal: options.abortSignal, marker }, options), { include: typeof options.include === "string" ? [options.include] : options.include, tracingOptions: updatedOptions.tracingOptions })));
+        });
     }
     /**
      * The Filter Blobs operation enables callers to list blobs across all containers whose tags
@@ -357,9 +272,14 @@ export class BlobServiceClient extends StorageClient {
      * @param options - Options to find blobs by tags.
      */
     async findBlobsByTagsSegment(tagFilterSqlExpression, marker, options = {}) {
-        const { span, updatedOptions } = createSpan("BlobServiceClient-findBlobsByTagsSegment", options);
-        try {
-            const response = await this.serviceContext.filterBlobs(Object.assign({ abortSignal: options.abortSignal, where: tagFilterSqlExpression, marker, maxPageSize: options.maxPageSize }, convertTracingToRequestOptionsBase(updatedOptions)));
+        return tracingClient.withSpan("BlobServiceClient-findBlobsByTagsSegment", options, async (updatedOptions) => {
+            const response = assertResponse(await this.serviceContext.filterBlobs({
+                abortSignal: options.abortSignal,
+                where: tagFilterSqlExpression,
+                marker,
+                maxPageSize: options.maxPageSize,
+                tracingOptions: updatedOptions.tracingOptions,
+            }));
             const wrappedResponse = Object.assign(Object.assign({}, response), { _response: response._response, blobs: response.blobs.map((blob) => {
                     var _a;
                     let tagValue = "";
@@ -369,17 +289,7 @@ export class BlobServiceClient extends StorageClient {
                     return Object.assign(Object.assign({}, blob), { tags: toTags(blob.tags), tagValue });
                 }) });
             return wrappedResponse;
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Returns an AsyncIterableIterator for ServiceFindBlobsByTagsSegmentResponse.
@@ -716,12 +626,14 @@ export class BlobServiceClient extends StorageClient {
      * @param expiresOn -     The end time for the user delegation SAS. Must be within 7 days of the current time
      */
     async getUserDelegationKey(startsOn, expiresOn, options = {}) {
-        const { span, updatedOptions } = createSpan("BlobServiceClient-getUserDelegationKey", options);
-        try {
-            const response = await this.serviceContext.getUserDelegationKey({
+        return tracingClient.withSpan("BlobServiceClient-getUserDelegationKey", options, async (updatedOptions) => {
+            const response = assertResponse(await this.serviceContext.getUserDelegationKey({
                 startsOn: truncatedISO8061Date(startsOn, false),
                 expiresOn: truncatedISO8061Date(expiresOn, false),
-            }, Object.assign({ abortSignal: options.abortSignal }, convertTracingToRequestOptionsBase(updatedOptions)));
+            }, {
+                abortSignal: options.abortSignal,
+                tracingOptions: updatedOptions.tracingOptions,
+            }));
             const userDelegationKey = {
                 signedObjectId: response.signedObjectId,
                 signedTenantId: response.signedTenantId,
@@ -733,17 +645,7 @@ export class BlobServiceClient extends StorageClient {
             };
             const res = Object.assign({ _response: response._response, requestId: response.requestId, clientRequestId: response.clientRequestId, version: response.version, date: response.date, errorCode: response.errorCode }, userDelegationKey);
             return res;
-        }
-        catch (e) {
-            span.setStatus({
-                code: SpanStatusCode.ERROR,
-                message: e.message,
-            });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Creates a BlobBatchClient object to conduct batch operations.