diff --git a/fe/fe-core/src/main/java/org/apache/doris/analysis/FunctionCallExpr.java b/fe/fe-core/src/main/java/org/apache/doris/analysis/FunctionCallExpr.java index 9cfce9e67de8ad..3978c3802bc1e4 100644 --- a/fe/fe-core/src/main/java/org/apache/doris/analysis/FunctionCallExpr.java +++ b/fe/fe-core/src/main/java/org/apache/doris/analysis/FunctionCallExpr.java @@ -1658,6 +1658,22 @@ && collectChildReturnTypes()[0].isDecimalV3()) { } fn = getBuiltinFunction(fnName.getFunction(), argTypes, Function.CompareMode.IS_NONSTRICT_SUPERTYPE_OF); + } else if (fnName.getFunction().equalsIgnoreCase("date_add") + || fnName.getFunction().equalsIgnoreCase("days_add") + || fnName.getFunction().equalsIgnoreCase("adddate") + || fnName.getFunction().equalsIgnoreCase("date_sub") + || fnName.getFunction().equalsIgnoreCase("days_sub") + || fnName.getFunction().equalsIgnoreCase("subdate")) { + Type[] childTypes = collectChildReturnTypes(); + argTypes[0] = childTypes[0]; + argTypes[1] = childTypes[1]; + if (childTypes[1] == Type.TINYINT || childTypes[1] == Type.SMALLINT) { + // be only support second param as int type + uncheckedCastChild(Type.INT, 1); + argTypes[1] = Type.INT; + } + fn = getBuiltinFunction(fnName.getFunction(), argTypes, + Function.CompareMode.IS_NONSTRICT_SUPERTYPE_OF); } else { // now first find table function in table function sets if (isTableFnCall) { diff --git a/gensrc/script/doris_builtins_functions.py b/gensrc/script/doris_builtins_functions.py index 841799f0a13b85..b7912d08904775 100644 --- a/gensrc/script/doris_builtins_functions.py +++ b/gensrc/script/doris_builtins_functions.py @@ -1080,6 +1080,8 @@ [['weeks_sub'], 'DATEV2', ['DATEV2', 'INT'], ''], [['days_add', 'date_add', 'adddate'], 'DATEV2', ['DATEV2', 'INT'], ''], [['days_sub', 'date_sub', 'subdate'], 'DATEV2', ['DATEV2', 'INT'], ''], + [['days_add', 'date_add', 'adddate'], 'DATE', ['DATE', 'INT'], ''], + [['days_sub', 'date_sub', 'subdate'], 'DATE', ['DATE', 'INT'], ''], [['hours_add'], 'DATETIMEV2', ['DATEV2', 'INT'], ''], [['hours_sub'], 'DATETIMEV2', ['DATEV2', 'INT'], ''], [['minutes_add'], 'DATETIMEV2', ['DATEV2', 'INT'], ''], diff --git a/regression-test/suites/correctness/test_date_function_const.groovy b/regression-test/suites/correctness/test_date_function_const.groovy index 85d8e5ba711370..df05738b1bf674 100644 --- a/regression-test/suites/correctness/test_date_function_const.groovy +++ b/regression-test/suites/correctness/test_date_function_const.groovy @@ -57,5 +57,8 @@ suite("test_date_function_const") { qt_select10 """ select hours_add(cast('2023-03-30 22:23:45.23452' as datetimev2(6)),8) """ - + explain { + sql("""select date_add(CURRENT_DATE(),-2);""") + notContains("00:00:00") + } } \ No newline at end of file