From 3611964b17ada62a51bd02dabd7876fd11d715c7 Mon Sep 17 00:00:00 2001
From: akvlad <akvlad90@gmail.com>
Date: Wed, 21 Aug 2024 20:43:04 +0300
Subject: [PATCH] fix rust library bug

---
 pyroscope/pprof-bin/pkg/pprof_bin_bg.wasm | Bin 225243 -> 225258 bytes
 pyroscope/pprof-bin/src/lib.rs            |   5 +++++
 pyroscope/pprof.js                        |   3 ---
 pyroscope/pyroscope.js                    |   1 -
 pyroscope/select_series.js                |   1 -
 5 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/pyroscope/pprof-bin/pkg/pprof_bin_bg.wasm b/pyroscope/pprof-bin/pkg/pprof_bin_bg.wasm
index a110fa20b1bcf70162a8c29840339d66ce3b1f53..6380006b4212600682e3b878038e4d8b20ecde61 100644
GIT binary patch
delta 2230
zcmZwIOGs2<6bJA#w+Yf35z=1x>Z)mJsaa;3HDg*tZh{DvHZBSkaVU%yl^4-yK*(sU
z+0?=)Fp!cYrdotWln~Lw9*crm2pY8stW84nKgFH<_oKzky}xt5b06pXzL`&nz}G}z
z?6Kb$9e?8g{j>0k-&)+W#Ib@_Fi%2x9ifrF2}wWF5D>7mqu7#$fG_&)*O1R2ZJ!VI
zjn2nM=Y6x&6-^>ijL(=aT~cvu#%F`!&E%HY(}b1pTtFzF!h?n{!P%4_#rb-}E6BHT
zUSfC^c?{>9JUj9h*y08Lx{y4D^Pu4h!$ss7oU^~dt115-=UyKik!J{Or2*=VhPgxH
z6Um@_8cc5_h450!AB1i40XWMw?k|l9tich7<m}Y~JexVNzD0MqyS1~ctks?6(}?rK
z_QKr$2)v!HI}DeTof`;nyPw0{t{3n!DtHN}kzd2R$y0Ex;TH0H<Y_%P@&N%}Kp){!
zDp2$KNWk2gCQD=;jc<mz1L{i3x4;ESqyHM<8>;n&lPT%=_NJr@@l5hr!+mfL<<G-w
z4fDc{Xr5aeiFHm}MZSU95Dv~D(kWElwrJr-3>!WU=hApJZ&dXj%6Ay!)imA_4RApg
z%o{5kuA&R(z}4hj*e2U>jp5`zK)e8XZy{h}G@vsV?ruAJf(lCz=OtDS^Mh5wyh5tr
zom8*hPu>RaGTey!CpTj?2CO94!n~=1|8a=MhmhZ4xQ<+hJWrk4;Wyli056?=Ft6z!
zSS$R-BQWoQTZZq#G}kh2c*epWx9JH#$LeRS8Yhl%9KH|N(=avvSv)YtKZLpSYMbgG
z!Sv%G>M}3y$QuNBvrihHfceF*!o0Vy!TH_=<uc4Oqvm`RjrR8Za*;a=0wVX~7iX<k
zsm?JJ+PH9k#=n2xHds5%)+OV*9eI6F*C&noQ%1d->wEAVwpVXUF9P~I?p{!>|7x!L
hugqDQp@<P3fb(e7Fsysnt-1~$BHxT3p0m!K`2!V0{5}8x

delta 2217
zcmZwIO-NKx6bJA-SA)i~gfu_MoZggLnx<A*qvco@fjh@dJ29a|GDM;e@j_!lLJ=5D
z3&X%5DhkPBh*@b6LS<izs1_oULM>WIn+Br)Dc-#QTw2V$_dDmF_i^sMGvDU3zszS(
zPG|atre-pK|IGcAsaFlIHo7UG!j;|STD4@W(XYR5_%cQ>B*R)aMH9(*BI#RRtZ7q9
zmE$w%%TZZ4M)28Txt81>`!KJA<_bbV3XfR63KvuUF3z`G-blWO^9sua<Oevf@@&*=
zV7C|e>uuyYoHtpnv0O%8z&U#fE~NZ-oO^vRN=+lMhX#muTILQVk;|pL-<S4ADgdvc
zd@JmbkHDK;<NoR>0!29Dkea=EfM>H1mbd8d?eFM0UESf%@>#_BVFzGte-z$F*Byg*
zk<B;)-0m5e+w}roO9iv=I`T{SAbAdMw0xZW7I|6^M!iFT7tnjSk_yDUK0d?TnKrFd
zDUFZ7+yQYN<=f${DWm@q;2VnNhO_=0e0x)J3h_MhIm?6aX3AfJw^-(d8<jk_HX6IQ
zsLOoAu?ZZ^2-1Ek@61Twh~1WZ;1U`y=8Y;oMENdjyqLxtB>^tThk0Waz<cR}g>XH&
z2zJQDaL96MA0S?WytfcAF%poO>+L_;)klS;i1QMwfce3yU|u0La0AtgkCAKP1D0EH
z|I}uz!+?BpJ<OXb^dE<5d;{|PEjN=Jk>{xs8-Bwu0=#sZVP4ZeuvYkuCt%(KcPu}I
zX|B~H%L^KN+@UA@7;9V7A@dmHI6MU(reR|Kvv^{Se+qNw#SYa!gXzaXiL1T5QLhl-
z&HmhS0_GQw!MwL`!a?tX>N?CbBj)@yG}_zq8<e`A7QhmTk1y*tS*8Pp4lX>D`|sbk
z6P6CMWy!ccfxJAZ>oZn;pH(mB`tx`W$E$bL00Qzm?p{zV|7xyHZ2gMP3rDT+FdU#^
VW3X&ux9k=iA>WR7t?0`Ge*h`f_d5Up

diff --git a/pyroscope/pprof-bin/src/lib.rs b/pyroscope/pprof-bin/src/lib.rs
index e3985831..c07523a7 100644
--- a/pyroscope/pprof-bin/src/lib.rs
+++ b/pyroscope/pprof-bin/src/lib.rs
@@ -23,6 +23,8 @@ use std::vec::Vec;
 use wasm_bindgen::prelude::*;
 use std::sync::Arc;
 
+//TODO: REMOVE
+use std::fs;
 
 pub mod pprof_pb {
 
@@ -80,6 +82,9 @@ struct Tree {
 impl Tree {
     pub fn total(&self) -> i64 {
         let mut total: i64 = 0;
+        if !self.nodes.contains_key(&0) {
+            return  0 as i64;
+        }
         for c in 0..self.nodes.get(&0).unwrap().len() {
             let _c = &self.nodes.get(&0).unwrap()[c];
             total += _c.total[0];
diff --git a/pyroscope/pprof.js b/pyroscope/pprof.js
index e261d79e..144e3f3d 100644
--- a/pyroscope/pprof.js
+++ b/pyroscope/pprof.js
@@ -136,7 +136,6 @@ const bfs = (t) => {
  * @param {string} sampleType
  */
 const createFlameGraph = (pprofBinaries, sampleType) => {
-  console.log(`got ${pprofBinaries.length} profiles`)
   const tree = new Tree()
   tree.sampleType = sampleType
   let start = Date.now()
@@ -144,10 +143,8 @@ const createFlameGraph = (pprofBinaries, sampleType) => {
     const prof = messages.Profile.deserializeBinary(p)
     tree.merge(prof)
   }
-  console.log(`ds + merge took ${Date.now() - start} ms`)
   start = Date.now()
   const levels = bfs(tree)
-  console.log(`bfs took ${Date.now() - start} ms`)
   return { levels: levels, names: tree.names, total: parseInt(tree.root.total), maxSelf: parseInt(tree.maxSelf) }
 }
 
diff --git a/pyroscope/pyroscope.js b/pyroscope/pyroscope.js
index f979e10a..57609adc 100644
--- a/pyroscope/pyroscope.js
+++ b/pyroscope/pyroscope.js
@@ -402,7 +402,6 @@ const analyzeQuery = async (req, res) => {
   const toTimeSec = Math.floor(req.getEnd && req.getEnd()
     ? parseInt(req.getEnd()) / 1000
     : Date.now() / 1000)
-  console.log(query)
 
   const scope = new messages.QueryScope()
   scope.setComponentType('store')
diff --git a/pyroscope/select_series.js b/pyroscope/select_series.js
index 0f406654..f25db50c 100644
--- a/pyroscope/select_series.js
+++ b/pyroscope/select_series.js
@@ -133,7 +133,6 @@ const selectSeriesImpl = async (fromTimeSec, toTimeSec, payload) => {
 
   const resp = new messages.SelectSeriesResponse()
   resp.setSeriesList(seriesList)
-  console.log(`Queried ${seriesList.length} series`)
   return resp
 }