From b1fd70a3c3eb2919ae42f04a07dc445d5bf38b4a Mon Sep 17 00:00:00 2001
From: Pritesh Nandgaonkar <prit91@fb.com>
Date: Tue, 30 Jul 2019 06:47:12 -0700
Subject: [PATCH] Show status message with percentage for the seriailization

Reviewed By: passy

Differential Revision: D16382096

fbshipit-source-id: 17c44a7befcef82815dc9d2379898e23a36a05c8
---
 src/utils/exportData.js    |  6 +++++-
 src/utils/serialization.js | 26 +++++++++++++++++++++++---
 2 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/src/utils/exportData.js b/src/utils/exportData.js
index 09fe330ac..305d5f066 100644
--- a/src/utils/exportData.js
+++ b/src/utils/exportData.js
@@ -405,7 +405,11 @@ export function exportStore(
       }
       try {
         statusUpdate && statusUpdate('Serializing Flipper data...');
-        const serializedString = await serialize(exportData, idler);
+        const serializedString = await serialize(
+          exportData,
+          idler,
+          statusUpdate,
+        );
         if (serializedString.length <= 0) {
           reject(new Error('Serialize function returned empty string'));
         }
diff --git a/src/utils/serialization.js b/src/utils/serialization.js
index 932c2104c..2bdd51661 100644
--- a/src/utils/serialization.js
+++ b/src/utils/serialization.js
@@ -6,8 +6,14 @@
  */
 
 import {Idler} from './Idler';
-export async function serialize(obj: Object, idler?: Idler): Promise<string> {
-  return makeObjectSerializable(obj, idler).then(obj => JSON.stringify(obj));
+export async function serialize(
+  obj: Object,
+  idler?: Idler,
+  statusUpdate?: (msg: string) => void,
+): Promise<string> {
+  return makeObjectSerializable(obj, idler, statusUpdate).then(obj =>
+    JSON.stringify(obj),
+  );
 }
 
 export function deserialize(str: string): Object {
@@ -112,12 +118,19 @@ export function processObjectToBeSerialized(
   }
   return {childNeedsIteration, outputObject: obj};
 }
-export async function makeObjectSerializable(obj: any, idler?: Idler): any {
+export async function makeObjectSerializable(
+  obj: any,
+  idler?: Idler,
+  statusUpdate?: (msg: string) => void,
+): any {
   if (!(obj instanceof Object)) {
     return obj;
   }
   const stack = [obj];
   const dict: Map<any, any> = new Map();
+  let numIterations = 0;
+  let prevStackLength = stack.length;
+  let accumulator = prevStackLength;
   while (stack.length > 0) {
     if (idler) {
       await idler.idle();
@@ -178,6 +191,13 @@ export async function makeObjectSerializable(obj: any, idler?: Idler): any {
       dict.set(element, outputObject);
     }
     stack.pop();
+    ++numIterations;
+    accumulator +=
+      stack.length >= prevStackLength ? stack.length - prevStackLength + 1 : 0;
+    const percentage = (numIterations / accumulator) * 100;
+    statusUpdate &&
+      statusUpdate(`Serializing Flipper (${percentage.toFixed(2)}%) `);
+    prevStackLength = stack.length;
   }
   return dict.get(obj);
 }