8000 Handle OID column inheritance correctly in ALTER TABLE ... INHERIT. · hackingwu/postgres@6c4cf2b · GitHub
[go: up one dir, main page]
Skip to content

Commit 6c4cf2b

Browse files
tglsfdctglsfdc
committed
Handle OID column inheritance correctly in ALTER TABLE ... INHERIT.
Inheritance operations must treat the OID column, if any, much like regular user columns. But MergeAttributesIntoExisting() neglected to do that, leading to weird results after a table with OIDs is associated to a parent with OIDs via ALTER TABLE ... INHERIT. Report and patch by Amit Langote, reviewed by Ashutosh Bapat, some adjustments by me. It's been broken all along, so back-patch to all supported branches. Discussion: https://postgr.es/m/cb13cfe7-a48c-5720-c383-bb843ab28298@lab.ntt.co.jp
1 parent 83a25a5 commit 6c4cf2b

File tree

3 files changed

+108
-0
lines changed

3 files changed

+108
-0
lines changed

‎src/backend/commands/tablecmds.c

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9263,6 +9263,39 @@ MergeAttributesIntoExisting(Relation child_rel, Relation parent_rel)
92639263
}
92649264
}
92659265

9266+
/*
9267+
* If the parent has an OID column, so must the child, and we'd better
9268+
* update the child's attinhcount and attislocal the same as for normal
9269+
* columns. We needn't check data type or not-nullness though.
9270+
*/
9271+
if (tupleDesc->tdhasoid)
9272+
{
9273+
/*
9274+
* Here we match by column number not name; the match *must* be the
9275+
* system column, not some random column named "oid".
9276+
*/
9277+
tuple = SearchSysCacheCopy2(ATTNUM,
9278+
ObjectIdGetDatum(RelationGetRelid(child_rel)),
9279+
Int16GetDatum(ObjectIdAttributeNumber));
9280+
if (HeapTupleIsValid(tuple))
9281+
{
9282+
Form_pg_attribute childatt = (Form_pg_attribute) GETSTRUCT(tuple);
9283+
9284+
/* See comments above; these changes should be the same */
9285+
childatt->attinhcount++;
9286+
simple_heap_update(attrrel, &tuple->t_self, tuple);
9287+
CatalogUpdateIndexes(attrrel, tuple);
9288+
heap_freetuple(tuple);
9289+
}
9290+
else
9291+
{
9292+
ereport(ERROR,
9293+
(errcode(ERRCODE_DATATYPE_MISMATCH),
9294+
errmsg("child table is missing column \"%s\"",
9295+
"oid")));
9296+
}
9297+
}
9298+
92669299
heap_close(attrrel, RowExclusiveLock);
92679300
}
92689301

‎src/test/regress/expected/inherit.out

Lines changed: 49 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -613,6 +613,55 @@ select * from d;
613613
32 | one | two | three
614614
(1 row)
615615

616+
-- check that oid column is handled properly during alter table inherit
617+
create table oid_parent (a int) with oids;
618+
create table oid_child () inherits (oid_parent);
619+
select attinhcount, attislocal from pg_attribute
620+
where attrelid = 'oid_child'::regclass and attname = 'oid';
621+
attinhcount | attislocal
622+
-------------+------------
623+
1 | f
624+
(1 row)
625+
626+
drop table oid_child;
627+
create table oid_child (a int) without oids;
628+
alter table oid_child inherit oid_parent; -- fail
629+
ERROR: table "oid_child" without OIDs cannot inherit from table "oid_parent" with OIDs
630+
alter table oid_child set with oids;
631+
select attinhcount, attislocal from pg_attribute
632+
where attrelid = 'oid_child'::regclass and attname = 'oid';
633+
attinhcount | attislocal
634+
-------------+------------
635+
0 | t
636+
(1 row)
637+
638+
alter table oid_child inherit oid_parent;
639+
select attinhcount, attislocal from pg_attribute
640+
where attrelid = 'oid_child'::regclass and attname = 'oid';
641+
attinhcount | attislocal
642+
-------------+------------
643+
1 | t
644+
(1 row)
645+
646+
alter table oid_child set without oids; -- fail
647+
ERROR: cannot drop inherited column "oid"
648+
alter table oid_parent set without oids;
649+
select attinhcount, attislocal from pg_attribute
650+
where attrelid = 'oid_child'::regclass and attname = 'oid';
651+
attinhcount | attislocal
652+
-------------+------------
653+
0 | t
654+
(1 row)
655+
656+
alter table oid_child set without oids;
657+
select attinhcount, attislocal from pg_attribute
658+
where attrelid = 'oid_child'::regclass and attname = 'oid';
659+
attinhcount | attislocal
660+
-------------+------------
661+
(0 rows)
662+
663+
drop table oid_parent cascade;
664+
NOTICE: drop cascades to table oid_child
616665
-- Test non-inheritable parent constraints
617666
create table p1(ff1 int);
618667
alter table p1 add constraint p1chk check (ff1 > 0) no inherit;

‎src/test/regress/sql/inherit.sql

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -145,6 +145,32 @@ insert into d values('test','one','two','three');
145145
alter table a alter column aa type integer using bit_length(aa);
146146
select * from d;
147147

148+
-- check that oid column is handled properly during alter table inherit
149+
create table oid_parent (a int) with oids;
150+
151+
create table oid_child () inherits (oid_parent);
152+
select attinhcount, attislocal from pg_attribute
153+
where attrelid = 'oid_child'::regclass and attname = 'oid';
154+
drop table oid_child;
155+
156+
create table oid_child (a int) without oids;
157+
alter table oid_child inherit oid_parent; -- fail
158+
alter table oid_child set with oids;
159+
select attinhcount, attislocal from pg_attribute
160+
where attrelid = 'oid_child'::regclass and attname = 'oid';
161+
alter table oid_child inherit oid_parent;
162+
select attinhcount, attislocal from pg_attribute
163+
where attrelid = 'oid_child'::regclass and attname = 'oid';
164+
alter table oid_child set without oids; -- fail
165+
alter table oid_parent set without oids;
166+
select attinhcount, attislocal from pg_attribute
167+
where attrelid = 'oid_child'::regclass and attname = 'oid';
168+
alter table oid_child set without oids;
169+
select attinhcount, attislocal from pg_attribute
170+
where attrelid = 'oid_child'::regclass and attname = 'oid';
171+
172+
drop table oid_parent cascade;
173+
148174
-- Test non-inheritable parent constraints
149175
create table p1(ff1 int);
150176
alter table p1 add constraint p1chk check (ff1 > 0) no inherit;

0 commit comments

Comments
 (0)
0