feat(qlsrc): support sort cpu flat/cum time and flamegraph root node cum time

Author: Lslightly

qlsrc/pprof_ext/cpu_flamegraph_root_cum_desc.ql

@@ -0,0 +1,6 @@
+import cpuprofile
+
+from CPUProfile p, string rootFuncFName
+where rootFuncFName = p.flameGraphRootFuncName()
+select rootFuncFName, timeInUnit(p.cumTimeOfFunc(rootFuncFName), "ms", 4) as rootCumTime
+order by rootCumTime desc

qlsrc/pprof_ext/cpu_func_cum_desc.ql

@@ -0,0 +1,6 @@
+import cpuprofile
+
+from CPUProfile profile, string funcFName
+where profile.containsFunc(funcFName)
+select funcNamePart(funcFName) as funcName, timeInUnit(profile.cumTimeOfFunc(funcFName), "ms", 4) as cum
+order by cum desc

qlsrc/pprof_ext/cpu_func_flat_desc.ql

@@ -0,0 +1,6 @@
+import cpuprofile
+
+from CPUProfile profile, string funcFName
+where profile.containsFunc(funcFName)
+select funcNamePart(funcFName) as funcName, timeInUnit(profile.flatTimeOfFunc(funcFName), "ms", 4) as flat
+order by flat desc

qlsrc/pprof_ext/cpuprofile.qll

@@ -108,5 +108,11 @@ class CPUProfile extends Profile {
             | sample.getLocation(sample.locationNum()-1).getLastLine().getFunction().getName() = result
         )
     }
+
+    predicate containsFunc(string funcFullName) {
+        exists(Sample sample |
+            sample.containsFunc(funcFullName)
+        )
+    }
 }
 

qlsrc/pprof_ext/helper.qll

@@ -22,3 +22,11 @@ bindingset[pkgPath, funcName]
 string funcFullName(string pkgPath, string funcName) {
     result = pkgPath+"."+funcName
 }
+
+/**
+ * return funcName part of pkgPath+"."+funcName
+ */
+bindingset[fullname]
+string funcNamePart(string fullname) {
+    result = fullname.suffix(fullname.splitAt(".", 0).length()+1)
+}