@azure/cosmos 4.3.0 → 4.4.0

package/dist/commonjs/diagnostics/CosmosDbDiagnosticLevel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CosmosDbDiagnosticLevel.js","sourceRoot":"","sources":["../../../src/diagnostics/CosmosDbDiagnosticLevel.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAElC;;GAEG;AACH,IAAY,uBAIX;AAJD,WAAY,uBAAuB;IACjC,wCAAa,CAAA;IACb,0CAAe,CAAA;IACf,uDAA4B,CAAA;AAC9B,CAAC,EAJW,uBAAuB,uCAAvB,uBAAuB,QAIlC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\n/**\n * Cosmos DB Diagnostic Level\n */\nexport enum CosmosDbDiagnosticLevel {\n  info = \"info\",\n  debug = \"debug\",\n  debugUnsafe = \"debug-unsafe\",\n}\n"]}

package/dist/commonjs/diagnostics/CosmosDiagnosticsContext.d.ts

@@ -0,0 +1,36 @@
+import type { ClientSideRequestStatistics, EncryptionDiagnostics, GatewayStatistics, MetadataLookUpType } from "../CosmosDiagnostics.js";
+/**
+ * @hidden
+ * Internal class to hold CosmosDiagnostic aggregate information all through the lifecycle of a request.
+ * This object gathers diagnostic information throughout Client operation which may span across multiple
+ * Server call, retries etc.
+ * Functions - recordFailedAttempt, recordMetaDataQuery, recordEndpointContactEvent are used to ingest
+ * data into the context. At the end of operation, getDiagnostics() is used to
+ * get final CosmosDiagnostic object.
+ */
+export declare class CosmosDiagnosticContext {
+    private requestStartTimeUTCinMs;
+    private failedAttempts;
+    private metadataLookups;
+    private gatewayStatistics;
+    locationEndpointsContacted: Set<string>;
+    encryptionDiagnostics: EncryptionDiagnostics;
+    constructor();
+    recordFailedAttempt(gatewayStatistics: GatewayStatistics, retryAttemptNumber: number): void;
+    recordNetworkCall(gatewayStatistics: GatewayStatistics): void;
+    recordEncryptionDiagnostics(encryptionDiagnostics: EncryptionDiagnostics): void;
+    /**
+     * Merge given DiagnosticContext to current node's DiagnosticContext, Treating GatewayRequests of
+     * given DiagnosticContext, as metadata requests.
+     */
+    mergeDiagnostics(childDiagnostics: CosmosDiagnosticContext, metadataType: MetadataLookUpType): void;
+    /**
+     * Merge given DiagnosticContext to current node's DiagnosticContext for bulk
+     */
+    mergeBulkDiagnostics(childDiagnostics: CosmosDiagnosticContext): void;
+    getClientSideStats(endTimeUTCInMs?: number): ClientSideRequestStatistics;
+    getTotalRequestPayloadLength(): number;
+    getTotalResponsePayloadLength(): number;
+    recordEndpointResolution(location: string): void;
+}
+//# sourceMappingURL=CosmosDiagnosticsContext.d.ts.map

package/dist/commonjs/diagnostics/CosmosDiagnosticsContext.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"CosmosDiagnosticsContext.d.ts","sourceRoot":"","sources":["../../../src/diagnostics/CosmosDiagnosticsContext.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EACV,2BAA2B,EAC3B,qBAAqB,EAErB,iBAAiB,EAEjB,kBAAkB,EACnB,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;GAQG;AACH,qBAAa,uBAAuB;IAClC,OAAO,CAAC,uBAAuB,CAAS;IACxC,OAAO,CAAC,cAAc,CAAwC;IAC9D,OAAO,CAAC,eAAe,CAAkC;IACzD,OAAO,CAAC,iBAAiB,CAA2B;IAC7C,0BAA0B,EAAE,GAAG,CAAC,MAAM,CAAC,CAAa;IAC3D,qBAAqB,EAAE,qBAAqB,CAAC;;IAMtC,mBAAmB,CACxB,iBAAiB,EAAE,iBAAiB,EACpC,kBAAkB,EAAE,MAAM,GACzB,IAAI;IAgBA,iBAAiB,CAAC,iBAAiB,EAAE,iBAAiB,GAAG,IAAI;IAI7D,2BAA2B,CAAC,qBAAqB,EAAE,qBAAqB,GAAG,IAAI;IAStF;;;OAGG;IACI,gBAAgB,CACrB,gBAAgB,EAAE,uBAAuB,EACzC,YAAY,EAAE,kBAAkB,GAC/B,IAAI;IAqBP;;OAEG;IACI,oBAAoB,CAAC,gBAAgB,EAAE,uBAAuB,GAAG,IAAI;IAuBrE,kBAAkB,CACvB,cAAc,GAAE,MAAkC,GACjD,2BAA2B;IAkBvB,4BAA4B,IAAI,MAAM;IActC,6BAA6B,IAAI,MAAM;IAcvC,wBAAwB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;CAGxD"}

package/dist/commonjs/diagnostics/CosmosDiagnosticsContext.js

@@ -0,0 +1,128 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CosmosDiagnosticContext = void 0;
+const index_js_1 = require("../common/index.js");
+const time_js_1 = require("../utils/time.js");
+/**
+ * @hidden
+ * Internal class to hold CosmosDiagnostic aggregate information all through the lifecycle of a request.
+ * This object gathers diagnostic information throughout Client operation which may span across multiple
+ * Server call, retries etc.
+ * Functions - recordFailedAttempt, recordMetaDataQuery, recordEndpointContactEvent are used to ingest
+ * data into the context. At the end of operation, getDiagnostics() is used to
+ * get final CosmosDiagnostic object.
+ */
+class CosmosDiagnosticContext {
+    constructor() {
+        this.failedAttempts = [];
+        this.metadataLookups = [];
+        this.gatewayStatistics = [];
+        this.locationEndpointsContacted = new Set();
+        this.requestStartTimeUTCinMs = (0, time_js_1.getCurrentTimestampInMs)();
+    }
+    recordFailedAttempt(gatewayStatistics, retryAttemptNumber) {
+        const attempt = {
+            attemptNumber: retryAttemptNumber,
+            startTimeUTCInMs: gatewayStatistics.startTimeUTCInMs,
+            durationInMs: gatewayStatistics.durationInMs,
+            statusCode: gatewayStatistics.statusCode,
+            substatusCode: gatewayStatistics.subStatusCode,
+            requestPayloadLengthInBytes: gatewayStatistics.requestPayloadLengthInBytes,
+            responsePayloadLengthInBytes: gatewayStatistics.responsePayloadLengthInBytes,
+            activityId: gatewayStatistics.activityId,
+            operationType: gatewayStatistics.operationType,
+            resourceType: gatewayStatistics.resourceType,
+        };
+        this.failedAttempts.push(attempt);
+    }
+    recordNetworkCall(gatewayStatistics) {
+        this.gatewayStatistics.push(gatewayStatistics);
+    }
+    recordEncryptionDiagnostics(encryptionDiagnostics) {
+        var _a, _b;
+        const { encryptContent, decryptContent } = encryptionDiagnostics;
+        const encryptionDuration = (_a = encryptContent[index_js_1.Constants.Encryption.DiagnosticsDuration]) !== null && _a !== void 0 ? _a : 0;
+        const decryptionDuration = (_b = decryptContent[index_js_1.Constants.Encryption.DiagnosticsDuration]) !== null && _b !== void 0 ? _b : 0;
+        encryptionDiagnostics.processingDurationInMs = encryptionDuration + decryptionDuration;
+        this.encryptionDiagnostics = encryptionDiagnostics;
+    }
+    /**
+     * Merge given DiagnosticContext to current node's DiagnosticContext, Treating GatewayRequests of
+     * given DiagnosticContext, as metadata requests.
+     */
+    mergeDiagnostics(childDiagnostics, metadataType) {
+        // Copy Location endpoints contacted.
+        childDiagnostics.locationEndpointsContacted.forEach((endpoint) => this.locationEndpointsContacted.add(endpoint));
+        // Copy child nodes's GatewayStatistics to parent's metadata lookups.
+        childDiagnostics.gatewayStatistics.forEach((gateway) => this.metadataLookups.push({
+            activityId: gateway.activityId,
+            requestPayloadLengthInBytes: gateway.requestPayloadLengthInBytes,
+            responsePayloadLengthInBytes: gateway.responsePayloadLengthInBytes,
+            startTimeUTCInMs: gateway.startTimeUTCInMs,
+            operationType: gateway.operationType,
+            resourceType: gateway.resourceType,
+            durationInMs: gateway.durationInMs,
+            metaDataType: metadataType,
+        }));
+    }
+    /**
+     * Merge given DiagnosticContext to current node's DiagnosticContext for bulk
+     */
+    mergeBulkDiagnostics(childDiagnostics) {
+        // Copy Location endpoints contacted.
+        childDiagnostics.locationEndpointsContacted.forEach((endpoint) => this.locationEndpointsContacted.add(endpoint));
+        // Copy child nodes's GatewayStatistics to parent's gateway statistics.
+        childDiagnostics.gatewayStatistics.forEach((gateway) => this.gatewayStatistics.push(gateway));
+        // merge metadata lookups and failed attempts
+        childDiagnostics.metadataLookups.forEach((lookup) => this.metadataLookups.push(lookup));
+        childDiagnostics.failedAttempts.forEach((lookup) => this.failedAttempts.push(lookup));
+        if (!this.encryptionDiagnostics) {
+            this.encryptionDiagnostics = childDiagnostics.encryptionDiagnostics;
+        }
+        else if (childDiagnostics.encryptionDiagnostics) {
+            this.encryptionDiagnostics.decryptContent =
+                childDiagnostics.encryptionDiagnostics.decryptContent;
+            this.encryptionDiagnostics.processingDurationInMs =
+                (this.encryptionDiagnostics.processingDurationInMs || 0) +
+                    (childDiagnostics.encryptionDiagnostics.processingDurationInMs || 0);
+        }
+    }
+    getClientSideStats(endTimeUTCInMs = (0, time_js_1.getCurrentTimestampInMs)()) {
+        return {
+            requestStartTimeUTCInMs: this.requestStartTimeUTCinMs,
+            requestDurationInMs: endTimeUTCInMs - this.requestStartTimeUTCinMs,
+            totalRequestPayloadLengthInBytes: this.getTotalRequestPayloadLength(),
+            totalResponsePayloadLengthInBytes: this.getTotalResponsePayloadLength(),
+            locationEndpointsContacted: [...this.locationEndpointsContacted.values()],
+            metadataDiagnostics: {
+                metadataLookups: [...this.metadataLookups],
+            },
+            retryDiagnostics: {
+                failedAttempts: [...this.failedAttempts],
+            },
+            gatewayStatistics: this.gatewayStatistics,
+            encryptionDiagnostics: this.encryptionDiagnostics,
+        };
+    }
+    getTotalRequestPayloadLength() {
+        let totalRequestPayloadLength = 0;
+        this.gatewayStatistics.forEach((req) => (totalRequestPayloadLength += req.requestPayloadLengthInBytes));
+        this.metadataLookups.forEach((req) => (totalRequestPayloadLength += req.requestPayloadLengthInBytes));
+        this.failedAttempts.forEach((req) => (totalRequestPayloadLength += req.requestPayloadLengthInBytes));
+        return totalRequestPayloadLength;
+    }
+    getTotalResponsePayloadLength() {
+        let totalResponsePayloadLength = 0;
+        this.gatewayStatistics.forEach((req) => (totalResponsePayloadLength += req.responsePayloadLengthInBytes));
+        this.metadataLookups.forEach((req) => (totalResponsePayloadLength += req.responsePayloadLengthInBytes));
+        this.failedAttempts.forEach((req) => (totalResponsePayloadLength += req.responsePayloadLengthInBytes));
+        return totalResponsePayloadLength;
+    }
+    recordEndpointResolution(location) {
+        this.locationEndpointsContacted.add(location);
+    }
+}
+exports.CosmosDiagnosticContext = CosmosDiagnosticContext;
+//# sourceMappingURL=CosmosDiagnosticsContext.js.map

package/dist/commonjs/diagnostics/CosmosDiagnosticsContext.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CosmosDiagnosticsContext.js","sourceRoot":"","sources":["../../../src/diagnostics/CosmosDiagnosticsContext.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAElC,iDAA+C;AAS/C,8CAA2D;AAC3D;;;;;;;;GAQG;AACH,MAAa,uBAAuB;IAQlC;QANQ,mBAAc,GAAqC,EAAE,CAAC;QACtD,oBAAe,GAA+B,EAAE,CAAC;QACjD,sBAAiB,GAAwB,EAAE,CAAC;QAC7C,+BAA0B,GAAgB,IAAI,GAAG,EAAE,CAAC;QAIzD,IAAI,CAAC,uBAAuB,GAAG,IAAA,iCAAuB,GAAE,CAAC;IAC3D,CAAC;IAEM,mBAAmB,CACxB,iBAAoC,EACpC,kBAA0B;QAE1B,MAAM,OAAO,GAAmC;YAC9C,aAAa,EAAE,kBAAkB;YACjC,gBAAgB,EAAE,iBAAiB,CAAC,gBAAgB;YACpD,YAAY,EAAE,iBAAiB,CAAC,YAAY;YAC5C,UAAU,EAAE,iBAAiB,CAAC,UAAU;YACxC,aAAa,EAAE,iBAAiB,CAAC,aAAa;YAC9C,2BAA2B,EAAE,iBAAiB,CAAC,2BAA2B;YAC1E,4BAA4B,EAAE,iBAAiB,CAAC,4BAA4B;YAC5E,UAAU,EAAE,iBAAiB,CAAC,UAAU;YACxC,aAAa,EAAE,iBAAiB,CAAC,aAAa;YAC9C,YAAY,EAAE,iBAAiB,CAAC,YAAY;SAC7C,CAAC;QACF,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;IAEM,iBAAiB,CAAC,iBAAoC;QAC3D,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACjD,CAAC;IAEM,2BAA2B,CAAC,qBAA4C;;QAC7E,MAAM,EAAE,cAAc,EAAE,cAAc,EAAE,GAAG,qBAAqB,CAAC;QACjE,MAAM,kBAAkB,GAAG,MAAA,cAAc,CAAC,oBAAS,CAAC,UAAU,CAAC,mBAAmB,CAAC,mCAAI,CAAC,CAAC;QACzF,MAAM,kBAAkB,GAAG,MAAA,cAAc,CAAC,oBAAS,CAAC,UAAU,CAAC,mBAAmB,CAAC,mCAAI,CAAC,CAAC;QAEzF,qBAAqB,CAAC,sBAAsB,GAAG,kBAAkB,GAAG,kBAAkB,CAAC;QACvF,IAAI,CAAC,qBAAqB,GAAG,qBAAqB,CAAC;IACrD,CAAC;IAED;;;OAGG;IACI,gBAAgB,CACrB,gBAAyC,EACzC,YAAgC;QAEhC,qCAAqC;QACrC,gBAAgB,CAAC,0BAA0B,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE,CAC/D,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,QAAQ,CAAC,CAC9C,CAAC;QAEF,qEAAqE;QACrE,gBAAgB,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CACrD,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC;YACxB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,2BAA2B,EAAE,OAAO,CAAC,2BAA2B;YAChE,4BAA4B,EAAE,OAAO,CAAC,4BAA4B;YAClE,gBAAgB,EAAE,OAAO,CAAC,gBAAgB;YAC1C,aAAa,EAAE,OAAO,CAAC,aAAa;YACpC,YAAY,EAAE,OAAO,CAAC,YAAY;YAClC,YAAY,EAAE,OAAO,CAAC,YAAY;YAClC,YAAY,EAAE,YAAY;SAC3B,CAAC,CACH,CAAC;IACJ,CAAC;IAED;;OAEG;IACI,oBAAoB,CAAC,gBAAyC;QACnE,qCAAqC;QACrC,gBAAgB,CAAC,0BAA0B,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE,CAC/D,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,QAAQ,CAAC,CAC9C,CAAC;QAEF,uEAAuE;QACvE,gBAAgB,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;QAC9F,6CAA6C;QAC7C,gBAAgB,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACxF,gBAAgB,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QAEtF,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAChC,IAAI,CAAC,qBAAqB,GAAG,gBAAgB,CAAC,qBAAqB,CAAC;QACtE,CAAC;aAAM,IAAI,gBAAgB,CAAC,qBAAqB,EAAE,CAAC;YAClD,IAAI,CAAC,qBAAqB,CAAC,cAAc;gBACvC,gBAAgB,CAAC,qBAAqB,CAAC,cAAc,CAAC;YACxD,IAAI,CAAC,qBAAqB,CAAC,sBAAsB;gBAC/C,CAAC,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,IAAI,CAAC,CAAC;oBACxD,CAAC,gBAAgB,CAAC,qBAAqB,CAAC,sBAAsB,IAAI,CAAC,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;IAEM,kBAAkB,CACvB,iBAAyB,IAAA,iCAAuB,GAAE;QAElD,OAAO;YACL,uBAAuB,EAAE,IAAI,CAAC,uBAAuB;YACrD,mBAAmB,EAAE,cAAc,GAAG,IAAI,CAAC,uBAAuB;YAClE,gCAAgC,EAAE,IAAI,CAAC,4BAA4B,EAAE;YACrE,iCAAiC,EAAE,IAAI,CAAC,6BAA6B,EAAE;YACvE,0BAA0B,EAAE,CAAC,GAAG,IAAI,CAAC,0BAA0B,CAAC,MAAM,EAAE,CAAC;YACzE,mBAAmB,EAAE;gBACnB,eAAe,EAAE,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC;aAC3C;YACD,gBAAgB,EAAE;gBAChB,cAAc,EAAE,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC;aACzC;YACD,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;YACzC,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;SAClD,CAAC;IACJ,CAAC;IAEM,4BAA4B;QACjC,IAAI,yBAAyB,GAAG,CAAC,CAAC;QAClC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAC5B,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,yBAAyB,IAAI,GAAG,CAAC,2BAA2B,CAAC,CACxE,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,OAAO,CAC1B,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,yBAAyB,IAAI,GAAG,CAAC,2BAA2B,CAAC,CACxE,CAAC;QACF,IAAI,CAAC,cAAc,CAAC,OAAO,CACzB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,yBAAyB,IAAI,GAAG,CAAC,2BAA2B,CAAC,CACxE,CAAC;QACF,OAAO,yBAAyB,CAAC;IACnC,CAAC;IAEM,6BAA6B;QAClC,IAAI,0BAA0B,GAAG,CAAC,CAAC;QACnC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAC5B,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,0BAA0B,IAAI,GAAG,CAAC,4BAA4B,CAAC,CAC1E,CAAC;QACF,IAAI,CAAC,eAAe,CAAC,OAAO,CAC1B,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,0BAA0B,IAAI,GAAG,CAAC,4BAA4B,CAAC,CAC1E,CAAC;QACF,IAAI,CAAC,cAAc,CAAC,OAAO,CACzB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,0BAA0B,IAAI,GAAG,CAAC,4BAA4B,CAAC,CAC1E,CAAC;QACF,OAAO,0BAA0B,CAAC;IACpC,CAAC;IAEM,wBAAwB,CAAC,QAAgB;QAC9C,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;CACF;AArJD,0DAqJC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { Constants } from \"../common/index.js\";\nimport type {\n  ClientSideRequestStatistics,\n  EncryptionDiagnostics,\n  FailedRequestAttemptDiagnostic,\n  GatewayStatistics,\n  MetadataLookUpDiagnostic,\n  MetadataLookUpType,\n} from \"../CosmosDiagnostics.js\";\nimport { getCurrentTimestampInMs } from \"../utils/time.js\";\n/**\n * @hidden\n * Internal class to hold CosmosDiagnostic aggregate information all through the lifecycle of a request.\n * This object gathers diagnostic information throughout Client operation which may span across multiple\n * Server call, retries etc.\n * Functions - recordFailedAttempt, recordMetaDataQuery, recordEndpointContactEvent are used to ingest\n * data into the context. At the end of operation, getDiagnostics() is used to\n * get final CosmosDiagnostic object.\n */\nexport class CosmosDiagnosticContext {\n  private requestStartTimeUTCinMs: number;\n  private failedAttempts: FailedRequestAttemptDiagnostic[] = [];\n  private metadataLookups: MetadataLookUpDiagnostic[] = [];\n  private gatewayStatistics: GatewayStatistics[] = [];\n  public locationEndpointsContacted: Set<string> = new Set();\n  encryptionDiagnostics: EncryptionDiagnostics;\n\n  public constructor() {\n    this.requestStartTimeUTCinMs = getCurrentTimestampInMs();\n  }\n\n  public recordFailedAttempt(\n    gatewayStatistics: GatewayStatistics,\n    retryAttemptNumber: number,\n  ): void {\n    const attempt: FailedRequestAttemptDiagnostic = {\n      attemptNumber: retryAttemptNumber,\n      startTimeUTCInMs: gatewayStatistics.startTimeUTCInMs,\n      durationInMs: gatewayStatistics.durationInMs,\n      statusCode: gatewayStatistics.statusCode,\n      substatusCode: gatewayStatistics.subStatusCode,\n      requestPayloadLengthInBytes: gatewayStatistics.requestPayloadLengthInBytes,\n      responsePayloadLengthInBytes: gatewayStatistics.responsePayloadLengthInBytes,\n      activityId: gatewayStatistics.activityId,\n      operationType: gatewayStatistics.operationType,\n      resourceType: gatewayStatistics.resourceType,\n    };\n    this.failedAttempts.push(attempt);\n  }\n\n  public recordNetworkCall(gatewayStatistics: GatewayStatistics): void {\n    this.gatewayStatistics.push(gatewayStatistics);\n  }\n\n  public recordEncryptionDiagnostics(encryptionDiagnostics: EncryptionDiagnostics): void {\n    const { encryptContent, decryptContent } = encryptionDiagnostics;\n    const encryptionDuration = encryptContent[Constants.Encryption.DiagnosticsDuration] ?? 0;\n    const decryptionDuration = decryptContent[Constants.Encryption.DiagnosticsDuration] ?? 0;\n\n    encryptionDiagnostics.processingDurationInMs = encryptionDuration + decryptionDuration;\n    this.encryptionDiagnostics = encryptionDiagnostics;\n  }\n\n  /**\n   * Merge given DiagnosticContext to current node's DiagnosticContext, Treating GatewayRequests of\n   * given DiagnosticContext, as metadata requests.\n   */\n  public mergeDiagnostics(\n    childDiagnostics: CosmosDiagnosticContext,\n    metadataType: MetadataLookUpType,\n  ): void {\n    // Copy Location endpoints contacted.\n    childDiagnostics.locationEndpointsContacted.forEach((endpoint) =>\n      this.locationEndpointsContacted.add(endpoint),\n    );\n\n    // Copy child nodes's GatewayStatistics to parent's metadata lookups.\n    childDiagnostics.gatewayStatistics.forEach((gateway) =>\n      this.metadataLookups.push({\n        activityId: gateway.activityId,\n        requestPayloadLengthInBytes: gateway.requestPayloadLengthInBytes,\n        responsePayloadLengthInBytes: gateway.responsePayloadLengthInBytes,\n        startTimeUTCInMs: gateway.startTimeUTCInMs,\n        operationType: gateway.operationType,\n        resourceType: gateway.resourceType,\n        durationInMs: gateway.durationInMs,\n        metaDataType: metadataType,\n      }),\n    );\n  }\n\n  /**\n   * Merge given DiagnosticContext to current node's DiagnosticContext for bulk\n   */\n  public mergeBulkDiagnostics(childDiagnostics: CosmosDiagnosticContext): void {\n    // Copy Location endpoints contacted.\n    childDiagnostics.locationEndpointsContacted.forEach((endpoint) =>\n      this.locationEndpointsContacted.add(endpoint),\n    );\n\n    // Copy child nodes's GatewayStatistics to parent's gateway statistics.\n    childDiagnostics.gatewayStatistics.forEach((gateway) => this.gatewayStatistics.push(gateway));\n    // merge metadata lookups and failed attempts\n    childDiagnostics.metadataLookups.forEach((lookup) => this.metadataLookups.push(lookup));\n    childDiagnostics.failedAttempts.forEach((lookup) => this.failedAttempts.push(lookup));\n\n    if (!this.encryptionDiagnostics) {\n      this.encryptionDiagnostics = childDiagnostics.encryptionDiagnostics;\n    } else if (childDiagnostics.encryptionDiagnostics) {\n      this.encryptionDiagnostics.decryptContent =\n        childDiagnostics.encryptionDiagnostics.decryptContent;\n      this.encryptionDiagnostics.processingDurationInMs =\n        (this.encryptionDiagnostics.processingDurationInMs || 0) +\n        (childDiagnostics.encryptionDiagnostics.processingDurationInMs || 0);\n    }\n  }\n\n  public getClientSideStats(\n    endTimeUTCInMs: number = getCurrentTimestampInMs(),\n  ): ClientSideRequestStatistics {\n    return {\n      requestStartTimeUTCInMs: this.requestStartTimeUTCinMs,\n      requestDurationInMs: endTimeUTCInMs - this.requestStartTimeUTCinMs,\n      totalRequestPayloadLengthInBytes: this.getTotalRequestPayloadLength(),\n      totalResponsePayloadLengthInBytes: this.getTotalResponsePayloadLength(),\n      locationEndpointsContacted: [...this.locationEndpointsContacted.values()],\n      metadataDiagnostics: {\n        metadataLookups: [...this.metadataLookups],\n      },\n      retryDiagnostics: {\n        failedAttempts: [...this.failedAttempts],\n      },\n      gatewayStatistics: this.gatewayStatistics,\n      encryptionDiagnostics: this.encryptionDiagnostics,\n    };\n  }\n\n  public getTotalRequestPayloadLength(): number {\n    let totalRequestPayloadLength = 0;\n    this.gatewayStatistics.forEach(\n      (req) => (totalRequestPayloadLength += req.requestPayloadLengthInBytes),\n    );\n    this.metadataLookups.forEach(\n      (req) => (totalRequestPayloadLength += req.requestPayloadLengthInBytes),\n    );\n    this.failedAttempts.forEach(\n      (req) => (totalRequestPayloadLength += req.requestPayloadLengthInBytes),\n    );\n    return totalRequestPayloadLength;\n  }\n\n  public getTotalResponsePayloadLength(): number {\n    let totalResponsePayloadLength = 0;\n    this.gatewayStatistics.forEach(\n      (req) => (totalResponsePayloadLength += req.responsePayloadLengthInBytes),\n    );\n    this.metadataLookups.forEach(\n      (req) => (totalResponsePayloadLength += req.responsePayloadLengthInBytes),\n    );\n    this.failedAttempts.forEach(\n      (req) => (totalResponsePayloadLength += req.responsePayloadLengthInBytes),\n    );\n    return totalResponsePayloadLength;\n  }\n\n  public recordEndpointResolution(location: string): void {\n    this.locationEndpointsContacted.add(location);\n  }\n}\n"]}

package/dist/commonjs/diagnostics/DiagnosticFormatter.d.ts

@@ -0,0 +1,8 @@
+import type { CosmosDiagnostics } from "../CosmosDiagnostics.js";
+export interface DiagnosticFormatter {
+    format(cosmosDiagnostic: CosmosDiagnostics): string;
+}
+export declare class DefaultDiagnosticFormatter implements DiagnosticFormatter {
+    format(cosmosDiagnostic: CosmosDiagnostics): string;
+}
+//# sourceMappingURL=DiagnosticFormatter.d.ts.map

package/dist/commonjs/diagnostics/DiagnosticFormatter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"DiagnosticFormatter.d.ts","sourceRoot":"","sources":["../../../src/diagnostics/DiagnosticFormatter.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAEjE,MAAM,WAAW,mBAAmB;IAClC,MAAM,CAAC,gBAAgB,EAAE,iBAAiB,GAAG,MAAM,CAAC;CACrD;AAED,qBAAa,0BAA2B,YAAW,mBAAmB;IACpE,MAAM,CAAC,gBAAgB,EAAE,iBAAiB,GAAG,MAAM;CAGpD"}

package/dist/commonjs/diagnostics/DiagnosticFormatter.js

@@ -0,0 +1,12 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DefaultDiagnosticFormatter = void 0;
+class DefaultDiagnosticFormatter {
+    format(cosmosDiagnostic) {
+        return JSON.stringify(cosmosDiagnostic);
+    }
+}
+exports.DefaultDiagnosticFormatter = DefaultDiagnosticFormatter;
+//# sourceMappingURL=DiagnosticFormatter.js.map

package/dist/commonjs/diagnostics/DiagnosticFormatter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"DiagnosticFormatter.js","sourceRoot":"","sources":["../../../src/diagnostics/DiagnosticFormatter.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAQlC,MAAa,0BAA0B;IACrC,MAAM,CAAC,gBAAmC;QACxC,OAAO,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAC1C,CAAC;CACF;AAJD,gEAIC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type { CosmosDiagnostics } from \"../CosmosDiagnostics.js\";\n\nexport interface DiagnosticFormatter {\n  format(cosmosDiagnostic: CosmosDiagnostics): string;\n}\n\nexport class DefaultDiagnosticFormatter implements DiagnosticFormatter {\n  format(cosmosDiagnostic: CosmosDiagnostics): string {\n    return JSON.stringify(cosmosDiagnostic);\n  }\n}\n"]}

package/dist/commonjs/diagnostics/DiagnosticNodeInternal.d.ts

@@ -0,0 +1,72 @@
+import { DiagnosticNode, MetadataLookUpType } from "../CosmosDiagnostics.js";
+import { CosmosDbDiagnosticLevel } from "./CosmosDbDiagnosticLevel.js";
+import type { CosmosHeaders } from "../queryExecutionContext/CosmosHeaders.js";
+import type { OperationType, ResourceType } from "../common/index.js";
+/**
+ * @hidden
+ * This is Internal Representation for DiagnosticNode. It contains useful helper functions to collect
+ * diagnostic information throughout the lifetime of Diagnostic session.
+ * The functions toDiagnosticNode() & toDiagnostic() are given to convert it to public facing counterpart.
+ */
+export declare class DiagnosticNodeInternal implements DiagnosticNode {
+    id: string;
+    nodeType: DiagnosticNodeType;
+    parent: DiagnosticNodeInternal;
+    children: DiagnosticNodeInternal[];
+    data: Partial<DiagnosticDataValue>;
+    startTimeUTCInMs: number;
+    durationInMs: number;
+    diagnosticLevel: CosmosDbDiagnosticLevel;
+    private diagnosticCtx;
+    private encryptionDiagnostics;
+}
+/**
+ * @hidden
+ */
+export type DiagnosticDataValue = {
+    selectedLocation: string;
+    activityId: string;
+    requestAttempNumber: number;
+    requestPayloadLengthInBytes: number;
+    responsePayloadLengthInBytes: number;
+    responseStatus: number;
+    readFromCache: boolean;
+    operationType: OperationType;
+    metadatOperationType: MetadataLookUpType;
+    resourceType: ResourceType;
+    failedAttempty: boolean;
+    successfulRetryPolicy: string;
+    partitionKeyRangeId: string;
+    stateful: boolean;
+    queryRecordsRead: number;
+    queryMethodIdentifier: string;
+    log: string[];
+    failure: boolean;
+    startTimeUTCInMs: number;
+    durationInMs: number;
+    requestData: Partial<{
+        requestPayloadLengthInBytes: number;
+        responsePayloadLengthInBytes: number;
+        operationType: OperationType;
+        resourceType: ResourceType;
+        headers: CosmosHeaders;
+        requestBody: any;
+        responseBody: any;
+        url: string;
+    }>;
+};
+/**
+ * @hidden
+ */
+export declare enum DiagnosticNodeType {
+    CLIENT_REQUEST_NODE = "CLIENT_REQUEST_NODE",// Top most node representing client operations.
+    METADATA_REQUEST_NODE = "METADATA_REQUEST_NODE",// Node representing a metadata request.
+    HTTP_REQUEST = "HTTP_REQUEST",// Node representing REST call to backend services.
+    BATCH_REQUEST = "BATCH_REQUEST",// Node representing batch request.
+    PARALLEL_QUERY_NODE = "PARALLEL_QUERY_NODE",// Node representing parallel query execution.
+    DEFAULT_QUERY_NODE = "DEFAULT_QUERY_NODE",// Node representing default query execution.
+    QUERY_REPAIR_NODE = "QUERY_REPAIR_NODE",// Node representing query repair.
+    BACKGROUND_REFRESH_THREAD = "BACKGROUND_REFRESH_THREAD",// Node representing background refresh.
+    REQUEST_ATTEMPTS = "REQUEST_ATTEMPTS"
+}
+//# sourceMappingURL=DiagnosticNodeInternal.d.ts.map

package/dist/commonjs/diagnostics/DiagnosticNodeInternal.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"DiagnosticNodeInternal.d.ts","sourceRoot":"","sources":["../../../src/diagnostics/DiagnosticNodeInternal.ts"],"names":[],"mappings":"AAKA,OAAO,EACL,cAAc,EAEd,kBAAkB,EAEnB,MAAM,yBAAyB,CAAC;AAGjC,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AACvE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,2CAA2C,CAAC;AAE/E,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAKtE;;;;;GAKG;AACH,qBAAa,sBAAuB,YAAW,cAAc;IACpD,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,kBAAkB,CAAC;IAC7B,MAAM,EAAE,sBAAsB,CAAC;IAC/B,QAAQ,EAAE,sBAAsB,EAAE,CAAC;IACnC,IAAI,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACnC,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,uBAAuB,CAAC;IAChD,OAAO,CAAC,aAAa,CAA0B;IAC/C,OAAO,CAAC,qBAAqB,CAAwB;CAsVtD;AAED;;GAEG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAChC,gBAAgB,EAAE,MAAM,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;IACnB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,2BAA2B,EAAE,MAAM,CAAC;IACpC,4BAA4B,EAAE,MAAM,CAAC;IACrC,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,OAAO,CAAC;IACvB,aAAa,EAAE,aAAa,CAAC;IAC7B,oBAAoB,EAAE,kBAAkB,CAAC;IACzC,YAAY,EAAE,YAAY,CAAC;IAC3B,cAAc,EAAE,OAAO,CAAC;IACxB,qBAAqB,EAAE,MAAM,CAAC;IAC9B,mBAAmB,EAAE,MAAM,CAAC;IAC5B,QAAQ,EAAE,OAAO,CAAC;IAClB,gBAAgB,EAAE,MAAM,CAAC;IACzB,qBAAqB,EAAE,MAAM,CAAC;IAC9B,GAAG,EAAE,MAAM,EAAE,CAAC;IACd,OAAO,EAAE,OAAO,CAAC;IACjB,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,OAAO,CAAC;QACnB,2BAA2B,EAAE,MAAM,CAAC;QACpC,4BAA4B,EAAE,MAAM,CAAC;QACrC,aAAa,EAAE,aAAa,CAAC;QAC7B,YAAY,EAAE,YAAY,CAAC;QAC3B,OAAO,EAAE,aAAa,CAAC;QACvB,WAAW,EAAE,GAAG,CAAC;QACjB,YAAY,EAAE,GAAG,CAAC;QAClB,GAAG,EAAE,MAAM,CAAC;KACb,CAAC,CAAC;CACJ,CAAC;AAEF;;GAEG;AACH,oBAAY,kBAAkB;IAC5B,mBAAmB,wBAAwB,CAAE,gDAAgD;IAC7F,qBAAqB,0BAA0B,CAAE,wCAAwC;IACzF,YAAY,iBAAiB,CAAE,mDAAmD;IAClF,aAAa,kBAAkB,CAAE,mCAAmC;IACpE,mBAAmB,wBAAwB,CAAE,8CAA8C;IAC3F,kBAAkB,uBAAuB,CAAE,6CAA6C;IACxF,iBAAiB,sBAAsB,CAAE,kCAAkC;IAC3E,yBAAyB,8BAA8B,CAAE,wCAAwC;IACjG,gBAAgB,qBAAqB;CACtC"}

package/dist/commonjs/diagnostics/DiagnosticNodeInternal.js

@@ -0,0 +1,304 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DiagnosticNodeType = exports.DiagnosticNodeInternal = void 0;
+const CosmosDiagnosticsContext_js_1 = require("./CosmosDiagnosticsContext.js");
+const index_js_1 = require("../request/index.js");
+const CosmosDiagnostics_js_1 = require("../CosmosDiagnostics.js");
+const time_js_1 = require("../utils/time.js");
+const CosmosDbDiagnosticLevel_js_1 = require("./CosmosDbDiagnosticLevel.js");
+const index_js_2 = require("../common/index.js");
+const diagnosticLevelComparator_js_1 = require("./diagnosticLevelComparator.js");
+const core_util_1 = require("@azure/core-util");
+/**
+ * @hidden
+ * This is Internal Representation for DiagnosticNode. It contains useful helper functions to collect
+ * diagnostic information throughout the lifetime of Diagnostic session.
+ * The functions toDiagnosticNode() & toDiagnostic() are given to convert it to public facing counterpart.
+ */
+class DiagnosticNodeInternal {
+    /**
+     * @internal
+     */
+    constructor(diagnosticLevel, type, parent, data = {}, startTimeUTCInMs = (0, time_js_1.getCurrentTimestampInMs)(), ctx = new CosmosDiagnosticsContext_js_1.CosmosDiagnosticContext()) {
+        this.id = (0, core_util_1.randomUUID)();
+        this.nodeType = type;
+        this.startTimeUTCInMs = startTimeUTCInMs;
+        this.data = data;
+        this.children = [];
+        this.durationInMs = 0;
+        this.parent = parent;
+        this.diagnosticCtx = ctx;
+        this.diagnosticLevel = diagnosticLevel;
+        // Initialize EncryptionDiagnostics
+        this.encryptionDiagnostics = {
+            encryptContent: {},
+            decryptContent: {},
+            processingDurationInMs: 0,
+        };
+    }
+    /**
+     * @internal
+     */
+    addLog(msg) {
+        if (!this.data.log) {
+            this.data.log = [];
+        }
+        this.data.log.push(msg);
+    }
+    /**
+     * @internal
+     */
+    sanitizeHeaders(headers) {
+        return headers;
+    }
+    /**
+     * Updated durationInMs for node, based on endTimeUTCInMs provided.
+     * @internal
+     */
+    updateTimestamp(endTimeUTCInMs = (0, time_js_1.getCurrentTimestampInMs)()) {
+        this.durationInMs = endTimeUTCInMs - this.startTimeUTCInMs;
+    }
+    /**
+     * @internal
+     */
+    recordSuccessfulNetworkCall(startTimeUTCInMs, requestContext, pipelineResponse, substatus, url) {
+        const responseHeaders = pipelineResponse.headers.toJSON();
+        const gatewayRequest = {
+            activityId: responseHeaders[index_js_2.Constants.HttpHeaders.ActivityId],
+            correlateActivityId: requestContext.headers[index_js_2.Constants.HttpHeaders.CorrelatedActivityId],
+            startTimeUTCInMs,
+            durationInMs: (0, time_js_1.getCurrentTimestampInMs)() - startTimeUTCInMs,
+            statusCode: pipelineResponse.status,
+            subStatusCode: substatus,
+            requestPayloadLengthInBytes: calculateRequestPayloadLength(requestContext),
+            responsePayloadLengthInBytes: calculateResponsePayloadLength(pipelineResponse),
+            operationType: requestContext.operationType,
+            resourceType: requestContext.resourceType,
+            partitionKeyRangeId: requestContext.partitionKeyRangeId,
+        };
+        let requestData = {
+            OperationType: gatewayRequest.operationType,
+            resourceType: gatewayRequest.resourceType,
+            requestPayloadLengthInBytes: gatewayRequest.requestPayloadLengthInBytes,
+        };
+        if ((0, diagnosticLevelComparator_js_1.allowTracing)(CosmosDbDiagnosticLevel_js_1.CosmosDbDiagnosticLevel.debugUnsafe, this.diagnosticLevel)) {
+            requestData = Object.assign(Object.assign({}, requestData), { headers: this.sanitizeHeaders(requestContext.headers), requestBody: requestContext.body, responseBody: pipelineResponse.bodyAsText, url: url });
+        }
+        this.addData({
+            requestPayloadLengthInBytes: gatewayRequest.requestPayloadLengthInBytes,
+            responsePayloadLengthInBytes: gatewayRequest.responsePayloadLengthInBytes,
+            startTimeUTCInMs: gatewayRequest.startTimeUTCInMs,
+            durationInMs: gatewayRequest.durationInMs,
+            requestData,
+        });
+        this.diagnosticCtx.recordNetworkCall(gatewayRequest);
+    }
+    /**
+     * @internal
+     */
+    recordFailedNetworkCall(startTimeUTCInMs, requestContext, retryAttemptNumber, statusCode, substatusCode, responseHeaders) {
+        this.addData({ failedAttempty: true });
+        const requestPayloadLengthInBytes = calculateRequestPayloadLength(requestContext);
+        this.diagnosticCtx.recordFailedAttempt({
+            activityId: responseHeaders[index_js_2.Constants.HttpHeaders.ActivityId],
+            correlatedActivityId: requestContext.headers[index_js_2.Constants.HttpHeaders.CorrelatedActivityId],
+            startTimeUTCInMs,
+            durationInMs: (0, time_js_1.getCurrentTimestampInMs)() - startTimeUTCInMs,
+            statusCode,
+            subStatusCode: substatusCode,
+            requestPayloadLengthInBytes,
+            responsePayloadLengthInBytes: 0,
+            operationType: requestContext.operationType,
+            resourceType: requestContext.resourceType,
+        }, retryAttemptNumber);
+        let requestData = {
+            OperationType: requestContext.operationType,
+            resourceType: requestContext.resourceType,
+            requestPayloadLengthInBytes,
+        };
+        if ((0, diagnosticLevelComparator_js_1.allowTracing)(CosmosDbDiagnosticLevel_js_1.CosmosDbDiagnosticLevel.debugUnsafe, this.diagnosticLevel)) {
+            requestData = Object.assign(Object.assign({}, requestData), { headers: this.sanitizeHeaders(requestContext.headers), requestBody: requestContext.body, url: (0, index_js_2.prepareURL)(requestContext.endpoint, requestContext.path) });
+        }
+        this.addData({
+            failedAttempty: true,
+            requestData,
+        });
+    }
+    /**
+     * @internal
+     */
+    recordEndpointResolution(location) {
+        this.addData({ selectedLocation: location });
+        this.diagnosticCtx.recordEndpointResolution(location);
+    }
+    /**
+     * @internal
+     */
+    addData(data, msg, level = this.diagnosticLevel) {
+        if (level !== CosmosDbDiagnosticLevel_js_1.CosmosDbDiagnosticLevel.info) {
+            this.data = Object.assign(Object.assign({}, this.data), data);
+            if (msg) {
+                this.addLog(msg);
+            }
+        }
+    }
+    /**
+     * Merge given DiagnosticNodeInternal's context to current node's DiagnosticContext, Treating GatewayRequests of
+     * given DiagnosticContext, as metadata requests. Given DiagnosticNodeInternal becomes a child of this node.
+     * @internal
+     */
+    addChildNode(child, level, metadataType) {
+        this.diagnosticCtx.mergeDiagnostics(child.diagnosticCtx, metadataType);
+        if ((0, diagnosticLevelComparator_js_1.allowTracing)(level, this.diagnosticLevel)) {
+            child.parent = this;
+            this.children.push(child);
+        }
+        return child;
+    }
+    /**
+     * Merge given DiagnosticNodeInternal's context to current node's DiagnosticContext for bulk.
+     * Given DiagnosticNodeInternal becomes a child of this node.
+     * @internal
+     */
+    addBulkChildNode(child, level) {
+        this.diagnosticCtx.mergeBulkDiagnostics(child.diagnosticCtx);
+        if ((0, diagnosticLevelComparator_js_1.allowTracing)(level, this.diagnosticLevel)) {
+            child.parent = this;
+            this.children.push(child);
+        }
+        return child;
+    }
+    /**
+     * @internal
+     */
+    initializeChildNode(type, level, data = {}) {
+        if ((0, diagnosticLevelComparator_js_1.allowTracing)(level, this.diagnosticLevel)) {
+            const child = new DiagnosticNodeInternal(this.diagnosticLevel, type, this, data, (0, time_js_1.getCurrentTimestampInMs)(), this.diagnosticCtx);
+            this.children.push(child);
+            return child;
+        }
+        else {
+            return this;
+        }
+    }
+    /**
+     * @internal
+     */
+    recordQueryResult(resources, level) {
+        var _a;
+        if ((0, diagnosticLevelComparator_js_1.allowTracing)(level, this.diagnosticLevel)) {
+            const previousCount = (_a = this.data.queryRecordsRead) !== null && _a !== void 0 ? _a : 0;
+            if (Array.isArray(resources)) {
+                this.data.queryRecordsRead = previousCount + resources.length;
+            }
+        }
+    }
+    /**
+     * @internal
+     * record startTime for encryption in an operation
+     */
+    beginEncryptionDiagnostics(operation) {
+        const startTime = (0, time_js_1.getCurrentTimestampInMs)();
+        switch (operation) {
+            case index_js_2.Constants.Encryption.DiagnosticsEncryptOperation:
+                this.encryptionDiagnostics.encryptContent[index_js_2.Constants.Encryption.DiagnosticsStartTime] =
+                    startTime;
+                break;
+            case index_js_2.Constants.Encryption.DiagnosticsDecryptOperation:
+                this.encryptionDiagnostics.decryptContent[index_js_2.Constants.Encryption.DiagnosticsStartTime] =
+                    startTime;
+                break;
+            default:
+                throw new index_js_1.ErrorResponse("Invalid operation type for encryption diagnostics");
+        }
+    }
+    /**
+     * @internal
+     * record duration from startTime and properties count for encryption in an operation
+     */
+    endEncryptionDiagnostics(operation, propertiesCount) {
+        const endTime = (0, time_js_1.getCurrentTimestampInMs)();
+        let processingDuration = 0;
+        switch (operation) {
+            case index_js_2.Constants.Encryption.DiagnosticsEncryptOperation:
+                processingDuration =
+                    endTime -
+                        this.encryptionDiagnostics.encryptContent[index_js_2.Constants.Encryption.DiagnosticsStartTime];
+                this.encryptionDiagnostics.encryptContent[index_js_2.Constants.Encryption.DiagnosticsDuration] =
+                    processingDuration;
+                // will be undefined in case of bulk/batch
+                if (propertiesCount !== undefined) {
+                    this.encryptionDiagnostics.encryptContent[index_js_2.Constants.Encryption.DiagnosticsPropertiesEncryptedCount] = propertiesCount;
+                }
+                break;
+            case index_js_2.Constants.Encryption.DiagnosticsDecryptOperation:
+                processingDuration =
+                    endTime -
+                        this.encryptionDiagnostics.decryptContent[index_js_2.Constants.Encryption.DiagnosticsStartTime];
+                this.encryptionDiagnostics.decryptContent[index_js_2.Constants.Encryption.DiagnosticsDuration] =
+                    processingDuration;
+                if (propertiesCount !== undefined) {
+                    this.encryptionDiagnostics.decryptContent[index_js_2.Constants.Encryption.DiagnosticsPropertiesDecryptedCount] = propertiesCount;
+                }
+                break;
+            default:
+                throw new index_js_1.ErrorResponse("Invalid operation type for encryption diagnostics");
+        }
+        this.diagnosticCtx.recordEncryptionDiagnostics(this.encryptionDiagnostics);
+    }
+    /**
+     * Convert DiagnosticNodeInternal (internal representation) to DiagnosticNode (public, sanitized representation)
+     * @internal
+     */
+    toDiagnosticNode() {
+        return {
+            id: this.id,
+            nodeType: this.nodeType,
+            children: this.children.map((child) => child.toDiagnosticNode()),
+            data: this.data,
+            startTimeUTCInMs: this.startTimeUTCInMs,
+            durationInMs: this.durationInMs,
+        };
+    }
+    /**
+     * Convert to CosmosDiagnostics
+     * @internal
+     */
+    toDiagnostic(clientConfigDiagnostic) {
+        const rootNode = (0, CosmosDiagnostics_js_1.getRootNode)(this);
+        const diagnostiNode = (0, diagnosticLevelComparator_js_1.allowTracing)(CosmosDbDiagnosticLevel_js_1.CosmosDbDiagnosticLevel.debug, this.diagnosticLevel)
+            ? rootNode.toDiagnosticNode()
+            : undefined;
+        const clientConfig = (0, diagnosticLevelComparator_js_1.allowTracing)(CosmosDbDiagnosticLevel_js_1.CosmosDbDiagnosticLevel.debug, this.diagnosticLevel)
+            ? clientConfigDiagnostic
+            : undefined;
+        const cosmosDiagnostic = new CosmosDiagnostics_js_1.CosmosDiagnostics(this.diagnosticCtx.getClientSideStats(), diagnostiNode, clientConfig);
+        return cosmosDiagnostic;
+    }
+}
+exports.DiagnosticNodeInternal = DiagnosticNodeInternal;
+/**
+ * @hidden
+ */
+var DiagnosticNodeType;
+(function (DiagnosticNodeType) {
+    DiagnosticNodeType["CLIENT_REQUEST_NODE"] = "CLIENT_REQUEST_NODE";
+    DiagnosticNodeType["METADATA_REQUEST_NODE"] = "METADATA_REQUEST_NODE";
+    DiagnosticNodeType["HTTP_REQUEST"] = "HTTP_REQUEST";
+    DiagnosticNodeType["BATCH_REQUEST"] = "BATCH_REQUEST";
+    DiagnosticNodeType["PARALLEL_QUERY_NODE"] = "PARALLEL_QUERY_NODE";
+    DiagnosticNodeType["DEFAULT_QUERY_NODE"] = "DEFAULT_QUERY_NODE";
+    DiagnosticNodeType["QUERY_REPAIR_NODE"] = "QUERY_REPAIR_NODE";
+    DiagnosticNodeType["BACKGROUND_REFRESH_THREAD"] = "BACKGROUND_REFRESH_THREAD";
+    DiagnosticNodeType["REQUEST_ATTEMPTS"] = "REQUEST_ATTEMPTS";
+})(DiagnosticNodeType || (exports.DiagnosticNodeType = DiagnosticNodeType = {}));
+function calculateResponsePayloadLength(response) {
+    var _a;
+    return ((_a = response === null || response === void 0 ? void 0 : response.bodyAsText) === null || _a === void 0 ? void 0 : _a.length) || 0;
+}
+function calculateRequestPayloadLength(requestContext) {
+    return requestContext.body ? requestContext.body.length : 0;
+}
+//# sourceMappingURL=DiagnosticNodeInternal.js.map