From 34efbea4c5172678861bdad80267f8096411c135 Mon Sep 17 00:00:00 2001
From: Pavel Feldman <pavel.feldman@gmail.com>
Date: Fri, 10 Mar 2023 17:01:19 -0800
Subject: [PATCH] chore(ui): throttle tree updates (#21578)

---
 packages/trace-viewer/src/ui/watchMode.tsx | 26 +++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/packages/trace-viewer/src/ui/watchMode.tsx b/packages/trace-viewer/src/ui/watchMode.tsx
index 2def9415fd..3079411dd5 100644
--- a/packages/trace-viewer/src/ui/watchMode.tsx
+++ b/packages/trace-viewer/src/ui/watchMode.tsx
@@ -357,6 +357,22 @@ declare global {
 
 let receiver: TeleReporterReceiver | undefined;
 
+let throttleTimer: NodeJS.Timeout | undefined;
+let throttleData: { rootSuite: Suite, progress: Progress } | undefined;
+const throttledAction = () => {
+  clearTimeout(throttleTimer);
+  throttleTimer = undefined;
+  updateRootSuite(throttleData!.rootSuite, throttleData!.progress);
+};
+
+const throttleUpdateRootSuite = (rootSuite: Suite, progress: Progress, immediate = false) => {
+  throttleData = { rootSuite, progress };
+  if (immediate)
+    throttledAction();
+  else if (!throttleTimer)
+    throttleTimer = setTimeout(throttledAction, 250);
+};
+
 const refreshRootSuite = (eraseResults: boolean) => {
   if (!eraseResults) {
     sendMessageNoReply('list');
@@ -378,11 +394,15 @@ const refreshRootSuite = (eraseResults: boolean) => {
       progress.passed = 0;
       progress.failed = 0;
       progress.skipped = 0;
-      updateRootSuite(rootSuite, progress);
+      throttleUpdateRootSuite(rootSuite, progress, true);
+    },
+
+    onEnd: () => {
+      throttleUpdateRootSuite(rootSuite, progress, true);
     },
 
     onTestBegin: () => {
-      updateRootSuite(rootSuite, progress);
+      throttleUpdateRootSuite(rootSuite, progress);
     },
 
     onTestEnd: (test: TestCase) => {
@@ -392,7 +412,7 @@ const refreshRootSuite = (eraseResults: boolean) => {
         ++progress.failed;
       else
         ++progress.passed;
-      updateRootSuite(rootSuite, progress);
+      throttleUpdateRootSuite(rootSuite, progress);
       // This will update selected trace viewer.
       updateStepsProgress();
     },