fix overflowing int for files larger than 2GB

jbellis · May 10, 2024 · 54fe767 · 54fe767
1 parent c899285
commit 54fe767
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jvector-base/src/main/java/io/github/jbellis/jvector/graph/disk/OnDiskGraphIndexWriter.java b/jvector-base/src/main/java/io/github/jbellis/jvector/graph/disk/OnDiskGraphIndexWriter.java
@@ -121,7 +121,7 @@ public int getMaxOrdinal() {
 
     private long featureOffsetForOrdinal(int ordinal) {
         int edgeSize = Integer.BYTES * (1 + graph.maxDegree());
-        int inlineBytes = ordinal * (Integer.BYTES + featureMap.values().stream().mapToInt(Feature::inlineSize).sum() + edgeSize);
+        long inlineBytes = ordinal * (long) (Integer.BYTES + featureMap.values().stream().mapToInt(Feature::inlineSize).sum() + edgeSize);
         return startOffset
                 + headerSize
                 + inlineBytes // previous nodes