mirror of https://github.com/citusdata/citus.git
121 lines
2.6 KiB
C
121 lines
2.6 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* qualify_publication_stmt.c
|
|
* Functions specialized in fully qualifying all publication statements. These
|
|
* functions are dispatched from qualify.c
|
|
*
|
|
* Copyright (c), Citus Data, Inc.
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
|
|
#include "postgres.h"
|
|
|
|
#include "catalog/namespace.h"
|
|
#include "nodes/nodes.h"
|
|
#include "utils/guc.h"
|
|
#include "utils/lsyscache.h"
|
|
|
|
#include "distributed/deparser.h"
|
|
#include "distributed/listutils.h"
|
|
|
|
#if (PG_VERSION_NUM >= PG_VERSION_15)
|
|
static void QualifyPublicationObjects(List *publicationObjects);
|
|
#else
|
|
static void QualifyTables(List *tables);
|
|
#endif
|
|
static void QualifyPublicationRangeVar(RangeVar *publication);
|
|
|
|
|
|
/*
|
|
* QualifyCreatePublicationStmt quailifies the publication names of the
|
|
* CREATE PUBLICATION statement.
|
|
*/
|
|
void
|
|
QualifyCreatePublicationStmt(Node *node)
|
|
{
|
|
CreatePublicationStmt *stmt = castNode(CreatePublicationStmt, node);
|
|
|
|
#if (PG_VERSION_NUM >= PG_VERSION_15)
|
|
QualifyPublicationObjects(stmt->pubobjects);
|
|
#else
|
|
QualifyTables(stmt->tables);
|
|
#endif
|
|
}
|
|
|
|
|
|
#if (PG_VERSION_NUM >= PG_VERSION_15)
|
|
|
|
/*
|
|
* QualifyPublicationObjects ensures all table names in a list of
|
|
* publication objects are fully qualified.
|
|
*/
|
|
static void
|
|
QualifyPublicationObjects(List *publicationObjects)
|
|
{
|
|
PublicationObjSpec *publicationObject = NULL;
|
|
|
|
foreach_ptr(publicationObject, publicationObjects)
|
|
{
|
|
if (publicationObject->pubobjtype == PUBLICATIONOBJ_TABLE)
|
|
{
|
|
/* FOR TABLE ... */
|
|
PublicationTable *publicationTable = publicationObject->pubtable;
|
|
|
|
QualifyPublicationRangeVar(publicationTable->relation);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
#else
|
|
|
|
/*
|
|
* QualifyTables ensures all table names in a list are fully qualified.
|
|
*/
|
|
static void
|
|
QualifyTables(List *tables)
|
|
{
|
|
RangeVar *rangeVar = NULL;
|
|
|
|
foreach_ptr(rangeVar, tables)
|
|
{
|
|
QualifyPublicationRangeVar(rangeVar);
|
|
}
|
|
}
|
|
|
|
|
|
#endif
|
|
|
|
|
|
/*
|
|
* QualifyPublicationObjects ensures all table names in a list of
|
|
* publication objects are fully qualified.
|
|
*/
|
|
void
|
|
QualifyAlterPublicationStmt(Node *node)
|
|
{
|
|
AlterPublicationStmt *stmt = castNode(AlterPublicationStmt, node);
|
|
|
|
#if (PG_VERSION_NUM >= PG_VERSION_15)
|
|
QualifyPublicationObjects(stmt->pubobjects);
|
|
#else
|
|
QualifyTables(stmt->tables);
|
|
#endif
|
|
}
|
|
|
|
|
|
/*
|
|
* QualifyPublicationRangeVar qualifies the given publication RangeVar if it is not qualified.
|
|
*/
|
|
static void
|
|
QualifyPublicationRangeVar(RangeVar *publication)
|
|
{
|
|
if (publication->schemaname == NULL)
|
|
{
|
|
Oid publicationOid = RelnameGetRelid(publication->relname);
|
|
Oid schemaOid = get_rel_namespace(publicationOid);
|
|
publication->schemaname = get_namespace_name(schemaOid);
|
|
}
|
|
}
|