Skip to content

Commit

Permalink
Add test for init plan traces
Browse files Browse the repository at this point in the history
  • Loading branch information
@bonnefoa
bonnefoa committed Apr 16, 2024
1 parent e709908 commit b6adb3a
Show file tree
Hide file tree
Showing 5 changed files with 158 additions and 1 deletion.
2 changes: 1 addition & 1 deletion Makefile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ OBJS = \
src/pg_tracing_span.o

REGRESSCHECKS = setup utility select extended insert trigger sample \
planstate planstate_bitmap planstate_hash planstate_projectset \
planstate planstate_bitmap planstate_hash planstate_projectset planstate_subplans \
parallel subxact full_buffer \
nested wal cleanup
REGRESSCHECKS_OPTS = --no-locale --encoding=UTF8 --temp-config pg_tracing.conf
Expand Down
83 changes: 83 additions & 0 deletions expected/planstate_subplans.out
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,83 @@
/*dddbs='postgres.db',traceparent='00-00000000000000000000000000000001-0000000000000001-01'*/ WITH cte_init AS MATERIALIZED (SELECT 1 a, 'cte_init val' b)
MERGE INTO m USING (select 1 k, 'merge source InitPlan' v offset 0) o ON m.k=o.k
WHEN MATCHED THEN UPDATE SET v = (SELECT b || ' merge update' FROM cte_init WHERE a = 1 LIMIT 1)
WHEN NOT MATCHED THEN INSERT VALUES(o.k, o.v);
SELECT span_operation, deparse_info, parameters, lvl from peek_ordered_spans where trace_id='00000000000000000000000000000001' AND lvl < 6;
span_operation | deparse_info | parameters | lvl
---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------+--------------------+-----------------------------------------------------------------------------------------------------------------+-----
WITH cte_init AS MATERIALIZED (SELECT $1 a, $2 b)MERGE INTO m USING (select $3 k, $4 v offset $5) o ON m.k=o.kWHEN MATCHED THEN UPDATE SET v = (SELECT b || $6 FROM cte_init WHERE a = $7 LIMIT $8)WHEN NOT MATCHED THEN INSERT VALUES(o.k, o.v); | | $1 = 1, $2 = 'cte_init val', $3 = 1, $4 = 'merge source InitPlan', $5 = 0, $6 = ' merge update', $7 = 1, $8 = 1 | 1
Planner | | | 2
ExecutorRun | | | 2
Merge on m | | | 3
Hash Right Join | Hash Cond: (k = k) | | 4
Hash | | | 5
SeqScan on m | | | 5
InitPlan 2 (returns $1) | | | 4
Limit | | | 5
CTE cte_init | | | 4
Result | | | 5
(11 rows)

-- +----------------------------------------------------------+
-- | A: Merge on m |
-- ++---------------------------------------------------------+
-- | B: Hash Right Join |
-- +---------------------------------------------------+
-- +-------------+
-- | F: CTE init |
-- +-------------+
-- | G: Result |
-- +-------------+
-- +-------------------------------------+
-- | C: InitPlan |
-- +-----------------------------+-------+
-- | D: Limit |
-- +----+------------------------+
-- | E: CTEScan on cte_init |
-- +------------------------+
SELECT span_id AS span_a_id,
get_epoch(span_start) as span_a_start,
get_epoch(span_end) as span_a_end
from pg_tracing_peek_spans
where trace_id='00000000000000000000000000000001' AND span_operation='Merge on m' \gset
SELECT span_id AS span_b_id,
get_epoch(span_start) as span_b_start,
get_epoch(span_end) as span_b_end
from pg_tracing_peek_spans
where parent_id =:'span_a_id' and span_operation='Hash Right Join' \gset
SELECT span_id AS span_c_id,
get_epoch(span_start) as span_c_start,
get_epoch(span_end) as span_c_end
from pg_tracing_peek_spans
where parent_id =:'span_a_id' and span_operation='InitPlan 2 (returns $1)' \gset
SELECT span_id AS span_d_id,
get_epoch(span_start) as span_d_start,
get_epoch(span_end) as span_d_end
from pg_tracing_peek_spans
where parent_id =:'span_c_id' and span_operation='Limit' \gset
SELECT span_id AS span_e_id,
get_epoch(span_start) as span_e_start,
get_epoch(span_end) as span_e_end
from pg_tracing_peek_spans
where parent_id =:'span_d_id' and span_operation='CTEScan on cte_init' \gset
SELECT span_id AS span_f_id,
get_epoch(span_start) as span_f_start,
get_epoch(span_end) as span_f_end
from pg_tracing_peek_spans
where parent_id =:'span_a_id' and span_operation='CTE cte_init' \gset
SELECT span_id AS span_g_id,
get_epoch(span_start) as span_g_start,
get_epoch(span_end) as span_g_end
from pg_tracing_peek_spans
where parent_id =:'span_f_id' and span_operation='Result' \gset
SELECT :span_a_end >= :span_c_end as merge_ends_after_init_plan,
:span_c_end >= :span_d_start as bitmap_or_third_child_start_after_second,
:span_e_start <= :span_f_start as cte_scan_starts_before_cte_init,
:span_f_start >= :span_e_start as cte_init_starts_after_cte_scan;
merge_ends_after_init_plan | bitmap_or_third_child_start_after_second | cte_scan_starts_before_cte_init | cte_init_starts_after_cte_scan
----------------------------+------------------------------------------+---------------------------------+--------------------------------
t | t | t | t
(1 row)

-- Clean created spans
CALL clean_spans();
2 changes: 2 additions & 0 deletions expected/setup.out
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,3 +14,5 @@ CREATE TABLE pg_tracing_test (a int, b char(20));
CREATE INDEX pg_tracing_index ON pg_tracing_test (a);
INSERT INTO pg_tracing_test VALUES(generate_series(1, 10000), 'aaa');
ANALYZE pg_tracing_test;
CREATE TABLE m AS SELECT i AS k, (i || ' v')::text v FROM generate_series(1, 16, 3) i;
ALTER TABLE m ADD UNIQUE (k);
69 changes: 69 additions & 0 deletions sql/planstate_subplans.sql
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,69 @@
/*dddbs='postgres.db',traceparent='00-00000000000000000000000000000001-0000000000000001-01'*/ WITH cte_init AS MATERIALIZED (SELECT 1 a, 'cte_init val' b)
MERGE INTO m USING (select 1 k, 'merge source InitPlan' v offset 0) o ON m.k=o.k
WHEN MATCHED THEN UPDATE SET v = (SELECT b || ' merge update' FROM cte_init WHERE a = 1 LIMIT 1)
WHEN NOT MATCHED THEN INSERT VALUES(o.k, o.v);

SELECT span_operation, deparse_info, parameters, lvl from peek_ordered_spans where trace_id='00000000000000000000000000000001' AND lvl < 6;

-- +----------------------------------------------------------+
-- | A: Merge on m |
-- ++---------------------------------------------------------+
-- | B: Hash Right Join |
-- +---------------------------------------------------+
-- +-------------+
-- | F: CTE init |
-- +-------------+
-- | G: Result |
-- +-------------+
-- +-------------------------------------+
-- | C: InitPlan |
-- +-----------------------------+-------+
-- | D: Limit |
-- +----+------------------------+
-- | E: CTEScan on cte_init |
-- +------------------------+


SELECT span_id AS span_a_id,
get_epoch(span_start) as span_a_start,
get_epoch(span_end) as span_a_end
from pg_tracing_peek_spans
where trace_id='00000000000000000000000000000001' AND span_operation='Merge on m' \gset
SELECT span_id AS span_b_id,
get_epoch(span_start) as span_b_start,
get_epoch(span_end) as span_b_end
from pg_tracing_peek_spans
where parent_id =:'span_a_id' and span_operation='Hash Right Join' \gset
SELECT span_id AS span_c_id,
get_epoch(span_start) as span_c_start,
get_epoch(span_end) as span_c_end
from pg_tracing_peek_spans
where parent_id =:'span_a_id' and span_operation='InitPlan 2 (returns $1)' \gset
SELECT span_id AS span_d_id,
get_epoch(span_start) as span_d_start,
get_epoch(span_end) as span_d_end
from pg_tracing_peek_spans
where parent_id =:'span_c_id' and span_operation='Limit' \gset
SELECT span_id AS span_e_id,
get_epoch(span_start) as span_e_start,
get_epoch(span_end) as span_e_end
from pg_tracing_peek_spans
where parent_id =:'span_d_id' and span_operation='CTEScan on cte_init' \gset
SELECT span_id AS span_f_id,
get_epoch(span_start) as span_f_start,
get_epoch(span_end) as span_f_end
from pg_tracing_peek_spans
where parent_id =:'span_a_id' and span_operation='CTE cte_init' \gset
SELECT span_id AS span_g_id,
get_epoch(span_start) as span_g_start,
get_epoch(span_end) as span_g_end
from pg_tracing_peek_spans
where parent_id =:'span_f_id' and span_operation='Result' \gset

SELECT :span_a_end >= :span_c_end as merge_ends_after_init_plan,
:span_c_end >= :span_d_start as bitmap_or_third_child_start_after_second,
:span_e_start <= :span_f_start as cte_scan_starts_before_cte_init,
:span_f_start >= :span_e_start as cte_init_starts_after_cte_scan;

-- Clean created spans
CALL clean_spans();
3 changes: 3 additions & 0 deletions sql/setup.sql
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,3 +16,6 @@ CREATE TABLE pg_tracing_test (a int, b char(20));
CREATE INDEX pg_tracing_index ON pg_tracing_test (a);
INSERT INTO pg_tracing_test VALUES(generate_series(1, 10000), 'aaa');
ANALYZE pg_tracing_test;

CREATE TABLE m AS SELECT i AS k, (i || ' v')::text v FROM generate_series(1, 16, 3) i;
ALTER TABLE m ADD UNIQUE (k);

0 comments on commit b6adb3a

Please sign in to comment.