Merge branch 'feature/apache#6766-json-startree-index-to-api' of gith…

…ub.com:kriti-sc/incubator-pinot into feature/apache#6766-json-startree-index-to-api
kriti-sc · May 16, 2021 · eafd833 · eafd833
2 parents e8781e0 + 9ead5e1
commit eafd833
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/pinot-server/src/main/java/org/apache/pinot/server/api/resources/SegmentMetadataFetcher.java b/pinot-server/src/main/java/org/apache/pinot/server/api/resources/SegmentMetadataFetcher.java
@@ -86,11 +86,12 @@ public static String getSegmentMetadata(SegmentDataManager segmentDataManager, L
    */
   private static Map<String, Map<String, String>> getIndexesForSegmentColumns(SegmentDataManager segmentDataManager) {
     Map<String, Map<String, String>> columnIndexMap = null;
-    ImmutableSegmentDataManager immutableSegmentDataManager = (ImmutableSegmentDataManager) segmentDataManager;
-    ImmutableSegment immutableSegment = immutableSegmentDataManager.getSegment();
-    ImmutableSegmentImpl immutableSegmentImpl = (ImmutableSegmentImpl) immutableSegment;
-    Map<String, ColumnIndexContainer> columnIndexContainerMap = immutableSegmentImpl.getIndexContainerMap();
-    columnIndexMap = getImmutableSegmentColumnIndexes(columnIndexContainerMap);
+    IndexSegment segment = segmentDataManager.getSegment();
+    if (segment instanceof ImmutableSegmentImpl) {
+      return getImmutableSegmentColumnIndexes(((ImmutableSegmentImpl) segment).getIndexContainerMap());
+    } else {
+      return null;
+    }
     return columnIndexMap;
   }