Prevent failure when heap section is missing in JVMStatisticsData (#223)

Co-authored-by: William Montaz <[email protected]>
criteo · Oct 24, 2022 · cfe0aa1 · cfe0aa1
1 parent e3ae055
commit cfe0aa1
Showing 1 changed file with 23 additions and 22 deletions.
diff --git a/readers/hdfs/src/main/java/com/criteo/hadoop/garmadon/hdfs/ReaderFactory.java b/readers/hdfs/src/main/java/com/criteo/hadoop/garmadon/hdfs/ReaderFactory.java
@@ -132,38 +132,39 @@ private static void addJvmStatHeapMessage(Map<Integer, GarmadonEventDescriptor>
             JVMStatisticsExplodedProtos.JvmStatisticsHeap.newBuilder(),
             body -> {
                 JVMStatisticsEventsProtos.JVMStatisticsData jvmStatisticsData = (JVMStatisticsEventsProtos.JVMStatisticsData) body;
-                JVMStatisticsEventsProtos.JVMStatisticsData.Section heapSection = jvmStatisticsData.getSectionList()
+                Optional<JVMStatisticsEventsProtos.JVMStatisticsData.Section> maybeHeapSection = jvmStatisticsData.getSectionList()
                             .stream()
                             .filter(section -> section.getName().equals("heap"))
-                            .findFirst()
-                            .orElseThrow(() -> new RuntimeException("JVMStatisticsData is supposed to have a heap section but could not find one"));
-
-                long init = 0;
-                long committed = 0;
-                long used = 0;
-                long max = 0;
-                for (JVMStatisticsEventsProtos.JVMStatisticsData.Property property : heapSection.getPropertyList()) {
-                    if (property.getName().equals("init")) {
-                        init = Long.parseLong(property.getValue());
-                    }
-                    if (property.getName().equals("committed")) {
-                        committed = Long.parseLong(property.getValue());
-                    }
-                    if (property.getName().equals("used")) {
-                        used = Long.parseLong(property.getValue());
-                    }
-                    if (property.getName().equals("max")) {
-                        max = Long.parseLong(property.getValue());
+                            .findFirst();
+
+                return maybeHeapSection.map(heapSection -> {
+                    long init = 0;
+                    long committed = 0;
+                    long used = 0;
+                    long max = 0;
+                    for (JVMStatisticsEventsProtos.JVMStatisticsData.Property property : heapSection.getPropertyList()) {
+                        if (property.getName().equals("init")) {
+                            init = Long.parseLong(property.getValue());
+                        }
+                        if (property.getName().equals("committed")) {
+                            committed = Long.parseLong(property.getValue());
+                        }
+                        if (property.getName().equals("used")) {
+                            used = Long.parseLong(property.getValue());
+                        }
+                        if (property.getName().equals("max")) {
+                            max = Long.parseLong(property.getValue());
+                        }
                     }
-                }
 
-                return JVMStatisticsExplodedProtos.JvmStatisticsHeap
+                    return JVMStatisticsExplodedProtos.JvmStatisticsHeap
                             .newBuilder()
                             .setInit(init)
                             .setCommitted(committed)
                             .setUsed(used)
                             .setMax(max)
                             .build();
+                }).orElseGet(() -> JVMStatisticsExplodedProtos.JvmStatisticsHeap.newBuilder().build());
             }
         );
     }