Add Literal field to the OrderAndLimit struct used for tracking expressions that use vector indexes.

Author: nicktobey

sql/analyzer/replace_order_by_distance.go

@@ -60,13 +60,16 @@ func replaceIdxOrderByDistanceHelper(ctx *sql.Context, scope *plan.Scope, node s
 			return n, transform.SameTree, nil
 		}
 		var column sql.Expression
+		var literal sql.Expression
 		_, leftIsLiteral := distance.LeftChild.(*expression.Literal)
 		if leftIsLiteral {
 			column = distance.RightChild
+			literal = distance.LeftChild
 		} else {
 			_, rightIsLiteral := distance.RightChild.(*expression.Literal)
 			if rightIsLiteral {
 				column = distance.LeftChild
+				literal = distance.RightChild
 			} else {
 				return n, transform.SameTree, nil
 			}
@@ -99,6 +102,7 @@ func replaceIdxOrderByDistanceHelper(ctx *sql.Context, scope *plan.Scope, node s
 			VectorOrderAndLimit: sql.OrderAndLimit{
 				OrderBy: distance,
 				Limit:   limit,
+				Literal: literal,
 			},
 		}
 		nn, err := plan.NewStaticIndexedAccessForTableNode(n, lookup)

sql/core.go

@@ -894,6 +894,7 @@ func IncrementStatusVariable(ctx *Context, name string, val int) {
 type OrderAndLimit struct {
 	OrderBy       Expression
 	Limit         Expression
+	Literal       Expression
 	CalcFoundRows bool
 }