From 3b25ba065176b1ebcd73bd32019f1ee40b068e12 Mon Sep 17 00:00:00 2001
From: Yan Chen <48968912+chenyan-dfinity@users.noreply.github.com>
Date: Tue, 17 Oct 2023 12:23:53 -0700
Subject: [PATCH] streaming profiling (#477)

---
 tools/ui/src/candid.ts | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/tools/ui/src/candid.ts b/tools/ui/src/candid.ts
index 4b025cf6..bbba1cda 100644
--- a/tools/ui/src/candid.ts
+++ b/tools/ui/src/candid.ts
@@ -98,7 +98,7 @@ export async function fetchActor(canisterId: Principal): Promise<ActorSubclass>
 
 export function getProfilerActor(canisterId: Principal): ActorSubclass {
   const profiler_interface: IDL.InterfaceFactory = ({ IDL }) => IDL.Service({
-    __get_profiling: IDL.Func([], [IDL.Vec(IDL.Tuple(IDL.Int32, IDL.Int64))], ['query']),
+    __get_profiling: IDL.Func([IDL.Int32], [IDL.Vec(IDL.Tuple(IDL.Int32, IDL.Int64)), IDL.Opt(IDL.Int32)], ['query']),
     __get_cycles: IDL.Func([], [IDL.Int64], ['query']),
   });
   return Actor.createActor(profiler_interface, { agent, canisterId });
@@ -155,7 +155,19 @@ async function getDidJsFromMetadata(canisterId: Principal): Promise<undefined |
 export async function getProfiling(canisterId: Principal): Promise<Array<[number, bigint]>|undefined> {
   try {
     const actor = getProfilerActor(canisterId);
-    const info = await actor.__get_profiling() as Array<[number, bigint]>;
+    let info: Array<[number, bigint]> = [];
+    let idx = 0;
+    let cnt = 0;
+    while (cnt < 50) {
+      const [res, next] = await actor.__get_profiling(idx) as [Array<[number, bigint]>, [number]|[]];
+      info = info.concat(res);
+      if (next.length === 1) {
+        idx = next[0];
+        cnt++;
+      } else {
+        break;
+      }
+    }
     return info;
   } catch(err) {
     console.log(err);