feat: fail if last_insert_id is used in other contexts that SELECT

Signed-off-by: Andres Taylor <andres@planetscale.com>
planetscale · Dec 10, 2024 · d0764df · d0764df
1 parent ae9c5a8
commit d0764df
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/go/vt/vtgate/planbuilder/operators/query_planning.go b/go/vt/vtgate/planbuilder/operators/query_planning.go
@@ -849,7 +849,9 @@ func handleLastInsertIDColumns(ctx *plancontext.PlanningContext, output Operator
 		newExpr := sqlparser.CopyAndReplaceExpr(ae.Expr, replaceFn)
 		ae.Expr = newExpr.(sqlparser.Expr)
 	}
-
+	if offset == -1 {
+		panic(vterrors.VT12001("last_insert_id(<expr>) only supported in the select list"))
+	}
 	if topLevel {
 		return &SaveToSession{
 			unaryOperator: unaryOperator{

diff --git a/go/vt/vtgate/planbuilder/testdata/unsupported_cases.json b/go/vt/vtgate/planbuilder/testdata/unsupported_cases.json
@@ -383,5 +383,10 @@
     "comment": "SOME/ANY/ALL comparison operator not supported for unsharded queries",
     "query": "select 1 from user where foo = ALL (select 1 from user_extra where foo = 1)",
     "plan": "VT12001: unsupported: ANY/ALL/SOME comparison operator"
+  },
+  {
+    "comment": "last_insert_id(<expr>) only supported in the select list",
+    "query": "select id, foo from user where last_insert_id(bar) = 12",
+    "plan": "VT12001: unsupported: last_insert_id(<expr>) only supported in the select list"
   }
 ]