database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[fix](planner) inplement constant folding for function to_monday() (#20708)

Browse Source
This commit is contained in:
mch_ucchi
2023-06-13 11:40:44 +08:00
committed by GitHub
parent e010fa8d4f
commit eaa13e66f9
1 changed files with 35 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
fe/fe-core/src/main/java/org/apache/doris/rewrite/FEFunctions.java
View File

@ -324,6 +324,41 @@ public class FEFunctions {
return null;
}
@FEFunction(name = "to_monday", argTypes = {"DATETIME"}, returnType = "DATE")
public static DateLiteral toMonday(LiteralExpr arg) {
if (arg instanceof DateLiteral && (arg.getType().isDate() || arg.getType().isDatetime())) {
DateLiteral dateLiteral = ((DateLiteral) arg);
LocalDateTime dateTime = LocalDateTime.of(
((int) dateLiteral.getYear()), ((int) dateLiteral.getMonth()), ((int) dateLiteral.getDay()),
0, 0, 0);
dateTime = toMonday(dateTime);
return new DateLiteral(dateTime.getYear(), dateTime.getMonthValue(), dateTime.getDayOfMonth(), Type.DATE);
}
return null;
}
@FEFunction(name = "to_monday", argTypes = {"DATETIMEV2"}, returnType = "DATEV2")
public static DateLiteral toMondayV2(LiteralExpr arg) {
if (arg instanceof DateLiteral && (arg.getType().isDateV2() || arg.getType().isDatetimeV2())) {
DateLiteral dateLiteral = ((DateLiteral) arg);
LocalDateTime dateTime = LocalDateTime.of(
((int) dateLiteral.getYear()), ((int) dateLiteral.getMonth()), ((int) dateLiteral.getDay()),
0, 0, 0);
dateTime = toMonday(dateTime);
return new DateLiteral(dateTime.getYear(), dateTime.getMonthValue(), dateTime.getDayOfMonth(), Type.DATEV2);
}
return null;
}
private static LocalDateTime toMonday(LocalDateTime dateTime) {
LocalDateTime specialUpperBound = LocalDateTime.of(1970, 1, 4, 0, 0, 0);
LocalDateTime specialLowerBound = LocalDateTime.of(1970, 1, 1, 0, 0, 0);
if (dateTime.isAfter(specialUpperBound) || dateTime.isBefore(specialLowerBound)) {
return dateTime.plusDays(-dateTime.getDayOfWeek().getValue() + 1);
}
return specialLowerBound;
}
/**
------------------------------------------------------------------------------
*/