如何在 Postgres 中创建包含上限的新日期范围类型？

Question

Postgres 带有一个名为Range Types 的好功能，它提供了有用的范围功能（重叠、包含等）。

我希望使用daterange 类型，但是我认为该类型是通过一个尴尬的选择实现的：日期范围的上限是排除。这意味着，如果我将值定义为 2014/01/01 - 2014/01/31，则会显示为 [2014/01/01, 2014/01/31)，并且 1 月 31 日将排除范围！

我认为这是错误的默认选择。我想不出现实生活中的任何应用程序或参考资料都假定日期范围的结束日期被排除在外。至少我的经验不是。

我想为包含下限和上限的日期实现一个范围类型，但我遇到了 Postgres 文档墙：关于如何创建新的离散范围类型的参考资料很神秘，并且缺少任何示例（取自文档:Creating a canonical function is a bit tricky, since it must be defined before the range type can be declared)。

有人可以提供一些帮助吗？甚至直接实现本身；应该是 5-10 行代码，但是将这 5-10 行代码放在一起是一项严肃的研究工作。

编辑：澄清：我正在寻找有关如何创建正确类型的信息，以便插入[2014/01/01, 2014/01/31] 导致upper(daterange) = '2014/01/31'。对于现有的 daterange 类型，此值“转换”为 [2014/01/01, 2014/02/01) 并给出 upper(daterange) = '2014/02/01'

Answer 1

我设法为日期范围创建了一个自定义类型：

    CREATE or replace FUNCTION to_timestamptz(arg1 timestamptz, arg2 timestamptz) RETURNS float8 AS
    'select extract(epoch from (arg2 - arg1));' LANGUAGE sql STRICT
                                                             IMMUTABLE;
    ;
    create type tsrangetz AS RANGE
    (
      subtype = timestamptz,
      subtype_diff =
      to_timestamptz
    )
    ;
    select tsrangetz(current_date, current_date + 1)
    --["2020-10-05 00:00:00+07","2020-10-06 00:00:00+07")
    ;

Answer 2

使用 PostgresSQL 11，您可以使用 upper_inc 函数解决演示部分，例如：

select
    WHEN upper_inc(mydaterange) THEN upper(mydaterange)
    ELSE date(upper(mydaterange)- INTERVAL '1 day')
 END 

Answer 3

按照 Postgres 文档中的说明，我想出了以下代码来创建我需要的类型。但是它不起作用（请继续阅读）。

CREATE TYPE daterange_;

CREATE FUNCTION date_minus(date1 date, date2 date) RETURNS float AS $$
    SELECT cast(date1 - date2 as float);
$$ LANGUAGE sql immutable;

CREATE FUNCTION dr_canonical(dr daterange_) RETURNS daterange_ AS $$
BEGIN
    IF NOT lower_inc(dr) THEN
        dr := daterange_(lower(dr) + 1, upper(dr), '[]');
    END IF;
    IF NOT upper_inc(dr) THEN
        dr := daterange_(lower(dr), upper(dr) - 1, '[]');
    END IF;
    RETURN dr;
END;
$$ LANGUAGE plpgsql;

CREATE TYPE daterange_ AS RANGE (
    SUBTYPE = date,
    SUBTYPE_DIFF = date_minus,
    CANONICAL = dr_canonical    
);

据我所知，这个定义完全符合规范。但是，它无法用ERROR: SQL function cannot accept shell type daterange_ 声明dr_canonical 函数。

It looks like (code also) 不可能使用 C 以外的任何语言声明规范函数！因此，几乎不可能声明一个新的离散范围类型，特别是如果您使用的 Postgres 云服务无法访问运行它的机器。 Postgres 玩得很好。

Answer 4

注意第三个构造函数参数：

select daterange('2014/01/01', '2014/01/31', '[]');
        daterange        
-------------------------
 [2014-01-01,2014-02-01)

或包含上限的直接强制转换：

select '[2014/01/01, 2014/01/31]'::daterange;
        daterange        
-------------------------
 [2014-01-01,2014-02-01)

编辑

不是新类型（错误的方法恕我直言）而是正确的功能：

create function inclusive_upper_daterange(dtr daterange)
returns date as $$

    select upper(dtr) - 1;

$$ language sql immutable;

select inclusive_upper_daterange('[2014/01/01, 2014/01/31]'::daterange);
 inclusive_upper_daterange 
---------------------------
 2014-01-31