@azure/data-tables 13.0.2-alpha.20220214.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

@@ -761,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.
@@ -3462,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));
         }
@@ -3480,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 });