mirror of https://github.com/citusdata/citus.git
377 lines
8.7 KiB
PL/PgSQL
377 lines
8.7 KiB
PL/PgSQL
--
|
|
-- MULTI_PREPARE_PLSQL
|
|
--
|
|
|
|
-- Many of the queries are taken from other regression test files
|
|
-- and converted into both plain SQL and PL/pgsql functions, which
|
|
-- use prepared statements internally.
|
|
|
|
|
|
ALTER SEQUENCE pg_catalog.pg_dist_shardid_seq RESTART 780000;
|
|
ALTER SEQUENCE pg_catalog.pg_dist_jobid_seq RESTART 780000;
|
|
|
|
|
|
CREATE FUNCTION sql_test_no_1() RETURNS bigint AS '
|
|
SELECT
|
|
count(*)
|
|
FROM
|
|
orders;
|
|
' LANGUAGE SQL;
|
|
|
|
CREATE FUNCTION sql_test_no_2() RETURNS bigint AS '
|
|
SELECT
|
|
count(*)
|
|
FROM
|
|
orders, lineitem
|
|
WHERE
|
|
o_orderkey = l_orderkey;
|
|
' LANGUAGE SQL;
|
|
|
|
CREATE FUNCTION sql_test_no_3() RETURNS bigint AS '
|
|
SELECT
|
|
count(*)
|
|
FROM
|
|
orders, customer
|
|
WHERE
|
|
o_custkey = c_custkey;
|
|
' LANGUAGE SQL;
|
|
|
|
CREATE FUNCTION sql_test_no_4() RETURNS bigint AS '
|
|
SELECT
|
|
count(*)
|
|
FROM
|
|
orders, customer, lineitem
|
|
WHERE
|
|
o_custkey = c_custkey AND
|
|
o_orderkey = l_orderkey;
|
|
' LANGUAGE SQL;
|
|
|
|
CREATE OR REPLACE FUNCTION sql_test_no_6(integer) RETURNS bigint AS $$
|
|
SELECT
|
|
count(*)
|
|
FROM
|
|
orders, lineitem
|
|
WHERE
|
|
o_orderkey = l_orderkey AND
|
|
l_suppkey > $1;
|
|
$$ LANGUAGE SQL RETURNS NULL ON NULL INPUT;
|
|
|
|
CREATE OR REPLACE FUNCTION plpgsql_test_1() RETURNS TABLE(count bigint) AS $$
|
|
DECLARE
|
|
BEGIN
|
|
RETURN QUERY
|
|
SELECT
|
|
count(*)
|
|
FROM
|
|
orders;
|
|
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
CREATE OR REPLACE FUNCTION plpgsql_test_2() RETURNS TABLE(count bigint) AS $$
|
|
DECLARE
|
|
BEGIN
|
|
RETURN QUERY
|
|
SELECT
|
|
count(*)
|
|
FROM
|
|
orders, lineitem
|
|
WHERE
|
|
o_orderkey = l_orderkey;
|
|
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
CREATE OR REPLACE FUNCTION plpgsql_test_3() RETURNS TABLE(count bigint) AS $$
|
|
DECLARE
|
|
BEGIN
|
|
RETURN QUERY
|
|
SELECT
|
|
count(*)
|
|
FROM
|
|
orders, customer
|
|
WHERE
|
|
o_custkey = c_custkey;
|
|
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
CREATE OR REPLACE FUNCTION plpgsql_test_4() RETURNS TABLE(count bigint) AS $$
|
|
DECLARE
|
|
BEGIN
|
|
RETURN QUERY
|
|
SELECT
|
|
count(*)
|
|
FROM
|
|
orders, customer, lineitem
|
|
WHERE
|
|
o_custkey = c_custkey AND
|
|
o_orderkey = l_orderkey;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
CREATE OR REPLACE FUNCTION plpgsql_test_5() RETURNS TABLE(count bigint) AS $$
|
|
DECLARE
|
|
BEGIN
|
|
RETURN QUERY
|
|
SELECT
|
|
count(*)
|
|
FROM
|
|
lineitem, customer
|
|
WHERE
|
|
l_partkey = c_nationkey;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
CREATE OR REPLACE FUNCTION plpgsql_test_6(int) RETURNS TABLE(count bigint) AS $$
|
|
DECLARE
|
|
BEGIN
|
|
RETURN QUERY
|
|
SELECT
|
|
count(*)
|
|
FROM
|
|
orders, lineitem
|
|
WHERE
|
|
o_orderkey = l_orderkey AND
|
|
l_suppkey > $1;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
CREATE OR REPLACE FUNCTION plpgsql_test_7(text, text) RETURNS TABLE(supp_natadsion text, cusasdt_nation text, l_yeasdar int, sasdaum double precision) AS $$
|
|
DECLARE
|
|
BEGIN
|
|
RETURN QUERY
|
|
SELECT
|
|
supp_nation::text,
|
|
cust_nation::text,
|
|
l_year::int,
|
|
sum(volume)::double precision AS revenue
|
|
FROM
|
|
(
|
|
SELECT
|
|
supp_nation,
|
|
cust_nation,
|
|
extract(year FROM l_shipdate) AS l_year,
|
|
l_extendedprice * (1 - l_discount) AS volume
|
|
FROM
|
|
supplier,
|
|
lineitem,
|
|
orders,
|
|
customer,
|
|
(
|
|
SELECT
|
|
n1.n_nationkey AS supp_nation_key,
|
|
n2.n_nationkey AS cust_nation_key,
|
|
n1.n_name AS supp_nation,
|
|
n2.n_name AS cust_nation
|
|
FROM
|
|
nation n1,
|
|
nation n2
|
|
WHERE
|
|
(
|
|
(n1.n_name = $1 AND n2.n_name = $2)
|
|
OR (n1.n_name = $2 AND n2.n_name = $1)
|
|
)
|
|
) AS temp
|
|
WHERE
|
|
s_suppkey = l_suppkey
|
|
AND o_orderkey = l_orderkey
|
|
AND c_custkey = o_custkey
|
|
AND s_nationkey = supp_nation_key
|
|
AND c_nationkey = cust_nation_key
|
|
AND l_shipdate between date '1995-01-01' AND date '1996-12-31'
|
|
) AS shipping
|
|
GROUP BY
|
|
supp_nation,
|
|
cust_nation,
|
|
l_year
|
|
ORDER BY
|
|
supp_nation,
|
|
cust_nation,
|
|
l_year;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
SET citus.task_executor_type TO 'task-tracker';
|
|
SET client_min_messages TO INFO;
|
|
|
|
-- now, run plain SQL functions
|
|
SELECT sql_test_no_1();
|
|
SELECT sql_test_no_2();
|
|
SELECT sql_test_no_3();
|
|
SELECT sql_test_no_4();
|
|
|
|
-- now, run PL/pgsql functions
|
|
SELECT plpgsql_test_1();
|
|
SELECT plpgsql_test_2();
|
|
SELECT plpgsql_test_3();
|
|
SELECT plpgsql_test_4();
|
|
SELECT plpgsql_test_5();
|
|
|
|
-- run PL/pgsql functions with different parameters
|
|
SELECT plpgsql_test_6(155);
|
|
SELECT plpgsql_test_6(1555);
|
|
SELECT plpgsql_test_7('UNITED KINGDOM', 'CHINA');
|
|
SELECT plpgsql_test_7('FRANCE', 'GERMANY');
|
|
|
|
-- now, PL/pgsql functions with random order
|
|
SELECT plpgsql_test_6(155);
|
|
SELECT plpgsql_test_3();
|
|
SELECT plpgsql_test_7('FRANCE', 'GERMANY');
|
|
SELECT plpgsql_test_5();
|
|
SELECT plpgsql_test_1();
|
|
SELECT plpgsql_test_6(1555);
|
|
SELECT plpgsql_test_4();
|
|
SELECT plpgsql_test_7('UNITED KINGDOM', 'CHINA');
|
|
SELECT plpgsql_test_2();
|
|
|
|
-- run the tests which do not require re-partition
|
|
-- with real-time executor
|
|
SET citus.task_executor_type TO 'real-time';
|
|
|
|
-- now, run plain SQL functions
|
|
SELECT sql_test_no_1();
|
|
SELECT sql_test_no_2();
|
|
|
|
-- plain SQL functions with parameters cannot be executed
|
|
-- FIXME: temporarily disabled, bad error message - waiting for proper parametrized query
|
|
-- FIXME: support
|
|
-- SELECT sql_test_no_6(155);
|
|
|
|
-- now, run PL/pgsql functions
|
|
SELECT plpgsql_test_1();
|
|
SELECT plpgsql_test_2();
|
|
|
|
-- run PL/pgsql functions with different parameters
|
|
-- FIXME: temporarily disabled, waiting for proper parametrized query support
|
|
-- SELECT plpgsql_test_6(155);
|
|
-- SELECT plpgsql_test_6(1555);
|
|
|
|
-- test router executor parameterized PL/pgsql functions
|
|
CREATE TABLE temp_table (
|
|
key int,
|
|
value int
|
|
);
|
|
SELECT master_create_distributed_table('temp_table','key','hash');
|
|
SELECT master_create_worker_shards('temp_table',4,1);
|
|
|
|
CREATE OR REPLACE FUNCTION no_parameter_insert() RETURNS void as $$
|
|
BEGIN
|
|
INSERT INTO temp_table (key) VALUES (0);
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
-- execute 6 times to trigger prepared statement usage
|
|
SELECT no_parameter_insert();
|
|
SELECT no_parameter_insert();
|
|
SELECT no_parameter_insert();
|
|
SELECT no_parameter_insert();
|
|
SELECT no_parameter_insert();
|
|
SELECT no_parameter_insert();
|
|
|
|
CREATE OR REPLACE FUNCTION single_parameter_insert(key_arg int) RETURNS void as $$
|
|
BEGIN
|
|
INSERT INTO temp_table (key) VALUES (key_arg);
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
-- execute 6 times to trigger prepared statement usage
|
|
SELECT single_parameter_insert(1);
|
|
SELECT single_parameter_insert(2);
|
|
SELECT single_parameter_insert(3);
|
|
SELECT single_parameter_insert(4);
|
|
SELECT single_parameter_insert(5);
|
|
SELECT single_parameter_insert(6);
|
|
|
|
CREATE OR REPLACE FUNCTION double_parameter_insert(key_arg int, value_arg int) RETURNS void as $$
|
|
BEGIN
|
|
INSERT INTO temp_table (key, value) VALUES (key_arg, value_arg);
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
-- execute 6 times to trigger prepared statement usage
|
|
SELECT double_parameter_insert(1, 10);
|
|
SELECT double_parameter_insert(2, 20);
|
|
SELECT double_parameter_insert(3, 30);
|
|
SELECT double_parameter_insert(4, 40);
|
|
SELECT double_parameter_insert(5, 50);
|
|
SELECT double_parameter_insert(6, 60);
|
|
|
|
-- check inserted values
|
|
SELECT * FROM temp_table ORDER BY key, value;
|
|
|
|
-- check router executor select
|
|
CREATE OR REPLACE FUNCTION partition_column_select(key_arg int) RETURNS TABLE(key int, value int) AS $$
|
|
DECLARE
|
|
BEGIN
|
|
RETURN QUERY
|
|
SELECT
|
|
temp_table.key,
|
|
temp_table.value
|
|
FROM
|
|
temp_table
|
|
WHERE
|
|
temp_table.key = key_arg
|
|
ORDER BY
|
|
key,
|
|
value;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
SELECT partition_column_select(1);
|
|
SELECT partition_column_select(2);
|
|
SELECT partition_column_select(3);
|
|
SELECT partition_column_select(4);
|
|
SELECT partition_column_select(5);
|
|
|
|
-- 6th execution is failing. We don't want to run the failing test because of
|
|
-- changing output. After implementing this feature, uncomment this.
|
|
-- SELECT partition_column_select(6);
|
|
|
|
-- check real-time executor
|
|
CREATE OR REPLACE FUNCTION non_partition_column_select(value_arg int) RETURNS TABLE(key int, value int) AS $$
|
|
DECLARE
|
|
BEGIN
|
|
RETURN QUERY
|
|
SELECT
|
|
temp_table.key,
|
|
temp_table.value
|
|
FROM
|
|
temp_table
|
|
WHERE
|
|
temp_table.value = value_arg
|
|
ORDER BY
|
|
key,
|
|
value;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
SELECT non_partition_column_select(10);
|
|
SELECT non_partition_column_select(20);
|
|
SELECT non_partition_column_select(30);
|
|
SELECT non_partition_column_select(40);
|
|
SELECT non_partition_column_select(50);
|
|
|
|
-- 6th execution is failing. We don't want to run the failing test because of
|
|
-- changing output. After implementing this feature, uncomment this.
|
|
-- SELECT partition_column_select(6);
|
|
|
|
-- clean-up functions
|
|
DROP FUNCTION sql_test_no_1();
|
|
DROP FUNCTION sql_test_no_2();
|
|
DROP FUNCTION sql_test_no_3();
|
|
DROP FUNCTION sql_test_no_4();
|
|
DROP FUNCTION sql_test_no_6(int);
|
|
DROP FUNCTION plpgsql_test_1();
|
|
DROP FUNCTION plpgsql_test_2();
|
|
DROP FUNCTION plpgsql_test_3();
|
|
DROP FUNCTION plpgsql_test_4();
|
|
DROP FUNCTION plpgsql_test_5();
|
|
DROP FUNCTION plpgsql_test_6(int);
|
|
DROP FUNCTION plpgsql_test_7(text, text);
|
|
DROP FUNCTION no_parameter_insert();
|
|
DROP FUNCTION single_parameter_insert(int);
|
|
DROP FUNCTION double_parameter_insert(int, int);
|
|
DROP FUNCTION partition_column_select(int);
|
|
DROP FUNCTION non_partition_column_select(int);
|