加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 百科 > 正文

使用时间戳OVERLAPS和“PARTITION BY”加速PostgreSQL查询

发布时间:2020-12-13 15:55:41 所属栏目:百科 来源:网络整理
导读:我在PostgreSQL 9.0中有一个相当大的表(500K – 1M行),它包含通用的“时间片”信息,也就是说,它确定另一个表中的行(“特征”)何时有效.定义看起来像这样(略微简化): CREATE TABLE feature_timeslice( timeslice_id int NOT NULL,feature_id int NOT NULL,va
我在PostgreSQL 9.0中有一个相当大的表(500K – 1M行),它包含通用的“时间片”信息,也就是说,它确定另一个表中的行(“特征”)何时有效.定义看起来像这样(略微简化):

CREATE TABLE feature_timeslice
(
  timeslice_id int NOT NULL,feature_id int NOT NULL,valid_time_begin timestamp NOT NULL,valid_time_end timestamp,sequence_number smallint,-- Some other columns
  CONSTRAINT pk_feature_timeslice PRIMARY KEY (timeslice_id)
  -- Some other constraints
)

CREATE INDEX ix_feature_timeslice_feature_id
ON feature_timeslice USING btree (feature_id);

然后,在timeslice_id上??将许多其他特定功能表连接到它:

CREATE TABLE specific_feature_timeslice
(
  timeslice_id int NOT NULL,-- Other columns
  CONSTRAINT pk_specific_feature_timeslice PRIMARY KEY (timeslice_id),CONSTRAINT fk_specific_feature_timeslice_feature_timeslice FOREIGN KEY (timeslice_id) REFERENCES feature_timeslice (timeslice_id)
)

可能存在具有重叠有效时段(开始/结束时间)的多个时间片,但具有最高sequence_number的时间片优先(再次,略微简化,但足够接近).我想有效地找到每个feature_id的当前有效行,所以我定义了一个视图,如下所示:

CREATE VIEW feature_timeslice_id_now
AS
    SELECT timeslice_id
    FROM
    (
        SELECT timeslice_id,rank() OVER
        (
            PARTITION BY feature_id
            ORDER BY sequence_number DESC,timeslice_id DESC
        )
        FROM feature_timeslice
        WHERE (current_timestamp AT TIME ZONE 'UTC','0'::interval) OVERLAPS (valid_time_begin,COALESCE(valid_time_end,'infinity'::timestamp))
    ) subq 
    WHERE subq.rank = 1

它通常像这样查询:

SELECT *
FROM specific_feature_timeslice sf
JOIN feature_timeslice_id_now n USING (timeslice_id)
WHERE sf.name = 'SOMETHING'

这可行,但它仍然有点太慢 – 需要1-2秒,即使可能只返回1-5行,因为specific_feature_timeslice标准通常会将其缩小很多. (加入多个功能视图的更复杂的查询会非常快速地变慢.)我无法弄清楚如何让PostgreSQL更有效地执行此操作.查询计划如下所示:

Join Filter: ((r.timeslice_id)::integer = (subq.timeslice_id)::integer)
  ->  Subquery Scan on subq  (cost=32034.36..37876.98 rows=835 width=4) (actual time=2086.125..5243.467 rows=250918 loops=1)
        Filter: (subq.rank = 1)
        ->  WindowAgg  (cost=32034.36..35790.33 rows=166932 width=10) (actual time=2086.110..4066.351 rows=250918 loops=1)
              ->  Sort  (cost=32034.36..32451.69 rows=166932 width=10) (actual time=2086.065..2654.971 rows=250918 loops=1)
                    Sort Key: feature_timeslice.feature_id,feature_timeslice.sequence_number,feature_timeslice.timeslice_id
                    Sort Method:  quicksort  Memory: 13898kB
                    ->  Seq Scan on feature_timeslice  (cost=0.00..17553.93 rows=166932 width=10) (actual time=287.270..1225.595 rows=250918 loops=1)
                          Filter: overlaps(timezone('UTC'::text,now()),(timezone('UTC'::text,now()) + '00:00:00'::interval),(valid_time_begin)::timestamp without time zone,COALESCE((valid_time_end)::timestamp without time zone,'infinity'::timestamp without time zone))
  ->  Materialize  (cost=0.00..1093.85 rows=2 width=139) (actual time=0.002..0.007 rows=2 loops=250918)
        ->  Seq Scan on specific_feature_timeslice sf  (cost=0.00..1093.84 rows=2 width=139) (actual time=1.958..7.674 rows=2 loops=1)
              Filter: ((name)::text = 'SOMETHING'::text)
Total runtime: 10319.875 ms

实际上,我想在任何给定时间内进行此查询,而不仅仅是当前时间.我有一个为此定义的函数,它将时间作为参数,但查询“现在”是最常见的情况,所以即使我只能加快速度,这将是一个很大的改进.

==编辑==

好的,我已经尝试按照两个答案的建议对表进行规范化 – 也就是说,我将valid_time_begin和valid_time_end移动到一个单独的表time_period中.我还用WHERE NOT EXISTS([更好的候选时间片])替换了窗口函数.在这个过程中,我也升级到了PostgreSQL 9.1.所有这些查询现在都快了两倍.查询计划与wildplasser的答案看起来相同.这很好,但不如我希望的那么好 – 从一个功能表中选择仍需要一秒钟.

理想情况下,我想利用功能WHERE条件的选择性,正如Erwin Brandstetter所说.如果我手工制作一个查询来做到这一点,我得到的时间是15-30毫秒.现在,它更像是它!手工制作的查询看起来像这样:

WITH filtered_feature AS
(
    SELECT *
    FROM specific_feature_timeslice sf
    JOIN feature_timeslice ft USING (timeslice_id)
    WHERE sf.name = 'SOMETHING'
)
SELECT *
FROM filtered_feature ff
JOIN
(
    SELECT timeslice_id
    FROM filtered_feature candidate
    JOIN time_period candidate_time ON candidate.valid_time_period_id = candidate_time.id
    WHERE ('2011-09-26','0'::interval) OVERLAPS (candidate_time.valid_time_begin,COALESCE(candidate_time.valid_time_end,'infinity'::timestamp))
        AND NOT EXISTS
        (
            SELECT *
            FROM filtered_feature better
            JOIN time_period better_time ON better.valid_time_period_id = better_time.id
            WHERE ('2011-09-26','0'::interval) OVERLAPS (better_time.valid_time_begin,COALESCE(better_time.valid_time_end,'infinity'::timestamp))
                AND better.feature_id = candidate.feature_id AND better.timeslice_id != candidate.timeslice_id
                AND better.sequence_number > candidate.sequence_number
        )
) AS ft ON ff.timeslice_id = ft.timeslice_id

不幸的是,这在普通查询中使用太大而且太复杂,它们可能会连接许多其他表.我需要一些方法将这个逻辑封装在一个函数(任意时间)或至少一个视图(当前时间),但我仍然无法弄清楚如何执行此操作,同时仍然让查询规划器首先过滤特定的功能.如果我只能将行集传递给函数 – 但据我所知,PostgreSQL不允许这样做.有任何想法吗?

==结论==

我最终使用PostgreSQL继承来解决这个问题(请参阅我的回答),但如果不是因为Erwin Brandstetter的答案,我就不会想出这个想法,所以赏金归于他. wildplasser的答案也非常有用,因为它允许我消除不必要的窗口功能,从而进一步加快它的速度.非常感谢你们俩!

解决方法

我最终使用PostgreSQL继承来解决这个问题,因此每个specific_feature_timeslice表都继承自feature_timeslice(而不是像之前那样引用它).这允许“特征的选择性可以首先生效” – 查询计划首先将其缩小到我想要的几行.因此,架构现在看起来像这样:

CREATE TABLE feature_timeslice
(
  timeslice_id int NOT NULL,-- Some other columns
  CONSTRAINT pk_feature_timeslice PRIMARY KEY (timeslice_id)
  -- Some other constraints
)

CREATE TABLE specific_feature_timeslice
(
  -- Feature-specific columns only,eg.
  name character varying(100),CONSTRAINT pk_specific_feature_timeslice PRIMARY KEY (timeslice_id)
)
INHERITS (feature_timeslice);

CREATE INDEX ix_specific_feature_timeslice_feature_id
ON specific_feature_timeslice (feature_id);

每个这样的派生表都有自己的函数来选择指定时间的行电流:

CREATE FUNCTION specific_feature_asof(effective_time timestamp)
RETURNS SETOF specific_feature_timeslice
AS $BODY$
    SELECT candidate.*
    FROM specific_feature_timeslice candidate
    WHERE ($1,'0'::interval) OVERLAPS (candidate.valid_time_begin,COALESCE(candidate.valid_time_end,'infinity'::timestamp))
        AND NOT EXISTS
        (
            SELECT *
            FROM specific_feature_timeslice better
            WHERE ($1,'0'::interval) OVERLAPS (better.valid_time_begin,COALESCE(better.valid_time_end,'infinity'::timestamp))
                AND better.feature_id = candidate.feature_id AND better.timeslice_id != candidate.timeslice_id AND better.sequence_number > candidate.sequence_number
        )
$BODY$LANGUAGE SQL STABLE;

我自动生成这些函数 – 当然 – 除了表名之外它们是相同的.然后典型的查询变为:

SELECT *
FROM specific_feature_asof('2011-09-30')
WHERE name = 'SOMETHING'

并且查询计划如下所示:

Nested Loop Anti Join  (cost=0.00..412.84 rows=3 width=177) (actual time=0.044..7.038 rows=10 loops=1)
  Join Filter: (((better.timeslice_id)::integer <> (candidate.timeslice_id)::integer) AND ((better.sequence_number)::smallint > (candidate.sequence_number)::smallint))
  ->  Seq Scan on specific_feature_timeslice candidate  (cost=0.00..379.66 rows=3 width=177) (actual time=0.018..6.688 rows=10 loops=1)
        Filter: (((name)::text = 'SOMETHING'::text) AND overlaps(('2011-09-30 00:00:00'::timestamp without time zone)::timestamp without time zone,(('2011-09-30 00:00:00'::timestamp without time zone)::timestamp without time zone + '00:00:00'::interval),'infinity'::timestamp without time zone)))
  ->  Index Scan using ix_specific_feature_timeslice_feature_id on specific_feature_timeslice better  (cost=0.00..8.28 rows=1 width=14) (actual time=0.008..0.011 rows=1 loops=10)
        Index Cond: ((feature_id)::integer = (candidate.feature_id)::integer)
        Filter: overlaps(('2011-09-30 00:00:00'::timestamp without time zone)::timestamp without time zone,'infinity'::timestamp without time zone))
Total runtime: 7.150 ms

性能差异非常显着:像上面的查询一样简单的选择需要30-60毫秒.加入两个这样的函数需要300-400毫秒,这比我预期的要多一点,但仍然可以接受.

通过这些更改,我认为不再需要规范化feature_timeslice,即.将有效的开始/结束时间提取到一个单独的表中,所以我没有这样做.

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读