chore: added warning message on HNSB index if corrupted

ArcadeData · Sep 20, 2023 · 08123a1 · 08123a1
1 parent 616843d
commit 08123a1
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/engine/src/main/java/com/arcadedb/index/vector/HnswVectorIndex.java b/engine/src/main/java/com/arcadedb/index/vector/HnswVectorIndex.java
@@ -352,8 +352,10 @@ public boolean add(Vertex vertex) {
         if (currObj != null) {
           if (vertexMaxLevel < entryPointCopyMaxLevel) {
             final TVector vector = getVectorFromVertex(currObj);
-            if (vector == null)
+            if (vector == null) {
+              LogManager.instance().log(this, Level.WARNING, "Vector not found in vertex %s", currObj);
               throw new IndexException("Embeddings not found in object " + currObj);
+            }
 
             TDistance curDist = distanceFunction.distance(vertexVector, vector);
             for (int activeLevel = entryPointCopyMaxLevel; activeLevel > vertexMaxLevel; activeLevel--) {
@@ -368,9 +370,11 @@ public boolean add(Vertex vertex) {
                     final Vertex candidateNode = candidateConnections.next();
 
                     final TVector candidateNodeVector = getVectorFromVertex(candidateNode);
-                    if (candidateNodeVector == null)
+                    if (candidateNodeVector == null) {
                       // INVALID
+                      LogManager.instance().log(this, Level.WARNING, "Vector not found in vertex %s", candidateNode);
                       continue;
+                    }
 
                     final TDistance candidateDistance = distanceFunction.distance(vertexVector, candidateNodeVector);
 
@@ -512,8 +516,10 @@ public List<SearchResult<Vertex, TDistance>> findNearest(final TVector destinati
     Vertex currObj = entryPointCopy;
 
     final TVector vector = getVectorFromVertex(currObj);
-    if (vector == null)
+    if (vector == null) {
+      LogManager.instance().log(this, Level.WARNING, "Vector not found in vertex %s", currObj);
       return Collections.emptyList();
+    }
 
     TDistance curDist = distanceFunction.distance(destination, vector);
 
@@ -598,9 +604,11 @@ private PriorityQueue<NodeIdAndDistance<TDistance>> searchBaseLayer(final Vertex
           visitedNodes.add(candidateNode.getIdentity());
 
           final TVector vector = getVectorFromVertex(candidateNode);
-          if (vector == null)
+          if (vector == null) {
             // INVALID
+            LogManager.instance().log(this, Level.WARNING, "Vector not found in vertex %s", candidateNode);
             continue;
+          }
 
           final TDistance candidateDistance = distanceFunction.distance(destination, vector);
           if (topCandidates.size() < k || gt(lowerBound, candidateDistance)) {