SQL HowTo: ekri yon bouk pandan dirèkteman nan rechèch la, oswa "Elemantè twa etap"

Detanzantan, travay pou chèche done ki gen rapò lè l sèvi avèk yon seri kle rive. jiskaske nou jwenn kantite total dosye ki nesesè yo.

Egzanp ki pi "lavi reyèl" se montre 20 pi ansyen pwoblèm, ki nan lis sou lis anplwaye yo (pa egzanp, nan yon divizyon). Pou divès kalite "Dashboards" jesyon ak rezime kout nan zòn travay yo, yon sijè ki sanble obligatwa byen souvan.

SQL HowTo: ekri yon bouk pandan dirèkteman nan rechèch la, oswa "Elemantè twa etap"

Nan atik sa a nou pral gade nan aplikasyon an nan PostgreSQL nan yon solisyon "nayif" nan yon pwoblèm konsa, yon algorithm "pi entelijan" ak trè konplèks. "bouk" nan SQL ak yon kondisyon sòti nan done yo jwenn yo, ki ka itil tou de pou devlopman jeneral ak pou itilize nan lòt ka menm jan an.

Ann pran yon seri done tès nan atik anvan. Pou anpeche dosye ki parèt yo "sote" de tan zan tan lè valè yo klase kowenside, elaji endèks sijè a lè w ajoute yon kle prensipal. An menm tan an, sa a pral imedyatman ba li inik ak garanti nou ke lòd la klasman an pa klè:

CREATE INDEX ON task(owner_id, task_date, id);
-- а старый - удалим
DROP INDEX task_owner_id_task_date_idx;

Jan yo tande sa, se konsa yo ekri

Premyèman, se pou nou fè desen vèsyon ki pi senp nan demann lan, pase ID yo nan pèfòmè yo etalaj kòm paramèt opinyon:

SELECT
  *
FROM
  task
WHERE
  owner_id = ANY('{1,2,4,8,16,32,64,128,256,512}'::integer[])
ORDER BY
  task_date, id
LIMIT 20;

SQL HowTo: ekri yon bouk pandan dirèkteman nan rechèch la, oswa "Elemantè twa etap"
[wè nan eksplike.tensor.ru]

Yon ti kras tris - nou sèlman kòmande 20 dosye, men Index Scan te retounen li ba nou 960 liy, ki Lè sa a, tou te dwe klase... Ann eseye li mwens.

unnest + ARRAY

Premye konsiderasyon ki pral ede nou se si nou bezwen sèlman 20 klase dosye, Lè sa a, jis li pa plis pase 20 klase nan menm lòd pou chak kle. bon, endis apwopriye (owner_id, task_date, id) nou genyen.

Ann sèvi ak menm mekanis pou ekstrè ak "gaye nan kolòn" dosye entegral tab la, tankou nan dènye atik. Nou ka aplike tou plisman nan yon etalaj lè l sèvi avèk fonksyon an ARRAY():

WITH T AS (
  SELECT
    unnest(ARRAY(
      SELECT
        t
      FROM
        task t
      WHERE
        owner_id = unnest
      ORDER BY
        task_date, id
      LIMIT 20 -- ограничиваем тут...
    )) r
  FROM
    unnest('{1,2,4,8,16,32,64,128,256,512}'::integer[])
)
SELECT
  (r).*
FROM
  T
ORDER BY
  (r).task_date, (r).id
LIMIT 20; -- ... и тут - тоже

SQL HowTo: ekri yon bouk pandan dirèkteman nan rechèch la, oswa "Elemantè twa etap"
[wè nan eksplike.tensor.ru]

Oh, pi bon deja! 40% pi vit ak 4.5 fwa mwens done Mwen te oblije li li.

Materyalizasyon dosye tab atravè CTEKite m atire atansyon w sou lefèt ke nan kèk ka Yon tantativ pou travay imedyatman ak jaden yon dosye apre w fin chèche li nan yon subquery, san yo pa "vlope" li nan yon CTE, ka mennen nan "miltipliye" InitPlan pwopòsyonèl ak kantite menm jaden sa yo:

SELECT
  ((
    SELECT
      t
    FROM
      task t
    WHERE
      owner_id = 1
    ORDER BY
      task_date, id
    LIMIT 1
  ).*);

Result  (cost=4.77..4.78 rows=1 width=16) (actual time=0.063..0.063 rows=1 loops=1)
  Buffers: shared hit=16
  InitPlan 1 (returns $0)
    ->  Limit  (cost=0.42..1.19 rows=1 width=48) (actual time=0.031..0.032 rows=1 loops=1)
          Buffers: shared hit=4
          ->  Index Scan using task_owner_id_task_date_id_idx on task t  (cost=0.42..387.57 rows=500 width=48) (actual time=0.030..0.030 rows=1 loops=1)
                Index Cond: (owner_id = 1)
                Buffers: shared hit=4
  InitPlan 2 (returns $1)
    ->  Limit  (cost=0.42..1.19 rows=1 width=48) (actual time=0.008..0.009 rows=1 loops=1)
          Buffers: shared hit=4
          ->  Index Scan using task_owner_id_task_date_id_idx on task t_1  (cost=0.42..387.57 rows=500 width=48) (actual time=0.008..0.008 rows=1 loops=1)
                Index Cond: (owner_id = 1)
                Buffers: shared hit=4
  InitPlan 3 (returns $2)
    ->  Limit  (cost=0.42..1.19 rows=1 width=48) (actual time=0.008..0.008 rows=1 loops=1)
          Buffers: shared hit=4
          ->  Index Scan using task_owner_id_task_date_id_idx on task t_2  (cost=0.42..387.57 rows=500 width=48) (actual time=0.008..0.008 rows=1 loops=1)
                Index Cond: (owner_id = 1)
                Buffers: shared hit=4"
  InitPlan 4 (returns $3)
    ->  Limit  (cost=0.42..1.19 rows=1 width=48) (actual time=0.009..0.009 rows=1 loops=1)
          Buffers: shared hit=4
          ->  Index Scan using task_owner_id_task_date_id_idx on task t_3  (cost=0.42..387.57 rows=500 width=48) (actual time=0.009..0.009 rows=1 loops=1)
                Index Cond: (owner_id = 1)
                Buffers: shared hit=4

Menm dosye a te "gade" 4 fwa... Jiska PostgreSQL 11, konpòtman sa a rive regilyèman, ak solisyon an se "vlope" li nan yon CTE, ki se yon limit absoli pou optimiseur a nan vèsyon sa yo.

Akimilatè rekursif

Nan vèsyon anvan an, nan total nou li 200 liy pou dedomajman pou yo mande 20. Pa 960, men menm mwens - èske li posib?

Ann eseye sèvi ak konesans nou bezwen an total 20 dosye. Sa vle di, nou pral repete lekti done sèlman jiskaske nou rive nan kantite lajan nou bezwen an.

Etap 1: Kòmanse Lis

Li evidan, lis "sib" nou an ki gen 20 dosye yo ta dwe kòmanse ak "premye" dosye yo pou youn nan kle owner_id nou yo. Se poutèt sa, premye nou pral jwenn sa yo "trè premye" pou chak nan kle yo epi ajoute li nan lis la, klasman li nan lòd nou vle - (task_date, id).

SQL HowTo: ekri yon bouk pandan dirèkteman nan rechèch la, oswa "Elemantè twa etap"

Etap 2: Jwenn "pwochen" antre yo

Koulye a, si nou pran premye antre nan lis nou an epi kòmanse "etap" pi lwen sou endèks la prezève kle owner_id la, Lè sa a, tout dosye yo jwenn yo se egzakteman sa ki vin apre yo nan seleksyon an ki kapab lakòz. Natirèlman, sèlman jiskaske nou travèse kle a bou dezyèm antre nan lis la.

Si li sanble ke nou "travèse" dezyèm dosye a, lè sa a dènye antre li ta dwe ajoute nan lis la olye pou yo premye a (ak menm owner_id), apre sa nou re-triye lis la ankò.

SQL HowTo: ekri yon bouk pandan dirèkteman nan rechèch la, oswa "Elemantè twa etap"

Sa vle di, nou toujou jwenn ke lis la pa gen plis pase yon antre pou chak nan kle yo (si antre yo fini epi nou pa "kwaze", Lè sa a, premye antre nan lis la pral tou senpleman disparèt epi pa gen anyen yo pral ajoute. ), epi yo toujou klase nan lòd monte kle aplikasyon an (task_date, id).

SQL HowTo: ekri yon bouk pandan dirèkteman nan rechèch la, oswa "Elemantè twa etap"

Etap 3: filtre ak "agrandi" dosye yo

Nan kèk nan ranje yo nan seleksyon rekursif nou an, kèk dosye rv yo kopi - premye nou jwenn tankou "travèse fwontyè a nan 2yèm antre nan lis la", epi answit ranplase li kòm premye nan lis la. Se konsa, premye ensidan an bezwen filtre.

Rekèt final la redoutable

WITH RECURSIVE T AS (
  -- #1 : заносим в список "первые" записи по каждому из ключей набора
  WITH wrap AS ( -- "материализуем" record'ы, чтобы обращение к полям не вызывало умножения InitPlan/SubPlan
    WITH T AS (
      SELECT
        (
          SELECT
            r
          FROM
            task r
          WHERE
            owner_id = unnest
          ORDER BY
            task_date, id
          LIMIT 1
        ) r
      FROM
        unnest('{1,2,4,8,16,32,64,128,256,512}'::integer[])
    )
    SELECT
      array_agg(r ORDER BY (r).task_date, (r).id) list -- сортируем список в нужном порядке
    FROM
      T
  )
  SELECT
    list
  , list[1] rv
  , FALSE not_cross
  , 0 size
  FROM
    wrap
UNION ALL
  -- #2 : вычитываем записи 1-го по порядку ключа, пока не перешагнем через запись 2-го
  SELECT
    CASE
      -- если ничего не найдено для ключа 1-й записи
      WHEN X._r IS NOT DISTINCT FROM NULL THEN
        T.list[2:] -- убираем ее из списка
      -- если мы НЕ пересекли прикладной ключ 2-й записи
      WHEN X.not_cross THEN
        T.list -- просто протягиваем тот же список без модификаций
      -- если в списке уже нет 2-й записи
      WHEN T.list[2] IS NULL THEN
        -- просто возвращаем пустой список
        '{}'
      -- пересортировываем словарь, убирая 1-ю запись и добавляя последнюю из найденных
      ELSE (
        SELECT
          coalesce(T.list[2] || array_agg(r ORDER BY (r).task_date, (r).id), '{}')
        FROM
          unnest(T.list[3:] || X._r) r
      )
    END
  , X._r
  , X.not_cross
  , T.size + X.not_cross::integer
  FROM
    T
  , LATERAL(
      WITH wrap AS ( -- "материализуем" record
        SELECT
          CASE
            -- если все-таки "перешагнули" через 2-ю запись
            WHEN NOT T.not_cross
              -- то нужная запись - первая из спписка
              THEN T.list[1]
            ELSE ( -- если не пересекли, то ключ остался как в предыдущей записи - отталкиваемся от нее
              SELECT
                _r
              FROM
                task _r
              WHERE
                owner_id = (rv).owner_id AND
                (task_date, id) > ((rv).task_date, (rv).id)
              ORDER BY
                task_date, id
              LIMIT 1
            )
          END _r
      )
      SELECT
        _r
      , CASE
          -- если 2-й записи уже нет в списке, но мы хоть что-то нашли
          WHEN list[2] IS NULL AND _r IS DISTINCT FROM NULL THEN
            TRUE
          ELSE -- ничего не нашли или "перешагнули"
            coalesce(((_r).task_date, (_r).id) < ((list[2]).task_date, (list[2]).id), FALSE)
        END not_cross
      FROM
        wrap
    ) X
  WHERE
    T.size < 20 AND -- ограничиваем тут количество
    T.list IS DISTINCT FROM '{}' -- или пока список не кончился
)
-- #3 : "разворачиваем" записи - порядок гарантирован по построению
SELECT
  (rv).*
FROM
  T
WHERE
  not_cross; -- берем только "непересекающие" записи

SQL HowTo: ekri yon bouk pandan dirèkteman nan rechèch la, oswa "Elemantè twa etap"
[wè nan eksplike.tensor.ru]

Kidonk, nou te fè kòmès 50% nan done lekti pou 20% nan tan ekzekisyon. Sa vle di, si ou gen rezon pou kwè ke lekti ka pran yon bon bout tan (pa egzanp, done yo souvan pa nan kachèt la, epi ou dwe ale nan disk pou li), Lè sa a, nan fason sa a ou ka depann mwens sou lekti. .

Nan nenpòt ka, tan ekzekisyon an te vin pi bon pase nan premye opsyon "nayif". Men, kilès nan 3 opsyon sa yo pou itilize se ou menm.

Sous: www.habr.com

Add nouvo kòmantè