thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-20 00:30:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix overflow

Browse Source
This commit is contained in:
Yarik Briukhovetskyi 2024-01-05 23:43:58 +00:00
parent 861421d27a
commit 1117284be7
1 changed files with 17 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/Functions/DateTimeTransforms.h
View File

@ -697,7 +697,12 @@ struct ToStartOfInterval<IntervalKind::Week>
if (origin == 0) if (origin == 0)
return time_zone.toStartOfWeekInterval(time_zone.toDayNum(t / scale_multiplier), weeks); return time_zone.toStartOfWeekInterval(time_zone.toDayNum(t / scale_multiplier), weeks);
else else
return ToStartOfInterval<IntervalKind::Day>::execute(t, weeks * 7, time_zone, scale_multiplier, origin); {
if (const auto weeks_to_days = weeks * 7; weeks_to_days / 7 == weeks) // Check if multiplication doesn't overflow Int64 value
return ToStartOfInterval<IntervalKind::Day>::execute(t, weeks_to_days, time_zone, scale_multiplier, origin);
else
throw Exception(ErrorCodes::VALUE_IS_OUT_OF_RANGE_OF_DATA_TYPE, "Value {} * 7 is out of bounds for type Int64", weeks);
}
} }
}; };
@ -728,11 +733,8 @@ struct ToStartOfInterval<IntervalKind::Month>
months_to_add = days < 0 ? months_to_add - 1 : months_to_add; months_to_add = days < 0 ? months_to_add - 1 : months_to_add;
months_to_add += years * 12; months_to_add += years * 12;
Int64 month_multiplier = (months_to_add / months) * months; Int64 month_multiplier = (months_to_add / months) * months;
Int64 a = 0;
a = time_zone.addMonths(time_zone.toDate(origin), month_multiplier); return time_zone.addMonths(time_zone.toDate(origin), month_multiplier) - time_zone.toDate(origin);
// a += time_zone.toTime(origin);
return a - time_zone.toDate(origin);
} }
} }
}; };
@ -757,7 +759,12 @@ struct ToStartOfInterval<IntervalKind::Quarter>
if (origin == 0) if (origin == 0)
return time_zone.toStartOfQuarterInterval(time_zone.toDayNum(t / scale_multiplier), quarters); return time_zone.toStartOfQuarterInterval(time_zone.toDayNum(t / scale_multiplier), quarters);
else else
return ToStartOfInterval<IntervalKind::Month>::execute(t, quarters * 3, time_zone, scale_multiplier, origin); {
if (const auto quarters_to_months = quarters * 3; quarters_to_months / 3 == quarters) // Check if multiplication doesn't overflow Int64 value
return ToStartOfInterval<IntervalKind::Month>::execute(t, quarters_to_months, time_zone, scale_multiplier, origin);
else
throw Exception(ErrorCodes::VALUE_IS_OUT_OF_RANGE_OF_DATA_TYPE, "Value {} * 3 is out of bounds for type Int64", quarters);
}
} }
}; };
@ -782,8 +789,10 @@ struct ToStartOfInterval<IntervalKind::Year>
return time_zone.toStartOfYearInterval(time_zone.toDayNum(t / scale_multiplier), years); return time_zone.toStartOfYearInterval(time_zone.toDayNum(t / scale_multiplier), years);
else else
{ {
auto a = ToStartOfInterval<IntervalKind::Month>::execute(t, years * 12, time_zone, scale_multiplier, origin); if (const auto years_to_months = years * 12; years_to_months / 12 == years) // Check if multiplication doesn't overflow Int64 value
return a; return ToStartOfInterval<IntervalKind::Month>::execute(t, years_to_months, time_zone, scale_multiplier, origin);
else
throw Exception(ErrorCodes::VALUE_IS_OUT_OF_RANGE_OF_DATA_TYPE, "Value {} * 12 is out of bounds for type Int64", years);
} }
} }
}; };