@azure/data-tables 13.0.2-alpha.20220125.1 → 13.1.0-alpha.20220216.2

package/dist/index.js

@@ -3,17 +3,37 @@
 Object.defineProperty(exports, '__esModule', { value: true });
 
 var tslib = require('tslib');
-var coreAuth = require('@azure/core-auth');
-var crypto = require('crypto');
 require('@azure/core-paging');
+var coreAuth = require('@azure/core-auth');
 var coreXml = require('@azure/core-xml');
 var coreClient = require('@azure/core-client');
-var coreTracing = require('@azure/core-tracing');
 var url$1 = require('url');
 var logger$1 = require('@azure/logger');
+var crypto = require('crypto');
+var coreTracing = require('@azure/core-tracing');
 var coreRestPipeline = require('@azure/core-rest-pipeline');
 var uuid = require('uuid');
 
+function _interopNamespace(e) {
+    if (e && e.__esModule) return e;
+    var n = Object.create(null);
+    if (e) {
+        Object.keys(e).forEach(function (k) {
+            if (k !== 'default') {
+                var d = Object.getOwnPropertyDescriptor(e, k);
+                Object.defineProperty(n, k, d.get ? d : {
+                    enumerable: true,
+                    get: function () { return e[k]; }
+                });
+            }
+        });
+    }
+    n["default"] = e;
+    return Object.freeze(n);
+}
+
+var coreClient__namespace = /*#__PURE__*/_interopNamespace(coreClient);
+
 /*
  * Copyright (c) Microsoft Corporation.
  * Licensed under the MIT License.
@@ -34,6 +54,8 @@ var KnownResponseFormat;
     KnownResponseFormat["ReturnNoContent"] = "return-no-content";
     KnownResponseFormat["ReturnContent"] = "return-content";
 })(KnownResponseFormat || (KnownResponseFormat = {}));
+/** Known values of {@link GeoReplicationStatusType} that the service accepts. */
+exports.KnownGeoReplicationStatusType = void 0;
 (function (KnownGeoReplicationStatusType) {
     KnownGeoReplicationStatusType["Live"] = "live";
     KnownGeoReplicationStatusType["Bootstrap"] = "bootstrap";
@@ -2596,8 +2618,8 @@ class TableImpl {
     }
 }
 // Operation Specifications
-const xmlSerializer = coreClient.createSerializer(Mappers, /* isXml */ true);
-const serializer = coreClient.createSerializer(Mappers, /* isXml */ false);
+const xmlSerializer$1 = coreClient__namespace.createSerializer(Mappers, /* isXml */ true);
+const serializer = coreClient__namespace.createSerializer(Mappers, /* isXml */ false);
 const queryOperationSpec = {
     path: "/Tables",
     httpMethod: "GET",
@@ -2897,7 +2919,7 @@ const getAccessPolicyOperationSpec = {
         accept2
     ],
     isXML: true,
-    serializer: xmlSerializer
+    serializer: xmlSerializer$1
 };
 const setAccessPolicyOperationSpec = {
     path: "/{table}",
@@ -2923,7 +2945,7 @@ const setAccessPolicyOperationSpec = {
     isXML: true,
     contentType: "application/xml; charset=utf-8",
     mediaType: "xml",
-    serializer: xmlSerializer
+    serializer: xmlSerializer$1
 };
 
 /*
@@ -2969,7 +2991,7 @@ class ServiceImpl {
     }
 }
 // Operation Specifications
-const xmlSerializer$1 = coreClient.createSerializer(Mappers, /* isXml */ true);
+const xmlSerializer = coreClient__namespace.createSerializer(Mappers, /* isXml */ true);
 const setPropertiesOperationSpec = {
     path: "/",
     httpMethod: "PUT",
@@ -2994,7 +3016,7 @@ const setPropertiesOperationSpec = {
     isXML: true,
     contentType: "application/xml; charset=utf-8",
     mediaType: "xml",
-    serializer: xmlSerializer$1
+    serializer: xmlSerializer
 };
 const getPropertiesOperationSpec = {
     path: "/",
@@ -3017,7 +3039,7 @@ const getPropertiesOperationSpec = {
         accept2
     ],
     isXML: true,
-    serializer: xmlSerializer$1
+    serializer: xmlSerializer
 };
 const getStatisticsOperationSpec = {
     path: "/",
@@ -3040,7 +3062,7 @@ const getStatisticsOperationSpec = {
         accept2
     ],
     isXML: true,
-    serializer: xmlSerializer$1
+    serializer: xmlSerializer
 };
 
 /*
@@ -3051,7 +3073,7 @@ const getStatisticsOperationSpec = {
  * Changes may cause incorrect behavior and will be lost if the code is regenerated.
  */
 /** @internal */
-class GeneratedClientContext extends coreClient.ServiceClient {
+class GeneratedClientContext extends coreClient__namespace.ServiceClient {
     /**
      * Initializes a new instance of the GeneratedClientContext class.
      * @param url The URL of the service account or table that is the target of the desired operation.
@@ -3068,7 +3090,7 @@ class GeneratedClientContext extends coreClient.ServiceClient {
         const defaults = {
             requestContentType: "application/json; charset=utf-8"
         };
-        const packageDetails = `azsdk-js-data-tables/13.0.2`;
+        const packageDetails = `azsdk-js-data-tables/13.1.0`;
         const userAgentPrefix = options.userAgentOptions && options.userAgentOptions.userAgentPrefix
             ? `${options.userAgentOptions.userAgentPrefix} ${packageDetails}`
             : `${packageDetails}`;
@@ -3104,16 +3126,6 @@ class GeneratedClient extends GeneratedClientContext {
     }
 }
 
-// Copyright (c) Microsoft Corporation.
-/**
- * Creates a span using the global tracer.
- * @internal
- */
-const createSpan = coreTracing.createSpanFunction({
-    packagePrefix: "Azure.Data.Tables",
-    namespace: "Microsoft.Data.Tables",
-});
-
 // Copyright (c) Microsoft Corporation.
 /**
  * Gets client parameters from an Account Connection String
@@ -3286,8 +3298,9 @@ function getAccountNameFromUrl(url) {
 }
 
 // Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
 function handleTableAlreadyExists(error, options = {}) {
-    var _a, _b, _c;
+    var _a, _b;
     const responseError = getErrorResponse(error);
     if (responseError &&
         responseError.status === 409 &&
@@ -3298,7 +3311,6 @@ function handleTableAlreadyExists(error, options = {}) {
         }
     }
     else {
-        options === null || options === void 0 ? void 0 : options.span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: (_c = error) === null || _c === void 0 ? void 0 : _c.message });
         throw error;
     }
 }
@@ -3416,6 +3428,17 @@ function signURLWithSAS(request, credential) {
     request.url = url.toString();
 }
 
+// Copyright (c) Microsoft Corporation.
+/**
+ * A tracing client that can be used to manage spans.
+ * @internal
+ */
+const tracingClient = coreTracing.createTracingClient({
+    namespace: "Microsoft.Data.Tables",
+    packageName: "@azure/data-tables",
+    packageVersion: "13.1.0",
+});
+
 // Copyright (c) Microsoft Corporation.
 /**
  * A TableServiceClient represents a Client to the Azure Tables service allowing you
@@ -3455,36 +3478,16 @@ class TableServiceClient {
      * secondary location endpoint when read-access geo-redundant replication is enabled for the account.
      * @param options - The options parameters.
      */
-    async getStatistics(options = {}) {
-        const { span, updatedOptions } = createSpan("TableServiceClient-getStatistics", options);
-        try {
-            return await this.service.getStatistics(updatedOptions);
-        }
-        catch (e) {
-            span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+    getStatistics(options = {}) {
+        return tracingClient.withSpan("TableServiceClient.getStatistics", options, (updatedOptions) => this.service.getStatistics(updatedOptions));
     }
     /**
      * Gets the properties of an account's Table service, including properties for Analytics and CORS
      * (Cross-Origin Resource Sharing) rules.
      * @param options - The options parameters.
      */
-    async getProperties(options = {}) {
-        const { span, updatedOptions } = createSpan("TableServiceClient-getProperties", options);
-        try {
-            return await this.service.getProperties(updatedOptions);
-        }
-        catch (e) {
-            span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+    getProperties(options = {}) {
+        return tracingClient.withSpan("TableServiceClient.getProperties", options, (updatedOptions) => this.service.getProperties(updatedOptions));
     }
     /**
      * Sets properties for an account's Table service endpoint, including properties for Analytics and CORS
@@ -3492,58 +3495,43 @@ class TableServiceClient {
      * @param properties - The Table Service properties.
      * @param options - The options parameters.
      */
-    async setProperties(properties, options = {}) {
-        const { span, updatedOptions } = createSpan("TableServiceClient-setProperties", options);
-        try {
-            return await this.service.setProperties(properties, updatedOptions);
-        }
-        catch (e) {
-            span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+    setProperties(properties, options = {}) {
+        return tracingClient.withSpan("TableServiceClient.setProperties", options, (updatedOptions) => this.service.setProperties(properties, updatedOptions));
     }
     /**
      * Creates a new table under the given account.
      * @param name - The name of the table.
      * @param options - The options parameters.
      */
-    async createTable(name, options = {}) {
-        const { span, updatedOptions } = createSpan("TableServiceClient-createTable", options);
-        try {
-            await this.table.create({ name }, Object.assign({}, updatedOptions));
-        }
-        catch (e) {
-            handleTableAlreadyExists(e, Object.assign(Object.assign({}, updatedOptions), { span, logger, tableName: name }));
-        }
-        finally {
-            span.end();
-        }
+    createTable(name, options = {}) {
+        return tracingClient.withSpan("TableServiceClient.createTable", options, async (updatedOptions) => {
+            try {
+                await this.table.create({ name }, updatedOptions);
+            }
+            catch (e) {
+                handleTableAlreadyExists(e, Object.assign(Object.assign({}, updatedOptions), { logger, tableName: name }));
+            }
+        });
     }
     /**
      * Operation permanently deletes the specified table.
      * @param name - The name of the table.
      * @param options - The options parameters.
      */
-    async deleteTable(name, options = {}) {
-        const { span, updatedOptions } = createSpan("TableServiceClient-deleteTable", options);
-        try {
-            await this.table.delete(name, updatedOptions);
-        }
-        catch (e) {
-            if (e.statusCode === 404) {
-                logger.info("TableServiceClient-deleteTable: Table doesn't exist");
+    deleteTable(name, options = {}) {
+        return tracingClient.withSpan("TableServiceClient.deleteTable", options, async (updatedOptions) => {
+            try {
+                await this.table.delete(name, updatedOptions);
             }
-            else {
-                span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
-                throw e;
+            catch (e) {
+                if (e.statusCode === 404) {
+                    logger.info("TableServiceClient-deleteTable: Table doesn't exist");
+                }
+                else {
+                    throw e;
+                }
             }
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Queries tables under the given account.
@@ -3595,22 +3583,15 @@ class TableServiceClient {
     }
     listTablesPage(options = {}) {
         return tslib.__asyncGenerator(this, arguments, function* listTablesPage_1() {
-            const { span, updatedOptions } = createSpan("TableServiceClient-listTablesPage", options);
-            try {
-                let result = yield tslib.__await(this._listTables(updatedOptions));
+            let result = yield tslib.__await(tracingClient.withSpan("TableServiceClient.listTablesPage", options, (updatedOptions) => this._listTables(updatedOptions)));
+            yield yield tslib.__await(result);
+            while (result.continuationToken) {
+                const optionsWithContinuation = Object.assign(Object.assign({}, options), { continuationToken: result.continuationToken });
+                result = yield tslib.__await(tracingClient.withSpan("TableServiceClient.listTablesPage", optionsWithContinuation, async (updatedOptions, span) => {
+                    span.setAttribute("continuationToken", updatedOptions.continuationToken);
+                    return this._listTables(updatedOptions);
+                }));
                 yield yield tslib.__await(result);
-                while (result.continuationToken) {
-                    const optionsWithContinuation = Object.assign(Object.assign({}, updatedOptions), { continuationToken: result.continuationToken });
-                    result = yield tslib.__await(this._listTables(optionsWithContinuation));
-                    yield yield tslib.__await(result);
-                }
-            }
-            catch (e) {
-                span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
-                throw e;
-            }
-            finally {
-                span.end();
             }
         });
     }
@@ -3852,9 +3833,8 @@ class InternalTableTransaction {
      * @param partitionKey - partition key
      * @param credential - credential to authenticate the transaction request
      */
-    constructor(url, partitionKey, transactionId, changesetId, clientOptions, interceptClient, credential, allowInsecureConnection = false) {
-        this.clientOptions = clientOptions;
-        this.credential = credential;
+    constructor(url, partitionKey, transactionId, changesetId, client, interceptClient, credential, allowInsecureConnection = false) {
+        this.client = client;
         this.url = url;
         this.interceptClient = interceptClient;
         this.allowInsecureConnection = allowInsecureConnection;
@@ -3948,43 +3928,19 @@ class InternalTableTransaction {
     async submitTransaction() {
         await Promise.all(this.resetableState.pendingOperations);
         const body = getTransactionHttpRequestBody(this.resetableState.bodyParts, this.resetableState.transactionId, this.resetableState.changesetId);
-        const options = this.clientOptions;
-        if (coreAuth.isTokenCredential(this.credential)) {
-            options.credentialScopes = STORAGE_SCOPE;
-            options.credential = this.credential;
-        }
-        const client = new coreClient.ServiceClient(options);
         const headers = getTransactionHeaders(this.resetableState.transactionId);
-        const { span, updatedOptions } = createSpan("TableTransaction-submitTransaction", {});
-        const request = coreRestPipeline.createPipelineRequest({
-            url: this.url,
-            method: "POST",
-            body,
-            headers: coreRestPipeline.createHttpHeaders(headers),
-            tracingOptions: updatedOptions.tracingOptions,
-            allowInsecureConnection: this.allowInsecureConnection,
-        });
-        if (coreAuth.isNamedKeyCredential(this.credential)) {
-            const authHeader = getAuthorizationHeader(request, this.credential);
-            request.headers.set("Authorization", authHeader);
-        }
-        else if (coreAuth.isSASCredential(this.credential)) {
-            signURLWithSAS(request, this.credential);
-        }
-        try {
-            const rawTransactionResponse = await client.sendRequest(request);
-            return parseTransactionResponse(rawTransactionResponse);
-        }
-        catch (error) {
-            span.setStatus({
-                code: coreTracing.SpanStatusCode.ERROR,
-                message: error.message,
+        return tracingClient.withSpan("TableTransaction.submitTransaction", {}, async (updatedOptions) => {
+            const request = coreRestPipeline.createPipelineRequest({
+                url: this.url,
+                method: "POST",
+                body,
+                headers: coreRestPipeline.createHttpHeaders(headers),
+                tracingOptions: updatedOptions.tracingOptions,
+                allowInsecureConnection: this.allowInsecureConnection,
             });
-            throw error;
-        }
-        finally {
-            span.end();
-        }
+            const rawTransactionResponse = await this.client.sendRequest(request);
+            return parseTransactionResponse(rawTransactionResponse);
+        });
     }
     checkPartitionKey(partitionKey) {
         if (this.resetableState.partitionKey !== partitionKey) {
@@ -4416,6 +4372,7 @@ class TableClient {
         if (isCosmosEndpoint(this.url)) {
             generatedClient.pipeline.addPolicy(cosmosPatchPolicy());
         }
+        this.generatedClient = generatedClient;
         this.table = generatedClient.table;
         this.pipeline = generatedClient.pipeline;
     }
@@ -4444,23 +4401,17 @@ class TableClient {
      * ```
      */
     // eslint-disable-next-line @azure/azure-sdk/ts-naming-options
-    async deleteTable(options = {}) {
-        const { span, updatedOptions } = createSpan("TableClient-deleteTable", options);
-        try {
-            await this.table.delete(this.tableName, updatedOptions);
-        }
-        catch (e) {
-            if (e.statusCode === 404) {
-                logger.info("TableClient-deleteTable: Table doesn't exist");
+    deleteTable(options = {}) {
+        return tracingClient.withSpan("TableClient.deleteTable", options, async (updatedOptions) => {
+            try {
+                await this.table.delete(this.tableName, updatedOptions);
             }
-            else {
-                span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
-                throw e;
+            catch (e) {
+                if (e.statusCode === 404) {
+                    logger.info("TableClient.deleteTable: Table doesn't exist");
+                }
             }
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      *  Creates a table with the tableName passed to the client constructor
@@ -4488,17 +4439,15 @@ class TableClient {
      * ```
      */
     // eslint-disable-next-line @azure/azure-sdk/ts-naming-options
-    async createTable(options = {}) {
-        const { span, updatedOptions } = createSpan("TableClient-createTable", options);
-        try {
-            await this.table.create({ name: this.tableName }, updatedOptions);
-        }
-        catch (e) {
-            handleTableAlreadyExists(e, Object.assign(Object.assign({}, updatedOptions), { span, logger, tableName: this.tableName }));
-        }
-        finally {
-            span.end();
-        }
+    createTable(options = {}) {
+        return tracingClient.withSpan("TableClient.createTable", options, async (updatedOptions) => {
+            try {
+                await this.table.create({ name: this.tableName }, updatedOptions);
+            }
+            catch (e) {
+                handleTableAlreadyExists(e, Object.assign(Object.assign({}, updatedOptions), { logger, tableName: this.tableName }));
+            }
+        });
     }
     /**
      * Returns a single entity in the table.
@@ -4527,30 +4476,22 @@ class TableClient {
      * console.log(entity);
      * ```
      */
-    async getEntity(partitionKey, rowKey, 
+    getEntity(partitionKey, rowKey, 
     // eslint-disable-next-line @azure/azure-sdk/ts-naming-options
     options = {}) {
-        const { span, updatedOptions } = createSpan("TableClient-getEntity", options);
-        let parsedBody;
-        function onResponse(rawResponse, flatResponse) {
-            parsedBody = rawResponse.parsedBody;
-            if (updatedOptions.onResponse) {
-                updatedOptions.onResponse(rawResponse, flatResponse);
+        return tracingClient.withSpan("TableClient.getEntity", options, async (updatedOptions) => {
+            let parsedBody;
+            function onResponse(rawResponse, flatResponse) {
+                parsedBody = rawResponse.parsedBody;
+                if (updatedOptions.onResponse) {
+                    updatedOptions.onResponse(rawResponse, flatResponse);
+                }
             }
-        }
-        try {
-            const _a = updatedOptions || {}, { disableTypeConversion, queryOptions } = _a, getEntityOptions = tslib.__rest(_a, ["disableTypeConversion", "queryOptions"]);
+            const { disableTypeConversion, queryOptions } = updatedOptions, getEntityOptions = tslib.__rest(updatedOptions, ["disableTypeConversion", "queryOptions"]);
             await this.table.queryEntitiesWithPartitionAndRowKey(this.tableName, escapeQuotes(partitionKey), escapeQuotes(rowKey), Object.assign(Object.assign({}, getEntityOptions), { queryOptions: serializeQueryOptions(queryOptions || {}), onResponse }));
             const tableEntity = deserialize(parsedBody, disableTypeConversion !== null && disableTypeConversion !== void 0 ? disableTypeConversion : false);
             return tableEntity;
-        }
-        catch (e) {
-            span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Queries entities in a table.
@@ -4628,25 +4569,15 @@ class TableClient {
     }
     listEntitiesPage(tableName, options = {}) {
         return tslib.__asyncGenerator(this, arguments, function* listEntitiesPage_1() {
-            const { span, updatedOptions } = createSpan("TableClient-listEntitiesPage", options);
-            try {
-                let result = yield tslib.__await(this._listEntities(tableName, updatedOptions));
+            let result = yield tslib.__await(tracingClient.withSpan("TableClient.listEntitiesPage", options, (updatedOptions) => this._listEntities(tableName, updatedOptions)));
+            yield yield tslib.__await(result);
+            while (result.continuationToken) {
+                const optionsWithContinuation = Object.assign(Object.assign({}, options), { continuationToken: result.continuationToken });
+                result = yield tslib.__await(tracingClient.withSpan("TableClient.listEntitiesPage", optionsWithContinuation, (updatedOptions, span) => {
+                    span.setAttribute("continuationToken", result.continuationToken);
+                    return this._listEntities(tableName, updatedOptions);
+                }));
                 yield yield tslib.__await(result);
-                while (result.continuationToken) {
-                    const optionsWithContinuation = Object.assign(Object.assign({}, updatedOptions), { continuationToken: result.continuationToken });
-                    result = yield tslib.__await(this._listEntities(tableName, optionsWithContinuation));
-                    yield yield tslib.__await(result);
-                }
-            }
-            catch (e) {
-                span.setStatus({
-                    code: coreTracing.SpanStatusCode.ERROR,
-                    message: e.message,
-                });
-                throw e;
-            }
-            finally {
-                span.end();
             }
         });
     }
@@ -4694,21 +4625,13 @@ class TableClient {
      * await client.createEntity({partitionKey: "p1", rowKey: "r1", foo: "Hello!"});
      * ```
      */
-    async createEntity(entity, 
+    createEntity(entity, 
     // eslint-disable-next-line @azure/azure-sdk/ts-naming-options
     options = {}) {
-        const { span, updatedOptions } = createSpan("TableClient-createEntity", options);
-        try {
+        return tracingClient.withSpan("TableClient.createEntity", options, (updatedOptions) => {
             const createTableEntity = tslib.__rest(updatedOptions || {}, []);
-            return await this.table.insertEntity(this.tableName, Object.assign(Object.assign({}, createTableEntity), { tableEntityProperties: serialize(entity), responsePreference: "return-no-content" }));
-        }
-        catch (e) {
-            span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+            return this.table.insertEntity(this.tableName, Object.assign(Object.assign({}, createTableEntity), { tableEntityProperties: serialize(entity), responsePreference: "return-no-content" }));
+        });
     }
     /**
      * Deletes the specified entity in the table.
@@ -4735,22 +4658,14 @@ class TableClient {
      * await client.deleteEntity("<partitionKey>", "<rowKey>")
      * ```
      */
-    async deleteEntity(partitionKey, rowKey, 
+    deleteEntity(partitionKey, rowKey, 
     // eslint-disable-next-line @azure/azure-sdk/ts-naming-options
     options = {}) {
-        const { span, updatedOptions } = createSpan("TableClient-deleteEntity", options);
-        try {
-            const _a = updatedOptions || {}, { etag = "*" } = _a, rest = tslib.__rest(_a, ["etag"]);
+        return tracingClient.withSpan("TableClient.deleteEntity", options, (updatedOptions) => {
+            const { etag = "*" } = updatedOptions, rest = tslib.__rest(updatedOptions, ["etag"]);
             const deleteOptions = Object.assign({}, rest);
-            return await this.table.deleteEntity(this.tableName, escapeQuotes(partitionKey), escapeQuotes(rowKey), etag, deleteOptions);
-        }
-        catch (e) {
-            span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+            return this.table.deleteEntity(this.tableName, escapeQuotes(partitionKey), escapeQuotes(rowKey), etag, deleteOptions);
+        });
     }
     /**
      * Update an entity in the table.
@@ -4791,29 +4706,25 @@ class TableClient {
      * await client.updateEntity(entity, "Replace")
      * ```
      */
-    async updateEntity(entity, mode = "Merge", 
+    updateEntity(entity, mode = "Merge", 
     // eslint-disable-next-line @azure/azure-sdk/ts-naming-options
     options = {}) {
-        const { span, updatedOptions } = createSpan(`TableClient-updateEntity-${mode}`, options);
-        try {
+        return tracingClient.withSpan("TableClient.updateEntity", options, async (updatedOptions) => {
             const partitionKey = escapeQuotes(entity.partitionKey);
             const rowKey = escapeQuotes(entity.rowKey);
             const _a = updatedOptions || {}, { etag = "*" } = _a, updateEntityOptions = tslib.__rest(_a, ["etag"]);
             if (mode === "Merge") {
-                return await this.table.mergeEntity(this.tableName, partitionKey, rowKey, Object.assign({ tableEntityProperties: serialize(entity), ifMatch: etag }, updateEntityOptions));
+                return this.table.mergeEntity(this.tableName, partitionKey, rowKey, Object.assign({ tableEntityProperties: serialize(entity), ifMatch: etag }, updateEntityOptions));
             }
             if (mode === "Replace") {
-                return await this.table.updateEntity(this.tableName, partitionKey, rowKey, Object.assign({ tableEntityProperties: serialize(entity), ifMatch: etag }, updateEntityOptions));
+                return this.table.updateEntity(this.tableName, partitionKey, rowKey, Object.assign({ tableEntityProperties: serialize(entity), ifMatch: etag }, updateEntityOptions));
             }
             throw new Error(`Unexpected value for update mode: ${mode}`);
-        }
-        catch (e) {
-            span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        }, {
+            spanAttributes: {
+                updateEntityMode: mode,
+            },
+        });
     }
     /**
      * Upsert an entity in the table.
@@ -4850,66 +4761,46 @@ class TableClient {
      * await client.upsertEntity(entity, "Replace")
      * ```
      */
-    async upsertEntity(entity, mode = "Merge", 
+    upsertEntity(entity, mode = "Merge", 
     // eslint-disable-next-line @azure/azure-sdk/ts-naming-options
     options = {}) {
-        const { span, updatedOptions } = createSpan(`TableClient-upsertEntity-${mode}`, options);
-        try {
+        return tracingClient.withSpan("TableClient.upsertEntity", options, async (updatedOptions) => {
             const partitionKey = escapeQuotes(entity.partitionKey);
             const rowKey = escapeQuotes(entity.rowKey);
             if (mode === "Merge") {
-                return await this.table.mergeEntity(this.tableName, partitionKey, rowKey, Object.assign({ tableEntityProperties: serialize(entity) }, updatedOptions));
+                return this.table.mergeEntity(this.tableName, partitionKey, rowKey, Object.assign({ tableEntityProperties: serialize(entity) }, updatedOptions));
             }
             if (mode === "Replace") {
-                return await this.table.updateEntity(this.tableName, partitionKey, rowKey, Object.assign({ tableEntityProperties: serialize(entity) }, updatedOptions));
+                return this.table.updateEntity(this.tableName, partitionKey, rowKey, Object.assign({ tableEntityProperties: serialize(entity) }, updatedOptions));
             }
             throw new Error(`Unexpected value for update mode: ${mode}`);
-        }
-        catch (e) {
-            span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        }, {
+            spanAttributes: {
+                upsertEntityMode: mode,
+            },
+        });
     }
     /**
      * Retrieves details about any stored access policies specified on the table that may be used with
      * Shared Access Signatures.
      * @param options - The options parameters.
      */
-    async getAccessPolicy(options = {}) {
-        const { span, updatedOptions } = createSpan("TableClient-getAccessPolicy", options);
-        try {
+    getAccessPolicy(options = {}) {
+        return tracingClient.withSpan("TableClient.getAccessPolicy", options, async (updatedOptions) => {
             const signedIdentifiers = await this.table.getAccessPolicy(this.tableName, updatedOptions);
             return deserializeSignedIdentifier(signedIdentifiers);
-        }
-        catch (e) {
-            span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+        });
     }
     /**
      * Sets stored access policies for the table that may be used with Shared Access Signatures.
      * @param tableAcl - The Access Control List for the table.
      * @param options - The options parameters.
      */
-    async setAccessPolicy(tableAcl, options = {}) {
-        const { span, updatedOptions } = createSpan("TableClient-setAccessPolicy", options);
-        try {
+    setAccessPolicy(tableAcl, options = {}) {
+        return tracingClient.withSpan("TableClient.setAccessPolicy", options, (updatedOptions) => {
             const serlializedAcl = serializeSignedIdentifiers(tableAcl);
-            return await this.table.setAccessPolicy(this.tableName, Object.assign(Object.assign({}, updatedOptions), { tableAcl: serlializedAcl }));
-        }
-        catch (e) {
-            span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
-            throw e;
-        }
-        finally {
-            span.end();
-        }
+            return this.table.setAccessPolicy(this.tableName, Object.assign(Object.assign({}, updatedOptions), { tableAcl: serlializedAcl }));
+        });
     }
     /**
      * Submits a Transaction which is composed of a set of actions. You can provide the actions as a list
@@ -4952,7 +4843,7 @@ class TableClient {
         const changesetId = Uuid.generateUuid();
         if (!this.transactionClient) {
             // Add pipeline
-            this.transactionClient = new InternalTableTransaction(this.url, partitionKey, transactionId, changesetId, this.clientOptions, new TableClient(this.url, this.tableName), this.credential, this.allowInsecureConnection);
+            this.transactionClient = new InternalTableTransaction(this.url, partitionKey, transactionId, changesetId, this.generatedClient, new TableClient(this.url, this.tableName), this.credential, this.allowInsecureConnection);
         }
         else {
             this.transactionClient.reset(transactionId, changesetId, partitionKey);
@@ -5003,21 +4894,15 @@ class TableClient {
 
 Object.defineProperty(exports, 'AzureNamedKeyCredential', {
     enumerable: true,
-    get: function () {
-        return coreAuth.AzureNamedKeyCredential;
-    }
+    get: function () { return coreAuth.AzureNamedKeyCredential; }
 });
 Object.defineProperty(exports, 'AzureSASCredential', {
     enumerable: true,
-    get: function () {
-        return coreAuth.AzureSASCredential;
-    }
+    get: function () { return coreAuth.AzureSASCredential; }
 });
 Object.defineProperty(exports, 'RestError', {
     enumerable: true,
-    get: function () {
-        return coreRestPipeline.RestError;
-    }
+    get: function () { return coreRestPipeline.RestError; }
 });
 exports.TableClient = TableClient;
 exports.TableServiceClient = TableServiceClient;