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