Allow using proximityIter for non-covering indexes.

dolthub · Jan 15, 2025 · e86d7b0 · e86d7b0
1 parent 513ab1a
commit e86d7b0
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/go/libraries/doltcore/sqle/index/index_reader.go b/go/libraries/doltcore/sqle/index/index_reader.go
@@ -711,13 +711,19 @@ func (ib *nonCoveringIndexImplBuilder) NewRangeMapIter(ctx context.Context, r pr
 
 // NewPartitionRowIter implements IndexScanBuilder
 func (ib *nonCoveringIndexImplBuilder) NewPartitionRowIter(ctx *sql.Context, part sql.Partition) (sql.RowIter, error) {
-	rangeIter, err := ib.rangeIter(ctx, part)
+	var indexIter prolly.MapIter
+	var err error
+	if proximityPartition, ok := part.(vectorPartitionIter); ok {
+		indexIter, err = ib.proximityIter(ctx, proximityPartition)
+	} else {
+		indexIter, err = ib.rangeIter(ctx, part)
+	}
 	if err != nil {
 		return nil, err
 	}
 	return prollyIndexIter{
 		idx:         ib.idx,
-		indexIter:   rangeIter,
+		indexIter:   indexIter,
 		primary:     ib.pri,
 		pkBld:       ib.pkBld,
 		pkMap:       ib.pkMap,