expression: support push down tikv supported functions (#32348)

ref pingcap/tidb#31846

expression/expr_to_pb_test.go

@@ -1327,6 +1327,31 @@ func TestExprPushDownToTiKV(t *testing.T) {
 			retType:      types.NewFieldType(mysql.TypeString),
 			args:         []Expression{stringColumn, intColumn},
 		},
+		{
+			functionName: ast.Date,
+			retType:      types.NewFieldType(mysql.TypeDate),
+			args:         []Expression{dateColumn},
+		},
+		{
+			functionName: ast.Week,
+			retType:      types.NewFieldType(mysql.TypeDate),
+			args:         []Expression{dateColumn},
+		},
+		{
+			functionName: ast.YearWeek,
+			retType:      types.NewFieldType(mysql.TypeDate),
+			args:         []Expression{dateColumn},
+		},
+		{
+			functionName: ast.ToSeconds,
+			retType:      types.NewFieldType(mysql.TypeDate),
+			args:         []Expression{dateColumn},
+		},
+		{
+			functionName: ast.DateDiff,
+			retType:      types.NewFieldType(mysql.TypeDate),
+			args:         []Expression{dateColumn, dateColumn},
+		},
 	}
 
 	for _, tc := range testcases {

expression/expression.go

@@ -983,10 +983,13 @@ func scalarExprSupportedByTiKV(sf *ScalarFunction) bool {
 		ast.JSONUnquote,
 
 		// date functions.
-		ast.DateFormat,                                               /* Date */
-		ast.Hour, ast.Minute, ast.Second, ast.MicroSecond, ast.Month, /* ast.MonthName */
-		/* ast.DayName */ ast.DayOfMonth, ast.DayOfWeek, ast.DayOfYear, /* ast.Week */
-		ast.Weekday, ast.WeekOfYear, ast.Year, /* ast.YearWeek */
+		ast.Date, ast.Week, ast.YearWeek, ast.ToSeconds, ast.DateDiff,
+		/* ast.TimeDiff, ast.AddTime,  ast.SubTime, */
+		ast.MonthName, ast.MakeDate, ast.TimeToSec, ast.MakeTime,
+		ast.DateFormat,
+		ast.Hour, ast.Minute, ast.Second, ast.MicroSecond, ast.Month,
+		/* ast.DayName */ ast.DayOfMonth, ast.DayOfWeek, ast.DayOfYear,
+		ast.Weekday, ast.WeekOfYear, ast.Year,
 		ast.FromDays, ast.ToDays,
 		ast.PeriodAdd, ast.PeriodDiff, /*ast.TimestampDiff, ast.DateAdd, ast.FromUnixTime,*/
 		ast.LastDay,

expression/integration_serial_test.go

@@ -1912,7 +1912,7 @@ func TestTimeBuiltin(t *testing.T) {
 	_, err = tk.Exec(`update t set a = week("aa", 1)`)
 	require.True(t, terror.ErrorEqual(err, types.ErrWrongValue))
 	_, err = tk.Exec(`delete from t where a = week("aa", 1)`)
-	require.True(t, terror.ErrorEqual(err, types.ErrWrongValue))
+	require.Equal(t, types.ErrWrongValue.Code(), errors.Cause(err).(*terror.Error).Code(), "err %v", err)
 
 	// for weekofyear
 	result = tk.MustQuery(`select weekofyear("2012-12-22"), weekofyear("2008-02-20"), weekofyear("aa"), weekofyear(null), weekofyear(11), weekofyear(12.99);`)

planner/core/integration_test.go

@@ -2690,6 +2690,26 @@ func (s *testIntegrationSuite) TestScalarFunctionPushDown(c *C) {
 	tk.MustQuery("explain analyze select /*+read_from_storage(tikv[t])*/ * from t where round(b,2)").
 		CheckAt([]int{0, 3, 6}, rows)
 
+	rows[1][2] = "date(test.t.d)"
+	tk.MustQuery("explain analyze select /*+read_from_storage(tikv[t])*/ * from t where date(d)").
+		CheckAt([]int{0, 3, 6}, rows)
+
+	rows[1][2] = "week(test.t.d)"
+	tk.MustQuery("explain analyze select /*+read_from_storage(tikv[t])*/ * from t where week(d)").
+		CheckAt([]int{0, 3, 6}, rows)
+
+	rows[1][2] = "yearweek(test.t.d)"
+	tk.MustQuery("explain analyze select /*+read_from_storage(tikv[t])*/ * from t where yearweek(d)").
+		CheckAt([]int{0, 3, 6}, rows)
+
+	rows[1][2] = "to_seconds(test.t.d)"
+	tk.MustQuery("explain analyze select /*+read_from_storage(tikv[t])*/ * from t where to_seconds(d)").
+		CheckAt([]int{0, 3, 6}, rows)
+
+	rows[1][2] = "datediff(test.t.d, test.t.d)"
+	tk.MustQuery("explain analyze select /*+read_from_storage(tikv[t])*/ * from t where datediff(d,d)").
+		CheckAt([]int{0, 3, 6}, rows)
+
 }
 
 func (s *testIntegrationSuite) TestDistinctScalarFunctionPushDown(c *C) {