@azure/storage-blob 12.13.0-alpha.20230202.2 → 12.13.0-beta.1

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

@@ -1,16 +1,17 @@
 import { __asyncDelegator, __asyncGenerator, __asyncValues, __await } from "tslib";
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT license.
-import { isTokenCredential } from "@azure/core-auth";
-import { getDefaultProxySettings } from "@azure/core-rest-pipeline";
-import { isNode } from "@azure/core-util";
+import { isTokenCredential, isNode, getDefaultProxySettings, } from "@azure/core-http";
+import { SpanStatusCode } from "@azure/core-tracing";
+import { Container, Service } from "./generated/src/operations";
 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 { truncatedISO8061Date, assertResponse } from "./utils/utils.common";
-import { tracingClient } from "./utils/tracing";
+import "@azure/core-paging";
+import { truncatedISO8061Date } from "./utils/utils.common";
+import { convertTracingToRequestOptionsBase, createSpan } from "./utils/tracing";
 import { BlobBatchClient } from "./BlobBatchClient";
 import { StorageClient } from "./StorageClient";
 import { AccountSASPermissions } from "./sas/AccountSASPermissions";
@@ -39,7 +40,7 @@ export class BlobServiceClient extends StorageClient {
             pipeline = newPipeline(new AnonymousCredential(), options);
         }
         super(url, pipeline);
-        this.serviceContext = this.storageClientContext.service;
+        this.serviceContext = new Service(this.storageClientContext);
     }
     /**
      *
@@ -103,14 +104,25 @@ export class BlobServiceClient extends StorageClient {
      * @returns Container creation response and the corresponding container client.
      */
     async createContainer(containerName, options = {}) {
-        return tracingClient.withSpan("BlobServiceClient-createContainer", options, async (updatedOptions) => {
+        const { span, updatedOptions } = createSpan("BlobServiceClient-createContainer", options);
+        try {
             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.
@@ -120,10 +132,21 @@ export class BlobServiceClient extends StorageClient {
      * @returns Container deletion response.
      */
     async deleteContainer(containerName, options = {}) {
-        return tracingClient.withSpan("BlobServiceClient-deleteContainer", options, async (updatedOptions) => {
+        const { span, updatedOptions } = createSpan("BlobServiceClient-deleteContainer", options);
+        try {
             const containerClient = this.getContainerClient(containerName);
-            return containerClient.delete(updatedOptions);
-        });
+            return await containerClient.delete(updatedOptions);
+        }
+        catch (e) {
+            span.setStatus({
+                code: SpanStatusCode.ERROR,
+                message: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
     }
     /**
      * Restore a previously deleted Blob container.
@@ -135,17 +158,25 @@ export class BlobServiceClient extends StorageClient {
      * @returns Container deletion response.
      */
     async undeleteContainer(deletedContainerName, deletedContainerVersion, options = {}) {
-        return tracingClient.withSpan("BlobServiceClient-undeleteContainer", options, async (updatedOptions) => {
+        const { span, updatedOptions } = createSpan("BlobServiceClient-undeleteContainer", options);
+        try {
             const containerClient = this.getContainerClient(options.destinationContainerName || deletedContainerName);
             // Hack to access a protected member.
-            const containerContext = containerClient["storageClientContext"].container;
-            const containerUndeleteResponse = assertResponse(await containerContext.restore({
-                deletedContainerName,
-                deletedContainerVersion,
-                tracingOptions: updatedOptions.tracingOptions,
-            }));
+            const containerContext = new Container(containerClient["storageClientContext"]);
+            const containerUndeleteResponse = await containerContext.restore(Object.assign({ deletedContainerName,
+                deletedContainerVersion }, updatedOptions));
             return { containerClient, containerUndeleteResponse };
-        });
+        }
+        catch (e) {
+            span.setStatus({
+                code: SpanStatusCode.ERROR,
+                message: e.message,
+            });
+            throw e;
+        }
+        finally {
+            span.end();
+        }
     }
     /**
      * Rename an existing Blob Container.
@@ -157,14 +188,25 @@ 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 = {}) {
-        return tracingClient.withSpan("BlobServiceClient-renameContainer", options, async (updatedOptions) => {
-            var _a;
+        var _a;
+        const { span, updatedOptions } = createSpan("BlobServiceClient-renameContainer", options);
+        try {
             const containerClient = this.getContainerClient(destinationContainerName);
             // Hack to access a protected member.
-            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 })));
+            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 }));
             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
@@ -175,12 +217,20 @@ export class BlobServiceClient extends StorageClient {
      * @returns Response data for the Service Get Properties operation.
      */
     async getProperties(options = {}) {
-        return tracingClient.withSpan("BlobServiceClient-getProperties", options, async (updatedOptions) => {
-            return assertResponse(await this.serviceContext.getProperties({
-                abortSignal: options.abortSignal,
-                tracingOptions: updatedOptions.tracingOptions,
-            }));
-        });
+        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();
+        }
     }
     /**
      * Sets properties for a storage account’s Blob service endpoint, including properties
@@ -192,12 +242,20 @@ export class BlobServiceClient extends StorageClient {
      * @returns Response data for the Service Set Properties operation.
      */
     async setProperties(properties, options = {}) {
-        return tracingClient.withSpan("BlobServiceClient-setProperties", options, async (updatedOptions) => {
-            return assertResponse(await this.serviceContext.setProperties(properties, {
-                abortSignal: options.abortSignal,
-                tracingOptions: updatedOptions.tracingOptions,
-            }));
-        });
+        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();
+        }
     }
     /**
      * Retrieves statistics related to replication for the Blob service. It is only
@@ -209,12 +267,20 @@ export class BlobServiceClient extends StorageClient {
      * @returns Response data for the Service Get Statistics operation.
      */
     async getStatistics(options = {}) {
-        return tracingClient.withSpan("BlobServiceClient-getStatistics", options, async (updatedOptions) => {
-            return assertResponse(await this.serviceContext.getStatistics({
-                abortSignal: options.abortSignal,
-                tracingOptions: updatedOptions.tracingOptions,
-            }));
-        });
+        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();
+        }
     }
     /**
      * The Get Account Information operation returns the sku name and account kind
@@ -227,12 +293,20 @@ export class BlobServiceClient extends StorageClient {
      * @returns Response data for the Service Get Account Info operation.
      */
     async getAccountInfo(options = {}) {
-        return tracingClient.withSpan("BlobServiceClient-getAccountInfo", options, async (updatedOptions) => {
-            return assertResponse(await this.serviceContext.getAccountInfo({
-                abortSignal: options.abortSignal,
-                tracingOptions: updatedOptions.tracingOptions,
-            }));
-        });
+        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();
+        }
     }
     /**
      * Returns a list of the containers under the specified account.
@@ -249,9 +323,20 @@ export class BlobServiceClient extends StorageClient {
      * @returns Response data for the Service List Container Segment operation.
      */
     async listContainersSegment(marker, options = {}) {
-        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 })));
-        });
+        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();
+        }
     }
     /**
      * The Filter Blobs operation enables callers to list blobs across all containers whose tags
@@ -272,14 +357,9 @@ export class BlobServiceClient extends StorageClient {
      * @param options - Options to find blobs by tags.
      */
     async findBlobsByTagsSegment(tagFilterSqlExpression, marker, options = {}) {
-        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 { 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)));
             const wrappedResponse = Object.assign(Object.assign({}, response), { _response: response._response, blobs: response.blobs.map((blob) => {
                     var _a;
                     let tagValue = "";
@@ -289,7 +369,17 @@ 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.
@@ -626,14 +716,12 @@ 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 = {}) {
-        return tracingClient.withSpan("BlobServiceClient-getUserDelegationKey", options, async (updatedOptions) => {
-            const response = assertResponse(await this.serviceContext.getUserDelegationKey({
+        const { span, updatedOptions } = createSpan("BlobServiceClient-getUserDelegationKey", options);
+        try {
+            const response = await this.serviceContext.getUserDelegationKey({
                 startsOn: truncatedISO8061Date(startsOn, false),
                 expiresOn: truncatedISO8061Date(expiresOn, false),
-            }, {
-                abortSignal: options.abortSignal,
-                tracingOptions: updatedOptions.tracingOptions,
-            }));
+            }, Object.assign({ abortSignal: options.abortSignal }, convertTracingToRequestOptionsBase(updatedOptions)));
             const userDelegationKey = {
                 signedObjectId: response.signedObjectId,
                 signedTenantId: response.signedTenantId,
@@ -645,7 +733,17 @@ 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.