1-- 2-- Test for pg_get_object_address 3-- 4 5-- Clean up in case a prior regression run failed 6SET client_min_messages TO 'warning'; 7DROP ROLE IF EXISTS regress_addr_user; 8RESET client_min_messages; 9 10CREATE USER regress_addr_user; 11 12-- Test generic object addressing/identification functions 13CREATE SCHEMA addr_nsp; 14SET search_path TO 'addr_nsp'; 15CREATE FOREIGN DATA WRAPPER addr_fdw; 16CREATE SERVER addr_fserv FOREIGN DATA WRAPPER addr_fdw; 17CREATE TEXT SEARCH DICTIONARY addr_ts_dict (template=simple); 18CREATE TEXT SEARCH CONFIGURATION addr_ts_conf (copy=english); 19CREATE TEXT SEARCH TEMPLATE addr_ts_temp (lexize=dsimple_lexize); 20CREATE TEXT SEARCH PARSER addr_ts_prs 21 (start = prsd_start, gettoken = prsd_nexttoken, end = prsd_end, lextypes = prsd_lextype); 22CREATE TABLE addr_nsp.gentable ( 23 a serial primary key CONSTRAINT a_chk CHECK (a > 0), 24 b text DEFAULT 'hello'); 25CREATE TABLE addr_nsp.parttable ( 26 a int PRIMARY KEY 27) PARTITION BY RANGE (a); 28CREATE VIEW addr_nsp.genview AS SELECT * from addr_nsp.gentable; 29CREATE MATERIALIZED VIEW addr_nsp.genmatview AS SELECT * FROM addr_nsp.gentable; 30CREATE TYPE addr_nsp.gencomptype AS (a int); 31CREATE TYPE addr_nsp.genenum AS ENUM ('one', 'two'); 32CREATE FOREIGN TABLE addr_nsp.genftable (a int) SERVER addr_fserv; 33CREATE AGGREGATE addr_nsp.genaggr(int4) (sfunc = int4pl, stype = int4); 34CREATE DOMAIN addr_nsp.gendomain AS int4 CONSTRAINT domconstr CHECK (value > 0); 35CREATE FUNCTION addr_nsp.trig() RETURNS TRIGGER LANGUAGE plpgsql AS $$ BEGIN END; $$; 36CREATE TRIGGER t BEFORE INSERT ON addr_nsp.gentable FOR EACH ROW EXECUTE PROCEDURE addr_nsp.trig(); 37CREATE POLICY genpol ON addr_nsp.gentable; 38CREATE PROCEDURE addr_nsp.proc(int4) LANGUAGE SQL AS $$ $$; 39CREATE SERVER "integer" FOREIGN DATA WRAPPER addr_fdw; 40CREATE USER MAPPING FOR regress_addr_user SERVER "integer"; 41ALTER DEFAULT PRIVILEGES FOR ROLE regress_addr_user IN SCHEMA public GRANT ALL ON TABLES TO regress_addr_user; 42ALTER DEFAULT PRIVILEGES FOR ROLE regress_addr_user REVOKE DELETE ON TABLES FROM regress_addr_user; 43-- this transform would be quite unsafe to leave lying around, 44-- except that the SQL language pays no attention to transforms: 45CREATE TRANSFORM FOR int LANGUAGE SQL ( 46 FROM SQL WITH FUNCTION prsd_lextype(internal), 47 TO SQL WITH FUNCTION int4recv(internal)); 48-- suppress warning that depends on wal_level 49SET client_min_messages = 'ERROR'; 50CREATE PUBLICATION addr_pub FOR TABLE addr_nsp.gentable; 51RESET client_min_messages; 52CREATE SUBSCRIPTION regress_addr_sub CONNECTION '' PUBLICATION bar WITH (connect = false, slot_name = NONE); 53CREATE STATISTICS addr_nsp.gentable_stat ON a, b FROM addr_nsp.gentable; 54 55-- test some error cases 56SELECT pg_get_object_address('stone', '{}', '{}'); 57SELECT pg_get_object_address('table', '{}', '{}'); 58SELECT pg_get_object_address('table', '{NULL}', '{}'); 59 60-- unrecognized object types 61DO $$ 62DECLARE 63 objtype text; 64BEGIN 65 FOR objtype IN VALUES ('toast table'), ('index column'), ('sequence column'), 66 ('toast table column'), ('view column'), ('materialized view column') 67 LOOP 68 BEGIN 69 PERFORM pg_get_object_address(objtype, '{one}', '{}'); 70 EXCEPTION WHEN invalid_parameter_value THEN 71 RAISE WARNING 'error for %: %', objtype, sqlerrm; 72 END; 73 END LOOP; 74END; 75$$; 76 77-- miscellaneous other errors 78select * from pg_get_object_address('operator of access method', '{btree,integer_ops,1}', '{int4,bool}'); 79select * from pg_get_object_address('operator of access method', '{btree,integer_ops,99}', '{int4,int4}'); 80select * from pg_get_object_address('function of access method', '{btree,integer_ops,1}', '{int4,bool}'); 81select * from pg_get_object_address('function of access method', '{btree,integer_ops,99}', '{int4,int4}'); 82 83DO $$ 84DECLARE 85 objtype text; 86 names text[]; 87 args text[]; 88BEGIN 89 FOR objtype IN VALUES 90 ('table'), ('index'), ('sequence'), ('view'), 91 ('materialized view'), ('foreign table'), 92 ('table column'), ('foreign table column'), 93 ('aggregate'), ('function'), ('procedure'), ('type'), ('cast'), 94 ('table constraint'), ('domain constraint'), ('conversion'), ('default value'), 95 ('operator'), ('operator class'), ('operator family'), ('rule'), ('trigger'), 96 ('text search parser'), ('text search dictionary'), 97 ('text search template'), ('text search configuration'), 98 ('policy'), ('user mapping'), ('default acl'), ('transform'), 99 ('operator of access method'), ('function of access method'), 100 ('publication relation') 101 LOOP 102 FOR names IN VALUES ('{eins}'), ('{addr_nsp, zwei}'), ('{eins, zwei, drei}') 103 LOOP 104 FOR args IN VALUES ('{}'), ('{integer}') 105 LOOP 106 BEGIN 107 PERFORM pg_get_object_address(objtype, names, args); 108 EXCEPTION WHEN OTHERS THEN 109 RAISE WARNING 'error for %,%,%: %', objtype, names, args, sqlerrm; 110 END; 111 END LOOP; 112 END LOOP; 113 END LOOP; 114END; 115$$; 116 117-- these object types cannot be qualified names 118SELECT pg_get_object_address('language', '{one}', '{}'); 119SELECT pg_get_object_address('language', '{one,two}', '{}'); 120SELECT pg_get_object_address('large object', '{123}', '{}'); 121SELECT pg_get_object_address('large object', '{123,456}', '{}'); 122SELECT pg_get_object_address('large object', '{blargh}', '{}'); 123SELECT pg_get_object_address('schema', '{one}', '{}'); 124SELECT pg_get_object_address('schema', '{one,two}', '{}'); 125SELECT pg_get_object_address('role', '{one}', '{}'); 126SELECT pg_get_object_address('role', '{one,two}', '{}'); 127SELECT pg_get_object_address('database', '{one}', '{}'); 128SELECT pg_get_object_address('database', '{one,two}', '{}'); 129SELECT pg_get_object_address('tablespace', '{one}', '{}'); 130SELECT pg_get_object_address('tablespace', '{one,two}', '{}'); 131SELECT pg_get_object_address('foreign-data wrapper', '{one}', '{}'); 132SELECT pg_get_object_address('foreign-data wrapper', '{one,two}', '{}'); 133SELECT pg_get_object_address('server', '{one}', '{}'); 134SELECT pg_get_object_address('server', '{one,two}', '{}'); 135SELECT pg_get_object_address('extension', '{one}', '{}'); 136SELECT pg_get_object_address('extension', '{one,two}', '{}'); 137SELECT pg_get_object_address('event trigger', '{one}', '{}'); 138SELECT pg_get_object_address('event trigger', '{one,two}', '{}'); 139SELECT pg_get_object_address('access method', '{one}', '{}'); 140SELECT pg_get_object_address('access method', '{one,two}', '{}'); 141SELECT pg_get_object_address('publication', '{one}', '{}'); 142SELECT pg_get_object_address('publication', '{one,two}', '{}'); 143SELECT pg_get_object_address('subscription', '{one}', '{}'); 144SELECT pg_get_object_address('subscription', '{one,two}', '{}'); 145 146-- test successful cases 147WITH objects (type, name, args) AS (VALUES 148 ('table', '{addr_nsp, gentable}'::text[], '{}'::text[]), 149 ('table', '{addr_nsp, parttable}'::text[], '{}'::text[]), 150 ('index', '{addr_nsp, gentable_pkey}', '{}'), 151 ('index', '{addr_nsp, parttable_pkey}', '{}'), 152 ('sequence', '{addr_nsp, gentable_a_seq}', '{}'), 153 -- toast table 154 ('view', '{addr_nsp, genview}', '{}'), 155 ('materialized view', '{addr_nsp, genmatview}', '{}'), 156 ('foreign table', '{addr_nsp, genftable}', '{}'), 157 ('table column', '{addr_nsp, gentable, b}', '{}'), 158 ('foreign table column', '{addr_nsp, genftable, a}', '{}'), 159 ('aggregate', '{addr_nsp, genaggr}', '{int4}'), 160 ('function', '{pg_catalog, pg_identify_object}', '{pg_catalog.oid, pg_catalog.oid, int4}'), 161 ('procedure', '{addr_nsp, proc}', '{int4}'), 162 ('type', '{pg_catalog._int4}', '{}'), 163 ('type', '{addr_nsp.gendomain}', '{}'), 164 ('type', '{addr_nsp.gencomptype}', '{}'), 165 ('type', '{addr_nsp.genenum}', '{}'), 166 ('cast', '{int8}', '{int4}'), 167 ('collation', '{default}', '{}'), 168 ('table constraint', '{addr_nsp, gentable, a_chk}', '{}'), 169 ('domain constraint', '{addr_nsp.gendomain}', '{domconstr}'), 170 ('conversion', '{pg_catalog, koi8_r_to_mic}', '{}'), 171 ('default value', '{addr_nsp, gentable, b}', '{}'), 172 ('language', '{plpgsql}', '{}'), 173 -- large object 174 ('operator', '{+}', '{int4, int4}'), 175 ('operator class', '{btree, int4_ops}', '{}'), 176 ('operator family', '{btree, integer_ops}', '{}'), 177 ('operator of access method', '{btree,integer_ops,1}', '{integer,integer}'), 178 ('function of access method', '{btree,integer_ops,2}', '{integer,integer}'), 179 ('rule', '{addr_nsp, genview, _RETURN}', '{}'), 180 ('trigger', '{addr_nsp, gentable, t}', '{}'), 181 ('schema', '{addr_nsp}', '{}'), 182 ('text search parser', '{addr_ts_prs}', '{}'), 183 ('text search dictionary', '{addr_ts_dict}', '{}'), 184 ('text search template', '{addr_ts_temp}', '{}'), 185 ('text search configuration', '{addr_ts_conf}', '{}'), 186 ('role', '{regress_addr_user}', '{}'), 187 -- database 188 -- tablespace 189 ('foreign-data wrapper', '{addr_fdw}', '{}'), 190 ('server', '{addr_fserv}', '{}'), 191 ('user mapping', '{regress_addr_user}', '{integer}'), 192 ('default acl', '{regress_addr_user,public}', '{r}'), 193 ('default acl', '{regress_addr_user}', '{r}'), 194 -- extension 195 -- event trigger 196 ('policy', '{addr_nsp, gentable, genpol}', '{}'), 197 ('transform', '{int}', '{sql}'), 198 ('access method', '{btree}', '{}'), 199 ('publication', '{addr_pub}', '{}'), 200 ('publication relation', '{addr_nsp, gentable}', '{addr_pub}'), 201 ('subscription', '{regress_addr_sub}', '{}'), 202 ('statistics object', '{addr_nsp, gentable_stat}', '{}') 203 ) 204SELECT (pg_identify_object(addr1.classid, addr1.objid, addr1.objsubid)).*, 205 -- test roundtrip through pg_identify_object_as_address 206 ROW(pg_identify_object(addr1.classid, addr1.objid, addr1.objsubid)) = 207 ROW(pg_identify_object(addr2.classid, addr2.objid, addr2.objsubid)) 208 FROM objects, pg_get_object_address(type, name, args) addr1, 209 pg_identify_object_as_address(classid, objid, objsubid) ioa(typ,nms,args), 210 pg_get_object_address(typ, nms, ioa.args) as addr2 211 ORDER BY addr1.classid, addr1.objid, addr1.objsubid; 212 213--- 214--- Cleanup resources 215--- 216DROP FOREIGN DATA WRAPPER addr_fdw CASCADE; 217DROP PUBLICATION addr_pub; 218DROP SUBSCRIPTION regress_addr_sub; 219 220DROP SCHEMA addr_nsp CASCADE; 221 222DROP OWNED BY regress_addr_user; 223DROP USER regress_addr_user; 224