Fixing variable names

Signed-off-by: Harsh Garg <[email protected]>
opensearch-project · Sep 10, 2024 · 68c720c · 68c720c
1 parent 8e86828
commit 68c720c
Showing 1 changed file with 19 additions and 11 deletions.
diff --git a/...er/src/main/java/org/opensearch/action/admin/indices/segments/IndicesSegmentResponse.java b/...er/src/main/java/org/opensearch/action/admin/indices/segments/IndicesSegmentResponse.java
@@ -84,22 +84,30 @@ public Map<String, IndexSegments> getIndices() {
         if (indicesSegments != null) {
             return indicesSegments;
         }
-
-        Arrays.sort(shards, Comparator.comparing(shardSegment -> shardSegment.getShardRouting().getIndexName()));
         Map<String, IndexSegments> indicesSegments = new HashMap<>();
+        if (shards.length == 0) {
+            this.indicesSegments = indicesSegments;
+            return indicesSegments;
+        }
 
-        int lastIdx = 0;
+        Arrays.sort(shards, Comparator.comparing(shardSegment -> shardSegment.getShardRouting().getIndexName()));
+        int startIndexPos = 0;
+        String startIndexName = shards[startIndexPos].getShardRouting().getIndexName();
         for (int i = 0; i < shards.length; i++) {
-            String lastIndexName = shards[lastIdx].getShardRouting().getIndexName();
-            String currentIndexName = shards[i].getShardRouting().getIndexName();
-            if (!currentIndexName.equals(lastIndexName)) {
-                indicesSegments.put(lastIndexName, new IndexSegments(lastIndexName, Arrays.copyOfRange(shards, lastIdx, i)));
-                lastIdx = i;
-            }
-            if (i == shards.length - 1) {
-                indicesSegments.put(currentIndexName, new IndexSegments(currentIndexName, Arrays.copyOfRange(shards, lastIdx, i + 1)));
+            if (!shards[i].getShardRouting().getIndexName().equals(startIndexName)) {
+                indicesSegments.put(startIndexName, new IndexSegments(startIndexName, Arrays.copyOfRange(shards, startIndexPos, i)));
+                startIndexPos = i;
+                startIndexName = shards[startIndexPos].getShardRouting().getIndexName();
             }
         }
+        // Add the last shardSegment from shards list which would have got missed in the loop above
+        indicesSegments.put(
+            shards[shards.length - 1].getShardRouting().getIndexName(),
+            new IndexSegments(
+                shards[shards.length - 1].getShardRouting().getIndexName(),
+                Arrays.copyOfRange(shards, startIndexPos, shards.length)
+            )
+        );
         this.indicesSegments = indicesSegments;
         return indicesSegments;
     }