Get executables for flamegraph nodes

elastic · Jul 5, 2022 · 2f52e94 · 2f52e94
1 parent df2a2c4
commit 2f52e94
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/src/plugins/profiling/server/routes/flamegraph.ts b/src/plugins/profiling/server/routes/flamegraph.ts
@@ -7,11 +7,12 @@
  */
 
 export class FlameGraph {
-  constructor(events, totalEvents, stackTraces, stackFrames) {
+  constructor(events, totalEvents, stackTraces, stackFrames, executables) {
     this.events = events;
     this.totalEvents = totalEvents;
     this.stacktraces = stackTraces;
     this.stackframes = stackFrames;
+    this.executables = executables;
   }
 
   toElastic() {

diff --git a/src/plugins/profiling/server/routes/search_flameChart.ts b/src/plugins/profiling/server/routes/search_flameChart.ts
@@ -151,11 +151,26 @@ export function registerFlameChartSearchRoute(router: IRouter<DataRequestHandler
           body: { ids: [...stackFrameDocIDs] },
         });
 
+        const executableDocIDs = new Set<string>();
+        for (const trace of resStackTraces.body.docs) {
+          if (trace.found) {
+            for (const fileID of trace._source.FileID) {
+              executableDocIDs.add(fileID);
+            }
+          }
+        }
+
+        const resExecutables = await esClient.mget<any>({
+          index: 'profiling-executables',
+          body: { ids: [...executableDocIDs] },
+        });
+
         const flamegraph = new FlameGraph(
           resEvents.body,
           resTotalEvents.body,
           resStackTraces.body.docs,
-          resStackFrames.body.docs
+          resStackFrames.body.docs,
+          resExecutables.body.docs
         );
 
         return response.ok({