@azure/data-tables 13.0.2-alpha.20220125.1 → 13.0.2-alpha.20220217.1

package/CHANGELOG.md

@@ -8,6 +8,8 @@
 
 ### Bugs Fixed
 
+- Fix issue with `getStatistics()` operation consistently failing and added test. [#20398](https://github.com/Azure/azure-sdk-for-js/pull/20398)
+
 ### Other Changes
 
 ## 13.0.1 (2022-01-12)

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 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";
@@ -739,6 +761,52 @@ function generateTableSas(tableName, credential, options = {}) {
     return sas;
 }
 
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+/**
+ * The programmatic identifier of the tablesSecondaryEndpointPolicy.
+ */
+const tablesSecondaryEndpointPolicyName = "tablesSecondaryEndpointPolicy";
+const SecondaryLocationHeaderName = "tables-secondary-endpoint";
+const SecondaryLocationAccountSuffix = "-secondary";
+/**
+ * Policy that would replace the Primary Endpoint with the secondary endpoint
+ * when the `tables-secondary-endpoint` is set in the request
+ */
+const tablesSecondaryEndpointPolicy = {
+    name: tablesSecondaryEndpointPolicyName,
+    sendRequest: async (req, next) => {
+        // Only replace the URL if the SecondaryLocationHeader is set
+        if (req.headers.get(SecondaryLocationHeaderName)) {
+            // Since the header is for internal use only, clean it up.
+            req.headers.delete(SecondaryLocationHeaderName);
+            // Calculate and update the secondary url
+            req.url = getSecondaryUrlFromPrimary(req.url);
+        }
+        return next(req);
+    },
+};
+/**
+ * Utility function that injects the SecondaryEndpointHeader into an operation options
+ */
+function injectSecondaryEndpointHeader(options) {
+    var _a;
+    const headerToInject = { [SecondaryLocationHeaderName]: "true" };
+    return Object.assign(Object.assign({}, options), { requestOptions: Object.assign(Object.assign({}, options.requestOptions), { customHeaders: Object.assign(Object.assign({}, (_a = options.requestOptions) === null || _a === void 0 ? void 0 : _a.customHeaders), headerToInject) }) });
+}
+/**
+ * Utility function that calculates the secondary URL for a table instance given the primary URL.
+ */
+function getSecondaryUrlFromPrimary(primaryUrl) {
+    const parsedPrimaryUrl = new URL(primaryUrl);
+    const host = parsedPrimaryUrl.hostname.split(".");
+    if (host.length > 1) {
+        host[0] = `${host[0]}${SecondaryLocationAccountSuffix}`;
+    }
+    parsedPrimaryUrl.hostname = host.join(".");
+    return parsedPrimaryUrl.toString();
+}
+
 /*
  * Copyright (c) Microsoft Corporation.
  * Licensed under the MIT License.
@@ -2596,8 +2664,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 +2965,7 @@ const getAccessPolicyOperationSpec = {
         accept2
     ],
     isXML: true,
-    serializer: xmlSerializer
+    serializer: xmlSerializer$1
 };
 const setAccessPolicyOperationSpec = {
     path: "/{table}",
@@ -2923,7 +2991,7 @@ const setAccessPolicyOperationSpec = {
     isXML: true,
     contentType: "application/xml; charset=utf-8",
     mediaType: "xml",
-    serializer: xmlSerializer
+    serializer: xmlSerializer$1
 };
 
 /*
@@ -2969,7 +3037,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 +3062,7 @@ const setPropertiesOperationSpec = {
     isXML: true,
     contentType: "application/xml; charset=utf-8",
     mediaType: "xml",
-    serializer: xmlSerializer$1
+    serializer: xmlSerializer
 };
 const getPropertiesOperationSpec = {
     path: "/",
@@ -3017,7 +3085,7 @@ const getPropertiesOperationSpec = {
         accept2
     ],
     isXML: true,
-    serializer: xmlSerializer$1
+    serializer: xmlSerializer
 };
 const getStatisticsOperationSpec = {
     path: "/",
@@ -3040,7 +3108,7 @@ const getStatisticsOperationSpec = {
         accept2
     ],
     isXML: true,
-    serializer: xmlSerializer$1
+    serializer: xmlSerializer
 };
 
 /*
@@ -3051,7 +3119,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.
@@ -3440,6 +3508,7 @@ class TableServiceClient {
             },
         }), (coreAuth.isTokenCredential(credential) && { credential, credentialScopes: STORAGE_SCOPE }));
         const client = new GeneratedClient(this.url, internalPipelineOptions);
+        client.pipeline.addPolicy(tablesSecondaryEndpointPolicy);
         if (coreAuth.isNamedKeyCredential(credential)) {
             client.pipeline.addPolicy(tablesNamedKeyCredentialPolicy(credential));
         }
@@ -3458,7 +3527,7 @@ class TableServiceClient {
     async getStatistics(options = {}) {
         const { span, updatedOptions } = createSpan("TableServiceClient-getStatistics", options);
         try {
-            return await this.service.getStatistics(updatedOptions);
+            return await this.service.getStatistics(injectSecondaryEndpointHeader(updatedOptions));
         }
         catch (e) {
             span.setStatus({ code: coreTracing.SpanStatusCode.ERROR, message: e.message });
@@ -3852,9 +3921,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,12 +4016,6 @@ 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({
@@ -3964,15 +4026,8 @@ class InternalTableTransaction {
             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);
+            const rawTransactionResponse = await this.client.sendRequest(request);
             return parseTransactionResponse(rawTransactionResponse);
         }
         catch (error) {
@@ -4416,6 +4471,7 @@ class TableClient {
         if (isCosmosEndpoint(this.url)) {
             generatedClient.pipeline.addPolicy(cosmosPatchPolicy());
         }
+        this.generatedClient = generatedClient;
         this.table = generatedClient.table;
         this.pipeline = generatedClient.pipeline;
     }
@@ -4952,7 +5008,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 +5059,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;