mirror of https://github.com/citusdata/citus.git
46 lines
1.6 KiB
Plaintext
46 lines
1.6 KiB
Plaintext
--
|
|
-- MULTI_REPARTITIONED_SUBQUERY_UDF
|
|
--
|
|
ALTER SEQUENCE pg_catalog.pg_dist_shardid_seq RESTART 830000;
|
|
-- Create UDF in master and workers
|
|
\c - - - :master_port
|
|
DROP FUNCTION IF EXISTS median(double precision[]);
|
|
NOTICE: function median(pg_catalog.float8[]) does not exist, skipping
|
|
CREATE FUNCTION median(double precision[]) RETURNS double precision
|
|
LANGUAGE sql IMMUTABLE AS $_$
|
|
SELECT AVG(val) FROM
|
|
(SELECT val FROM unnest($1) val
|
|
ORDER BY 1 LIMIT 2 - MOD(array_upper($1, 1), 2)
|
|
OFFSET CEIL(array_upper($1, 1) / 2.0) - 1) sub;
|
|
$_$;
|
|
\c - - - :worker_1_port
|
|
DROP FUNCTION IF EXISTS median(double precision[]);
|
|
NOTICE: function median(pg_catalog.float8[]) does not exist, skipping
|
|
CREATE FUNCTION median(double precision[]) RETURNS double precision
|
|
LANGUAGE sql IMMUTABLE AS $_$
|
|
SELECT AVG(val) FROM
|
|
(SELECT val FROM unnest($1) val
|
|
ORDER BY 1 LIMIT 2 - MOD(array_upper($1, 1), 2)
|
|
OFFSET CEIL(array_upper($1, 1) / 2.0) - 1) sub;
|
|
$_$;
|
|
\c - - - :worker_2_port
|
|
DROP FUNCTION IF EXISTS median(double precision[]);
|
|
NOTICE: function median(pg_catalog.float8[]) does not exist, skipping
|
|
CREATE FUNCTION median(double precision[]) RETURNS double precision
|
|
LANGUAGE sql IMMUTABLE AS $_$
|
|
SELECT AVG(val) FROM
|
|
(SELECT val FROM unnest($1) val
|
|
ORDER BY 1 LIMIT 2 - MOD(array_upper($1, 1), 2)
|
|
OFFSET CEIL(array_upper($1, 1) / 2.0) - 1) sub;
|
|
$_$;
|
|
-- Run query on master
|
|
\c - - - :master_port
|
|
SET citus.task_executor_type TO 'task-tracker';
|
|
SELECT * FROM (SELECT median(ARRAY[1,2,sum(l_suppkey)]) as median, count(*)
|
|
FROM lineitem GROUP BY l_partkey) AS a
|
|
WHERE median > 2;
|
|
median | count
|
|
--------+-------
|
|
(0 rows)
|
|
|