mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-08-22 14:13:23 +02:00
In pg_stat_statements, savepoint names now show up as constants with a parameter symbol, using as base query string the one added as a new entry to the PGSS hash table, leading to: RELEASE $1 ROLLBACK TO $1 SAVEPOINT $1 Applying constants to these query parts is a huge advantage for workloads that generate randomly savepoint points, like ORMs (Django is at the origin of this patch). The ODBC driver is a second layer that likes a lot savepoints, though it does not use a random naming pattern. A "location" field is added to TransactionStmt, now set only for savepoints. The savepoint name is ignored by the query jumbling. The location can be extended to other query patterns, if required, like 2PC commands. Some tests are added to pg_stat_statements for all the query patterns supported by the parser. ROLLBACK, ROLLBACK TO SAVEPOINT and ROLLBACK TRANSACTION TO SAVEPOINT have the same Node representation, so all these are equivalents. The same happens for RELEASE and RELEASE SAVEPOINT. Author: Greg Sabino Mullane Discussion: https://postgr.es/m/CAKAnmm+2s9PA4OaumwMJReWHk8qvJ_-g1WqxDRDAN1BSUfxyTw@mail.gmail.com
297 lines
9.5 KiB
PL/PgSQL
297 lines
9.5 KiB
PL/PgSQL
--
|
|
-- Utility commands
|
|
--
|
|
|
|
-- These tests require track_utility to be enabled.
|
|
SET pg_stat_statements.track_utility = TRUE;
|
|
SELECT pg_stat_statements_reset();
|
|
|
|
-- Tables, indexes, triggers
|
|
CREATE TEMP TABLE tab_stats (a int, b char(20));
|
|
CREATE INDEX index_stats ON tab_stats(b, (b || 'data1'), (b || 'data2')) WHERE a > 0;
|
|
ALTER TABLE tab_stats ALTER COLUMN b set default 'a';
|
|
ALTER TABLE tab_stats ALTER COLUMN b TYPE text USING 'data' || b;
|
|
ALTER TABLE tab_stats ADD CONSTRAINT a_nonzero CHECK (a <> 0);
|
|
DROP TABLE tab_stats \;
|
|
DROP TABLE IF EXISTS tab_stats \;
|
|
-- This DROP query uses two different strings, still they count as one entry.
|
|
DROP TABLE IF EXISTS tab_stats \;
|
|
Drop Table If Exists tab_stats \;
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
SELECT pg_stat_statements_reset();
|
|
|
|
-- Partitions
|
|
CREATE TABLE pt_stats (a int, b int) PARTITION BY range (a);
|
|
CREATE TABLE pt_stats1 (a int, b int);
|
|
ALTER TABLE pt_stats ATTACH PARTITION pt_stats1 FOR VALUES FROM (0) TO (100);
|
|
CREATE TABLE pt_stats2 PARTITION OF pt_stats FOR VALUES FROM (100) TO (200);
|
|
CREATE INDEX pt_stats_index ON ONLY pt_stats (a);
|
|
CREATE INDEX pt_stats2_index ON ONLY pt_stats2 (a);
|
|
ALTER INDEX pt_stats_index ATTACH PARTITION pt_stats2_index;
|
|
DROP TABLE pt_stats;
|
|
|
|
-- Views
|
|
CREATE VIEW view_stats AS SELECT 1::int AS a, 2::int AS b;
|
|
ALTER VIEW view_stats ALTER COLUMN a SET DEFAULT 2;
|
|
DROP VIEW view_stats;
|
|
|
|
-- Foreign tables
|
|
CREATE FOREIGN DATA WRAPPER wrapper_stats;
|
|
CREATE SERVER server_stats FOREIGN DATA WRAPPER wrapper_stats;
|
|
CREATE FOREIGN TABLE foreign_stats (a int) SERVER server_stats;
|
|
ALTER FOREIGN TABLE foreign_stats ADD COLUMN b integer DEFAULT 1;
|
|
ALTER FOREIGN TABLE foreign_stats ADD CONSTRAINT b_nonzero CHECK (b <> 0);
|
|
DROP FOREIGN TABLE foreign_stats;
|
|
DROP SERVER server_stats;
|
|
DROP FOREIGN DATA WRAPPER wrapper_stats;
|
|
|
|
-- Functions
|
|
CREATE FUNCTION func_stats(a text DEFAULT 'a_data', b text DEFAULT lower('b_data'))
|
|
RETURNS text AS $$ SELECT $1::text || '_' || $2::text; $$ LANGUAGE SQL;
|
|
DROP FUNCTION func_stats;
|
|
|
|
-- Rules
|
|
CREATE TABLE tab_rule_stats (a int, b int);
|
|
CREATE TABLE tab_rule_stats_2 (a int, b int, c int, d int);
|
|
CREATE RULE rules_stats AS ON INSERT TO tab_rule_stats DO INSTEAD
|
|
INSERT INTO tab_rule_stats_2 VALUES(new.*, 1, 2);
|
|
DROP RULE rules_stats ON tab_rule_stats;
|
|
DROP TABLE tab_rule_stats, tab_rule_stats_2;
|
|
|
|
-- Types
|
|
CREATE TYPE stats_type as (f1 numeric(35, 6), f2 numeric(35, 2));
|
|
DROP TYPE stats_type;
|
|
|
|
-- Triggers
|
|
CREATE TABLE trigger_tab_stats (a int, b int);
|
|
CREATE FUNCTION trigger_func_stats () RETURNS trigger LANGUAGE plpgsql
|
|
AS $$ BEGIN return OLD; end; $$;
|
|
CREATE TRIGGER trigger_tab_stats
|
|
AFTER UPDATE ON trigger_tab_stats
|
|
FOR EACH ROW WHEN (OLD.a < 0 AND OLD.b < 1 AND true)
|
|
EXECUTE FUNCTION trigger_func_stats();
|
|
DROP TABLE trigger_tab_stats;
|
|
|
|
-- Policies
|
|
CREATE TABLE tab_policy_stats (a int, b int);
|
|
CREATE POLICY policy_stats ON tab_policy_stats USING (a = 5) WITH CHECK (b < 5);
|
|
DROP TABLE tab_policy_stats;
|
|
|
|
-- Statistics
|
|
CREATE TABLE tab_expr_stats (a int, b int);
|
|
CREATE STATISTICS tab_expr_stats_1 (mcv) ON a, (2*a), (3*b) FROM tab_expr_stats;
|
|
DROP TABLE tab_expr_stats;
|
|
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
SELECT pg_stat_statements_reset();
|
|
|
|
-- Transaction statements
|
|
BEGIN;
|
|
ABORT;
|
|
BEGIN;
|
|
ROLLBACK;
|
|
-- WORK
|
|
BEGIN WORK;
|
|
COMMIT WORK;
|
|
BEGIN WORK;
|
|
ABORT WORK;
|
|
-- TRANSACTION
|
|
BEGIN TRANSACTION;
|
|
COMMIT TRANSACTION;
|
|
BEGIN TRANSACTION;
|
|
ABORT TRANSACTION;
|
|
-- More isolation levels
|
|
BEGIN TRANSACTION DEFERRABLE;
|
|
COMMIT TRANSACTION AND NO CHAIN;
|
|
BEGIN ISOLATION LEVEL SERIALIZABLE;
|
|
COMMIT;
|
|
BEGIN TRANSACTION ISOLATION LEVEL SERIALIZABLE;
|
|
COMMIT;
|
|
-- List of A_Const nodes, same lists.
|
|
BEGIN TRANSACTION READ ONLY, READ WRITE, DEFERRABLE, NOT DEFERRABLE;
|
|
COMMIT;
|
|
BEGIN TRANSACTION NOT DEFERRABLE, READ ONLY, READ WRITE, DEFERRABLE;
|
|
COMMIT;
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
SELECT pg_stat_statements_reset();
|
|
|
|
-- Savepoints
|
|
BEGIN;
|
|
SAVEPOINT sp1;
|
|
SAVEPOINT sp2;
|
|
SAVEPOINT sp3;
|
|
SAVEPOINT sp4;
|
|
ROLLBACK TO sp4;
|
|
ROLLBACK TO SAVEPOINT sp4;
|
|
ROLLBACK TRANSACTION TO SAVEPOINT sp3;
|
|
RELEASE sp3;
|
|
RELEASE SAVEPOINT sp2;
|
|
ROLLBACK TO sp1;
|
|
RELEASE SAVEPOINT sp1;
|
|
COMMIT;
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
SELECT pg_stat_statements_reset();
|
|
|
|
-- EXPLAIN statements
|
|
-- A Query is used, normalized by the query jumbling.
|
|
EXPLAIN (costs off) SELECT 1;
|
|
EXPLAIN (costs off) SELECT 2;
|
|
EXPLAIN (costs off) SELECT a FROM generate_series(1,10) AS tab(a) WHERE a = 3;
|
|
EXPLAIN (costs off) SELECT a FROM generate_series(1,10) AS tab(a) WHERE a = 7;
|
|
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
|
|
-- CALL
|
|
CREATE OR REPLACE PROCEDURE sum_one(i int) AS $$
|
|
DECLARE
|
|
r int;
|
|
BEGIN
|
|
SELECT (i + i)::int INTO r;
|
|
END; $$ LANGUAGE plpgsql;
|
|
CREATE OR REPLACE PROCEDURE sum_two(i int, j int) AS $$
|
|
DECLARE
|
|
r int;
|
|
BEGIN
|
|
SELECT (i + j)::int INTO r;
|
|
END; $$ LANGUAGE plpgsql;
|
|
SELECT pg_stat_statements_reset();
|
|
CALL sum_one(3);
|
|
CALL sum_one(199);
|
|
CALL sum_two(1,1);
|
|
CALL sum_two(1,2);
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
|
|
-- COPY
|
|
CREATE TABLE copy_stats (a int, b int);
|
|
SELECT pg_stat_statements_reset();
|
|
-- Some queries with A_Const nodes.
|
|
COPY (SELECT 1) TO STDOUT;
|
|
COPY (SELECT 2) TO STDOUT;
|
|
COPY (INSERT INTO copy_stats VALUES (1, 1) RETURNING *) TO STDOUT;
|
|
COPY (INSERT INTO copy_stats VALUES (2, 2) RETURNING *) TO STDOUT;
|
|
COPY (UPDATE copy_stats SET b = b + 1 RETURNING *) TO STDOUT;
|
|
COPY (UPDATE copy_stats SET b = b + 2 RETURNING *) TO STDOUT;
|
|
COPY (DELETE FROM copy_stats WHERE a = 1 RETURNING *) TO STDOUT;
|
|
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
DROP TABLE copy_stats;
|
|
SELECT pg_stat_statements_reset();
|
|
|
|
-- CREATE TABLE AS
|
|
-- SELECT queries are normalized, creating matching query IDs.
|
|
CREATE TABLE ctas_stats_1 AS SELECT 1 AS a;
|
|
DROP TABLE ctas_stats_1;
|
|
CREATE TABLE ctas_stats_1 AS SELECT 2 AS a;
|
|
DROP TABLE ctas_stats_1;
|
|
CREATE TABLE ctas_stats_2 AS
|
|
SELECT a AS col1, 2::int AS col2
|
|
FROM generate_series(1, 10) AS tab(a) WHERE a < 5 AND a > 2;
|
|
DROP TABLE ctas_stats_2;
|
|
CREATE TABLE ctas_stats_2 AS
|
|
SELECT a AS col1, 4::int AS col2
|
|
FROM generate_series(1, 5) AS tab(a) WHERE a < 4 AND a > 1;
|
|
DROP TABLE ctas_stats_2;
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
SELECT pg_stat_statements_reset();
|
|
|
|
-- CREATE MATERIALIZED VIEW
|
|
-- SELECT queries are normalized, creating matching query IDs.
|
|
CREATE MATERIALIZED VIEW matview_stats_1 AS
|
|
SELECT a AS col1, 2::int AS col2
|
|
FROM generate_series(1, 10) AS tab(a) WHERE a < 5 AND a > 2;
|
|
DROP MATERIALIZED VIEW matview_stats_1;
|
|
CREATE MATERIALIZED VIEW matview_stats_1 AS
|
|
SELECT a AS col1, 4::int AS col2
|
|
FROM generate_series(1, 5) AS tab(a) WHERE a < 4 AND a > 3;
|
|
DROP MATERIALIZED VIEW matview_stats_1;
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
SELECT pg_stat_statements_reset();
|
|
|
|
-- CREATE VIEW
|
|
CREATE VIEW view_stats_1 AS
|
|
SELECT a AS col1, 2::int AS col2
|
|
FROM generate_series(1, 10) AS tab(a) WHERE a < 5 AND a > 2;
|
|
DROP VIEW view_stats_1;
|
|
CREATE VIEW view_stats_1 AS
|
|
SELECT a AS col1, 4::int AS col2
|
|
FROM generate_series(1, 5) AS tab(a) WHERE a < 4 AND a > 3;
|
|
DROP VIEW view_stats_1;
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
SELECT pg_stat_statements_reset();
|
|
|
|
-- Domains
|
|
CREATE DOMAIN domain_stats AS int CHECK (VALUE > 0);
|
|
ALTER DOMAIN domain_stats SET DEFAULT '3';
|
|
ALTER DOMAIN domain_stats ADD CONSTRAINT higher_than_one CHECK (VALUE > 1);
|
|
DROP DOMAIN domain_stats;
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
SELECT pg_stat_statements_reset();
|
|
|
|
-- SET statements.
|
|
-- These use two different strings, still they count as one entry.
|
|
SET work_mem = '1MB';
|
|
Set work_mem = '1MB';
|
|
SET work_mem = '2MB';
|
|
RESET work_mem;
|
|
SET enable_seqscan = off;
|
|
SET enable_seqscan = on;
|
|
RESET enable_seqscan;
|
|
-- SET TRANSACTION ISOLATION
|
|
BEGIN;
|
|
SET TRANSACTION ISOLATION LEVEL READ COMMITTED;
|
|
SET TRANSACTION ISOLATION LEVEL REPEATABLE READ;
|
|
SET TRANSACTION ISOLATION LEVEL SERIALIZABLE;
|
|
COMMIT;
|
|
-- SET SESSION CHARACTERISTICS
|
|
SET SESSION SESSION AUTHORIZATION DEFAULT;
|
|
RESET SESSION AUTHORIZATION;
|
|
BEGIN;
|
|
SET LOCAL SESSION AUTHORIZATION DEFAULT;
|
|
RESET SESSION AUTHORIZATION;
|
|
COMMIT;
|
|
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
SELECT pg_stat_statements_reset();
|
|
|
|
--
|
|
-- Track the total number of rows retrieved or affected by the utility
|
|
-- commands of COPY, FETCH, CREATE TABLE AS, CREATE MATERIALIZED VIEW,
|
|
-- REFRESH MATERIALIZED VIEW and SELECT INTO
|
|
--
|
|
CREATE TABLE pgss_ctas AS SELECT a, 'ctas' b FROM generate_series(1, 10) a;
|
|
SELECT generate_series(1, 10) c INTO pgss_select_into;
|
|
COPY pgss_ctas (a, b) FROM STDIN;
|
|
11 copy
|
|
12 copy
|
|
13 copy
|
|
\.
|
|
CREATE MATERIALIZED VIEW pgss_matv AS SELECT * FROM pgss_ctas;
|
|
REFRESH MATERIALIZED VIEW pgss_matv;
|
|
BEGIN;
|
|
DECLARE pgss_cursor CURSOR FOR SELECT * FROM pgss_matv;
|
|
FETCH NEXT pgss_cursor;
|
|
FETCH FORWARD 5 pgss_cursor;
|
|
FETCH FORWARD ALL pgss_cursor;
|
|
COMMIT;
|
|
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
|
|
DROP MATERIALIZED VIEW pgss_matv;
|
|
DROP TABLE pgss_ctas;
|
|
DROP TABLE pgss_select_into;
|
|
|
|
SELECT pg_stat_statements_reset();
|
|
|
|
-- SET statements.
|
|
-- These use two different strings, still they count as one entry.
|
|
SET work_mem = '1MB';
|
|
Set work_mem = '1MB';
|
|
SET work_mem = '2MB';
|
|
RESET work_mem;
|
|
SET enable_seqscan = off;
|
|
SET enable_seqscan = on;
|
|
RESET enable_seqscan;
|
|
|
|
SELECT calls, rows, query FROM pg_stat_statements ORDER BY query COLLATE "C";
|
|
SELECT pg_stat_statements_reset();
|