From 5cc8f437692a35bb9bc7b678bec98f9976b20105 Mon Sep 17 00:00:00 2001
From: Pavel Ivanov <pafelka@gmail.com>
Date: Mon, 16 Dec 2024 12:05:36 +0200
Subject: [PATCH] fix: improve render count and timing tracking accuracy

---
 packages/scan/src/core/index.ts | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/packages/scan/src/core/index.ts b/packages/scan/src/core/index.ts
index a92d65d3..f3f87f39 100644
--- a/packages/scan/src/core/index.ts
+++ b/packages/scan/src/core/index.ts
@@ -240,7 +240,11 @@ export const reportRender = (fiber: Fiber, renders: Array<Render>) => {
   Store.lastReportTime.value = performance.now();
 
   if (prevRenderData) {
+    prevRenderData.count = ((prevRenderData.count) ?? 0) + renders.length;
     prevRenderData.renders.push(...renders);
+    if (renders[0]?.time) {
+      prevRenderData.time = ((prevRenderData.time) ?? 0) + (renders[0].time);
+    }
   } else {
     const { selfTime } = getTimings(fiber);
 
@@ -259,7 +263,11 @@ export const reportRender = (fiber: Fiber, renders: Array<Render>) => {
     const prevLegacyRenderData = Store.legacyReportData.get(displayName);
 
     if (prevLegacyRenderData) {
+      prevLegacyRenderData.count = ((prevLegacyRenderData.count) ?? 0) + renders.length;
       prevLegacyRenderData.renders.push(...renders);
+      if (renders[0]?.time) {
+        prevLegacyRenderData.time = ((prevLegacyRenderData.time) ?? 0) + (renders[0].time);
+      }
     } else {
       const { selfTime } = getTimings(fiber);