PostgreSQL Source Code  git master
pg_publication.c File Reference
#include "postgres.h"
#include "access/genam.h"
#include "access/heapam.h"
#include "access/htup_details.h"
#include "access/tableam.h"
#include "catalog/catalog.h"
#include "catalog/dependency.h"
#include "catalog/indexing.h"
#include "catalog/namespace.h"
#include "catalog/objectaddress.h"
#include "catalog/partition.h"
#include "catalog/pg_inherits.h"
#include "catalog/pg_namespace.h"
#include "catalog/pg_publication.h"
#include "catalog/pg_publication_namespace.h"
#include "catalog/pg_publication_rel.h"
#include "catalog/pg_type.h"
#include "commands/publicationcmds.h"
#include "funcapi.h"
#include "utils/array.h"
#include "utils/builtins.h"
#include "utils/catcache.h"
#include "utils/fmgroids.h"
#include "utils/lsyscache.h"
#include "utils/rel.h"
#include "utils/syscache.h"
Include dependency graph for pg_publication.c:

Go to the source code of this file.

Data Structures

struct  published_rel
 

Macros

#define NUM_PUBLICATION_TABLES_ELEM   4
 

Functions

static void publication_translate_columns (Relation targetrel, List *columns, int *natts, AttrNumber **attrs)
 
static void check_publication_add_relation (Relation targetrel)
 
static void check_publication_add_schema (Oid schemaid)
 
static bool is_publishable_class (Oid relid, Form_pg_class reltuple)
 
bool is_publishable_relation (Relation rel)
 
Datum pg_relation_is_publishable (PG_FUNCTION_ARGS)
 
static bool is_ancestor_member_tableinfos (Oid ancestor, List *table_infos)
 
static void filter_partitions (List *table_infos)
 
bool is_schema_publication (Oid pubid)
 
ListGetPubPartitionOptionRelations (List *result, PublicationPartOpt pub_partopt, Oid relid)
 
Oid GetTopMostAncestorInPublication (Oid puboid, List *ancestors, int *ancestor_level)
 
ObjectAddress publication_add_relation (Oid pubid, PublicationRelInfo *pri, bool if_not_exists)
 
static int compare_int16 (const void *a, const void *b)
 
Bitmapsetpub_collist_to_bitmapset (Bitmapset *columns, Datum pubcols, MemoryContext mcxt)
 
ObjectAddress publication_add_schema (Oid pubid, Oid schemaid, bool if_not_exists)
 
ListGetRelationPublications (Oid relid)
 
ListGetPublicationRelations (Oid pubid, PublicationPartOpt pub_partopt)
 
ListGetAllTablesPublications (void)
 
ListGetAllTablesPublicationRelations (bool pubviaroot)
 
ListGetPublicationSchemas (Oid pubid)
 
ListGetSchemaPublications (Oid schemaid)
 
ListGetSchemaPublicationRelations (Oid schemaid, PublicationPartOpt pub_partopt)
 
ListGetAllSchemaPublicationRelations (Oid pubid, PublicationPartOpt pub_partopt)
 
PublicationGetPublication (Oid pubid)
 
PublicationGetPublicationByName (const char *pubname, bool missing_ok)
 
Datum pg_get_publication_tables (PG_FUNCTION_ARGS)
 

Macro Definition Documentation

◆ NUM_PUBLICATION_TABLES_ELEM

#define NUM_PUBLICATION_TABLES_ELEM   4

Function Documentation

◆ check_publication_add_relation()

static void check_publication_add_relation ( Relation  targetrel)
static

Definition at line 59 of file pg_publication.c.

60 {
61  /* Must be a regular or partitioned table */
62  if (RelationGetForm(targetrel)->relkind != RELKIND_RELATION &&
63  RelationGetForm(targetrel)->relkind != RELKIND_PARTITIONED_TABLE)
64  ereport(ERROR,
65  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
66  errmsg("cannot add relation \"%s\" to publication",
67  RelationGetRelationName(targetrel)),
68  errdetail_relkind_not_supported(RelationGetForm(targetrel)->relkind)));
69 
70  /* Can't be system table */
71  if (IsCatalogRelation(targetrel))
72  ereport(ERROR,
73  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
74  errmsg("cannot add relation \"%s\" to publication",
75  RelationGetRelationName(targetrel)),
76  errdetail("This operation is not supported for system tables.")));
77 
78  /* UNLOGGED and TEMP relations cannot be part of publication. */
79  if (targetrel->rd_rel->relpersistence == RELPERSISTENCE_TEMP)
80  ereport(ERROR,
81  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
82  errmsg("cannot add relation \"%s\" to publication",
83  RelationGetRelationName(targetrel)),
84  errdetail("This operation is not supported for temporary tables.")));
85  else if (targetrel->rd_rel->relpersistence == RELPERSISTENCE_UNLOGGED)
86  ereport(ERROR,
87  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
88  errmsg("cannot add relation \"%s\" to publication",
89  RelationGetRelationName(targetrel)),
90  errdetail("This operation is not supported for unlogged tables.")));
91 }
bool IsCatalogRelation(Relation relation)
Definition: catalog.c:103
int errdetail(const char *fmt,...)
Definition: elog.c:1205
int errcode(int sqlerrcode)
Definition: elog.c:859
int errmsg(const char *fmt,...)
Definition: elog.c:1072
#define ERROR
Definition: elog.h:39
#define ereport(elevel,...)
Definition: elog.h:149
int errdetail_relkind_not_supported(char relkind)
Definition: pg_class.c:24
#define RelationGetForm(relation)
Definition: rel.h:499
#define RelationGetRelationName(relation)
Definition: rel.h:539
Form_pg_class rd_rel
Definition: rel.h:111

References ereport, errcode(), errdetail(), errdetail_relkind_not_supported(), errmsg(), ERROR, IsCatalogRelation(), RelationData::rd_rel, RelationGetForm, and RelationGetRelationName.

Referenced by publication_add_relation().

◆ check_publication_add_schema()

static void check_publication_add_schema ( Oid  schemaid)
static

Definition at line 98 of file pg_publication.c.

99 {
100  /* Can't be system namespace */
101  if (IsCatalogNamespace(schemaid) || IsToastNamespace(schemaid))
102  ereport(ERROR,
103  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
104  errmsg("cannot add schema \"%s\" to publication",
105  get_namespace_name(schemaid)),
106  errdetail("This operation is not supported for system schemas.")));
107 
108  /* Can't be temporary namespace */
109  if (isAnyTempNamespace(schemaid))
110  ereport(ERROR,
111  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
112  errmsg("cannot add schema \"%s\" to publication",
113  get_namespace_name(schemaid)),
114  errdetail("Temporary schemas cannot be replicated.")));
115 }
bool IsToastNamespace(Oid namespaceId)
Definition: catalog.c:200
bool IsCatalogNamespace(Oid namespaceId)
Definition: catalog.c:182
char * get_namespace_name(Oid nspid)
Definition: lsyscache.c:3322
bool isAnyTempNamespace(Oid namespaceId)
Definition: namespace.c:3672

References ereport, errcode(), errdetail(), errmsg(), ERROR, get_namespace_name(), isAnyTempNamespace(), IsCatalogNamespace(), and IsToastNamespace().

Referenced by publication_add_schema().

◆ compare_int16()

static int compare_int16 ( const void *  a,
const void *  b 
)
static

Definition at line 481 of file pg_publication.c.

482 {
483  int av = *(const int16 *) a;
484  int bv = *(const int16 *) b;
485 
486  /* this can't overflow if int is wider than int16 */
487  return (av - bv);
488 }
signed short int16
Definition: c.h:480
int b
Definition: isn.c:70
int a
Definition: isn.c:69
struct @10::@11 av[32]

References a, av, and b.

Referenced by publication_translate_columns().

◆ filter_partitions()

static void filter_partitions ( List table_infos)
static

Definition at line 201 of file pg_publication.c.

202 {
203  ListCell *lc;
204 
205  foreach(lc, table_infos)
206  {
207  bool skip = false;
208  List *ancestors = NIL;
209  ListCell *lc2;
210  published_rel *table_info = (published_rel *) lfirst(lc);
211 
212  if (get_rel_relispartition(table_info->relid))
213  ancestors = get_partition_ancestors(table_info->relid);
214 
215  foreach(lc2, ancestors)
216  {
217  Oid ancestor = lfirst_oid(lc2);
218 
219  if (is_ancestor_member_tableinfos(ancestor, table_infos))
220  {
221  skip = true;
222  break;
223  }
224  }
225 
226  if (skip)
227  table_infos = foreach_delete_current(table_infos, lc);
228  }
229 }
bool get_rel_relispartition(Oid relid)
Definition: lsyscache.c:2005
List * get_partition_ancestors(Oid relid)
Definition: partition.c:134
static const struct exclude_list_item skip[]
Definition: pg_checksums.c:108
#define lfirst(lc)
Definition: pg_list.h:172
#define NIL
Definition: pg_list.h:68
#define foreach_delete_current(lst, var_or_cell)
Definition: pg_list.h:391
#define lfirst_oid(lc)
Definition: pg_list.h:174
static bool is_ancestor_member_tableinfos(Oid ancestor, List *table_infos)
unsigned int Oid
Definition: postgres_ext.h:31
Definition: pg_list.h:54

References foreach_delete_current, get_partition_ancestors(), get_rel_relispartition(), is_ancestor_member_tableinfos(), lfirst, lfirst_oid, NIL, published_rel::relid, and skip.

Referenced by pg_get_publication_tables().

◆ GetAllSchemaPublicationRelations()

List* GetAllSchemaPublicationRelations ( Oid  pubid,
PublicationPartOpt  pub_partopt 
)

Definition at line 982 of file pg_publication.c.

983 {
984  List *result = NIL;
985  List *pubschemalist = GetPublicationSchemas(pubid);
986  ListCell *cell;
987 
988  foreach(cell, pubschemalist)
989  {
990  Oid schemaid = lfirst_oid(cell);
991  List *schemaRels = NIL;
992 
993  schemaRels = GetSchemaPublicationRelations(schemaid, pub_partopt);
994  result = list_concat(result, schemaRels);
995  }
996 
997  return result;
998 }
List * list_concat(List *list1, const List *list2)
Definition: list.c:561
List * GetPublicationSchemas(Oid pubid)
List * GetSchemaPublicationRelations(Oid schemaid, PublicationPartOpt pub_partopt)

References GetPublicationSchemas(), GetSchemaPublicationRelations(), lfirst_oid, list_concat(), and NIL.

Referenced by AlterPublicationOptions(), and pg_get_publication_tables().

◆ GetAllTablesPublicationRelations()

List* GetAllTablesPublicationRelations ( bool  pubviaroot)

Definition at line 800 of file pg_publication.c.

801 {
802  Relation classRel;
803  ScanKeyData key[1];
804  TableScanDesc scan;
805  HeapTuple tuple;
806  List *result = NIL;
807 
808  classRel = table_open(RelationRelationId, AccessShareLock);
809 
810  ScanKeyInit(&key[0],
811  Anum_pg_class_relkind,
812  BTEqualStrategyNumber, F_CHAREQ,
813  CharGetDatum(RELKIND_RELATION));
814 
815  scan = table_beginscan_catalog(classRel, 1, key);
816 
817  while ((tuple = heap_getnext(scan, ForwardScanDirection)) != NULL)
818  {
819  Form_pg_class relForm = (Form_pg_class) GETSTRUCT(tuple);
820  Oid relid = relForm->oid;
821 
822  if (is_publishable_class(relid, relForm) &&
823  !(relForm->relispartition && pubviaroot))
824  result = lappend_oid(result, relid);
825  }
826 
827  table_endscan(scan);
828 
829  if (pubviaroot)
830  {
831  ScanKeyInit(&key[0],
832  Anum_pg_class_relkind,
833  BTEqualStrategyNumber, F_CHAREQ,
834  CharGetDatum(RELKIND_PARTITIONED_TABLE));
835 
836  scan = table_beginscan_catalog(classRel, 1, key);
837 
838  while ((tuple = heap_getnext(scan, ForwardScanDirection)) != NULL)
839  {
840  Form_pg_class relForm = (Form_pg_class) GETSTRUCT(tuple);
841  Oid relid = relForm->oid;
842 
843  if (is_publishable_class(relid, relForm) &&
844  !relForm->relispartition)
845  result = lappend_oid(result, relid);
846  }
847 
848  table_endscan(scan);
849  }
850 
851  table_close(classRel, AccessShareLock);
852  return result;
853 }
HeapTuple heap_getnext(TableScanDesc sscan, ScanDirection direction)
Definition: heapam.c:1082
#define GETSTRUCT(TUP)
Definition: htup_details.h:653
List * lappend_oid(List *list, Oid datum)
Definition: list.c:375
#define AccessShareLock
Definition: lockdefs.h:36
FormData_pg_class * Form_pg_class
Definition: pg_class.h:153
static bool is_publishable_class(Oid relid, Form_pg_class reltuple)
static Datum CharGetDatum(char X)
Definition: postgres.h:122
void ScanKeyInit(ScanKey entry, AttrNumber attributeNumber, StrategyNumber strategy, RegProcedure procedure, Datum argument)
Definition: scankey.c:76
@ ForwardScanDirection
Definition: sdir.h:28
#define BTEqualStrategyNumber
Definition: stratnum.h:31
void table_close(Relation relation, LOCKMODE lockmode)
Definition: table.c:126
Relation table_open(Oid relationId, LOCKMODE lockmode)
Definition: table.c:40
TableScanDesc table_beginscan_catalog(Relation relation, int nkeys, struct ScanKeyData *key)
Definition: tableam.c:112
static void table_endscan(TableScanDesc scan)
Definition: tableam.h:1009

References AccessShareLock, BTEqualStrategyNumber, CharGetDatum(), ForwardScanDirection, GETSTRUCT, heap_getnext(), is_publishable_class(), sort-test::key, lappend_oid(), NIL, ScanKeyInit(), table_beginscan_catalog(), table_close(), table_endscan(), and table_open().

Referenced by pg_get_publication_tables().

◆ GetAllTablesPublications()

List* GetAllTablesPublications ( void  )

Definition at line 759 of file pg_publication.c.

760 {
761  List *result;
762  Relation rel;
763  ScanKeyData scankey;
764  SysScanDesc scan;
765  HeapTuple tup;
766 
767  /* Find all publications that are marked as for all tables. */
768  rel = table_open(PublicationRelationId, AccessShareLock);
769 
770  ScanKeyInit(&scankey,
771  Anum_pg_publication_puballtables,
772  BTEqualStrategyNumber, F_BOOLEQ,
773  BoolGetDatum(true));
774 
775  scan = systable_beginscan(rel, InvalidOid, false,
776  NULL, 1, &scankey);
777 
778  result = NIL;
779  while (HeapTupleIsValid(tup = systable_getnext(scan)))
780  {
781  Oid oid = ((Form_pg_publication) GETSTRUCT(tup))->oid;
782 
783  result = lappend_oid(result, oid);
784  }
785 
786  systable_endscan(scan);
788 
789  return result;
790 }
void systable_endscan(SysScanDesc sysscan)
Definition: genam.c:596
HeapTuple systable_getnext(SysScanDesc sysscan)
Definition: genam.c:503
SysScanDesc systable_beginscan(Relation heapRelation, Oid indexId, bool indexOK, Snapshot snapshot, int nkeys, ScanKey key)
Definition: genam.c:384
#define HeapTupleIsValid(tuple)
Definition: htup.h:78
FormData_pg_publication * Form_pg_publication
static Datum BoolGetDatum(bool X)
Definition: postgres.h:102
#define InvalidOid
Definition: postgres_ext.h:36

References AccessShareLock, BoolGetDatum(), BTEqualStrategyNumber, GETSTRUCT, HeapTupleIsValid, InvalidOid, lappend_oid(), NIL, ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by RelationBuildPublicationDesc().

◆ GetPublication()

Publication* GetPublication ( Oid  pubid)

Definition at line 1006 of file pg_publication.c.

1007 {
1008  HeapTuple tup;
1009  Publication *pub;
1010  Form_pg_publication pubform;
1011 
1012  tup = SearchSysCache1(PUBLICATIONOID, ObjectIdGetDatum(pubid));
1013  if (!HeapTupleIsValid(tup))
1014  elog(ERROR, "cache lookup failed for publication %u", pubid);
1015 
1016  pubform = (Form_pg_publication) GETSTRUCT(tup);
1017 
1018  pub = (Publication *) palloc(sizeof(Publication));
1019  pub->oid = pubid;
1020  pub->name = pstrdup(NameStr(pubform->pubname));
1021  pub->alltables = pubform->puballtables;
1022  pub->pubactions.pubinsert = pubform->pubinsert;
1023  pub->pubactions.pubupdate = pubform->pubupdate;
1024  pub->pubactions.pubdelete = pubform->pubdelete;
1025  pub->pubactions.pubtruncate = pubform->pubtruncate;
1026  pub->pubviaroot = pubform->pubviaroot;
1027 
1028  ReleaseSysCache(tup);
1029 
1030  return pub;
1031 }
#define NameStr(name)
Definition: c.h:733
#define elog(elevel,...)
Definition: elog.h:224
char * pstrdup(const char *in)
Definition: mcxt.c:1683
void * palloc(Size size)
Definition: mcxt.c:1304
static Datum ObjectIdGetDatum(Oid X)
Definition: postgres.h:252
PublicationActions pubactions
void ReleaseSysCache(HeapTuple tuple)
Definition: syscache.c:266
HeapTuple SearchSysCache1(int cacheId, Datum key1)
Definition: syscache.c:218

References Publication::alltables, elog, ERROR, GETSTRUCT, HeapTupleIsValid, Publication::name, NameStr, ObjectIdGetDatum(), Publication::oid, palloc(), pstrdup(), Publication::pubactions, PublicationActions::pubdelete, PublicationActions::pubinsert, PublicationActions::pubtruncate, PublicationActions::pubupdate, Publication::pubviaroot, ReleaseSysCache(), and SearchSysCache1().

Referenced by GetPublicationByName(), pg_get_publication_tables(), publication_add_relation(), and publication_add_schema().

◆ GetPublicationByName()

Publication* GetPublicationByName ( const char *  pubname,
bool  missing_ok 
)

Definition at line 1037 of file pg_publication.c.

1038 {
1039  Oid oid;
1040 
1041  oid = get_publication_oid(pubname, missing_ok);
1042 
1043  return OidIsValid(oid) ? GetPublication(oid) : NULL;
1044 }
#define OidIsValid(objectId)
Definition: c.h:762
Oid get_publication_oid(const char *pubname, bool missing_ok)
Definition: lsyscache.c:3581
Publication * GetPublication(Oid pubid)

References get_publication_oid(), GetPublication(), and OidIsValid.

Referenced by get_object_address_publication_rel(), get_object_address_publication_schema(), LoadPublications(), and pg_get_publication_tables().

◆ GetPublicationRelations()

List* GetPublicationRelations ( Oid  pubid,
PublicationPartOpt  pub_partopt 
)

Definition at line 716 of file pg_publication.c.

717 {
718  List *result;
719  Relation pubrelsrel;
720  ScanKeyData scankey;
721  SysScanDesc scan;
722  HeapTuple tup;
723 
724  /* Find all publications associated with the relation. */
725  pubrelsrel = table_open(PublicationRelRelationId, AccessShareLock);
726 
727  ScanKeyInit(&scankey,
728  Anum_pg_publication_rel_prpubid,
729  BTEqualStrategyNumber, F_OIDEQ,
730  ObjectIdGetDatum(pubid));
731 
732  scan = systable_beginscan(pubrelsrel, PublicationRelPrpubidIndexId,
733  true, NULL, 1, &scankey);
734 
735  result = NIL;
736  while (HeapTupleIsValid(tup = systable_getnext(scan)))
737  {
739 
740  pubrel = (Form_pg_publication_rel) GETSTRUCT(tup);
741  result = GetPubPartitionOptionRelations(result, pub_partopt,
742  pubrel->prrelid);
743  }
744 
745  systable_endscan(scan);
746  table_close(pubrelsrel, AccessShareLock);
747 
748  /* Now sort and de-duplicate the result list */
749  list_sort(result, list_oid_cmp);
750  list_deduplicate_oid(result);
751 
752  return result;
753 }
void list_sort(List *list, list_sort_comparator cmp)
Definition: list.c:1674
void list_deduplicate_oid(List *list)
Definition: list.c:1495
int list_oid_cmp(const ListCell *p1, const ListCell *p2)
Definition: list.c:1703
List * GetPubPartitionOptionRelations(List *result, PublicationPartOpt pub_partopt, Oid relid)
FormData_pg_publication_rel * Form_pg_publication_rel

References AccessShareLock, BTEqualStrategyNumber, GetPubPartitionOptionRelations(), GETSTRUCT, HeapTupleIsValid, list_deduplicate_oid(), list_oid_cmp(), list_sort(), NIL, ObjectIdGetDatum(), ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by AlterPublicationOptions(), AlterPublicationSchemas(), AlterPublicationTables(), and pg_get_publication_tables().

◆ GetPublicationSchemas()

List* GetPublicationSchemas ( Oid  pubid)

Definition at line 861 of file pg_publication.c.

862 {
863  List *result = NIL;
864  Relation pubschsrel;
865  ScanKeyData scankey;
866  SysScanDesc scan;
867  HeapTuple tup;
868 
869  /* Find all schemas associated with the publication */
870  pubschsrel = table_open(PublicationNamespaceRelationId, AccessShareLock);
871 
872  ScanKeyInit(&scankey,
873  Anum_pg_publication_namespace_pnpubid,
874  BTEqualStrategyNumber, F_OIDEQ,
875  ObjectIdGetDatum(pubid));
876 
877  scan = systable_beginscan(pubschsrel,
878  PublicationNamespacePnnspidPnpubidIndexId,
879  true, NULL, 1, &scankey);
880  while (HeapTupleIsValid(tup = systable_getnext(scan)))
881  {
883 
885 
886  result = lappend_oid(result, pubsch->pnnspid);
887  }
888 
889  systable_endscan(scan);
890  table_close(pubschsrel, AccessShareLock);
891 
892  return result;
893 }
FormData_pg_publication_namespace * Form_pg_publication_namespace

References AccessShareLock, BTEqualStrategyNumber, GETSTRUCT, HeapTupleIsValid, lappend_oid(), NIL, ObjectIdGetDatum(), ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by AlterPublicationSchemas(), and GetAllSchemaPublicationRelations().

◆ GetPubPartitionOptionRelations()

List* GetPubPartitionOptionRelations ( List result,
PublicationPartOpt  pub_partopt,
Oid  relid 
)

Definition at line 267 of file pg_publication.c.

269 {
270  if (get_rel_relkind(relid) == RELKIND_PARTITIONED_TABLE &&
271  pub_partopt != PUBLICATION_PART_ROOT)
272  {
273  List *all_parts = find_all_inheritors(relid, NoLock,
274  NULL);
275 
276  if (pub_partopt == PUBLICATION_PART_ALL)
277  result = list_concat(result, all_parts);
278  else if (pub_partopt == PUBLICATION_PART_LEAF)
279  {
280  ListCell *lc;
281 
282  foreach(lc, all_parts)
283  {
284  Oid partOid = lfirst_oid(lc);
285 
286  if (get_rel_relkind(partOid) != RELKIND_PARTITIONED_TABLE)
287  result = lappend_oid(result, partOid);
288  }
289  }
290  else
291  Assert(false);
292  }
293  else
294  result = lappend_oid(result, relid);
295 
296  return result;
297 }
Assert(fmt[strlen(fmt) - 1] !='\n')
#define NoLock
Definition: lockdefs.h:34
char get_rel_relkind(Oid relid)
Definition: lsyscache.c:1981
List * find_all_inheritors(Oid parentrelId, LOCKMODE lockmode, List **numparents)
Definition: pg_inherits.c:255
@ PUBLICATION_PART_LEAF
@ PUBLICATION_PART_ROOT
@ PUBLICATION_PART_ALL

References Assert(), find_all_inheritors(), get_rel_relkind(), lappend_oid(), lfirst_oid, list_concat(), NoLock, PUBLICATION_PART_ALL, PUBLICATION_PART_LEAF, and PUBLICATION_PART_ROOT.

Referenced by AlterPublicationOptions(), GetPublicationRelations(), GetSchemaPublicationRelations(), publication_add_relation(), and RemovePublicationRelById().

◆ GetRelationPublications()

List* GetRelationPublications ( Oid  relid)

Definition at line 687 of file pg_publication.c.

688 {
689  List *result = NIL;
690  CatCList *pubrellist;
691  int i;
692 
693  /* Find all publications associated with the relation. */
694  pubrellist = SearchSysCacheList1(PUBLICATIONRELMAP,
695  ObjectIdGetDatum(relid));
696  for (i = 0; i < pubrellist->n_members; i++)
697  {
698  HeapTuple tup = &pubrellist->members[i]->tuple;
699  Oid pubid = ((Form_pg_publication_rel) GETSTRUCT(tup))->prpubid;
700 
701  result = lappend_oid(result, pubid);
702  }
703 
704  ReleaseSysCacheList(pubrellist);
705 
706  return result;
707 }
int i
Definition: isn.c:73
CatCTup * members[FLEXIBLE_ARRAY_MEMBER]
Definition: catcache.h:178
int n_members
Definition: catcache.h:176
HeapTupleData tuple
Definition: catcache.h:121
#define ReleaseSysCacheList(x)
Definition: syscache.h:129
#define SearchSysCacheList1(cacheId, key1)
Definition: syscache.h:122

References GETSTRUCT, i, lappend_oid(), catclist::members, catclist::n_members, NIL, ObjectIdGetDatum(), ReleaseSysCacheList, SearchSysCacheList1, and catctup::tuple.

Referenced by ATPrepChangePersistence(), get_rel_sync_entry(), GetTopMostAncestorInPublication(), and RelationBuildPublicationDesc().

◆ GetSchemaPublicationRelations()

List* GetSchemaPublicationRelations ( Oid  schemaid,
PublicationPartOpt  pub_partopt 
)

Definition at line 925 of file pg_publication.c.

926 {
927  Relation classRel;
928  ScanKeyData key[1];
929  TableScanDesc scan;
930  HeapTuple tuple;
931  List *result = NIL;
932 
933  Assert(OidIsValid(schemaid));
934 
935  classRel = table_open(RelationRelationId, AccessShareLock);
936 
937  ScanKeyInit(&key[0],
938  Anum_pg_class_relnamespace,
939  BTEqualStrategyNumber, F_OIDEQ,
940  schemaid);
941 
942  /* get all the relations present in the specified schema */
943  scan = table_beginscan_catalog(classRel, 1, key);
944  while ((tuple = heap_getnext(scan, ForwardScanDirection)) != NULL)
945  {
946  Form_pg_class relForm = (Form_pg_class) GETSTRUCT(tuple);
947  Oid relid = relForm->oid;
948  char relkind;
949 
950  if (!is_publishable_class(relid, relForm))
951  continue;
952 
953  relkind = get_rel_relkind(relid);
954  if (relkind == RELKIND_RELATION)
955  result = lappend_oid(result, relid);
956  else if (relkind == RELKIND_PARTITIONED_TABLE)
957  {
958  List *partitionrels = NIL;
959 
960  /*
961  * It is quite possible that some of the partitions are in a
962  * different schema than the parent table, so we need to get such
963  * partitions separately.
964  */
965  partitionrels = GetPubPartitionOptionRelations(partitionrels,
966  pub_partopt,
967  relForm->oid);
968  result = list_concat_unique_oid(result, partitionrels);
969  }
970  }
971 
972  table_endscan(scan);
973  table_close(classRel, AccessShareLock);
974  return result;
975 }
List * list_concat_unique_oid(List *list1, const List *list2)
Definition: list.c:1469

References AccessShareLock, Assert(), BTEqualStrategyNumber, ForwardScanDirection, get_rel_relkind(), GetPubPartitionOptionRelations(), GETSTRUCT, heap_getnext(), is_publishable_class(), sort-test::key, lappend_oid(), list_concat_unique_oid(), NIL, OidIsValid, ScanKeyInit(), table_beginscan_catalog(), table_close(), table_endscan(), and table_open().

Referenced by GetAllSchemaPublicationRelations(), publication_add_schema(), and RemovePublicationSchemaById().

◆ GetSchemaPublications()

List* GetSchemaPublications ( Oid  schemaid)

Definition at line 899 of file pg_publication.c.

900 {
901  List *result = NIL;
902  CatCList *pubschlist;
903  int i;
904 
905  /* Find all publications associated with the schema */
906  pubschlist = SearchSysCacheList1(PUBLICATIONNAMESPACEMAP,
907  ObjectIdGetDatum(schemaid));
908  for (i = 0; i < pubschlist->n_members; i++)
909  {
910  HeapTuple tup = &pubschlist->members[i]->tuple;
911  Oid pubid = ((Form_pg_publication_namespace) GETSTRUCT(tup))->pnpubid;
912 
913  result = lappend_oid(result, pubid);
914  }
915 
916  ReleaseSysCacheList(pubschlist);
917 
918  return result;
919 }

References GETSTRUCT, i, lappend_oid(), catclist::members, catclist::n_members, NIL, ObjectIdGetDatum(), ReleaseSysCacheList, SearchSysCacheList1, and catctup::tuple.

Referenced by get_rel_sync_entry(), GetTopMostAncestorInPublication(), and RelationBuildPublicationDesc().

◆ GetTopMostAncestorInPublication()

Oid GetTopMostAncestorInPublication ( Oid  puboid,
List ancestors,
int *  ancestor_level 
)

Definition at line 311 of file pg_publication.c.

312 {
313  ListCell *lc;
314  Oid topmost_relid = InvalidOid;
315  int level = 0;
316 
317  /*
318  * Find the "topmost" ancestor that is in this publication.
319  */
320  foreach(lc, ancestors)
321  {
322  Oid ancestor = lfirst_oid(lc);
323  List *apubids = GetRelationPublications(ancestor);
324  List *aschemaPubids = NIL;
325 
326  level++;
327 
328  if (list_member_oid(apubids, puboid))
329  {
330  topmost_relid = ancestor;
331 
332  if (ancestor_level)
333  *ancestor_level = level;
334  }
335  else
336  {
337  aschemaPubids = GetSchemaPublications(get_rel_namespace(ancestor));
338  if (list_member_oid(aschemaPubids, puboid))
339  {
340  topmost_relid = ancestor;
341 
342  if (ancestor_level)
343  *ancestor_level = level;
344  }
345  }
346 
347  list_free(apubids);
348  list_free(aschemaPubids);
349  }
350 
351  return topmost_relid;
352 }
void list_free(List *list)
Definition: list.c:1546
bool list_member_oid(const List *list, Oid datum)
Definition: list.c:722
Oid get_rel_namespace(Oid relid)
Definition: lsyscache.c:1930
List * GetSchemaPublications(Oid schemaid)
List * GetRelationPublications(Oid relid)

References get_rel_namespace(), GetRelationPublications(), GetSchemaPublications(), InvalidOid, lfirst_oid, list_free(), list_member_oid(), and NIL.

Referenced by get_rel_sync_entry(), pub_collist_contains_invalid_column(), and pub_rf_contains_invalid_column().

◆ is_ancestor_member_tableinfos()

static bool is_ancestor_member_tableinfos ( Oid  ancestor,
List table_infos 
)
static

Definition at line 182 of file pg_publication.c.

183 {
184  ListCell *lc;
185 
186  foreach(lc, table_infos)
187  {
188  Oid relid = ((published_rel *) lfirst(lc))->relid;
189 
190  if (relid == ancestor)
191  return true;
192  }
193 
194  return false;
195 }

References lfirst.

Referenced by filter_partitions().

◆ is_publishable_class()

static bool is_publishable_class ( Oid  relid,
Form_pg_class  reltuple 
)
static

Definition at line 137 of file pg_publication.c.

138 {
139  return (reltuple->relkind == RELKIND_RELATION ||
140  reltuple->relkind == RELKIND_PARTITIONED_TABLE) &&
141  !IsCatalogRelationOid(relid) &&
142  reltuple->relpersistence == RELPERSISTENCE_PERMANENT &&
143  relid >= FirstNormalObjectId;
144 }
bool IsCatalogRelationOid(Oid relid)
Definition: catalog.c:120
#define FirstNormalObjectId
Definition: transam.h:197

References FirstNormalObjectId, and IsCatalogRelationOid().

Referenced by GetAllTablesPublicationRelations(), GetSchemaPublicationRelations(), is_publishable_relation(), and pg_relation_is_publishable().

◆ is_publishable_relation()

bool is_publishable_relation ( Relation  rel)

Definition at line 150 of file pg_publication.c.

151 {
152  return is_publishable_class(RelationGetRelid(rel), rel->rd_rel);
153 }
#define RelationGetRelid(relation)
Definition: rel.h:505

References is_publishable_class(), RelationData::rd_rel, and RelationGetRelid.

Referenced by pgoutput_change(), pgoutput_truncate(), and RelationBuildPublicationDesc().

◆ is_schema_publication()

bool is_schema_publication ( Oid  pubid)

Definition at line 236 of file pg_publication.c.

237 {
238  Relation pubschsrel;
239  ScanKeyData scankey;
240  SysScanDesc scan;
241  HeapTuple tup;
242  bool result = false;
243 
244  pubschsrel = table_open(PublicationNamespaceRelationId, AccessShareLock);
245  ScanKeyInit(&scankey,
246  Anum_pg_publication_namespace_pnpubid,
247  BTEqualStrategyNumber, F_OIDEQ,
248  ObjectIdGetDatum(pubid));
249 
250  scan = systable_beginscan(pubschsrel,
251  PublicationNamespacePnnspidPnpubidIndexId,
252  true, NULL, 1, &scankey);
253  tup = systable_getnext(scan);
254  result = HeapTupleIsValid(tup);
255 
256  systable_endscan(scan);
257  table_close(pubschsrel, AccessShareLock);
258 
259  return result;
260 }

References AccessShareLock, BTEqualStrategyNumber, HeapTupleIsValid, ObjectIdGetDatum(), ScanKeyInit(), systable_beginscan(), systable_endscan(), systable_getnext(), table_close(), and table_open().

Referenced by AlterPublicationOwner_internal(), and AlterPublicationTables().

◆ pg_get_publication_tables()

Datum pg_get_publication_tables ( PG_FUNCTION_ARGS  )

Definition at line 1052 of file pg_publication.c.

1053 {
1054 #define NUM_PUBLICATION_TABLES_ELEM 4
1055  FuncCallContext *funcctx;
1056  List *table_infos = NIL;
1057 
1058  /* stuff done only on the first call of the function */
1059  if (SRF_IS_FIRSTCALL())
1060  {
1061  TupleDesc tupdesc;
1062  MemoryContext oldcontext;
1063  ArrayType *arr;
1064  Datum *elems;
1065  int nelems,
1066  i;
1067  bool viaroot = false;
1068 
1069  /* create a function context for cross-call persistence */
1070  funcctx = SRF_FIRSTCALL_INIT();
1071 
1072  /* switch to memory context appropriate for multiple function calls */
1073  oldcontext = MemoryContextSwitchTo(funcctx->multi_call_memory_ctx);
1074 
1075  /*
1076  * Deconstruct the parameter into elements where each element is a
1077  * publication name.
1078  */
1079  arr = PG_GETARG_ARRAYTYPE_P(0);
1080  deconstruct_array(arr, TEXTOID, -1, false, TYPALIGN_INT,
1081  &elems, NULL, &nelems);
1082 
1083  /* Get Oids of tables from each publication. */
1084  for (i = 0; i < nelems; i++)
1085  {
1086  Publication *pub_elem;
1087  List *pub_elem_tables = NIL;
1088  ListCell *lc;
1089 
1090  pub_elem = GetPublicationByName(TextDatumGetCString(elems[i]), false);
1091 
1092  /*
1093  * Publications support partitioned tables. If
1094  * publish_via_partition_root is false, all changes are replicated
1095  * using leaf partition identity and schema, so we only need
1096  * those. Otherwise, get the partitioned table itself.
1097  */
1098  if (pub_elem->alltables)
1099  pub_elem_tables = GetAllTablesPublicationRelations(pub_elem->pubviaroot);
1100  else
1101  {
1102  List *relids,
1103  *schemarelids;
1104 
1105  relids = GetPublicationRelations(pub_elem->oid,
1106  pub_elem->pubviaroot ?
1109  schemarelids = GetAllSchemaPublicationRelations(pub_elem->oid,
1110  pub_elem->pubviaroot ?
1113  pub_elem_tables = list_concat_unique_oid(relids, schemarelids);
1114  }
1115 
1116  /*
1117  * Record the published table and the corresponding publication so
1118  * that we can get row filters and column lists later.
1119  *
1120  * When a table is published by multiple publications, to obtain
1121  * all row filters and column lists, the structure related to this
1122  * table will be recorded multiple times.
1123  */
1124  foreach(lc, pub_elem_tables)
1125  {
1126  published_rel *table_info = (published_rel *) palloc(sizeof(published_rel));
1127 
1128  table_info->relid = lfirst_oid(lc);
1129  table_info->pubid = pub_elem->oid;
1130  table_infos = lappend(table_infos, table_info);
1131  }
1132 
1133  /* At least one publication is using publish_via_partition_root. */
1134  if (pub_elem->pubviaroot)
1135  viaroot = true;
1136  }
1137 
1138  /*
1139  * If the publication publishes partition changes via their respective
1140  * root partitioned tables, we must exclude partitions in favor of
1141  * including the root partitioned tables. Otherwise, the function
1142  * could return both the child and parent tables which could cause
1143  * data of the child table to be double-published on the subscriber
1144  * side.
1145  */
1146  if (viaroot)
1147  filter_partitions(table_infos);
1148 
1149  /* Construct a tuple descriptor for the result rows. */
1151  TupleDescInitEntry(tupdesc, (AttrNumber) 1, "pubid",
1152  OIDOID, -1, 0);
1153  TupleDescInitEntry(tupdesc, (AttrNumber) 2, "relid",
1154  OIDOID, -1, 0);
1155  TupleDescInitEntry(tupdesc, (AttrNumber) 3, "attrs",
1156  INT2VECTOROID, -1, 0);
1157  TupleDescInitEntry(tupdesc, (AttrNumber) 4, "qual",
1158  PG_NODE_TREEOID, -1, 0);
1159 
1160  funcctx->tuple_desc = BlessTupleDesc(tupdesc);
1161  funcctx->user_fctx = (void *) table_infos;
1162 
1163  MemoryContextSwitchTo(oldcontext);
1164  }
1165 
1166  /* stuff done on every call of the function */
1167  funcctx = SRF_PERCALL_SETUP();
1168  table_infos = (List *) funcctx->user_fctx;
1169 
1170  if (funcctx->call_cntr < list_length(table_infos))
1171  {
1172  HeapTuple pubtuple = NULL;
1173  HeapTuple rettuple;
1174  Publication *pub;
1175  published_rel *table_info = (published_rel *) list_nth(table_infos, funcctx->call_cntr);
1176  Oid relid = table_info->relid;
1177  Oid schemaid = get_rel_namespace(relid);
1179  bool nulls[NUM_PUBLICATION_TABLES_ELEM] = {0};
1180 
1181  /*
1182  * Form tuple with appropriate data.
1183  */
1184 
1185  pub = GetPublication(table_info->pubid);
1186 
1187  values[0] = ObjectIdGetDatum(pub->oid);
1188  values[1] = ObjectIdGetDatum(relid);
1189 
1190  /*
1191  * We don't consider row filters or column lists for FOR ALL TABLES or
1192  * FOR TABLES IN SCHEMA publications.
1193  */
1194  if (!pub->alltables &&
1195  !SearchSysCacheExists2(PUBLICATIONNAMESPACEMAP,
1196  ObjectIdGetDatum(schemaid),
1197  ObjectIdGetDatum(pub->oid)))
1198  pubtuple = SearchSysCacheCopy2(PUBLICATIONRELMAP,
1199  ObjectIdGetDatum(relid),
1200  ObjectIdGetDatum(pub->oid));
1201 
1202  if (HeapTupleIsValid(pubtuple))
1203  {
1204  /* Lookup the column list attribute. */
1205  values[2] = SysCacheGetAttr(PUBLICATIONRELMAP, pubtuple,
1206  Anum_pg_publication_rel_prattrs,
1207  &(nulls[2]));
1208 
1209  /* Null indicates no filter. */
1210  values[3] = SysCacheGetAttr(PUBLICATIONRELMAP, pubtuple,
1211  Anum_pg_publication_rel_prqual,
1212  &(nulls[3]));
1213  }
1214  else
1215  {
1216  nulls[2] = true;
1217  nulls[3] = true;
1218  }
1219 
1220  /* Show all columns when the column list is not specified. */
1221  if (nulls[2])
1222  {
1223  Relation rel = table_open(relid, AccessShareLock);
1224  int nattnums = 0;
1225  int16 *attnums;
1226  TupleDesc desc = RelationGetDescr(rel);
1227  int i;
1228 
1229  attnums = (int16 *) palloc(desc->natts * sizeof(int16));
1230 
1231  for (i = 0; i < desc->natts; i++)
1232  {
1233  Form_pg_attribute att = TupleDescAttr(desc, i);
1234 
1235  if (att->attisdropped || att->attgenerated)
1236  continue;
1237 
1238  attnums[nattnums++] = att->attnum;
1239  }
1240 
1241  if (nattnums > 0)
1242  {
1243  values[2] = PointerGetDatum(buildint2vector(attnums, nattnums));
1244  nulls[2] = false;
1245  }
1246 
1248  }
1249 
1250  rettuple = heap_form_tuple(funcctx->tuple_desc, values, nulls);
1251 
1252  SRF_RETURN_NEXT(funcctx, HeapTupleGetDatum(rettuple));
1253  }
1254 
1255  SRF_RETURN_DONE(funcctx);
1256 }
#define PG_GETARG_ARRAYTYPE_P(n)
Definition: array.h:263
void deconstruct_array(ArrayType *array, Oid elmtype, int elmlen, bool elmbyval, char elmalign, Datum **elemsp, bool **nullsp, int *nelemsp)
Definition: arrayfuncs.c:3612
int16 AttrNumber
Definition: attnum.h:21
static Datum values[MAXATTR]
Definition: bootstrap.c:152
#define TextDatumGetCString(d)
Definition: builtins.h:98
TupleDesc BlessTupleDesc(TupleDesc tupdesc)
Definition: execTuples.c:2070
#define SRF_IS_FIRSTCALL()
Definition: funcapi.h:304
#define SRF_PERCALL_SETUP()
Definition: funcapi.h:308
#define SRF_RETURN_NEXT(_funcctx, _result)
Definition: funcapi.h:310
#define SRF_FIRSTCALL_INIT()
Definition: funcapi.h:306
static Datum HeapTupleGetDatum(const HeapTupleData *tuple)
Definition: funcapi.h:230
#define SRF_RETURN_DONE(_funcctx)
Definition: funcapi.h:328
HeapTuple heap_form_tuple(TupleDesc tupleDescriptor, const Datum *values, const bool *isnull)
Definition: heaptuple.c:1116
int2vector * buildint2vector(const int16 *int2s, int n)
Definition: int.c:114
if(TABLE==NULL||TABLE_index==NULL)
Definition: isn.c:77
List * lappend(List *list, void *datum)
Definition: list.c:339
static MemoryContext MemoryContextSwitchTo(MemoryContext context)
Definition: palloc.h:124
FormData_pg_attribute * Form_pg_attribute
Definition: pg_attribute.h:209
static int list_length(const List *l)
Definition: pg_list.h:152
static void * list_nth(const List *list, int n)
Definition: pg_list.h:299
Publication * GetPublicationByName(const char *pubname, bool missing_ok)
List * GetPublicationRelations(Oid pubid, PublicationPartOpt pub_partopt)
static void filter_partitions(List *table_infos)
List * GetAllTablesPublicationRelations(bool pubviaroot)
#define NUM_PUBLICATION_TABLES_ELEM
List * GetAllSchemaPublicationRelations(Oid pubid, PublicationPartOpt pub_partopt)
static Datum PointerGetDatum(const void *X)
Definition: postgres.h:322
uintptr_t Datum
Definition: postgres.h:64
#define RelationGetDescr(relation)
Definition: rel.h:531
void * user_fctx
Definition: funcapi.h:82
uint64 call_cntr
Definition: funcapi.h:65
MemoryContext multi_call_memory_ctx
Definition: funcapi.h:101
TupleDesc tuple_desc
Definition: funcapi.h:112
Datum SysCacheGetAttr(int cacheId, HeapTuple tup, AttrNumber attributeNumber, bool *isNull)
Definition: syscache.c:479
#define SearchSysCacheCopy2(cacheId, key1, key2)
Definition: syscache.h:88
#define SearchSysCacheExists2(cacheId, key1, key2)
Definition: syscache.h:97
TupleDesc CreateTemplateTupleDesc(int natts)
Definition: tupdesc.c:67
void TupleDescInitEntry(TupleDesc desc, AttrNumber attributeNumber, const char *attributeName, Oid oidtypeid, int32 typmod, int attdim)
Definition: tupdesc.c:651
#define TupleDescAttr(tupdesc, i)
Definition: tupdesc.h:92

References AccessShareLock, Publication::alltables, BlessTupleDesc(), buildint2vector(), FuncCallContext::call_cntr, CreateTemplateTupleDesc(), deconstruct_array(), filter_partitions(), get_rel_namespace(), GetAllSchemaPublicationRelations(), GetAllTablesPublicationRelations(), GetPublication(), GetPublicationByName(), GetPublicationRelations(), heap_form_tuple(), HeapTupleGetDatum(), HeapTupleIsValid, i, if(), lappend(), lfirst_oid, list_concat_unique_oid(), list_length(), list_nth(), MemoryContextSwitchTo(), FuncCallContext::multi_call_memory_ctx, TupleDescData::natts, NIL, NUM_PUBLICATION_TABLES_ELEM, ObjectIdGetDatum(), Publication::oid, palloc(), PG_GETARG_ARRAYTYPE_P, PointerGetDatum(), published_rel::pubid, PUBLICATION_PART_LEAF, PUBLICATION_PART_ROOT, Publication::pubviaroot, RelationGetDescr, published_rel::relid, SearchSysCacheCopy2, SearchSysCacheExists2, SRF_FIRSTCALL_INIT, SRF_IS_FIRSTCALL, SRF_PERCALL_SETUP, SRF_RETURN_DONE, SRF_RETURN_NEXT, SysCacheGetAttr(), table_close(), table_open(), TextDatumGetCString, FuncCallContext::tuple_desc, TupleDescAttr, TupleDescInitEntry(), FuncCallContext::user_fctx, and values.

◆ pg_relation_is_publishable()

Datum pg_relation_is_publishable ( PG_FUNCTION_ARGS  )

Definition at line 163 of file pg_publication.c.

164 {
165  Oid relid = PG_GETARG_OID(0);
166  HeapTuple tuple;
167  bool result;
168 
169  tuple = SearchSysCache1(RELOID, ObjectIdGetDatum(relid));
170  if (!HeapTupleIsValid(tuple))
171  PG_RETURN_NULL();
172  result = is_publishable_class(relid, (Form_pg_class) GETSTRUCT(tuple));
173  ReleaseSysCache(tuple);
174  PG_RETURN_BOOL(result);
175 }
#define PG_GETARG_OID(n)
Definition: fmgr.h:275
#define PG_RETURN_NULL()
Definition: fmgr.h:345
#define PG_RETURN_BOOL(x)
Definition: fmgr.h:359

References GETSTRUCT, HeapTupleIsValid, is_publishable_class(), ObjectIdGetDatum(), PG_GETARG_OID, PG_RETURN_BOOL, PG_RETURN_NULL, ReleaseSysCache(), and SearchSysCache1().

◆ pub_collist_to_bitmapset()

Bitmapset* pub_collist_to_bitmapset ( Bitmapset columns,
Datum  pubcols,
MemoryContext  mcxt 
)

Definition at line 570 of file pg_publication.c.

571 {
572  Bitmapset *result = NULL;
573  ArrayType *arr;
574  int nelems;
575  int16 *elems;
576  MemoryContext oldcxt = NULL;
577 
578  /*
579  * If an existing bitmap was provided, use it. Otherwise just use NULL and
580  * build a new bitmap.
581  */
582  if (columns)
583  result = columns;
584 
585  arr = DatumGetArrayTypeP(pubcols);
586  nelems = ARR_DIMS(arr)[0];
587  elems = (int16 *) ARR_DATA_PTR(arr);
588 
589  /* If a memory context was specified, switch to it. */
590  if (mcxt)
591  oldcxt = MemoryContextSwitchTo(mcxt);
592 
593  for (int i = 0; i < nelems; i++)
594  result = bms_add_member(result, elems[i]);
595 
596  if (mcxt)
597  MemoryContextSwitchTo(oldcxt);
598 
599  return result;
600 }
#define ARR_DATA_PTR(a)
Definition: array.h:322
#define DatumGetArrayTypeP(X)
Definition: array.h:261
#define ARR_DIMS(a)
Definition: array.h:294
Bitmapset * bms_add_member(Bitmapset *a, int x)
Definition: bitmapset.c:815

References ARR_DATA_PTR, ARR_DIMS, bms_add_member(), DatumGetArrayTypeP, i, and MemoryContextSwitchTo().

Referenced by AlterPublicationTables(), pgoutput_column_list_init(), and pub_collist_contains_invalid_column().

◆ publication_add_relation()

ObjectAddress publication_add_relation ( Oid  pubid,
PublicationRelInfo pri,
bool  if_not_exists 
)

Definition at line 358 of file pg_publication.c.

360 {
361  Relation rel;
362  HeapTuple tup;
363  Datum values[Natts_pg_publication_rel];
364  bool nulls[Natts_pg_publication_rel];
365  Relation targetrel = pri->relation;
366  Oid relid = RelationGetRelid(targetrel);
367  Oid pubreloid;
368  Publication *pub = GetPublication(pubid);
369  AttrNumber *attarray = NULL;
370  int natts = 0;
371  ObjectAddress myself,
372  referenced;
373  List *relids = NIL;
374 
375  rel = table_open(PublicationRelRelationId, RowExclusiveLock);
376 
377  /*
378  * Check for duplicates. Note that this does not really prevent
379  * duplicates, it's here just to provide nicer error message in common
380  * case. The real protection is the unique key on the catalog.
381  */
382  if (SearchSysCacheExists2(PUBLICATIONRELMAP, ObjectIdGetDatum(relid),
383  ObjectIdGetDatum(pubid)))
384  {
386 
387  if (if_not_exists)
388  return InvalidObjectAddress;
389 
390  ereport(ERROR,
392  errmsg("relation \"%s\" is already member of publication \"%s\"",
393  RelationGetRelationName(targetrel), pub->name)));
394  }
395 
397 
398  /*
399  * Translate column names to attnums and make sure the column list
400  * contains only allowed elements (no system or generated columns etc.).
401  * Also build an array of attnums, for storing in the catalog.
402  */
404  &natts, &attarray);
405 
406  /* Form a tuple. */
407  memset(values, 0, sizeof(values));
408  memset(nulls, false, sizeof(nulls));
409 
410  pubreloid = GetNewOidWithIndex(rel, PublicationRelObjectIndexId,
411  Anum_pg_publication_rel_oid);
412  values[Anum_pg_publication_rel_oid - 1] = ObjectIdGetDatum(pubreloid);
413  values[Anum_pg_publication_rel_prpubid - 1] =
414  ObjectIdGetDatum(pubid);
415  values[Anum_pg_publication_rel_prrelid - 1] =
416  ObjectIdGetDatum(relid);
417 
418  /* Add qualifications, if available */
419  if (pri->whereClause != NULL)
420  values[Anum_pg_publication_rel_prqual - 1] = CStringGetTextDatum(nodeToString(pri->whereClause));
421  else
422  nulls[Anum_pg_publication_rel_prqual - 1] = true;
423 
424  /* Add column list, if available */
425  if (pri->columns)
426  values[Anum_pg_publication_rel_prattrs - 1] = PointerGetDatum(buildint2vector(attarray, natts));
427  else
428  nulls[Anum_pg_publication_rel_prattrs - 1] = true;
429 
430  tup = heap_form_tuple(RelationGetDescr(rel), values, nulls);
431 
432  /* Insert tuple into catalog. */
433  CatalogTupleInsert(rel, tup);
434  heap_freetuple(tup);
435 
436  /* Register dependencies as needed */
437  ObjectAddressSet(myself, PublicationRelRelationId, pubreloid);
438 
439  /* Add dependency on the publication */
440  ObjectAddressSet(referenced, PublicationRelationId, pubid);
441  recordDependencyOn(&myself, &referenced, DEPENDENCY_AUTO);
442 
443  /* Add dependency on the relation */
444  ObjectAddressSet(referenced, RelationRelationId, relid);
445  recordDependencyOn(&myself, &referenced, DEPENDENCY_AUTO);
446 
447  /* Add dependency on the objects mentioned in the qualifications */
448  if (pri->whereClause)
449  recordDependencyOnSingleRelExpr(&myself, pri->whereClause, relid,
451  false);
452 
453  /* Add dependency on the columns, if any are listed */
454  for (int i = 0; i < natts; i++)
455  {
456  ObjectAddressSubSet(referenced, RelationRelationId, relid, attarray[i]);
457  recordDependencyOn(&myself, &referenced, DEPENDENCY_NORMAL);
458  }
459 
460  /* Close the table. */
462 
463  /*
464  * Invalidate relcache so that publication info is rebuilt.
465  *
466  * For the partitioned tables, we must invalidate all partitions contained
467  * in the respective partition hierarchies, not just the one explicitly
468  * mentioned in the publication. This is required because we implicitly
469  * publish the child tables when the parent table is published.
470  */
472  relid);
473 
475 
476  return myself;
477 }
#define CStringGetTextDatum(s)
Definition: builtins.h:97
Oid GetNewOidWithIndex(Relation relation, Oid indexId, AttrNumber oidcolumn)
Definition: catalog.c:391
void recordDependencyOnSingleRelExpr(const ObjectAddress *depender, Node *expr, Oid relId, DependencyType behavior, DependencyType self_behavior, bool reverse_self)
Definition: dependency.c:1596
@ DEPENDENCY_AUTO
Definition: dependency.h:34
@ DEPENDENCY_NORMAL
Definition: dependency.h:33
void heap_freetuple(HeapTuple htup)
Definition: heaptuple.c:1434
void CatalogTupleInsert(Relation heapRel, HeapTuple tup)
Definition: indexing.c:233
#define RowExclusiveLock
Definition: lockdefs.h:38
const ObjectAddress InvalidObjectAddress
#define ObjectAddressSet(addr, class_id, object_id)
Definition: objectaddress.h:40
#define ObjectAddressSubSet(addr, class_id, object_id, object_sub_id)
Definition: objectaddress.h:33
char * nodeToString(const void *obj)
Definition: outfuncs.c:763
void recordDependencyOn(const ObjectAddress *depender, const ObjectAddress *referenced, DependencyType behavior)
Definition: pg_depend.c:44
static void check_publication_add_relation(Relation targetrel)
static void publication_translate_columns(Relation targetrel, List *columns, int *natts, AttrNumber **attrs)
void InvalidatePublicationRels(List *relids)
#define ERRCODE_DUPLICATE_OBJECT
Definition: streamutil.c:32

References buildint2vector(), CatalogTupleInsert(), check_publication_add_relation(), PublicationRelInfo::columns, CStringGetTextDatum, DEPENDENCY_AUTO, DEPENDENCY_NORMAL, ereport, errcode(), ERRCODE_DUPLICATE_OBJECT, errmsg(), ERROR, GetNewOidWithIndex(), GetPublication(), GetPubPartitionOptionRelations(), heap_form_tuple(), heap_freetuple(), i, InvalidatePublicationRels(), InvalidObjectAddress, Publication::name, NIL, nodeToString(), ObjectAddressSet, ObjectAddressSubSet, ObjectIdGetDatum(), PointerGetDatum(), PUBLICATION_PART_ALL, publication_translate_columns(), recordDependencyOn(), recordDependencyOnSingleRelExpr(), PublicationRelInfo::relation, RelationGetDescr, RelationGetRelationName, RelationGetRelid, RowExclusiveLock, SearchSysCacheExists2, table_close(), table_open(), values, and PublicationRelInfo::whereClause.

Referenced by PublicationAddTables().

◆ publication_add_schema()

ObjectAddress publication_add_schema ( Oid  pubid,
Oid  schemaid,
bool  if_not_exists 
)

Definition at line 606 of file pg_publication.c.

607 {
608  Relation rel;
609  HeapTuple tup;
610  Datum values[Natts_pg_publication_namespace];
611  bool nulls[Natts_pg_publication_namespace];
612  Oid psschid;
613  Publication *pub = GetPublication(pubid);
614  List *schemaRels = NIL;
615  ObjectAddress myself,
616  referenced;
617 
618  rel = table_open(PublicationNamespaceRelationId, RowExclusiveLock);
619 
620  /*
621  * Check for duplicates. Note that this does not really prevent
622  * duplicates, it's here just to provide nicer error message in common
623  * case. The real protection is the unique key on the catalog.
624  */
625  if (SearchSysCacheExists2(PUBLICATIONNAMESPACEMAP,
626  ObjectIdGetDatum(schemaid),
627  ObjectIdGetDatum(pubid)))
628  {
630 
631  if (if_not_exists)
632  return InvalidObjectAddress;
633 
634  ereport(ERROR,
636  errmsg("schema \"%s\" is already member of publication \"%s\"",
637  get_namespace_name(schemaid), pub->name)));
638  }
639 
641 
642  /* Form a tuple */
643  memset(values, 0, sizeof(values));
644  memset(nulls, false, sizeof(nulls));
645 
646  psschid = GetNewOidWithIndex(rel, PublicationNamespaceObjectIndexId,
647  Anum_pg_publication_namespace_oid);
648  values[Anum_pg_publication_namespace_oid - 1] = ObjectIdGetDatum(psschid);
649  values[Anum_pg_publication_namespace_pnpubid - 1] =
650  ObjectIdGetDatum(pubid);
651  values[Anum_pg_publication_namespace_pnnspid - 1] =
652  ObjectIdGetDatum(schemaid);
653 
654  tup = heap_form_tuple(RelationGetDescr(rel), values, nulls);
655 
656  /* Insert tuple into catalog */
657  CatalogTupleInsert(rel, tup);
658  heap_freetuple(tup);
659 
660  ObjectAddressSet(myself, PublicationNamespaceRelationId, psschid);
661 
662  /* Add dependency on the publication */
663  ObjectAddressSet(referenced, PublicationRelationId, pubid);
664  recordDependencyOn(&myself, &referenced, DEPENDENCY_AUTO);
665 
666  /* Add dependency on the schema */
667  ObjectAddressSet(referenced, NamespaceRelationId, schemaid);
668  recordDependencyOn(&myself, &referenced, DEPENDENCY_AUTO);
669 
670  /* Close the table */
672 
673  /*
674  * Invalidate relcache so that publication info is rebuilt. See
675  * publication_add_relation for why we need to consider all the
676  * partitions.
677  */
678  schemaRels = GetSchemaPublicationRelations(schemaid,
680  InvalidatePublicationRels(schemaRels);
681 
682  return myself;
683 }
static void check_publication_add_schema(Oid schemaid)

References CatalogTupleInsert(), check_publication_add_schema(), DEPENDENCY_AUTO, ereport, errcode(), ERRCODE_DUPLICATE_OBJECT, errmsg(), ERROR, get_namespace_name(), GetNewOidWithIndex(), GetPublication(), GetSchemaPublicationRelations(), heap_form_tuple(), heap_freetuple(), InvalidatePublicationRels(), InvalidObjectAddress, Publication::name, NIL, ObjectAddressSet, ObjectIdGetDatum(), PUBLICATION_PART_ALL, recordDependencyOn(), RelationGetDescr, RowExclusiveLock, SearchSysCacheExists2, table_close(), table_open(), and values.

Referenced by PublicationAddSchemas().

◆ publication_translate_columns()

static void publication_translate_columns ( Relation  targetrel,
List columns,
int *  natts,
AttrNumber **  attrs 
)
static

Definition at line 502 of file pg_publication.c.

504 {
505  AttrNumber *attarray = NULL;
506  Bitmapset *set = NULL;
507  ListCell *lc;
508  int n = 0;
509  TupleDesc tupdesc = RelationGetDescr(targetrel);
510 
511  /* Bail out when no column list defined. */
512  if (!columns)
513  return;
514 
515  /*
516  * Translate list of columns to attnums. We prohibit system attributes and
517  * make sure there are no duplicate columns.
518  */
519  attarray = palloc(sizeof(AttrNumber) * list_length(columns));
520  foreach(lc, columns)
521  {
522  char *colname = strVal(lfirst(lc));
523  AttrNumber attnum = get_attnum(RelationGetRelid(targetrel), colname);
524 
525  if (attnum == InvalidAttrNumber)
526  ereport(ERROR,
527  errcode(ERRCODE_UNDEFINED_COLUMN),
528  errmsg("column \"%s\" of relation \"%s\" does not exist",
529  colname, RelationGetRelationName(targetrel)));
530 
532  ereport(ERROR,
533  errcode(ERRCODE_INVALID_COLUMN_REFERENCE),
534  errmsg("cannot use system column \"%s\" in publication column list",
535  colname));
536 
537  if (TupleDescAttr(tupdesc, attnum - 1)->attgenerated)
538  ereport(ERROR,
539  errcode(ERRCODE_INVALID_COLUMN_REFERENCE),
540  errmsg("cannot use generated column \"%s\" in publication column list",
541  colname));
542 
543  if (bms_is_member(attnum, set))
544  ereport(ERROR,
546  errmsg("duplicate column \"%s\" in publication column list",
547  colname));
548 
549  set = bms_add_member(set, attnum);
550  attarray[n++] = attnum;
551  }
552 
553  /* Be tidy, so that the catalog representation is always sorted */
554  qsort(attarray, n, sizeof(AttrNumber), compare_int16);
555 
556  *natts = n;
557  *attrs = attarray;
558 
559  bms_free(set);
560 }
#define AttrNumberIsForUserDefinedAttr(attributeNumber)
Definition: attnum.h:41
#define InvalidAttrNumber
Definition: attnum.h:23
void bms_free(Bitmapset *a)
Definition: bitmapset.c:239
bool bms_is_member(int x, const Bitmapset *a)
Definition: bitmapset.c:510
AttrNumber get_attnum(Oid relid, const char *attname)
Definition: lsyscache.c:858
int16 attnum
Definition: pg_attribute.h:74
static int compare_int16(const void *a, const void *b)
#define qsort(a, b, c, d)
Definition: port.h:449
#define strVal(v)
Definition: value.h:82

References attnum, AttrNumberIsForUserDefinedAttr, bms_add_member(), bms_free(), bms_is_member(), compare_int16(), ereport, errcode(), ERRCODE_DUPLICATE_OBJECT, errmsg(), ERROR, get_attnum(), InvalidAttrNumber, lfirst, list_length(), palloc(), qsort, RelationGetDescr, RelationGetRelationName, RelationGetRelid, strVal, and TupleDescAttr.

Referenced by publication_add_relation().