[CHERRY PICK] Adding a additional check to just process first 50 obje…

…cts of hot store (#284) cr : https://code.amazon.com/reviews/CR-46506118
opendistro-for-elasticsearch · Mar 3, 2021 · 4ab4a0b · 4ab4a0b
1 parent 8cf28ba
commit 4ab4a0b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...o/elasticsearch/performanceanalyzer/collectors/ShardIndexingPressureMetricsCollector.java b/...o/elasticsearch/performanceanalyzer/collectors/ShardIndexingPressureMetricsCollector.java
@@ -55,6 +55,8 @@ public class ShardIndexingPressureMetricsCollector extends PerformanceAnalyzerMe
     public static final String SHARD_INDEXING_PRESSURE_STORE_FIELD_NAME = "shardIndexingPressureStore";
     public static final String SHARD_INDEXING_PRESSURE_HOT_STORE_FIELD_NAME = "shardIndexingPressureHotStore";
 
+    private static final Integer MAX_HOT_STORE_LIMIT = 50;
+
     private final ConfigOverridesWrapper configOverridesWrapper;
     private final PerformanceAnalyzerController controller;
     private StringBuilder value;
@@ -86,7 +88,7 @@ public void collectMetrics(long startTime) {
                             .get(shardIndexingPressureStore);
 
                     value.setLength(0);
-                    shardIndexingPressureHotStore.entrySet().stream().forEach(storeObject -> {
+                    shardIndexingPressureHotStore.entrySet().stream().limit(MAX_HOT_STORE_LIMIT).forEach(storeObject -> {
                         try {
                             JSONObject tracker = (JSONObject) parser.parse(mapper.writeValueAsString(storeObject.getValue()));
                             JSONObject shardId = (JSONObject) parser.parse(mapper.writeValueAsString(tracker.get("shardId")));