bugfix: actually use the jump

Signed-off-by: Andres Taylor <[email protected]>
vitessio · Dec 19, 2024 · 6fcd9e1 · 6fcd9e1
1 parent ec1f432
commit 6fcd9e1
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/go/vt/vtgate/evalengine/compiler_asm.go b/go/vt/vtgate/evalengine/compiler_asm.go
@@ -5153,3 +5153,9 @@ func (asm *assembler) Fn_LAST_INSERT_ID_NULL() {
 		return 1
 	}, "FN LAST_INSERT_ID NULL")
 }
+
+func (asm *assembler) addJump(end *jump) {
+	asm.emit(func(env *ExpressionEnv) int {
+		return end.offset()
+	}, "JUMP")
+}
diff --git a/go/vt/vtgate/evalengine/compiler_test.go b/go/vt/vtgate/evalengine/compiler_test.go
@@ -966,7 +966,7 @@ func runTest(t *testing.T, expr sqlparser.Expr, cfg *evalengine.Config, tc struc
 	vc := &testVcursor{env: vtenv.NewTestEnv()}
 	env := evalengine.NewExpressionEnv(context.Background(), nil, vc)
 
-	_, err = env.EvaluateAST(converted)
+	_, err = env.Evaluate(converted)
 	require.NoError(t, err)
 	if tc.missing {
 		require.Nil(t, vc.lastInsertID)

diff --git a/go/vt/vtgate/evalengine/fn_misc.go b/go/vt/vtgate/evalengine/fn_misc.go
@@ -224,6 +224,7 @@ func (call *builtinLastInsertID) compile(c *compiler) (ctype, error) {
 	c.compileToUint64(arg, 1)
 	c.asm.Fn_LAST_INSERT_ID()
 	end := c.asm.jumpFrom()
+	c.asm.addJump(end)
 
 	c.asm.jumpDestination(setZero)
 	c.asm.Fn_LAST_INSERT_ID_NULL()