mirror of https://github.com/citusdata/citus.git
185 lines
6.6 KiB
Plaintext
185 lines
6.6 KiB
Plaintext
--
|
|
-- Test the CREATE statements related to columnar.
|
|
--
|
|
-- Create uncompressed table
|
|
CREATE TABLE contestant (handle TEXT, birthdate DATE, rating INT,
|
|
percentile FLOAT, country CHAR(3), achievements TEXT[])
|
|
USING columnar;
|
|
ALTER TABLE contestant SET (columnar.compression = none);
|
|
CREATE INDEX contestant_idx on contestant(handle);
|
|
-- Create zstd compressed table
|
|
CREATE TABLE contestant_compressed (handle TEXT, birthdate DATE, rating INT,
|
|
percentile FLOAT, country CHAR(3), achievements TEXT[])
|
|
USING columnar;
|
|
-- Test that querying an empty table works
|
|
ANALYZE contestant;
|
|
SELECT count(*) FROM contestant;
|
|
count
|
|
---------------------------------------------------------------------
|
|
0
|
|
(1 row)
|
|
|
|
-- Should fail: unlogged tables not supported
|
|
CREATE UNLOGGED TABLE columnar_unlogged(i int) USING columnar;
|
|
ERROR: unlogged columnar tables are not supported
|
|
CREATE TABLE columnar_table_1 (a int) USING columnar;
|
|
INSERT INTO columnar_table_1 VALUES (1);
|
|
CREATE MATERIALIZED VIEW columnar_table_1_mv USING columnar
|
|
AS SELECT * FROM columnar_table_1;
|
|
SELECT columnar.get_storage_id(oid) AS columnar_table_1_mv_storage_id
|
|
FROM pg_class WHERE relname='columnar_table_1_mv' \gset
|
|
-- test columnar_relation_set_new_filenode
|
|
REFRESH MATERIALIZED VIEW columnar_table_1_mv;
|
|
SELECT columnar_test_helpers.columnar_metadata_has_storage_id(:columnar_table_1_mv_storage_id);
|
|
columnar_metadata_has_storage_id
|
|
---------------------------------------------------------------------
|
|
f
|
|
(1 row)
|
|
|
|
SELECT columnar.get_storage_id(oid) AS columnar_table_1_storage_id
|
|
FROM pg_class WHERE relname='columnar_table_1' \gset
|
|
BEGIN;
|
|
-- test columnar_relation_nontransactional_truncate
|
|
TRUNCATE columnar_table_1;
|
|
SELECT columnar_test_helpers.columnar_metadata_has_storage_id(:columnar_table_1_storage_id);
|
|
columnar_metadata_has_storage_id
|
|
---------------------------------------------------------------------
|
|
f
|
|
(1 row)
|
|
|
|
ROLLBACK;
|
|
-- since we rollback'ed above xact, should return true
|
|
SELECT columnar_test_helpers.columnar_metadata_has_storage_id(:columnar_table_1_storage_id);
|
|
columnar_metadata_has_storage_id
|
|
---------------------------------------------------------------------
|
|
t
|
|
(1 row)
|
|
|
|
-- test dropping columnar table
|
|
DROP TABLE columnar_table_1 CASCADE;
|
|
NOTICE: drop cascades to materialized view columnar_table_1_mv
|
|
SELECT columnar_test_helpers.columnar_metadata_has_storage_id(:columnar_table_1_storage_id);
|
|
columnar_metadata_has_storage_id
|
|
---------------------------------------------------------------------
|
|
f
|
|
(1 row)
|
|
|
|
-- test temporary columnar tables
|
|
-- Should work: temporary tables are supported
|
|
CREATE TEMPORARY TABLE columnar_temp(i int) USING columnar;
|
|
-- reserve some chunks and a stripe
|
|
INSERT INTO columnar_temp SELECT i FROM generate_series(1,5) i;
|
|
SELECT columnar.get_storage_id(oid) AS columnar_temp_storage_id
|
|
FROM pg_class WHERE relname='columnar_temp' \gset
|
|
SELECT pg_backend_pid() AS val INTO old_backend_pid;
|
|
\c - - - :master_port
|
|
-- wait until old backend to expire to make sure that temp table cleanup is complete
|
|
SELECT columnar_test_helpers.pg_waitpid(val) FROM old_backend_pid;
|
|
pg_waitpid
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
DROP TABLE old_backend_pid;
|
|
-- show that temporary table itself and its metadata is removed
|
|
SELECT COUNT(*)=0 FROM pg_class WHERE relname='columnar_temp';
|
|
?column?
|
|
---------------------------------------------------------------------
|
|
t
|
|
(1 row)
|
|
|
|
SELECT columnar_test_helpers.columnar_metadata_has_storage_id(:columnar_temp_storage_id);
|
|
columnar_metadata_has_storage_id
|
|
---------------------------------------------------------------------
|
|
f
|
|
(1 row)
|
|
|
|
-- connect to another session and create a temp table with same name
|
|
CREATE TEMPORARY TABLE columnar_temp(i int) USING columnar;
|
|
-- reserve some chunks and a stripe
|
|
INSERT INTO columnar_temp SELECT i FROM generate_series(1,5) i;
|
|
-- test basic select
|
|
SELECT COUNT(*) FROM columnar_temp WHERE i < 5;
|
|
count
|
|
---------------------------------------------------------------------
|
|
4
|
|
(1 row)
|
|
|
|
SELECT columnar.get_storage_id(oid) AS columnar_temp_storage_id
|
|
FROM pg_class WHERE relname='columnar_temp' \gset
|
|
BEGIN;
|
|
DROP TABLE columnar_temp;
|
|
-- show that we drop stripes properly
|
|
SELECT columnar_test_helpers.columnar_metadata_has_storage_id(:columnar_temp_storage_id);
|
|
columnar_metadata_has_storage_id
|
|
---------------------------------------------------------------------
|
|
f
|
|
(1 row)
|
|
|
|
ROLLBACK;
|
|
-- make sure that table is not dropped yet since we rollbacked above xact
|
|
SELECT COUNT(*)=1 FROM pg_class WHERE relname='columnar_temp';
|
|
?column?
|
|
---------------------------------------------------------------------
|
|
t
|
|
(1 row)
|
|
|
|
-- show that we preserve the stripe of the temp columanar table after rollback
|
|
SELECT columnar_test_helpers.columnar_metadata_has_storage_id(:columnar_temp_storage_id);
|
|
columnar_metadata_has_storage_id
|
|
---------------------------------------------------------------------
|
|
t
|
|
(1 row)
|
|
|
|
-- drop it for next tests
|
|
DROP TABLE columnar_temp;
|
|
BEGIN;
|
|
CREATE TEMPORARY TABLE columnar_temp(i int) USING columnar ON COMMIT DROP;
|
|
-- force flushing stripe
|
|
INSERT INTO columnar_temp SELECT i FROM generate_series(1,150000) i;
|
|
SELECT columnar.get_storage_id(oid) AS columnar_temp_storage_id
|
|
FROM pg_class WHERE relname='columnar_temp' \gset
|
|
COMMIT;
|
|
-- make sure that table & its stripe is dropped after commiting above xact
|
|
SELECT COUNT(*)=0 FROM pg_class WHERE relname='columnar_temp';
|
|
?column?
|
|
---------------------------------------------------------------------
|
|
t
|
|
(1 row)
|
|
|
|
SELECT columnar_test_helpers.columnar_metadata_has_storage_id(:columnar_temp_storage_id);
|
|
columnar_metadata_has_storage_id
|
|
---------------------------------------------------------------------
|
|
f
|
|
(1 row)
|
|
|
|
BEGIN;
|
|
CREATE TEMPORARY TABLE columnar_temp(i int) USING columnar ON COMMIT DELETE ROWS;
|
|
-- force flushing stripe
|
|
INSERT INTO columnar_temp SELECT i FROM generate_series(1,150000) i;
|
|
SELECT columnar.get_storage_id(oid) AS columnar_temp_storage_id
|
|
FROM pg_class WHERE relname='columnar_temp' \gset
|
|
COMMIT;
|
|
-- make sure that table is not dropped but its rows's are deleted after commiting above xact
|
|
SELECT COUNT(*)=1 FROM pg_class WHERE relname='columnar_temp';
|
|
?column?
|
|
---------------------------------------------------------------------
|
|
t
|
|
(1 row)
|
|
|
|
SELECT COUNT(*)=0 FROM columnar_temp;
|
|
?column?
|
|
---------------------------------------------------------------------
|
|
t
|
|
(1 row)
|
|
|
|
-- since we deleted all the rows, we shouldn't have any stripes for table
|
|
SELECT columnar_test_helpers.columnar_metadata_has_storage_id(:columnar_temp_storage_id);
|
|
columnar_metadata_has_storage_id
|
|
---------------------------------------------------------------------
|
|
f
|
|
(1 row)
|
|
|
|
-- make sure citus_columnar can be loaded
|
|
LOAD 'citus_columnar';
|