zclllyybb commented on code in PR #56241:
URL: https://github.com/apache/doris/pull/56241#discussion_r2378268645
##########
fe/fe-core/src/main/java/org/apache/doris/nereids/util/DateUtils.java:
##########
@@ -46,6 +47,204 @@ public class DateUtils {
"%Y-%m-%d %T", "%Y%m%d", "%Y%m");
private static final WeekFields weekFields =
WeekFields.of(DayOfWeek.SUNDAY, 7);
+ /**
+ * get the length of day of week string
+ */
+ public static int dayOfWeekLength(int dayOfWeek) {
+ if (dayOfWeek == 1 || dayOfWeek == 7 || dayOfWeek == 5) {
+ return 6;
+ } else if (dayOfWeek == 2) {
+ return 7;
+ } else if (dayOfWeek == 6 || dayOfWeek == 4) {
+ return 8;
+ } else {
+ return 10;
+ }
+ }
+
+ /**
+ * get the length of month string
+ */
+ public static int monthLength(long month) {
+ if (month == 5) {
+ return 3;
+ } else if (month == 6 || month == 7) {
+ return 4;
+ } else if (month == 4 || month == 3) {
+ return 5;
+ } else if (month == 8) {
+ return 6;
+ } else if (month == 10 || month == 1) {
+ return 7;
+ } else if (month == 2 || month == 11 || month == 12) {
+ return 8;
+ } else {
+ return 9;
+ }
+ }
+
+ /**
+ * format builder with length check.
+ * for some pattern, the length of result string is not fixed,
+ * so we need to check the length of result string to avoid parse error.
+ */
+ public static DateTimeFormatter dateTimeFormatterChecklength(String
pattern, DateTimeV2Literal date)
+ throws AnalysisException {
+ DateTimeFormatterBuilder builder = new DateTimeFormatterBuilder();
+ boolean escaped = false;
+ int length = 0;
+ for (int i = 0; i < pattern.length(); i++) {
+ char character = pattern.charAt(i);
+ if (length > 100) {
+ throw new AnalysisException("Date format string is too long");
+ }
+ if (escaped) {
+ switch (character) {
+ case 'a': // %a Abbreviated weekday name (Sun..Sat)
+ builder.appendText(ChronoField.DAY_OF_WEEK,
TextStyle.SHORT);
+ length += 3;
+ break;
+ case 'b': // %b Abbreviated month name (Jan..Dec)
+ builder.appendText(ChronoField.MONTH_OF_YEAR,
TextStyle.SHORT);
+ length += 3;
+ break;
+ case 'c': // %c Month, numeric (0..12)
+ builder.appendValue(ChronoField.MONTH_OF_YEAR);
+ if (date.getMonth() < 10) {
+ length += 1;
+ } else {
+ length += 2;
+ }
+ break;
+ case 'd': // %d Day of the month, numeric (00..31)
+ builder.appendValue(ChronoField.DAY_OF_MONTH, 2);
+ length += 2;
+ break;
+ case 'e': // %e Day of the month, numeric (0..31)
+ builder.appendValue(ChronoField.DAY_OF_MONTH);
+ if (date.getDay() < 10) {
+ length += 1;
+ } else {
+ length += 2;
+ }
+ break;
+ case 'H': // %H Hour (00..23)
+ builder.appendValue(ChronoField.HOUR_OF_DAY, 2);
+ length += 2;
+ break;
+ case 'h': // %h Hour (01..12)
+ case 'I': // %I Hour (01..12)
+ builder.appendValue(ChronoField.CLOCK_HOUR_OF_AMPM, 2);
+ length += 2;
+ break;
+ case 'i': // %i Minutes, numeric (00..59)
+ builder.appendValue(ChronoField.MINUTE_OF_HOUR, 2);
+ length += 2;
+ break;
+ case 'j': // %j Day of year (001..366)
+ builder.appendValue(ChronoField.DAY_OF_YEAR, 3);
+ length += 3;
+ break;
+ case 'k': // %k Hour (0..23)
+ builder.appendValue(ChronoField.HOUR_OF_DAY);
+ length += 2;
+ break;
+ case 'l': // %l Hour (1..12)
+ builder.appendValue(ChronoField.CLOCK_HOUR_OF_AMPM);
+ if ((date.getHour() % 24 + 11) % 12 + 1 >= 10) {
+ length += 2;
+ } else {
+ length += 1;
+ }
+ break;
+ case 'M': // %M Month name (January..December)
+ builder.appendText(ChronoField.MONTH_OF_YEAR,
TextStyle.FULL);
+ length += monthLength(date.getMonth());
+ break;
+ case 'm': // %m Month, numeric (00..12）
+ builder.appendValue(ChronoField.MONTH_OF_YEAR, 2);
+ length += 2;
+ break;
+ case 'p': // %p AM or PM
+ builder.appendText(ChronoField.AMPM_OF_DAY);
+ length += 2;
+ break;
+ case 'r': // %r Time, 12-hour (hh:mm:ss followed by AM or
PM)
+ builder.appendValue(ChronoField.CLOCK_HOUR_OF_AMPM, 2)
+ .appendLiteral(':')
+ .appendValue(ChronoField.MINUTE_OF_HOUR, 2)
+ .appendLiteral(':')
+ .appendValue(ChronoField.SECOND_OF_MINUTE, 2)
+ .appendLiteral(' ')
+ .appendText(ChronoField.AMPM_OF_DAY,
TextStyle.FULL)
+ .toFormatter();
+ length += 11;
+ break;
+ case 'S': // %S Seconds (00..59)
+ case 's': // %s Seconds (00..59)
+ builder.appendValue(ChronoField.SECOND_OF_MINUTE, 2);
+ length += 2;
+ break;
+ case 'T': // %T Time, 24-hour (HH:mm:ss)
+ builder.appendPattern("HH:mm:ss");
+ length += 8;
+ break;
+ case 'V': // %V Week (01..53), where Sunday is the first
day of the week; used with %X
+ builder.appendValue(weekFields.weekOfWeekBasedYear(),
2);
+ length += 2;
+ break;
+ case 'v': // %v Week (01..53), where Monday is the first
day of the week; used with %x
+ length += 2;
+ builder.appendValue(IsoFields.WEEK_OF_WEEK_BASED_YEAR,
2);
+ break;
+ case 'W': // %W Weekday name (Sunday..Saturday)
+ builder.appendText(ChronoField.DAY_OF_WEEK,
TextStyle.FULL);
+ length += dayOfWeekLength(date.getDayOfWeek());
+ break;
+ case 'x': // %x Year for the week where Monday is the
first day of the week,
+ builder.appendValue(IsoFields.WEEK_BASED_YEAR, 4);
+ length += 4;
+ break;
+ case 'X':
+ builder.appendValue(weekFields.weekBasedYear(), 4, 10,
SignStyle.EXCEEDS_PAD);
+ length += 4;
+ break;
+ case 'Y': // %Y Year, numeric, four digits
+ // %X Year for the week, where Sunday is the first day
of the week,
+ // numeric, four digits; used with %v
+ builder.appendValue(ChronoField.YEAR, 4);
+ length += 4;
+ break;
+ case 'y': // %y Year, numeric (two digits)
+ builder.appendValueReduced(ChronoField.YEAR, 2, 2,
1970);
+ length += 2;
+ break;
+ // TODO(Gabriel): support microseconds in date literal
+ case 'D': // %D Day of the month with English suffix (0th,
1st, 2nd, 3rd, …)
+ case 'f': // %f Microseconds (000000..999999)
+ case 'U': // %U Week (00..53), where Sunday is the first
day of the week
+ case 'u': // %u Week (00..53), where Monday is the first
day of the week
+ case 'w': // %w Day of the week (0=Sunday..6=Saturday)
+ throw new AnalysisException(String.format("%%%s not
supported in date format string",
+ character));
+ case '%': // %% A literal "%" character
+ builder.appendLiteral('%');
Review Comment:
A literal % also have length
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
