mirror of
https://github.com/postgres/postgres.git
synced 2025-06-14 18:42:34 +03:00
Support all SQL:2011 options for window frame clauses.
This patch adds the ability to use "RANGE offset PRECEDING/FOLLOWING" frame boundaries in window functions. We'd punted on that back in the original patch to add window functions, because it was not clear how to do it in a reasonably data-type-extensible fashion. That problem is resolved here by adding the ability for btree operator classes to provide an "in_range" support function that defines how to add or subtract the RANGE offset value. Factoring it this way also allows the operator class to avoid overflow problems near the ends of the datatype's range, if it wishes to expend effort on that. (In the committed patch, the integer opclasses handle that issue, but it did not seem worth the trouble to avoid overflow failures for datetime types.) The patch includes in_range support for the integer_ops opfamily (int2/int4/int8) as well as the standard datetime types. Support for other numeric types has been requested, but that seems like suitable material for a follow-on patch. In addition, the patch adds GROUPS mode which counts the offset in ORDER-BY peer groups rather than rows, and it adds the frame_exclusion options specified by SQL:2011. As far as I can see, we are now fully up to spec on window framing options. Existing behaviors remain unchanged, except that I changed the errcode for a couple of existing error reports to meet the SQL spec's expectation that negative "offset" values should be reported as SQLSTATE 22013. Internally and in relevant parts of the documentation, we now consistently use the terminology "offset PRECEDING/FOLLOWING" rather than "value PRECEDING/FOLLOWING", since the term "value" is confusingly vague. Oliver Ford, reviewed and whacked around some by me Discussion: https://postgr.es/m/CAGMVOdu9sivPAxbNN0X+q19Sfv9edEPv=HibOJhB14TJv_RCQg@mail.gmail.com
This commit is contained in:
@ -3258,6 +3258,110 @@ interval_div(PG_FUNCTION_ARGS)
|
||||
PG_RETURN_INTERVAL_P(result);
|
||||
}
|
||||
|
||||
|
||||
/*
|
||||
* in_range support functions for timestamps and intervals.
|
||||
*
|
||||
* Per SQL spec, we support these with interval as the offset type.
|
||||
* The spec's restriction that the offset not be negative is a bit hard to
|
||||
* decipher for intervals, but we choose to interpret it the same as our
|
||||
* interval comparison operators would.
|
||||
*/
|
||||
|
||||
Datum
|
||||
in_range_timestamptz_interval(PG_FUNCTION_ARGS)
|
||||
{
|
||||
TimestampTz val = PG_GETARG_TIMESTAMPTZ(0);
|
||||
TimestampTz base = PG_GETARG_TIMESTAMPTZ(1);
|
||||
Interval *offset = PG_GETARG_INTERVAL_P(2);
|
||||
bool sub = PG_GETARG_BOOL(3);
|
||||
bool less = PG_GETARG_BOOL(4);
|
||||
TimestampTz sum;
|
||||
|
||||
if (int128_compare(interval_cmp_value(offset), int64_to_int128(0)) < 0)
|
||||
ereport(ERROR,
|
||||
(errcode(ERRCODE_INVALID_PRECEDING_FOLLOWING_SIZE),
|
||||
errmsg("invalid preceding or following size in window function")));
|
||||
|
||||
/* We don't currently bother to avoid overflow hazards here */
|
||||
if (sub)
|
||||
sum = DatumGetTimestampTz(DirectFunctionCall2(timestamptz_mi_interval,
|
||||
TimestampTzGetDatum(base),
|
||||
IntervalPGetDatum(offset)));
|
||||
else
|
||||
sum = DatumGetTimestampTz(DirectFunctionCall2(timestamptz_pl_interval,
|
||||
TimestampTzGetDatum(base),
|
||||
IntervalPGetDatum(offset)));
|
||||
|
||||
if (less)
|
||||
PG_RETURN_BOOL(val <= sum);
|
||||
else
|
||||
PG_RETURN_BOOL(val >= sum);
|
||||
}
|
||||
|
||||
Datum
|
||||
in_range_timestamp_interval(PG_FUNCTION_ARGS)
|
||||
{
|
||||
Timestamp val = PG_GETARG_TIMESTAMP(0);
|
||||
Timestamp base = PG_GETARG_TIMESTAMP(1);
|
||||
Interval *offset = PG_GETARG_INTERVAL_P(2);
|
||||
bool sub = PG_GETARG_BOOL(3);
|
||||
bool less = PG_GETARG_BOOL(4);
|
||||
Timestamp sum;
|
||||
|
||||
if (int128_compare(interval_cmp_value(offset), int64_to_int128(0)) < 0)
|
||||
ereport(ERROR,
|
||||
(errcode(ERRCODE_INVALID_PRECEDING_FOLLOWING_SIZE),
|
||||
errmsg("invalid preceding or following size in window function")));
|
||||
|
||||
/* We don't currently bother to avoid overflow hazards here */
|
||||
if (sub)
|
||||
sum = DatumGetTimestamp(DirectFunctionCall2(timestamp_mi_interval,
|
||||
TimestampGetDatum(base),
|
||||
IntervalPGetDatum(offset)));
|
||||
else
|
||||
sum = DatumGetTimestamp(DirectFunctionCall2(timestamp_pl_interval,
|
||||
TimestampGetDatum(base),
|
||||
IntervalPGetDatum(offset)));
|
||||
|
||||
if (less)
|
||||
PG_RETURN_BOOL(val <= sum);
|
||||
else
|
||||
PG_RETURN_BOOL(val >= sum);
|
||||
}
|
||||
|
||||
Datum
|
||||
in_range_interval_interval(PG_FUNCTION_ARGS)
|
||||
{
|
||||
Interval *val = PG_GETARG_INTERVAL_P(0);
|
||||
Interval *base = PG_GETARG_INTERVAL_P(1);
|
||||
Interval *offset = PG_GETARG_INTERVAL_P(2);
|
||||
bool sub = PG_GETARG_BOOL(3);
|
||||
bool less = PG_GETARG_BOOL(4);
|
||||
Interval *sum;
|
||||
|
||||
if (int128_compare(interval_cmp_value(offset), int64_to_int128(0)) < 0)
|
||||
ereport(ERROR,
|
||||
(errcode(ERRCODE_INVALID_PRECEDING_FOLLOWING_SIZE),
|
||||
errmsg("invalid preceding or following size in window function")));
|
||||
|
||||
/* We don't currently bother to avoid overflow hazards here */
|
||||
if (sub)
|
||||
sum = DatumGetIntervalP(DirectFunctionCall2(interval_mi,
|
||||
IntervalPGetDatum(base),
|
||||
IntervalPGetDatum(offset)));
|
||||
else
|
||||
sum = DatumGetIntervalP(DirectFunctionCall2(interval_pl,
|
||||
IntervalPGetDatum(base),
|
||||
IntervalPGetDatum(offset)));
|
||||
|
||||
if (less)
|
||||
PG_RETURN_BOOL(interval_cmp_internal(val, sum) <= 0);
|
||||
else
|
||||
PG_RETURN_BOOL(interval_cmp_internal(val, sum) >= 0);
|
||||
}
|
||||
|
||||
|
||||
/*
|
||||
* interval_accum, interval_accum_inv, and interval_avg implement the
|
||||
* AVG(interval) aggregate.
|
||||
|
Reference in New Issue
Block a user