evalengine: jump inside instruction

Signed-off-by: Vicent Marti <[email protected]>
vitessio · Dec 20, 2024 · a18dc21 · a18dc21
1 parent 4d97507
commit a18dc21
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 27 deletions.
diff --git a/go/vt/vtgate/evalengine/compiler_asm.go b/go/vt/vtgate/evalengine/compiler_asm.go
@@ -5141,21 +5141,15 @@ func (asm *assembler) Introduce(offset int, t sqltypes.Type, col collations.Type
 
 func (asm *assembler) Fn_LAST_INSERT_ID() {
 	asm.emit(func(env *ExpressionEnv) int {
-		arg := env.vm.stack[env.vm.sp-1].(*evalUint64)
-		env.VCursor().SetLastInsertID(arg.u)
+		arg := env.vm.stack[env.vm.sp-1]
+		if arg == nil {
+			env.VCursor().SetLastInsertID(0)
+		} else {
+			iarg := evalToInt64(arg)
+			uarg := env.vm.arena.newEvalUint64(uint64(iarg.i))
+			env.vm.stack[env.vm.sp-1] = uarg
+			env.VCursor().SetLastInsertID(uarg.u)
+		}
 		return 1
 	}, "FN LAST_INSERT_ID UINT64(SP-1)")
 }
-
-func (asm *assembler) Fn_LAST_INSERT_ID_NULL() {
-	asm.emit(func(env *ExpressionEnv) int {
-		env.VCursor().SetLastInsertID(0)
-		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/fn_misc.go b/go/vt/vtgate/evalengine/fn_misc.go
@@ -219,19 +219,8 @@ func (call *builtinLastInsertID) compile(c *compiler) (ctype, error) {
 	if err != nil {
 		return ctype{}, err
 	}
-
-	setZero := c.compileNullCheck1(arg)
-	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()
-
-	c.asm.jumpDestination(end)
-
-	return ctype{Type: sqltypes.Uint64, Flag: flagNullable, Col: collationNumeric}, nil
+	return ctype{Type: sqltypes.Uint64, Flag: arg.Flag & flagNullable, Col: collationNumeric}, nil
 }
 
 func printIPv6AsIPv4(addr netip.Addr) (netip.Addr, bool) {