mirror of https://github.com/citusdata/citus.git
625 lines
23 KiB
Plaintext
625 lines
23 KiB
Plaintext
CREATE SCHEMA "extension'test";
|
|
-- use a schema name with escape character
|
|
SET search_path TO "extension'test";
|
|
SET client_min_messages TO WARNING;
|
|
-- create an extension on the given search_path
|
|
-- the extension is on contrib, so should be avaliable for the regression tests
|
|
CREATE EXTENSION seg;
|
|
-- make sure that both the schema and the extension is distributed
|
|
SELECT count(*) FROM citus.pg_dist_object WHERE objid = (SELECT oid FROM pg_extension WHERE extname = 'seg');
|
|
count
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
SELECT count(*) FROM citus.pg_dist_object WHERE objid = (SELECT oid FROM pg_namespace WHERE nspname = 'extension''test');
|
|
count
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
CREATE TABLE test_table (key int, value seg);
|
|
SELECT create_distributed_table('test_table', 'key');
|
|
create_distributed_table
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
-- make sure that the table is also distributed now
|
|
SELECT count(*) from pg_dist_partition where logicalrelid='extension''test.test_table'::regclass;
|
|
count
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
CREATE TYPE two_segs AS (seg_1 seg, seg_2 seg);
|
|
-- verify that the type that depends on the extension is also marked as distributed
|
|
SELECT count(*) FROM citus.pg_dist_object WHERE objid = (SELECT oid FROM pg_type WHERE typname = 'two_segs' AND typnamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'extension''test'));
|
|
count
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
-- now try to run CREATE EXTENSION within a transction block, all should work fine
|
|
BEGIN;
|
|
CREATE EXTENSION isn WITH SCHEMA public;
|
|
-- now, try create a reference table relying on the data types
|
|
-- this should not succeed as we do not distribute extension commands within transaction blocks
|
|
CREATE TABLE dist_table (key int, value public.issn);
|
|
SELECT create_distributed_table('dist_table', 'key');
|
|
create_distributed_table
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
-- we can even run queries (sequentially) over the distributed table
|
|
SELECT * FROM dist_table;
|
|
key | value
|
|
---------------------------------------------------------------------
|
|
(0 rows)
|
|
|
|
INSERT INTO dist_table VALUES (1, public.issn('1436-4522'));
|
|
INSERT INTO dist_table SELECT * FROM dist_table RETURNING *;
|
|
key | value
|
|
---------------------------------------------------------------------
|
|
1 | 1436-4522
|
|
(1 row)
|
|
|
|
COMMIT;
|
|
-- make sure that the extension is distributed even if we run create extension in a transaction block
|
|
SELECT count(*) FROM citus.pg_dist_object WHERE objid = (SELECT oid FROM pg_extension WHERE extname = 'isn');
|
|
count
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
SELECT run_command_on_workers($$SELECT count(*) FROM pg_extension WHERE extname = 'isn'$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,1)
|
|
(localhost,57638,t,1)
|
|
(2 rows)
|
|
|
|
CREATE TABLE ref_table (a public.issn);
|
|
-- now, create a reference table relying on the data types
|
|
SELECT create_reference_table('ref_table');
|
|
create_reference_table
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
-- now, drop the extension, recreate it with an older version and update it to latest version
|
|
DROP EXTENSION isn CASCADE;
|
|
CREATE EXTENSION isn WITH VERSION "1.1";
|
|
-- before updating the version, ensure the current version
|
|
SELECT run_command_on_workers($$SELECT extversion FROM pg_extension WHERE extname = 'isn'$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,1.1)
|
|
(localhost,57638,t,1.1)
|
|
(2 rows)
|
|
|
|
-- now, update to a newer version
|
|
ALTER EXTENSION isn UPDATE TO '1.2';
|
|
-- show that ALTER EXTENSION is propagated
|
|
SELECT run_command_on_workers($$SELECT extversion FROM pg_extension WHERE extname = 'isn'$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,1.2)
|
|
(localhost,57638,t,1.2)
|
|
(2 rows)
|
|
|
|
-- before changing the schema, ensure the current schmea
|
|
SELECT run_command_on_workers($$SELECT nspname from pg_namespace where oid=(SELECT extnamespace FROM pg_extension WHERE extname = 'isn')$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,extension'test)
|
|
(localhost,57638,t,extension'test)
|
|
(2 rows)
|
|
|
|
-- now change the schema
|
|
ALTER EXTENSION isn SET SCHEMA public;
|
|
-- switch back to public schema as we set extension's schema to public
|
|
SET search_path TO public;
|
|
-- make sure that the extension is distributed
|
|
SELECT count(*) FROM citus.pg_dist_object WHERE objid = (SELECT oid FROM pg_extension WHERE extname = 'isn');
|
|
count
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
-- show that the ALTER EXTENSION command is propagated
|
|
SELECT run_command_on_workers($$SELECT nspname from pg_namespace where oid=(SELECT extnamespace FROM pg_extension WHERE extname = 'isn')$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,public)
|
|
(localhost,57638,t,public)
|
|
(2 rows)
|
|
|
|
-- drop the extension finally
|
|
DROP EXTENSION isn CASCADE;
|
|
-- now make sure that the reference tables depending on an extension can be succesfully created.
|
|
-- we should also ensure that we replicate this reference table (and hence the extension)
|
|
-- to new nodes after calling master_activate_node.
|
|
-- now, first drop seg and existing objects before next test
|
|
DROP EXTENSION seg CASCADE;
|
|
-- but as we have only 2 ports in postgresql tests, let's remove one of the nodes first
|
|
-- before remove, first remove the existing relations (due to the other tests)
|
|
DROP SCHEMA "extension'test" CASCADE;
|
|
SELECT 1 from master_remove_node('localhost', :worker_2_port);
|
|
?column?
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
-- then create the extension
|
|
CREATE EXTENSION seg;
|
|
-- show that the extension is created on existing worker
|
|
SELECT run_command_on_workers($$SELECT count(extnamespace) FROM pg_extension WHERE extname = 'seg'$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,1)
|
|
(1 row)
|
|
|
|
SELECT workers.result = pg_extension.extversion AS same_version
|
|
FROM run_command_on_workers($$SELECT extversion FROM pg_extension WHERE extname = 'seg'$$) workers, pg_extension WHERE extname = 'seg';
|
|
same_version
|
|
---------------------------------------------------------------------
|
|
t
|
|
(1 row)
|
|
|
|
-- now create the reference table
|
|
CREATE TABLE ref_table_2 (x seg);
|
|
SELECT create_reference_table('ref_table_2');
|
|
create_reference_table
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
-- we also add an old style extension from before extensions which we upgrade to an extension
|
|
-- by exercising it before the add node we verify it will create the extension (without upgrading)
|
|
-- it on the new worker as well. For this we use the dict_int extension which is in contrib,
|
|
-- supports FROM unpackaged, and is relatively small
|
|
-- create objects for dict_int manually so we can upgrade from unpacked
|
|
CREATE FUNCTION dintdict_init(internal) RETURNS internal AS 'dict_int.so' LANGUAGE C STRICT;
|
|
CREATE FUNCTION dintdict_lexize(internal, internal, internal, internal) RETURNS internal AS 'dict_int.so' LANGUAGE C STRICT;
|
|
CREATE TEXT SEARCH TEMPLATE intdict_template (LEXIZE = dintdict_lexize, INIT = dintdict_init );
|
|
CREATE TEXT SEARCH DICTIONARY intdict (TEMPLATE = intdict_template);
|
|
COMMENT ON TEXT SEARCH DICTIONARY intdict IS 'dictionary for integers';
|
|
SELECT run_command_on_workers($$
|
|
CREATE TEXT SEARCH TEMPLATE intdict_template (LEXIZE = dintdict_lexize, INIT = dintdict_init );
|
|
$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,"CREATE TEXT SEARCH TEMPLATE")
|
|
(1 row)
|
|
|
|
SELECT run_command_on_workers($$
|
|
CREATE TEXT SEARCH DICTIONARY intdict (TEMPLATE = intdict_template);
|
|
$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,"CREATE TEXT SEARCH DICTIONARY")
|
|
(1 row)
|
|
|
|
SELECT run_command_on_workers($$
|
|
COMMENT ON TEXT SEARCH DICTIONARY intdict IS 'dictionary for integers';
|
|
$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,COMMENT)
|
|
(1 row)
|
|
|
|
CREATE EXTENSION dict_int FROM unpackaged;
|
|
SELECT run_command_on_workers($$SELECT count(extnamespace) FROM pg_extension WHERE extname = 'dict_int'$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,1)
|
|
(1 row)
|
|
|
|
SELECT run_command_on_workers($$SELECT extversion FROM pg_extension WHERE extname = 'dict_int'$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,1.0)
|
|
(1 row)
|
|
|
|
-- and add the other node
|
|
SELECT 1 from master_add_node('localhost', :worker_2_port);
|
|
?column?
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
-- show that the extension is created on both existing and new node
|
|
SELECT run_command_on_workers($$SELECT count(extnamespace) FROM pg_extension WHERE extname = 'seg'$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,1)
|
|
(localhost,57638,t,1)
|
|
(2 rows)
|
|
|
|
SELECT workers.result = pg_extension.extversion AS same_version
|
|
FROM run_command_on_workers($$SELECT extversion FROM pg_extension WHERE extname = 'seg'$$) workers, pg_extension WHERE extname = 'seg';
|
|
same_version
|
|
---------------------------------------------------------------------
|
|
t
|
|
t
|
|
(2 rows)
|
|
|
|
-- check for the unpackaged extension to be created correctly
|
|
SELECT run_command_on_workers($$SELECT count(extnamespace) FROM pg_extension WHERE extname = 'dict_int'$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,1)
|
|
(localhost,57638,t,1)
|
|
(2 rows)
|
|
|
|
SELECT run_command_on_workers($$SELECT extversion FROM pg_extension WHERE extname = 'dict_int'$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,1.0)
|
|
(localhost,57638,t,1.0)
|
|
(2 rows)
|
|
|
|
-- and similarly check for the reference table
|
|
select count(*) from pg_dist_partition where partmethod='n' and logicalrelid='ref_table_2'::regclass;
|
|
count
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
SELECT count(*) FROM pg_dist_shard WHERE logicalrelid='ref_table_2'::regclass;
|
|
count
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
DROP TABLE ref_table_2;
|
|
-- now test create extension in another transaction block but rollback this time
|
|
BEGIN;
|
|
CREATE EXTENSION isn WITH VERSION '1.1' SCHEMA public;
|
|
ROLLBACK;
|
|
-- at the end of the transaction block, we did not create isn extension in coordinator or worker nodes as we rollback'ed
|
|
-- make sure that the extension is not distributed
|
|
SELECT count(*) FROM citus.pg_dist_object WHERE objid = (SELECT oid FROM pg_extension WHERE extname = 'isn');
|
|
count
|
|
---------------------------------------------------------------------
|
|
0
|
|
(1 row)
|
|
|
|
-- and the extension does not exist on workers
|
|
SELECT run_command_on_workers($$SELECT count(*) FROM pg_extension WHERE extname = 'isn'$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,0)
|
|
(localhost,57638,t,0)
|
|
(2 rows)
|
|
|
|
-- give a notice for the following commands saying that it is not
|
|
-- propagated to the workers. the user should run it manually on the workers
|
|
CREATE TABLE t1 (A int);
|
|
CREATE VIEW v1 AS select * from t1;
|
|
ALTER EXTENSION seg ADD VIEW v1;
|
|
ALTER EXTENSION seg DROP VIEW v1;
|
|
DROP VIEW v1;
|
|
DROP TABLE t1;
|
|
-- drop multiple extensions at the same time
|
|
CREATE EXTENSION isn WITH VERSION '1.1' SCHEMA public;
|
|
-- let's create another extension locally
|
|
set citus.enable_ddl_propagation to 'off';
|
|
CREATE EXTENSION pg_buffercache;
|
|
set citus.enable_ddl_propagation to 'on';
|
|
DROP EXTENSION pg_buffercache, isn CASCADE;
|
|
SELECT count(*) FROM pg_extension WHERE extname IN ('pg_buffercache', 'isn');
|
|
count
|
|
---------------------------------------------------------------------
|
|
0
|
|
(1 row)
|
|
|
|
-- drop extension should just work
|
|
DROP EXTENSION seg CASCADE;
|
|
SELECT count(*) FROM citus.pg_dist_object WHERE objid = (SELECT oid FROM pg_extension WHERE extname = 'seg');
|
|
count
|
|
---------------------------------------------------------------------
|
|
0
|
|
(1 row)
|
|
|
|
SELECT run_command_on_workers($$SELECT count(*) FROM pg_extension WHERE extname = 'seg'$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,0)
|
|
(localhost,57638,t,0)
|
|
(2 rows)
|
|
|
|
-- make sure that the extension is not avaliable anymore as a distributed object
|
|
SELECT count(*) FROM citus.pg_dist_object WHERE objid = (SELECT oid FROM pg_extension WHERE extname IN ('seg', 'isn'));
|
|
count
|
|
---------------------------------------------------------------------
|
|
0
|
|
(1 row)
|
|
|
|
CREATE SCHEMA "extension'test";
|
|
SET search_path TO "extension'test";
|
|
-- check restriction for sequential execution
|
|
-- enable it and see that create command errors but continues its execution by changing citus.multi_shard_modify_mode TO 'off
|
|
BEGIN;
|
|
CREATE TABLE some_random_table (a int);
|
|
SELECT create_distributed_table('some_random_table', 'a');
|
|
create_distributed_table
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
CREATE EXTENSION seg;
|
|
CREATE TABLE some_random_table_2 (a int, b seg);
|
|
SELECT create_distributed_table('some_random_table_2', 'a');
|
|
create_distributed_table
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
ROLLBACK;
|
|
-- show that the CREATE EXTENSION command propagated even if the transaction
|
|
-- block is rollbacked, that's a shortcoming of dependency creation logic
|
|
SELECT COUNT(DISTINCT workers.result)
|
|
FROM run_command_on_workers($$SELECT extversion FROM pg_extension WHERE extname = 'seg'$$) workers;
|
|
count
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
-- drop the schema and all the objects
|
|
DROP SCHEMA "extension'test" CASCADE;
|
|
-- recreate for the next tests
|
|
CREATE SCHEMA "extension'test";
|
|
-- use a schema name with escape character
|
|
SET search_path TO "extension'test";
|
|
-- remove the node, we'll add back again
|
|
SELECT 1 from master_remove_node('localhost', :worker_2_port);
|
|
?column?
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
-- Test extension function incorrect distribution argument
|
|
CREATE TABLE test_extension_function(col varchar);
|
|
CREATE EXTENSION seg;
|
|
-- Missing distribution argument
|
|
SELECT create_distributed_function('seg_in(cstring)');
|
|
ERROR: Extension functions(seg_in) without distribution argument are not supported.
|
|
-- Missing colocation argument
|
|
SELECT create_distributed_function('seg_in(cstring)', '$1');
|
|
ERROR: cannot distribute the function "seg_in" since there is no table to colocate with
|
|
HINT: Provide a distributed table via "colocate_with" option to create_distributed_function()
|
|
-- Incorrect distribution argument
|
|
SELECT create_distributed_function('seg_in(cstring)', '$2', colocate_with:='test_extension_function');
|
|
ERROR: cannot distribute the function "seg_in" since the distribution argument is not valid
|
|
HINT: Either provide a valid function argument name or a valid "$paramIndex" to create_distributed_function()
|
|
-- Colocated table is not distributed
|
|
SELECT create_distributed_function('seg_in(cstring)', '$1', 'test_extension_function');
|
|
ERROR: relation test_extension_function is not distributed
|
|
DROP EXTENSION seg;
|
|
SET citus.shard_replication_factor TO 1;
|
|
SELECT create_distributed_table('test_extension_function', 'col', colocate_with := 'none');
|
|
create_distributed_table
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
-- now, create a type that depends on another type, which
|
|
-- finally depends on an extension
|
|
BEGIN;
|
|
CREATE EXTENSION seg;
|
|
CREATE EXTENSION isn;
|
|
CREATE TYPE test_type AS (a int, b seg);
|
|
CREATE TYPE test_type_2 AS (a int, b test_type);
|
|
CREATE TABLE t2 (a int, b test_type_2, c issn);
|
|
SELECT create_distributed_table('t2', 'a');
|
|
create_distributed_table
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
CREATE TYPE test_type_3 AS (a int, b test_type, c issn);
|
|
CREATE TABLE t3 (a int, b test_type_3);
|
|
SELECT create_reference_table('t3');
|
|
create_reference_table
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
-- Distribute an extension-function
|
|
SELECT create_distributed_function('seg_in(cstring)', '$1', 'test_extension_function');
|
|
create_distributed_function
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
COMMIT;
|
|
-- Check the pg_dist_object
|
|
SELECT pg_proc.proname as DistributedFunction
|
|
FROM citus.pg_dist_object, pg_proc
|
|
WHERE pg_proc.proname = 'seg_in' and
|
|
pg_proc.oid = citus.pg_dist_object.objid and
|
|
classid = 'pg_proc'::regclass;
|
|
distributedfunction
|
|
---------------------------------------------------------------------
|
|
seg_in
|
|
(1 row)
|
|
|
|
SELECT run_command_on_workers($$
|
|
SELECT count(*)
|
|
FROM citus.pg_dist_object, pg_proc
|
|
WHERE pg_proc.proname = 'seg_in' and
|
|
pg_proc.oid = citus.pg_dist_object.objid and
|
|
classid = 'pg_proc'::regclass;
|
|
$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,1)
|
|
(1 row)
|
|
|
|
-- add the node back
|
|
SELECT 1 from master_add_node('localhost', :worker_2_port);
|
|
?column?
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
-- make sure that both extensions are created on both nodes
|
|
SELECT count(*) FROM citus.pg_dist_object WHERE objid IN (SELECT oid FROM pg_extension WHERE extname IN ('seg', 'isn'));
|
|
count
|
|
---------------------------------------------------------------------
|
|
2
|
|
(1 row)
|
|
|
|
SELECT run_command_on_workers($$SELECT count(*) FROM pg_extension WHERE extname IN ('seg', 'isn')$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,2)
|
|
(localhost,57638,t,2)
|
|
(2 rows)
|
|
|
|
-- Check the pg_dist_object on the both nodes
|
|
SELECT run_command_on_workers($$
|
|
SELECT count(*)
|
|
FROM citus.pg_dist_object, pg_proc
|
|
WHERE pg_proc.proname = 'seg_in' and
|
|
pg_proc.oid = citus.pg_dist_object.objid and
|
|
classid = 'pg_proc'::regclass;
|
|
$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,1)
|
|
(localhost,57638,t,1)
|
|
(2 rows)
|
|
|
|
DROP EXTENSION seg CASCADE;
|
|
-- Recheck the pg_dist_object
|
|
SELECT pg_proc.proname as DistributedFunction
|
|
FROM citus.pg_dist_object, pg_proc
|
|
WHERE pg_proc.proname = 'seg_in' and
|
|
pg_proc.oid = citus.pg_dist_object.objid and
|
|
classid = 'pg_proc'::regclass;
|
|
distributedfunction
|
|
---------------------------------------------------------------------
|
|
(0 rows)
|
|
|
|
SELECT run_command_on_workers($$
|
|
SELECT count(*)
|
|
FROM citus.pg_dist_object, pg_proc
|
|
WHERE pg_proc.proname = 'seg_in' and
|
|
pg_proc.oid = citus.pg_dist_object.objid and
|
|
classid = 'pg_proc'::regclass;
|
|
$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,0)
|
|
(localhost,57638,t,0)
|
|
(2 rows)
|
|
|
|
-- Distribute an extension-function where extension is not in pg_dist_object
|
|
SET citus.enable_ddl_propagation TO false;
|
|
CREATE EXTENSION seg;
|
|
SET citus.enable_ddl_propagation TO true;
|
|
-- Check the extension in pg_dist_object
|
|
SELECT count(*) FROM citus.pg_dist_object WHERE classid = 'pg_catalog.pg_extension'::pg_catalog.regclass AND
|
|
objid = (SELECT oid FROM pg_extension WHERE extname = 'seg');
|
|
count
|
|
---------------------------------------------------------------------
|
|
0
|
|
(1 row)
|
|
|
|
SELECT run_command_on_workers($$
|
|
SELECT count(*)
|
|
FROM citus.pg_dist_object, pg_proc
|
|
WHERE pg_proc.proname = 'seg_in' and
|
|
pg_proc.oid = citus.pg_dist_object.objid and
|
|
classid = 'pg_proc'::regclass;
|
|
$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,0)
|
|
(localhost,57638,t,0)
|
|
(2 rows)
|
|
|
|
SELECT create_distributed_function('seg_in(cstring)', '$1', 'test_extension_function');
|
|
create_distributed_function
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
-- Recheck the extension in pg_dist_object
|
|
SELECT count(*) FROM citus.pg_dist_object WHERE classid = 'pg_catalog.pg_extension'::pg_catalog.regclass AND
|
|
objid = (SELECT oid FROM pg_extension WHERE extname = 'seg');
|
|
count
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
SELECT pg_proc.proname as DistributedFunction
|
|
FROM citus.pg_dist_object, pg_proc
|
|
WHERE pg_proc.proname = 'seg_in' and
|
|
pg_proc.oid = citus.pg_dist_object.objid and
|
|
classid = 'pg_proc'::regclass;
|
|
distributedfunction
|
|
---------------------------------------------------------------------
|
|
seg_in
|
|
(1 row)
|
|
|
|
SELECT run_command_on_workers($$
|
|
SELECT count(*)
|
|
FROM citus.pg_dist_object, pg_proc
|
|
WHERE pg_proc.proname = 'seg_in' and
|
|
pg_proc.oid = citus.pg_dist_object.objid and
|
|
classid = 'pg_proc'::regclass;
|
|
$$);
|
|
run_command_on_workers
|
|
---------------------------------------------------------------------
|
|
(localhost,57637,t,1)
|
|
(localhost,57638,t,1)
|
|
(2 rows)
|
|
|
|
DROP EXTENSION seg;
|
|
DROP TABLE test_extension_function;
|
|
-- Test extension function altering distribution argument
|
|
BEGIN;
|
|
SET citus.shard_replication_factor = 1;
|
|
CREATE TABLE test_extension_function(col1 float8[], col2 float8[]);
|
|
SELECT create_distributed_table('test_extension_function', 'col1', colocate_with := 'none');
|
|
create_distributed_table
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
CREATE EXTENSION cube;
|
|
SELECT create_distributed_function('cube(float8[], float8[])', '$1', 'test_extension_function');
|
|
create_distributed_function
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT distribution_argument_index FROM citus.pg_dist_object WHERE classid = 'pg_catalog.pg_proc'::pg_catalog.regclass AND
|
|
objid = (SELECT oid FROM pg_proc WHERE prosrc = 'cube_a_f8_f8');
|
|
distribution_argument_index
|
|
---------------------------------------------------------------------
|
|
0
|
|
(1 row)
|
|
|
|
SELECT create_distributed_function('cube(float8[], float8[])', '$2', 'test_extension_function');
|
|
create_distributed_function
|
|
---------------------------------------------------------------------
|
|
|
|
(1 row)
|
|
|
|
SELECT distribution_argument_index FROM citus.pg_dist_object WHERE classid = 'pg_catalog.pg_proc'::pg_catalog.regclass AND
|
|
objid = (SELECT oid FROM pg_proc WHERE prosrc = 'cube_a_f8_f8');
|
|
distribution_argument_index
|
|
---------------------------------------------------------------------
|
|
1
|
|
(1 row)
|
|
|
|
ROLLBACK;
|
|
-- drop the schema and all the objects
|
|
DROP SCHEMA "extension'test" CASCADE;
|