mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-10-03 03:26:51 +02:00
dd7a8f66ed
The original implementation of TABLESAMPLE modeled the tablesample method API on index access methods, which wasn't a good choice because, without specialized DDL commands, there's no way to build an extension that can implement a TSM. (Raw inserts into system catalogs are not an acceptable thing to do, because we can't undo them during DROP EXTENSION, nor will pg_upgrade behave sanely.) Instead adopt an API more like procedural language handlers or foreign data wrappers, wherein the only SQL-level support object needed is a single handler function identified by having a special return type. This lets us get rid of the supporting catalog altogether, so that no custom DDL support is needed for the feature. Adjust the API so that it can support non-constant tablesample arguments (the original coding assumed we could evaluate the argument expressions at ExecInitSampleScan time, which is undesirable even if it weren't outright unsafe), and discourage sampling methods from looking at invisible tuples. Make sure that the BERNOULLI and SYSTEM methods are genuinely repeatable within and across queries, as required by the SQL standard, and deal more honestly with methods that can't support that requirement. Make a full code-review pass over the tablesample additions, and fix assorted bugs, omissions, infelicities, and cosmetic issues (such as failure to put the added code stanzas in a consistent ordering). Improve EXPLAIN's output of tablesample plans, too. Back-patch to 9.5 so that we don't have to support the original API in production.
103 lines
3.8 KiB
PL/PgSQL
103 lines
3.8 KiB
PL/PgSQL
CREATE TABLE test_tablesample (id int, name text) WITH (fillfactor=10);
|
|
-- use fillfactor so we don't have to load too much data to get multiple pages
|
|
|
|
INSERT INTO test_tablesample
|
|
SELECT i, repeat(i::text, 200) FROM generate_series(0, 9) s(i);
|
|
|
|
SELECT t.id FROM test_tablesample AS t TABLESAMPLE SYSTEM (50) REPEATABLE (0);
|
|
SELECT id FROM test_tablesample TABLESAMPLE SYSTEM (100.0/11) REPEATABLE (0);
|
|
SELECT id FROM test_tablesample TABLESAMPLE SYSTEM (50) REPEATABLE (0);
|
|
SELECT id FROM test_tablesample TABLESAMPLE BERNOULLI (50) REPEATABLE (0);
|
|
SELECT id FROM test_tablesample TABLESAMPLE BERNOULLI (5.5) REPEATABLE (0);
|
|
|
|
-- 100% should give repeatable count results (ie, all rows) in any case
|
|
SELECT count(*) FROM test_tablesample TABLESAMPLE SYSTEM (100);
|
|
SELECT count(*) FROM test_tablesample TABLESAMPLE SYSTEM (100) REPEATABLE (1+2);
|
|
SELECT count(*) FROM test_tablesample TABLESAMPLE SYSTEM (100) REPEATABLE (0.4);
|
|
|
|
CREATE VIEW test_tablesample_v1 AS
|
|
SELECT id FROM test_tablesample TABLESAMPLE SYSTEM (10*2) REPEATABLE (2);
|
|
CREATE VIEW test_tablesample_v2 AS
|
|
SELECT id FROM test_tablesample TABLESAMPLE SYSTEM (99);
|
|
\d+ test_tablesample_v1
|
|
\d+ test_tablesample_v2
|
|
|
|
-- check a sampled query doesn't affect cursor in progress
|
|
BEGIN;
|
|
DECLARE tablesample_cur CURSOR FOR
|
|
SELECT id FROM test_tablesample TABLESAMPLE SYSTEM (50) REPEATABLE (0);
|
|
|
|
FETCH FIRST FROM tablesample_cur;
|
|
FETCH NEXT FROM tablesample_cur;
|
|
FETCH NEXT FROM tablesample_cur;
|
|
|
|
SELECT id FROM test_tablesample TABLESAMPLE SYSTEM (50) REPEATABLE (0);
|
|
|
|
FETCH NEXT FROM tablesample_cur;
|
|
FETCH NEXT FROM tablesample_cur;
|
|
FETCH NEXT FROM tablesample_cur;
|
|
|
|
FETCH FIRST FROM tablesample_cur;
|
|
FETCH NEXT FROM tablesample_cur;
|
|
FETCH NEXT FROM tablesample_cur;
|
|
FETCH NEXT FROM tablesample_cur;
|
|
FETCH NEXT FROM tablesample_cur;
|
|
FETCH NEXT FROM tablesample_cur;
|
|
|
|
CLOSE tablesample_cur;
|
|
END;
|
|
|
|
EXPLAIN (COSTS OFF)
|
|
SELECT id FROM test_tablesample TABLESAMPLE SYSTEM (50) REPEATABLE (2);
|
|
EXPLAIN (COSTS OFF)
|
|
SELECT * FROM test_tablesample_v1;
|
|
|
|
-- check inheritance behavior
|
|
explain (costs off)
|
|
select count(*) from person tablesample bernoulli (100);
|
|
select count(*) from person tablesample bernoulli (100);
|
|
select count(*) from person;
|
|
|
|
-- check that collations get assigned within the tablesample arguments
|
|
SELECT count(*) FROM test_tablesample TABLESAMPLE bernoulli (('1'::text < '0'::text)::int);
|
|
|
|
-- check behavior during rescans, as well as correct handling of min/max pct
|
|
select * from
|
|
(values (0),(100)) v(pct),
|
|
lateral (select count(*) from tenk1 tablesample bernoulli (pct)) ss;
|
|
select * from
|
|
(values (0),(100)) v(pct),
|
|
lateral (select count(*) from tenk1 tablesample system (pct)) ss;
|
|
explain (costs off)
|
|
select pct, count(unique1) from
|
|
(values (0),(100)) v(pct),
|
|
lateral (select * from tenk1 tablesample bernoulli (pct)) ss
|
|
group by pct;
|
|
select pct, count(unique1) from
|
|
(values (0),(100)) v(pct),
|
|
lateral (select * from tenk1 tablesample bernoulli (pct)) ss
|
|
group by pct;
|
|
select pct, count(unique1) from
|
|
(values (0),(100)) v(pct),
|
|
lateral (select * from tenk1 tablesample system (pct)) ss
|
|
group by pct;
|
|
|
|
-- errors
|
|
SELECT id FROM test_tablesample TABLESAMPLE FOOBAR (1);
|
|
|
|
SELECT id FROM test_tablesample TABLESAMPLE SYSTEM (NULL);
|
|
SELECT id FROM test_tablesample TABLESAMPLE SYSTEM (50) REPEATABLE (NULL);
|
|
|
|
SELECT id FROM test_tablesample TABLESAMPLE BERNOULLI (-1);
|
|
SELECT id FROM test_tablesample TABLESAMPLE BERNOULLI (200);
|
|
SELECT id FROM test_tablesample TABLESAMPLE SYSTEM (-1);
|
|
SELECT id FROM test_tablesample TABLESAMPLE SYSTEM (200);
|
|
|
|
SELECT id FROM test_tablesample_v1 TABLESAMPLE BERNOULLI (1);
|
|
INSERT INTO test_tablesample_v1 VALUES(1);
|
|
|
|
WITH query_select AS (SELECT * FROM test_tablesample)
|
|
SELECT * FROM query_select TABLESAMPLE BERNOULLI (5.5) REPEATABLE (1);
|
|
|
|
SELECT q.* FROM (SELECT * FROM test_tablesample) as q TABLESAMPLE BERNOULLI (5);
|