From 913095e950960d01cf8623fe488527173367c224 Mon Sep 17 00:00:00 2001
From: karishmas6 <carishmashukla28@gmail.com>
Date: Tue, 29 Oct 2024 04:26:19 +0530
Subject: [PATCH] feat: get count of items

---
 .../workflow-management/scheduler/index.ts    | 29 ++++++++++++++-----
 1 file changed, 22 insertions(+), 7 deletions(-)

diff --git a/server/src/workflow-management/scheduler/index.ts b/server/src/workflow-management/scheduler/index.ts
index fc45b7e0..75d9f0ef 100644
--- a/server/src/workflow-management/scheduler/index.ts
+++ b/server/src/workflow-management/scheduler/index.ts
@@ -134,10 +134,25 @@ async function executeRun(id: string) {
     });
 
     let totalRowsExtracted = 0;
-    run.dataValues.serializableOutput['item-0'].forEach((item: any) => {
-      totalRowsExtracted += Object.keys(item).length;
-    }
-    );
+      let extractedScreenshotsCount = 0;
+      let extractedItemsCount = 0;
+
+      if (run.dataValues.binaryOutput && run.dataValues.binaryOutput["item-0"]) {
+        extractedScreenshotsCount = 1;
+      }
+
+      if (run.dataValues.serializableOutput && run.dataValues.serializableOutput["item-0"]) {
+        const itemsArray = run.dataValues.serializableOutput["item-0"];
+        extractedItemsCount = itemsArray.length;
+
+        totalRowsExtracted = itemsArray.reduce((total, item) => {
+          return total + Object.keys(item).length;
+        }, 0);
+      }
+
+      console.log(`Extracted Items Count: ${extractedItemsCount}`);
+      console.log(`Extracted Screenshots Count: ${extractedScreenshotsCount}`);
+      console.log(`Total Rows Extracted: ${totalRowsExtracted}`);
 
     capture(
       'maxun-oss-run-created-scheduled',
@@ -145,9 +160,9 @@ async function executeRun(id: string) {
         runId: id,
         created_at: new Date().toISOString(),
         status: 'success',
-        extractedItemsCount: run.dataValues.serializableOutput['item-0'].length,
-        extractedRowsCount: totalRowsExtracted,
-        extractedScreenshotsCount: run.dataValues.binaryOutput['item-0'].length,
+        totalRowsExtracted,
+          extractedItemsCount,
+          extractedScreenshotsCount,
       }
     );