[fix][dingo-executor] Fix disk ann index state issue (#1331)

dingodb · Jan 9, 2025 · 079f1ce · 079f1ce
1 parent 17806c8
commit 079f1ce
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/dingo-store-proxy/src/main/java/io/dingodb/store/proxy/service/StoreService.java b/dingo-store-proxy/src/main/java/io/dingodb/store/proxy/service/StoreService.java
@@ -51,6 +51,7 @@
 import io.dingodb.sdk.service.DocumentService;
 import io.dingodb.sdk.service.IndexService;
 import io.dingodb.sdk.service.Services;
+import io.dingodb.sdk.service.entity.common.DiskANNState;
 import io.dingodb.sdk.service.entity.common.DocumentWithScore;
 import io.dingodb.sdk.service.entity.common.Location;
 import io.dingodb.sdk.service.entity.common.RangeWithOptions;
@@ -686,7 +687,9 @@ public String diskAnnBuild(long requestTs, CommonId indexId, long ts) {
             );
             Optional<VectorStateParameter> optionalState = Optional.ofNullable(vectorBuildResponse.getState());
             return optionalState
-                .map(state -> ((StateDiskAnnParam) state.getState()).getDiskannState().toString())
+                .map(state -> Optional.ofNullable(((StateDiskAnnParam) state.getState()).getDiskannState())
+                    .map(DiskANNState::toString)
+                    .orElse("DISKANN_INITIALIZED"))
                 .orElse("DISKANN_INITIALIZED");
         }
 
@@ -712,7 +715,9 @@ public String diskAnnLoad(long requestTs, CommonId indexId, int nodesCacheNum, b
             );
             Optional<VectorStateParameter> optionalState = Optional.ofNullable(vectorLoadResponse.getState());
             return optionalState
-                .map(state -> ((StateDiskAnnParam) state.getState()).getDiskannState().toString())
+                .map(state -> Optional.ofNullable(((StateDiskAnnParam) state.getState()).getDiskannState())
+                    .map(DiskANNState::toString)
+                    .orElse("DISKANN_INITIALIZED"))
                 .orElse("DISKANN_INITIALIZED");
         }
 
@@ -724,7 +729,9 @@ public String diskAnnStatus(long requestTs, CommonId indexId) {
             );
             Optional<VectorStateParameter> optionalState = Optional.ofNullable(vectorStatusResponse.getState());
             return optionalState
-                .map(state -> ((StateDiskAnnParam) state.getState()).getDiskannState().toString())
+                .map(state -> Optional.ofNullable(((StateDiskAnnParam) state.getState()).getDiskannState())
+                    .map(DiskANNState::toString)
+                    .orElse("DISKANN_INITIALIZED"))
                 .orElse("DISKANN_INITIALIZED");
         }
 
@@ -747,7 +754,9 @@ public String diskAnnReset(long requestTs, CommonId indexId) {
             );
             Optional<VectorStateParameter> optionalState = Optional.ofNullable(vectorResetResponse.getState());
             return optionalState
-                .map(state -> ((StateDiskAnnParam) state.getState()).getDiskannState().toString())
+                .map(state -> Optional.ofNullable(((StateDiskAnnParam) state.getState()).getDiskannState())
+                    .map(DiskANNState::toString)
+                    .orElse("DISKANN_INITIALIZED"))
                 .orElse("DISKANN_INITIALIZED");
         }