planner: set enumsetasint if original value is int in point get (#57550) (#59257)

close #56832

Author: ti-chi-bot

planner/core/point_get_plan.go

@@ -1670,7 +1670,11 @@ func buildOrderedList(ctx sessionctx.Context, plan Plan, list []*ast.Assignment,
 		if err != nil {
 			return nil, true
 		}
-		expr = expression.BuildCastFunction(ctx, expr, col.GetType())
+		castToTP := col.GetType()
+		if castToTP.GetType() == mysql.TypeEnum && assign.Expr.GetType().EvalType() == types.ETInt {
+			castToTP.AddFlag(mysql.EnumSetAsIntFlag)
+		}
+		expr = expression.BuildCastFunction(ctx, expr, castToTP)
 		if allAssignmentsAreConstant {
 			_, isConst := expr.(*expression.Constant)
 			allAssignmentsAreConstant = isConst

planner/core/point_get_plan_test.go

@@ -993,3 +993,14 @@ func TestIssue23511(t *testing.T) {
 	tk.MustQuery("select * from t2 where col1 = 0x000039;").Check(testkit.Rows("\x009"))
 	tk.MustExec("drop table t1, t2;")
 }
+
+func TestIssue56832(t *testing.T) {
+	store := testkit.CreateMockStore(t)
+	tk := testkit.NewTestKit(t, store)
+	tk.MustExec("use test")
+	tk.MustExec("drop table if exists t")
+	tk.MustExec("create table t (id int primary key, c enum('0', '1', '2'));")
+	tk.MustExec("insert into t values (0,'0'), (1,'1'), (2,'2');")
+	tk.MustExec("update t set c = 2 where id = 0;")
+	tk.MustQuery("select c from t where id = 0").Check(testkit.Rows("1"))
+}