feat: Add duration tracking for query execution success and failure in nextPage function (#2487)

src/Common/IteratorUtilities.ts

@@ -19,9 +19,16 @@ export interface MinimalQueryIterator {
 // Pick<QueryIterator<any>, "fetchNext">;
 
 export function nextPage(documentsIterator: MinimalQueryIterator, firstItemIndex: number): Promise<QueryResults> {
-  TelemetryProcessor.traceStart(Action.ExecuteQuery);
+  const startKey = TelemetryProcessor.traceStart(Action.ExecuteQuery);
-  return documentsIterator.fetchNext().then((response) => {
+  return documentsIterator
-    TelemetryProcessor.traceSuccess(Action.ExecuteQuery, { dataExplorerArea: Constants.Areas.Tab });
+    .fetchNext()
+    .then((response) => {
+      const durationMs = Date.now() - startKey;
+      TelemetryProcessor.traceSuccess(
+        Action.ExecuteQuery,
+        { dataExplorerArea: Constants.Areas.Tab, durationMs },
+        startKey,
+      );
       const documents = response.resources;
       // eslint-disable-next-line @typescript-eslint/no-explicit-any
       const headers = (response as any).headers || {}; // TODO this is a private key. Remove any
@@ -36,5 +43,14 @@ export function nextPage(documentsIterator: MinimalQueryIterator, firstItemIndex
         activityId: response.activityId,
         requestCharge: response.requestCharge,
       };
+    })
+    .catch((error) => {
+      const durationMs = Date.now() - startKey;
+      TelemetryProcessor.traceFailure(
+        Action.ExecuteQuery,
+        { dataExplorerArea: Constants.Areas.Tab, durationMs, error: error.message },
+        startKey,
+      );
+      throw error;
     });
 }