PostgreSQL Source Code  git master
createas.c File Reference
#include "postgres.h"
#include "access/heapam.h"
#include "access/htup_details.h"
#include "access/reloptions.h"
#include "access/sysattr.h"
#include "access/tableam.h"
#include "access/xact.h"
#include "access/xlog.h"
#include "catalog/namespace.h"
#include "catalog/toasting.h"
#include "commands/createas.h"
#include "commands/matview.h"
#include "commands/prepare.h"
#include "commands/tablecmds.h"
#include "commands/view.h"
#include "miscadmin.h"
#include "nodes/makefuncs.h"
#include "nodes/nodeFuncs.h"
#include "parser/parse_clause.h"
#include "rewrite/rewriteHandler.h"
#include "storage/smgr.h"
#include "tcop/tcopprot.h"
#include "utils/builtins.h"
#include "utils/lsyscache.h"
#include "utils/rel.h"
#include "utils/rls.h"
#include "utils/snapmgr.h"
Include dependency graph for createas.c:

Go to the source code of this file.

Data Structures

struct  DR_intorel
 

Functions

static ObjectAddress create_ctas_internal (List *attrList, IntoClause *into)
 
static ObjectAddress create_ctas_nodata (List *tlist, IntoClause *into)
 
static void intorel_startup (DestReceiver *self, int operation, TupleDesc typeinfo)
 
static bool intorel_receive (TupleTableSlot *slot, DestReceiver *self)
 
static void intorel_shutdown (DestReceiver *self)
 
static void intorel_destroy (DestReceiver *self)
 
ObjectAddress ExecCreateTableAs (ParseState *pstate, CreateTableAsStmt *stmt, ParamListInfo params, QueryEnvironment *queryEnv, QueryCompletion *qc)
 
int GetIntoRelEFlags (IntoClause *intoClause)
 
DestReceiverCreateIntoRelDestReceiver (IntoClause *intoClause)
 

Function Documentation

◆ create_ctas_internal()

static ObjectAddress create_ctas_internal ( List attrList,
IntoClause into 
)
static

Definition at line 85 of file createas.c.

References IntoClause::accessMethod, CreateStmt::accessMethod, CommandCounterIncrement(), CreateStmt::constraints, copyObject, DefineRelation(), HEAP_RELOPT_NAMESPACES, heap_reloptions(), CreateStmt::if_not_exists, CreateStmt::inhRelations, InvalidOid, makeNode, NewRelationCreateToastTable(), NIL, ObjectAddress::objectId, CreateStmt::ofTypename, IntoClause::onCommit, CreateStmt::oncommit, IntoClause::options, CreateStmt::options, IntoClause::rel, CreateStmt::relation, StoreViewQuery(), CreateStmt::tableElts, IntoClause::tableSpaceName, CreateStmt::tablespacename, transformRelOptions(), and IntoClause::viewQuery.

Referenced by create_ctas_nodata(), and intorel_startup().

86 {
87  CreateStmt *create = makeNode(CreateStmt);
88  bool is_matview;
89  char relkind;
90  Datum toast_options;
91  static char *validnsps[] = HEAP_RELOPT_NAMESPACES;
92  ObjectAddress intoRelationAddr;
93 
94  /* This code supports both CREATE TABLE AS and CREATE MATERIALIZED VIEW */
95  is_matview = (into->viewQuery != NULL);
96  relkind = is_matview ? RELKIND_MATVIEW : RELKIND_RELATION;
97 
98  /*
99  * Create the target relation by faking up a CREATE TABLE parsetree and
100  * passing it to DefineRelation.
101  */
102  create->relation = into->rel;
103  create->tableElts = attrList;
104  create->inhRelations = NIL;
105  create->ofTypename = NULL;
106  create->constraints = NIL;
107  create->options = into->options;
108  create->oncommit = into->onCommit;
109  create->tablespacename = into->tableSpaceName;
110  create->if_not_exists = false;
111  create->accessMethod = into->accessMethod;
112 
113  /*
114  * Create the relation. (This will error out if there's an existing view,
115  * so we don't need more code to complain if "replace" is false.)
116  */
117  intoRelationAddr = DefineRelation(create, relkind, InvalidOid, NULL, NULL);
118 
119  /*
120  * If necessary, create a TOAST table for the target table. Note that
121  * NewRelationCreateToastTable ends with CommandCounterIncrement(), so
122  * that the TOAST table will be visible for insertion.
123  */
125 
126  /* parse and validate reloptions for the toast table */
127  toast_options = transformRelOptions((Datum) 0,
128  create->options,
129  "toast",
130  validnsps,
131  true, false);
132 
133  (void) heap_reloptions(RELKIND_TOASTVALUE, toast_options, true);
134 
135  NewRelationCreateToastTable(intoRelationAddr.objectId, toast_options);
136 
137  /* Create the "view" part of a materialized view. */
138  if (is_matview)
139  {
140  /* StoreViewQuery scribbles on tree, so make a copy */
141  Query *query = (Query *) copyObject(into->viewQuery);
142 
143  StoreViewQuery(intoRelationAddr.objectId, query, false);
145  }
146 
147  return intoRelationAddr;
148 }
RangeVar * relation
Definition: parsenodes.h:2072
bytea * heap_reloptions(char relkind, Datum reloptions, bool validate)
Definition: reloptions.c:1996
#define NIL
Definition: pg_list.h:65
OnCommitAction oncommit
Definition: parsenodes.h:2081
List * inhRelations
Definition: parsenodes.h:2074
OnCommitAction onCommit
Definition: primnodes.h:116
Datum transformRelOptions(Datum oldOptions, List *defList, const char *namspace, char *validnsps[], bool acceptOidsOff, bool isReset)
Definition: reloptions.c:1132
char * tableSpaceName
Definition: primnodes.h:117
void NewRelationCreateToastTable(Oid relOid, Datum reloptions)
Definition: toasting.c:69
List * constraints
Definition: parsenodes.h:2079
bool if_not_exists
Definition: parsenodes.h:2084
char * accessMethod
Definition: parsenodes.h:2083
Node * viewQuery
Definition: primnodes.h:118
List * options
Definition: parsenodes.h:2080
#define HEAP_RELOPT_NAMESPACES
Definition: reloptions.h:61
ObjectAddress DefineRelation(CreateStmt *stmt, char relkind, Oid ownerId, ObjectAddress *typaddress, const char *queryString)
Definition: tablecmds.c:581
List * options
Definition: primnodes.h:115
char * tablespacename
Definition: parsenodes.h:2082
uintptr_t Datum
Definition: postgres.h:367
void CommandCounterIncrement(void)
Definition: xact.c:1021
void StoreViewQuery(Oid viewOid, Query *viewParse, bool replace)
Definition: view.c:565
#define InvalidOid
Definition: postgres_ext.h:36
List * tableElts
Definition: parsenodes.h:2073
#define makeNode(_type_)
Definition: nodes.h:577
RangeVar * rel
Definition: primnodes.h:112
#define copyObject(obj)
Definition: nodes.h:645
char * accessMethod
Definition: primnodes.h:114
TypeName * ofTypename
Definition: parsenodes.h:2078

◆ create_ctas_nodata()

static ObjectAddress create_ctas_nodata ( List tlist,
IntoClause into 
)
static

Definition at line 158 of file createas.c.

References ColumnDef::collOid, ColumnDef::colname, IntoClause::colNames, create_ctas_internal(), ereport, errcode(), errhint(), errmsg(), ERROR, TargetEntry::expr, exprCollation(), exprType(), exprTypmod(), format_type_be(), lappend(), lfirst, list_head(), lnext(), makeColumnDef(), NIL, OidIsValid, TargetEntry::resjunk, TargetEntry::resname, strVal, type_is_collatable(), ColumnDef::typeName, and TypeName::typeOid.

Referenced by ExecCreateTableAs().

159 {
160  List *attrList;
161  ListCell *t,
162  *lc;
163 
164  /*
165  * Build list of ColumnDefs from non-junk elements of the tlist. If a
166  * column name list was specified in CREATE TABLE AS, override the column
167  * names in the query. (Too few column names are OK, too many are not.)
168  */
169  attrList = NIL;
170  lc = list_head(into->colNames);
171  foreach(t, tlist)
172  {
173  TargetEntry *tle = (TargetEntry *) lfirst(t);
174 
175  if (!tle->resjunk)
176  {
177  ColumnDef *col;
178  char *colname;
179 
180  if (lc)
181  {
182  colname = strVal(lfirst(lc));
183  lc = lnext(into->colNames, lc);
184  }
185  else
186  colname = tle->resname;
187 
188  col = makeColumnDef(colname,
189  exprType((Node *) tle->expr),
190  exprTypmod((Node *) tle->expr),
191  exprCollation((Node *) tle->expr));
192 
193  /*
194  * It's possible that the column is of a collatable type but the
195  * collation could not be resolved, so double-check. (We must
196  * check this here because DefineRelation would adopt the type's
197  * default collation rather than complaining.)
198  */
199  if (!OidIsValid(col->collOid) &&
201  ereport(ERROR,
202  (errcode(ERRCODE_INDETERMINATE_COLLATION),
203  errmsg("no collation was derived for column \"%s\" with collatable type %s",
204  col->colname,
206  errhint("Use the COLLATE clause to set the collation explicitly.")));
207 
208  attrList = lappend(attrList, col);
209  }
210  }
211 
212  if (lc != NULL)
213  ereport(ERROR,
214  (errcode(ERRCODE_SYNTAX_ERROR),
215  errmsg("too many column names were specified")));
216 
217  /* Create the relation definition using the ColumnDef list */
218  return create_ctas_internal(attrList, into);
219 }
#define NIL
Definition: pg_list.h:65
Oid typeOid
Definition: parsenodes.h:211
int errhint(const char *fmt,...)
Definition: elog.c:1071
static ListCell * lnext(const List *l, const ListCell *c)
Definition: pg_list.h:321
int32 exprTypmod(const Node *expr)
Definition: nodeFuncs.c:275
Definition: nodes.h:529
#define strVal(v)
Definition: value.h:54
int errcode(int sqlerrcode)
Definition: elog.c:610
char * format_type_be(Oid type_oid)
Definition: format_type.c:339
char * resname
Definition: primnodes.h:1409
#define OidIsValid(objectId)
Definition: c.h:651
static ObjectAddress create_ctas_internal(List *attrList, IntoClause *into)
Definition: createas.c:85
bool resjunk
Definition: primnodes.h:1414
#define ERROR
Definition: elog.h:43
Oid collOid
Definition: parsenodes.h:661
static ListCell * list_head(const List *l)
Definition: pg_list.h:125
List * lappend(List *list, void *datum)
Definition: list.c:321
#define ereport(elevel,...)
Definition: elog.h:144
#define lfirst(lc)
Definition: pg_list.h:190
Expr * expr
Definition: primnodes.h:1407
Oid exprType(const Node *expr)
Definition: nodeFuncs.c:41
Oid exprCollation(const Node *expr)
Definition: nodeFuncs.c:719
TypeName * typeName
Definition: parsenodes.h:648
int errmsg(const char *fmt,...)
Definition: elog.c:824
char * colname
Definition: parsenodes.h:647
List * colNames
Definition: primnodes.h:113
bool type_is_collatable(Oid typid)
Definition: lsyscache.c:2958
Definition: pg_list.h:50
ColumnDef * makeColumnDef(const char *colname, Oid typeOid, int32 typmod, Oid collOid)
Definition: makefuncs.c:490

◆ CreateIntoRelDestReceiver()

DestReceiver* CreateIntoRelDestReceiver ( IntoClause intoClause)

Definition at line 411 of file createas.c.

References DestIntoRel, intorel_destroy(), intorel_receive(), intorel_shutdown(), intorel_startup(), and palloc0().

Referenced by CreateDestReceiver(), ExecCreateTableAs(), and ExplainOnePlan().

412 {
413  DR_intorel *self = (DR_intorel *) palloc0(sizeof(DR_intorel));
414 
415  self->pub.receiveSlot = intorel_receive;
416  self->pub.rStartup = intorel_startup;
417  self->pub.rShutdown = intorel_shutdown;
418  self->pub.rDestroy = intorel_destroy;
419  self->pub.mydest = DestIntoRel;
420  self->into = intoClause;
421  /* other private fields will be set during intorel_startup */
422 
423  return (DestReceiver *) self;
424 }
static void intorel_shutdown(DestReceiver *self)
Definition: createas.c:596
static bool intorel_receive(TupleTableSlot *slot, DestReceiver *self)
Definition: createas.c:568
void * palloc0(Size size)
Definition: mcxt.c:980
static void intorel_destroy(DestReceiver *self)
Definition: createas.c:613
static void intorel_startup(DestReceiver *self, int operation, TupleDesc typeinfo)
Definition: createas.c:430

◆ ExecCreateTableAs()

ObjectAddress ExecCreateTableAs ( ParseState pstate,
CreateTableAsStmt stmt,
ParamListInfo  params,
QueryEnvironment queryEnv,
QueryCompletion qc 
)

Definition at line 226 of file createas.c.

References Assert, AtEOXact_GUC(), castNode, CMD_SELECT, CMD_UTILITY, Query::commandType, copyObject, create_ctas_nodata(), CreateIntoRelDestReceiver(), CreateQueryDesc(), CURSOR_OPT_PARALLEL_OK, generate_unaccent_rules::dest, elog, ereport, errcode(), errmsg(), ERROR, EState::es_processed, QueryDesc::estate, ExecuteQuery(), ExecutorEnd(), ExecutorFinish(), ExecutorRun(), ExecutorStart(), ForwardScanDirection, FreeQueryDesc(), get_relname_relid(), GetActiveSnapshot(), GetIntoRelEFlags(), GetUserIdAndSecContext(), CreateTableAsStmt::if_not_exists, CreateTableAsStmt::into, InvalidObjectAddress, InvalidOid, InvalidSnapshot, IsA, linitial_node, list_length(), NewGUCNestLevel(), NOTICE, ParseState::p_sourcetext, pg_plan_query(), PopActiveSnapshot(), PushCopiedSnapshot(), CreateTableAsStmt::query, QueryRewrite(), RangeVarGetCreationNamespace(), IntoClause::rel, RangeVar::relname, SECURITY_RESTRICTED_OPERATION, SetQueryCompletion(), SetUserIdAndSecContext(), IntoClause::skipData, Query::targetList, UpdateActiveSnapshotCommandId(), Query::utilityStmt, and IntoClause::viewQuery.

Referenced by ProcessUtilitySlow().

229 {
230  Query *query = castNode(Query, stmt->query);
231  IntoClause *into = stmt->into;
232  bool is_matview = (into->viewQuery != NULL);
234  Oid save_userid = InvalidOid;
235  int save_sec_context = 0;
236  int save_nestlevel = 0;
237  ObjectAddress address;
238  List *rewritten;
239  PlannedStmt *plan;
240  QueryDesc *queryDesc;
241 
242  if (stmt->if_not_exists)
243  {
244  Oid nspid;
245 
246  nspid = RangeVarGetCreationNamespace(stmt->into->rel);
247 
248  if (get_relname_relid(stmt->into->rel->relname, nspid))
249  {
250  ereport(NOTICE,
251  (errcode(ERRCODE_DUPLICATE_TABLE),
252  errmsg("relation \"%s\" already exists, skipping",
253  stmt->into->rel->relname)));
254  return InvalidObjectAddress;
255  }
256  }
257 
258  /*
259  * Create the tuple receiver object and insert info it will need
260  */
261  dest = CreateIntoRelDestReceiver(into);
262 
263  /*
264  * The contained Query could be a SELECT, or an EXECUTE utility command.
265  * If the latter, we just pass it off to ExecuteQuery.
266  */
267  if (query->commandType == CMD_UTILITY &&
268  IsA(query->utilityStmt, ExecuteStmt))
269  {
270  ExecuteStmt *estmt = castNode(ExecuteStmt, query->utilityStmt);
271 
272  Assert(!is_matview); /* excluded by syntax */
273  ExecuteQuery(pstate, estmt, into, params, dest, qc);
274 
275  /* get object address that intorel_startup saved for us */
276  address = ((DR_intorel *) dest)->reladdr;
277 
278  return address;
279  }
280  Assert(query->commandType == CMD_SELECT);
281 
282  /*
283  * For materialized views, lock down security-restricted operations and
284  * arrange to make GUC variable changes local to this command. This is
285  * not necessary for security, but this keeps the behavior similar to
286  * REFRESH MATERIALIZED VIEW. Otherwise, one could create a materialized
287  * view not possible to refresh.
288  */
289  if (is_matview)
290  {
291  GetUserIdAndSecContext(&save_userid, &save_sec_context);
292  SetUserIdAndSecContext(save_userid,
293  save_sec_context | SECURITY_RESTRICTED_OPERATION);
294  save_nestlevel = NewGUCNestLevel();
295  }
296 
297  if (into->skipData)
298  {
299  /*
300  * If WITH NO DATA was specified, do not go through the rewriter,
301  * planner and executor. Just define the relation using a code path
302  * similar to CREATE VIEW. This avoids dump/restore problems stemming
303  * from running the planner before all dependencies are set up.
304  */
305  address = create_ctas_nodata(query->targetList, into);
306  }
307  else
308  {
309  /*
310  * Parse analysis was done already, but we still have to run the rule
311  * rewriter. We do not do AcquireRewriteLocks: we assume the query
312  * either came straight from the parser, or suitable locks were
313  * acquired by plancache.c.
314  *
315  * Because the rewriter and planner tend to scribble on the input, we
316  * make a preliminary copy of the source querytree. This prevents
317  * problems in the case that CTAS is in a portal or plpgsql function
318  * and is executed repeatedly. (See also the same hack in EXPLAIN and
319  * PREPARE.)
320  */
321  rewritten = QueryRewrite(copyObject(query));
322 
323  /* SELECT should never rewrite to more or less than one SELECT query */
324  if (list_length(rewritten) != 1)
325  elog(ERROR, "unexpected rewrite result for %s",
326  is_matview ? "CREATE MATERIALIZED VIEW" :
327  "CREATE TABLE AS SELECT");
328  query = linitial_node(Query, rewritten);
329  Assert(query->commandType == CMD_SELECT);
330 
331  /* plan the query */
332  plan = pg_plan_query(query, pstate->p_sourcetext,
333  CURSOR_OPT_PARALLEL_OK, params);
334 
335  /*
336  * Use a snapshot with an updated command ID to ensure this query sees
337  * results of any previously executed queries. (This could only
338  * matter if the planner executed an allegedly-stable function that
339  * changed the database contents, but let's do it anyway to be
340  * parallel to the EXPLAIN code path.)
341  */
344 
345  /* Create a QueryDesc, redirecting output to our tuple receiver */
346  queryDesc = CreateQueryDesc(plan, pstate->p_sourcetext,
348  dest, params, queryEnv, 0);
349 
350  /* call ExecutorStart to prepare the plan for execution */
351  ExecutorStart(queryDesc, GetIntoRelEFlags(into));
352 
353  /* run the plan to completion */
354  ExecutorRun(queryDesc, ForwardScanDirection, 0L, true);
355 
356  /* save the rowcount if we're given a qc to fill */
357  if (qc)
358  SetQueryCompletion(qc, CMDTAG_SELECT, queryDesc->estate->es_processed);
359 
360  /* get object address that intorel_startup saved for us */
361  address = ((DR_intorel *) dest)->reladdr;
362 
363  /* and clean up */
364  ExecutorFinish(queryDesc);
365  ExecutorEnd(queryDesc);
366 
367  FreeQueryDesc(queryDesc);
368 
370  }
371 
372  if (is_matview)
373  {
374  /* Roll back any GUC changes */
375  AtEOXact_GUC(false, save_nestlevel);
376 
377  /* Restore userid and security context */
378  SetUserIdAndSecContext(save_userid, save_sec_context);
379  }
380 
381  return address;
382 }
void UpdateActiveSnapshotCommandId(void)
Definition: snapmgr.c:783
#define IsA(nodeptr, _type_)
Definition: nodes.h:580
List * QueryRewrite(Query *parsetree)
EState * estate
Definition: execdesc.h:48
#define SECURITY_RESTRICTED_OPERATION
Definition: miscadmin.h:299
void SetUserIdAndSecContext(Oid userid, int sec_context)
Definition: miscinit.c:562
#define castNode(_type_, nodeptr)
Definition: nodes.h:598
void FreeQueryDesc(QueryDesc *qdesc)
Definition: pquery.c:105
void ExecutorStart(QueryDesc *queryDesc, int eflags)
Definition: execMain.c:143
Snapshot GetActiveSnapshot(void)
Definition: snapmgr.c:841
int errcode(int sqlerrcode)
Definition: elog.c:610
void PopActiveSnapshot(void)
Definition: snapmgr.c:814
bool skipData
Definition: primnodes.h:119
unsigned int Oid
Definition: postgres_ext.h:31
Node * utilityStmt
Definition: parsenodes.h:120
#define linitial_node(type, l)
Definition: pg_list.h:198
List * targetList
Definition: parsenodes.h:140
void ExecutorEnd(QueryDesc *queryDesc)
Definition: execMain.c:462
char * relname
Definition: primnodes.h:68
int GetIntoRelEFlags(IntoClause *intoClause)
Definition: createas.c:393
#define ERROR
Definition: elog.h:43
Node * viewQuery
Definition: primnodes.h:118
void ExecutorRun(QueryDesc *queryDesc, ScanDirection direction, uint64 count, bool execute_once)
Definition: execMain.c:301
void PushCopiedSnapshot(Snapshot snapshot)
Definition: snapmgr.c:771
QueryDesc * CreateQueryDesc(PlannedStmt *plannedstmt, const char *sourceText, Snapshot snapshot, Snapshot crosscheck_snapshot, DestReceiver *dest, ParamListInfo params, QueryEnvironment *queryEnv, int instrument_options)
Definition: pquery.c:67
Oid get_relname_relid(const char *relname, Oid relnamespace)
Definition: lsyscache.c:1797
void GetUserIdAndSecContext(Oid *userid, int *sec_context)
Definition: miscinit.c:555
void AtEOXact_GUC(bool isCommit, int nestLevel)
Definition: guc.c:5956
void ExecuteQuery(ParseState *pstate, ExecuteStmt *stmt, IntoClause *intoClause, ParamListInfo params, DestReceiver *dest, QueryCompletion *qc)
Definition: prepare.c:187
DestReceiver * CreateIntoRelDestReceiver(IntoClause *intoClause)
Definition: createas.c:411
const char * p_sourcetext
Definition: parse_node.h:179
void ExecutorFinish(QueryDesc *queryDesc)
Definition: execMain.c:402
#define InvalidSnapshot
Definition: snapshot.h:123
#define InvalidOid
Definition: postgres_ext.h:36
#define ereport(elevel,...)
Definition: elog.h:144
IntoClause * into
Definition: parsenodes.h:3277
#define NOTICE
Definition: elog.h:37
CmdType commandType
Definition: parsenodes.h:112
#define Assert(condition)
Definition: c.h:745
static void SetQueryCompletion(QueryCompletion *qc, CommandTag commandTag, uint64 nprocessed)
Definition: cmdtag.h:36
uint64 es_processed
Definition: execnodes.h:559
static int list_length(const List *l)
Definition: pg_list.h:169
static ObjectAddress create_ctas_nodata(List *tlist, IntoClause *into)
Definition: createas.c:158
int NewGUCNestLevel(void)
Definition: guc.c:5942
#define CURSOR_OPT_PARALLEL_OK
Definition: parsenodes.h:2715
const ObjectAddress InvalidObjectAddress
int errmsg(const char *fmt,...)
Definition: elog.c:824
RangeVar * rel
Definition: primnodes.h:112
#define elog(elevel,...)
Definition: elog.h:214
PlannedStmt * pg_plan_query(Query *querytree, const char *query_string, int cursorOptions, ParamListInfo boundParams)
Definition: postgres.c:857
#define copyObject(obj)
Definition: nodes.h:645
Definition: pg_list.h:50
Oid RangeVarGetCreationNamespace(const RangeVar *newRelation)
Definition: namespace.c:449

◆ GetIntoRelEFlags()

int GetIntoRelEFlags ( IntoClause intoClause)

Definition at line 393 of file createas.c.

References EXEC_FLAG_WITH_NO_DATA, and IntoClause::skipData.

Referenced by ExecCreateTableAs(), ExecuteQuery(), and ExplainOnePlan().

394 {
395  int flags = 0;
396 
397  if (intoClause->skipData)
398  flags |= EXEC_FLAG_WITH_NO_DATA;
399 
400  return flags;
401 }
#define EXEC_FLAG_WITH_NO_DATA
Definition: executor.h:61
bool skipData
Definition: primnodes.h:119

◆ intorel_destroy()

static void intorel_destroy ( DestReceiver self)
static

Definition at line 613 of file createas.c.

References pfree().

Referenced by CreateIntoRelDestReceiver().

614 {
615  pfree(self);
616 }
void pfree(void *pointer)
Definition: mcxt.c:1056

◆ intorel_receive()

static bool intorel_receive ( TupleTableSlot slot,
DestReceiver self 
)
static

Definition at line 568 of file createas.c.

References DR_intorel::bistate, DR_intorel::output_cid, DR_intorel::rel, table_tuple_insert(), and DR_intorel::ti_options.

Referenced by CreateIntoRelDestReceiver().

569 {
570  DR_intorel *myState = (DR_intorel *) self;
571 
572  /*
573  * Note that the input slot might not be of the type of the target
574  * relation. That's supported by table_tuple_insert(), but slightly less
575  * efficient than inserting with the right slot - but the alternative
576  * would be to copy into a slot of the right type, which would not be
577  * cheap either. This also doesn't allow accessing per-AM data (say a
578  * tuple's xmin), but since we don't do that here...
579  */
580 
581  table_tuple_insert(myState->rel,
582  slot,
583  myState->output_cid,
584  myState->ti_options,
585  myState->bistate);
586 
587  /* We know this is a newly created relation, so there are no indexes */
588 
589  return true;
590 }
int ti_options
Definition: createas.c:62
CommandId output_cid
Definition: createas.c:61
BulkInsertState bistate
Definition: createas.c:63
static void table_tuple_insert(Relation rel, TupleTableSlot *slot, CommandId cid, int options, struct BulkInsertStateData *bistate)
Definition: tableam.h:1178
Relation rel
Definition: createas.c:59

◆ intorel_shutdown()

static void intorel_shutdown ( DestReceiver self)
static

Definition at line 596 of file createas.c.

References DR_intorel::bistate, FreeBulkInsertState(), NoLock, DR_intorel::rel, table_close(), table_finish_bulk_insert(), and DR_intorel::ti_options.

Referenced by CreateIntoRelDestReceiver().

597 {
598  DR_intorel *myState = (DR_intorel *) self;
599 
600  FreeBulkInsertState(myState->bistate);
601 
602  table_finish_bulk_insert(myState->rel, myState->ti_options);
603 
604  /* close rel, but keep lock until commit */
605  table_close(myState->rel, NoLock);
606  myState->rel = NULL;
607 }
void table_close(Relation relation, LOCKMODE lockmode)
Definition: table.c:133
int ti_options
Definition: createas.c:62
static void table_finish_bulk_insert(Relation rel, int options)
Definition: tableam.h:1371
BulkInsertState bistate
Definition: createas.c:63
#define NoLock
Definition: lockdefs.h:34
Relation rel
Definition: createas.c:59
void FreeBulkInsertState(BulkInsertState bistate)
Definition: heapam.c:1813

◆ intorel_startup()

static void intorel_startup ( DestReceiver self,
int  operation,
TupleDesc  typeinfo 
)
static

Definition at line 430 of file createas.c.

References AccessExclusiveLock, ACL_INSERT, Assert, attnum, DR_intorel::bistate, bms_add_member(), check_enable_rls(), ColumnDef::collOid, ColumnDef::colname, IntoClause::colNames, create_ctas_internal(), ereport, errcode(), errhint(), errmsg(), ERROR, ExecCheckRTPerms(), FirstLowInvalidHeapAttributeNumber, format_type_be(), GetBulkInsertState(), GetCurrentCommandId(), RangeTblEntry::insertedCols, DR_intorel::into, InvalidBlockNumber, InvalidOid, lappend(), lfirst, list_head(), list_make1, lnext(), makeColumnDef(), makeNode, NameStr, TupleDescData::natts, NIL, ObjectAddress::objectId, OidIsValid, DR_intorel::output_cid, RelationData::rd_att, DR_intorel::rel, DR_intorel::reladdr, RelationGetTargetBlock, RangeTblEntry::relid, RangeTblEntry::relkind, RangeTblEntry::rellockmode, RangeTblEntry::requiredPerms, RLS_ENABLED, RowExclusiveLock, RTE_RELATION, RangeTblEntry::rtekind, SetMatViewPopulatedState(), IntoClause::skipData, strVal, TABLE_INSERT_SKIP_FSM, table_open(), DR_intorel::ti_options, TupleDescAttr, type_is_collatable(), ColumnDef::typeName, TypeName::typeOid, and IntoClause::viewQuery.

Referenced by CreateIntoRelDestReceiver().

431 {
432  DR_intorel *myState = (DR_intorel *) self;
433  IntoClause *into = myState->into;
434  bool is_matview;
435  char relkind;
436  List *attrList;
437  ObjectAddress intoRelationAddr;
438  Relation intoRelationDesc;
439  RangeTblEntry *rte;
440  ListCell *lc;
441  int attnum;
442 
443  Assert(into != NULL); /* else somebody forgot to set it */
444 
445  /* This code supports both CREATE TABLE AS and CREATE MATERIALIZED VIEW */
446  is_matview = (into->viewQuery != NULL);
447  relkind = is_matview ? RELKIND_MATVIEW : RELKIND_RELATION;
448 
449  /*
450  * Build column definitions using "pre-cooked" type and collation info. If
451  * a column name list was specified in CREATE TABLE AS, override the
452  * column names derived from the query. (Too few column names are OK, too
453  * many are not.)
454  */
455  attrList = NIL;
456  lc = list_head(into->colNames);
457  for (attnum = 0; attnum < typeinfo->natts; attnum++)
458  {
459  Form_pg_attribute attribute = TupleDescAttr(typeinfo, attnum);
460  ColumnDef *col;
461  char *colname;
462 
463  if (lc)
464  {
465  colname = strVal(lfirst(lc));
466  lc = lnext(into->colNames, lc);
467  }
468  else
469  colname = NameStr(attribute->attname);
470 
471  col = makeColumnDef(colname,
472  attribute->atttypid,
473  attribute->atttypmod,
474  attribute->attcollation);
475 
476  /*
477  * It's possible that the column is of a collatable type but the
478  * collation could not be resolved, so double-check. (We must check
479  * this here because DefineRelation would adopt the type's default
480  * collation rather than complaining.)
481  */
482  if (!OidIsValid(col->collOid) &&
484  ereport(ERROR,
485  (errcode(ERRCODE_INDETERMINATE_COLLATION),
486  errmsg("no collation was derived for column \"%s\" with collatable type %s",
487  col->colname,
489  errhint("Use the COLLATE clause to set the collation explicitly.")));
490 
491  attrList = lappend(attrList, col);
492  }
493 
494  if (lc != NULL)
495  ereport(ERROR,
496  (errcode(ERRCODE_SYNTAX_ERROR),
497  errmsg("too many column names were specified")));
498 
499  /*
500  * Actually create the target table
501  */
502  intoRelationAddr = create_ctas_internal(attrList, into);
503 
504  /*
505  * Finally we can open the target table
506  */
507  intoRelationDesc = table_open(intoRelationAddr.objectId, AccessExclusiveLock);
508 
509  /*
510  * Check INSERT permission on the constructed table.
511  *
512  * XXX: It would arguably make sense to skip this check if into->skipData
513  * is true.
514  */
515  rte = makeNode(RangeTblEntry);
516  rte->rtekind = RTE_RELATION;
517  rte->relid = intoRelationAddr.objectId;
518  rte->relkind = relkind;
520  rte->requiredPerms = ACL_INSERT;
521 
522  for (attnum = 1; attnum <= intoRelationDesc->rd_att->natts; attnum++)
525 
526  ExecCheckRTPerms(list_make1(rte), true);
527 
528  /*
529  * Make sure the constructed table does not have RLS enabled.
530  *
531  * check_enable_rls() will ereport(ERROR) itself if the user has requested
532  * something invalid, and otherwise will return RLS_ENABLED if RLS should
533  * be enabled here. We don't actually support that currently, so throw
534  * our own ereport(ERROR) if that happens.
535  */
536  if (check_enable_rls(intoRelationAddr.objectId, InvalidOid, false) == RLS_ENABLED)
537  ereport(ERROR,
538  (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
539  errmsg("policies not yet implemented for this command")));
540 
541  /*
542  * Tentatively mark the target as populated, if it's a matview and we're
543  * going to fill it; otherwise, no change needed.
544  */
545  if (is_matview && !into->skipData)
546  SetMatViewPopulatedState(intoRelationDesc, true);
547 
548  /*
549  * Fill private fields of myState for use by later routines
550  */
551  myState->rel = intoRelationDesc;
552  myState->reladdr = intoRelationAddr;
553  myState->output_cid = GetCurrentCommandId(true);
555  myState->bistate = GetBulkInsertState();
556 
557  /*
558  * Valid smgr_targblock implies something already wrote to the relation.
559  * This may be harmless, but this function hasn't planned for it.
560  */
561  Assert(RelationGetTargetBlock(intoRelationDesc) == InvalidBlockNumber);
562 }
#define NIL
Definition: pg_list.h:65
Oid typeOid
Definition: parsenodes.h:211
int errhint(const char *fmt,...)
Definition: elog.c:1071
static ListCell * lnext(const List *l, const ListCell *c)
Definition: pg_list.h:321
#define TupleDescAttr(tupdesc, i)
Definition: tupdesc.h:92
int ti_options
Definition: createas.c:62
#define strVal(v)
Definition: value.h:54
int errcode(int sqlerrcode)
Definition: elog.c:610
char * format_type_be(Oid type_oid)
Definition: format_type.c:339
CommandId output_cid
Definition: createas.c:61
#define FirstLowInvalidHeapAttributeNumber
Definition: sysattr.h:27
AclMode requiredPerms
Definition: parsenodes.h:1119
bool skipData
Definition: primnodes.h:119
BulkInsertState bistate
Definition: createas.c:63
#define OidIsValid(objectId)
Definition: c.h:651
#define RelationGetTargetBlock(relation)
Definition: rel.h:541
#define list_make1(x1)
Definition: pg_list.h:227
BulkInsertState GetBulkInsertState(void)
Definition: heapam.c:1799
ObjectAddress reladdr
Definition: createas.c:60
static ObjectAddress create_ctas_internal(List *attrList, IntoClause *into)
Definition: createas.c:85
#define ERROR
Definition: elog.h:43
Node * viewQuery
Definition: primnodes.h:118
Oid collOid
Definition: parsenodes.h:661
#define RowExclusiveLock
Definition: lockdefs.h:38
Relation rel
Definition: createas.c:59
static ListCell * list_head(const List *l)
Definition: pg_list.h:125
FormData_pg_attribute * Form_pg_attribute
Definition: pg_attribute.h:193
List * lappend(List *list, void *datum)
Definition: list.c:321
#define TABLE_INSERT_SKIP_FSM
Definition: tableam.h:133
TupleDesc rd_att
Definition: rel.h:110
#define InvalidOid
Definition: postgres_ext.h:36
int16 attnum
Definition: pg_attribute.h:79
#define ereport(elevel,...)
Definition: elog.h:144
int check_enable_rls(Oid relid, Oid checkAsUser, bool noError)
Definition: rls.c:52
#define makeNode(_type_)
Definition: nodes.h:577
#define Assert(condition)
Definition: c.h:745
#define lfirst(lc)
Definition: pg_list.h:190
#define ACL_INSERT
Definition: parsenodes.h:74
#define InvalidBlockNumber
Definition: block.h:33
TypeName * typeName
Definition: parsenodes.h:648
Bitmapset * bms_add_member(Bitmapset *a, int x)
Definition: bitmapset.c:736
IntoClause * into
Definition: createas.c:57
RTEKind rtekind
Definition: parsenodes.h:976
#define AccessExclusiveLock
Definition: lockdefs.h:45
int errmsg(const char *fmt,...)
Definition: elog.c:824
Bitmapset * insertedCols
Definition: parsenodes.h:1122
bool ExecCheckRTPerms(List *rangeTable, bool ereport_on_violation)
Definition: execMain.c:571
#define NameStr(name)
Definition: c.h:622
char * colname
Definition: parsenodes.h:647
CommandId GetCurrentCommandId(bool used)
Definition: xact.c:761
List * colNames
Definition: primnodes.h:113
Relation table_open(Oid relationId, LOCKMODE lockmode)
Definition: table.c:39
bool type_is_collatable(Oid typid)
Definition: lsyscache.c:2958
Definition: pg_list.h:50
void SetMatViewPopulatedState(Relation relation, bool newstate)
Definition: matview.c:84
ColumnDef * makeColumnDef(const char *colname, Oid typeOid, int32 typmod, Oid collOid)
Definition: makefuncs.c:490