@azure/data-tables 13.1.0-alpha.20220216.2 → 13.1.0-alpha.20220315.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
 
 - Updated our `@azure/core-tracing` dependency to the latest version (1.0.0-preview.14)

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.
@@ -3463,6 +3509,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));
         }
@@ -3478,8 +3525,8 @@ class TableServiceClient {
      * secondary location endpoint when read-access geo-redundant replication is enabled for the account.
      * @param options - The options parameters.
      */
-    getStatistics(options = {}) {
-        return tracingClient.withSpan("TableServiceClient.getStatistics", options, (updatedOptions) => this.service.getStatistics(updatedOptions));
+    async getStatistics(options = {}) {
+        return tracingClient.withSpan("TableServiceClient.getStatistics", options, (updatedOptions) => this.service.getStatistics(injectSecondaryEndpointHeader(updatedOptions)));
     }
     /**
      * Gets the properties of an account's Table service, including properties for Analytics and CORS
@@ -3525,7 +3572,7 @@ class TableServiceClient {
             }
             catch (e) {
                 if (e.statusCode === 404) {
-                    logger.info("TableServiceClient-deleteTable: Table doesn't exist");
+                    logger.info("TableServiceClient.deleteTable: Table doesn't exist");
                 }
                 else {
                     throw e;