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

close #56832

Author: ti-chi-bot

pkg/planner/core/point_get_plan.go

@@ -1730,7 +1730,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

pkg/planner/core/point_get_plan_test.go

@@ -272,3 +272,14 @@ func TestIssue18042(t *testing.T) {
 	require.Equal(t, uint64(100), tk.Session().GetSessionVars().StmtCtx.MaxExecutionTime)
 	tk.MustExec("drop table t")
 }
+
+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"))
+}