Check directory casting state

Signed-off-by: luyuncheng <[email protected]>
opensearch-project · Aug 15, 2024 · 2bcc139 · 2bcc139
1 parent 70a2ea4
commit 2bcc139
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/main/java/org/opensearch/knn/index/codec/KNN80Codec/KNN80DocValuesProducer.java b/src/main/java/org/opensearch/knn/index/codec/KNN80Codec/KNN80DocValuesProducer.java
@@ -58,7 +58,16 @@ public KNN80DocValuesProducer(DocValuesProducer delegate, SegmentReadState state
         if (state.directory instanceof KNN80CompoundDirectory) {
             directory = ((KNN80CompoundDirectory) state.directory).getDir();
         }
-        String directoryPath = ((FSDirectory) FilterDirectory.unwrap(directory)).getDirectory().toString();
+
+        FSDirectory fsDirectory = null;
+        try {
+             fsDirectory = ((FSDirectory) FilterDirectory.unwrap(directory));
+        } catch (ClassCastException ex) {
+            log.warn("{} can not casting to FSDirectory", directory.toString());
+            return;
+        }
+        assert fsDirectory != null;
+        String directoryPath = fsDirectory.getDirectory().toString();
         for (FieldInfo field : state.fieldInfos) {
             if (!field.attributes().containsKey(KNN_FIELD)) {
                 continue;