PostgreSQL Source Code  git master
test_ddl_deparse.c
Go to the documentation of this file.
1 /*----------------------------------------------------------------------
2  * test_ddl_deparse.c
3  * Support functions for the test_ddl_deparse module
4  *
5  * Copyright (c) 2014-2024, PostgreSQL Global Development Group
6  *
7  * IDENTIFICATION
8  * src/test/modules/test_ddl_deparse/test_ddl_deparse.c
9  *----------------------------------------------------------------------
10  */
11 #include "postgres.h"
12 
13 #include "catalog/pg_type.h"
14 #include "funcapi.h"
15 #include "nodes/execnodes.h"
16 #include "tcop/deparse_utility.h"
17 #include "tcop/utility.h"
18 #include "utils/builtins.h"
19 
21 
25 
26 /*
27  * Return the textual representation of the struct type used to represent a
28  * command in struct CollectedCommand format.
29  */
30 Datum
32 {
34  const char *type;
35 
36  switch (cmd->type)
37  {
38  case SCT_Simple:
39  type = "simple";
40  break;
41  case SCT_AlterTable:
42  type = "alter table";
43  break;
44  case SCT_Grant:
45  type = "grant";
46  break;
47  case SCT_AlterOpFamily:
48  type = "alter operator family";
49  break;
51  type = "alter default privileges";
52  break;
53  case SCT_CreateOpClass:
54  type = "create operator class";
55  break;
56  case SCT_AlterTSConfig:
57  type = "alter text search configuration";
58  break;
59  default:
60  type = "unknown command type";
61  break;
62  }
63 
65 }
66 
67 /*
68  * Return the command tag corresponding to a parse node contained in a
69  * CollectedCommand struct.
70  */
71 Datum
73 {
75 
76  if (!cmd->parsetree)
78 
80 }
81 
82 /*
83  * Return a text array representation of the subcommands of an ALTER TABLE
84  * command.
85  */
86 Datum
88 {
90  ListCell *cell;
91  ReturnSetInfo *rsinfo = (ReturnSetInfo *) fcinfo->resultinfo;
92 
93  if (cmd->type != SCT_AlterTable)
94  elog(ERROR, "command is not ALTER TABLE");
95 
96  InitMaterializedSRF(fcinfo, 0);
97 
98  if (cmd->d.alterTable.subcmds == NIL)
99  elog(ERROR, "empty alter table subcommand list");
100 
101  foreach(cell, cmd->d.alterTable.subcmds)
102  {
103  CollectedATSubcmd *sub = lfirst(cell);
105  const char *strtype = "unrecognized";
106  Datum values[2];
107  bool nulls[2];
108 
109  memset(values, 0, sizeof(values));
110  memset(nulls, 0, sizeof(nulls));
111 
112  switch (subcmd->subtype)
113  {
114  case AT_AddColumn:
115  strtype = "ADD COLUMN";
116  break;
117  case AT_AddColumnToView:
118  strtype = "ADD COLUMN TO VIEW";
119  break;
120  case AT_ColumnDefault:
121  strtype = "ALTER COLUMN SET DEFAULT";
122  break;
124  strtype = "ALTER COLUMN SET DEFAULT (precooked)";
125  break;
126  case AT_DropNotNull:
127  strtype = "DROP NOT NULL";
128  break;
129  case AT_SetNotNull:
130  strtype = "SET NOT NULL";
131  break;
132  case AT_SetExpression:
133  strtype = "SET EXPRESSION";
134  break;
135  case AT_DropExpression:
136  strtype = "DROP EXPRESSION";
137  break;
138  case AT_CheckNotNull:
139  strtype = "CHECK NOT NULL";
140  break;
141  case AT_SetStatistics:
142  strtype = "SET STATS";
143  break;
144  case AT_SetOptions:
145  strtype = "SET OPTIONS";
146  break;
147  case AT_ResetOptions:
148  strtype = "RESET OPTIONS";
149  break;
150  case AT_SetStorage:
151  strtype = "SET STORAGE";
152  break;
153  case AT_SetCompression:
154  strtype = "SET COMPRESSION";
155  break;
156  case AT_DropColumn:
157  strtype = "DROP COLUMN";
158  break;
159  case AT_AddIndex:
160  strtype = "ADD INDEX";
161  break;
162  case AT_ReAddIndex:
163  strtype = "(re) ADD INDEX";
164  break;
165  case AT_AddConstraint:
166  strtype = "ADD CONSTRAINT";
167  break;
168  case AT_ReAddConstraint:
169  strtype = "(re) ADD CONSTRAINT";
170  break;
172  strtype = "(re) ADD DOMAIN CONSTRAINT";
173  break;
174  case AT_AlterConstraint:
175  strtype = "ALTER CONSTRAINT";
176  break;
178  strtype = "VALIDATE CONSTRAINT";
179  break;
181  strtype = "ADD CONSTRAINT (using index)";
182  break;
183  case AT_DropConstraint:
184  strtype = "DROP CONSTRAINT";
185  break;
186  case AT_ReAddComment:
187  strtype = "(re) ADD COMMENT";
188  break;
189  case AT_AlterColumnType:
190  strtype = "ALTER COLUMN SET TYPE";
191  break;
193  strtype = "ALTER COLUMN SET OPTIONS";
194  break;
195  case AT_ChangeOwner:
196  strtype = "CHANGE OWNER";
197  break;
198  case AT_ClusterOn:
199  strtype = "CLUSTER";
200  break;
201  case AT_DropCluster:
202  strtype = "DROP CLUSTER";
203  break;
204  case AT_SetLogged:
205  strtype = "SET LOGGED";
206  break;
207  case AT_SetUnLogged:
208  strtype = "SET UNLOGGED";
209  break;
210  case AT_DropOids:
211  strtype = "DROP OIDS";
212  break;
213  case AT_SetAccessMethod:
214  strtype = "SET ACCESS METHOD";
215  break;
216  case AT_SetTableSpace:
217  strtype = "SET TABLESPACE";
218  break;
219  case AT_SetRelOptions:
220  strtype = "SET RELOPTIONS";
221  break;
222  case AT_ResetRelOptions:
223  strtype = "RESET RELOPTIONS";
224  break;
226  strtype = "REPLACE RELOPTIONS";
227  break;
228  case AT_EnableTrig:
229  strtype = "ENABLE TRIGGER";
230  break;
231  case AT_EnableAlwaysTrig:
232  strtype = "ENABLE TRIGGER (always)";
233  break;
235  strtype = "ENABLE TRIGGER (replica)";
236  break;
237  case AT_DisableTrig:
238  strtype = "DISABLE TRIGGER";
239  break;
240  case AT_EnableTrigAll:
241  strtype = "ENABLE TRIGGER (all)";
242  break;
243  case AT_DisableTrigAll:
244  strtype = "DISABLE TRIGGER (all)";
245  break;
246  case AT_EnableTrigUser:
247  strtype = "ENABLE TRIGGER (user)";
248  break;
249  case AT_DisableTrigUser:
250  strtype = "DISABLE TRIGGER (user)";
251  break;
252  case AT_EnableRule:
253  strtype = "ENABLE RULE";
254  break;
255  case AT_EnableAlwaysRule:
256  strtype = "ENABLE RULE (always)";
257  break;
259  strtype = "ENABLE RULE (replica)";
260  break;
261  case AT_DisableRule:
262  strtype = "DISABLE RULE";
263  break;
264  case AT_AddInherit:
265  strtype = "ADD INHERIT";
266  break;
267  case AT_DropInherit:
268  strtype = "DROP INHERIT";
269  break;
270  case AT_AddOf:
271  strtype = "OF";
272  break;
273  case AT_DropOf:
274  strtype = "NOT OF";
275  break;
276  case AT_ReplicaIdentity:
277  strtype = "REPLICA IDENTITY";
278  break;
280  strtype = "ENABLE ROW SECURITY";
281  break;
283  strtype = "DISABLE ROW SECURITY";
284  break;
285  case AT_ForceRowSecurity:
286  strtype = "FORCE ROW SECURITY";
287  break;
289  strtype = "NO FORCE ROW SECURITY";
290  break;
291  case AT_GenericOptions:
292  strtype = "SET OPTIONS";
293  break;
294  case AT_DetachPartition:
295  strtype = "DETACH PARTITION";
296  break;
297  case AT_AttachPartition:
298  strtype = "ATTACH PARTITION";
299  break;
301  strtype = "DETACH PARTITION ... FINALIZE";
302  break;
303  case AT_SplitPartition:
304  strtype = "SPLIT PARTITION";
305  break;
306  case AT_MergePartitions:
307  strtype = "MERGE PARTITIONS";
308  break;
309  case AT_AddIdentity:
310  strtype = "ADD IDENTITY";
311  break;
312  case AT_SetIdentity:
313  strtype = "SET IDENTITY";
314  break;
315  case AT_DropIdentity:
316  strtype = "DROP IDENTITY";
317  break;
318  case AT_ReAddStatistics:
319  strtype = "(re) ADD STATS";
320  break;
321  }
322 
323  if (subcmd->recurse)
324  values[0] = CStringGetTextDatum(psprintf("%s (and recurse)", strtype));
325  else
326  values[0] = CStringGetTextDatum(strtype);
327  if (OidIsValid(sub->address.objectId))
328  {
329  char *objdesc;
330 
331  objdesc = getObjectDescription((const ObjectAddress *) &sub->address, false);
332  values[1] = CStringGetTextDatum(objdesc);
333  }
334  else
335  nulls[1] = true;
336 
337  tuplestore_putvalues(rsinfo->setResult, rsinfo->setDesc, values, nulls);
338  }
339 
340  return (Datum) 0;
341 }
static Datum values[MAXATTR]
Definition: bootstrap.c:152
#define CStringGetTextDatum(s)
Definition: builtins.h:97
#define OidIsValid(objectId)
Definition: c.h:775
@ SCT_Simple
@ SCT_AlterTSConfig
@ SCT_AlterDefaultPrivileges
@ SCT_Grant
@ SCT_CreateOpClass
@ SCT_AlterOpFamily
@ SCT_AlterTable
#define ERROR
Definition: elog.h:39
#define elog(elevel,...)
Definition: elog.h:224
#define PG_GETARG_POINTER(n)
Definition: fmgr.h:276
#define PG_RETURN_NULL()
Definition: fmgr.h:345
#define PG_RETURN_TEXT_P(x)
Definition: fmgr.h:372
#define PG_FUNCTION_ARGS
Definition: fmgr.h:193
void InitMaterializedSRF(FunctionCallInfo fcinfo, bits32 flags)
Definition: funcapi.c:76
if(TABLE==NULL||TABLE_index==NULL)
Definition: isn.c:77
#define castNode(_type_, nodeptr)
Definition: nodes.h:176
char * getObjectDescription(const ObjectAddress *object, bool missing_ok)
@ AT_AddIndexConstraint
Definition: parsenodes.h:2375
@ AT_MergePartitions
Definition: parsenodes.h:2417
@ AT_DropOf
Definition: parsenodes.h:2406
@ AT_CheckNotNull
Definition: parsenodes.h:2361
@ AT_SetOptions
Definition: parsenodes.h:2363
@ AT_DropIdentity
Definition: parsenodes.h:2420
@ AT_DisableTrigUser
Definition: parsenodes.h:2398
@ AT_DropNotNull
Definition: parsenodes.h:2357
@ AT_AddOf
Definition: parsenodes.h:2405
@ AT_ResetOptions
Definition: parsenodes.h:2364
@ AT_ReplicaIdentity
Definition: parsenodes.h:2407
@ AT_ReplaceRelOptions
Definition: parsenodes.h:2390
@ AT_EnableRowSecurity
Definition: parsenodes.h:2408
@ AT_AddColumnToView
Definition: parsenodes.h:2354
@ AT_ResetRelOptions
Definition: parsenodes.h:2389
@ AT_EnableReplicaTrig
Definition: parsenodes.h:2393
@ AT_DropOids
Definition: parsenodes.h:2385
@ AT_SetIdentity
Definition: parsenodes.h:2419
@ AT_ReAddStatistics
Definition: parsenodes.h:2421
@ AT_SetUnLogged
Definition: parsenodes.h:2384
@ AT_DisableTrig
Definition: parsenodes.h:2394
@ AT_SetCompression
Definition: parsenodes.h:2366
@ AT_DropExpression
Definition: parsenodes.h:2360
@ AT_AddIndex
Definition: parsenodes.h:2368
@ AT_EnableReplicaRule
Definition: parsenodes.h:2401
@ AT_ReAddIndex
Definition: parsenodes.h:2369
@ AT_DropConstraint
Definition: parsenodes.h:2376
@ AT_SetNotNull
Definition: parsenodes.h:2358
@ AT_ClusterOn
Definition: parsenodes.h:2381
@ AT_AddIdentity
Definition: parsenodes.h:2418
@ AT_ForceRowSecurity
Definition: parsenodes.h:2410
@ AT_EnableAlwaysRule
Definition: parsenodes.h:2400
@ AT_SetAccessMethod
Definition: parsenodes.h:2386
@ AT_AlterColumnType
Definition: parsenodes.h:2378
@ AT_DetachPartitionFinalize
Definition: parsenodes.h:2415
@ AT_AddInherit
Definition: parsenodes.h:2403
@ AT_ReAddDomainConstraint
Definition: parsenodes.h:2372
@ AT_EnableTrig
Definition: parsenodes.h:2391
@ AT_DropColumn
Definition: parsenodes.h:2367
@ AT_ReAddComment
Definition: parsenodes.h:2377
@ AT_AlterColumnGenericOptions
Definition: parsenodes.h:2379
@ AT_DisableTrigAll
Definition: parsenodes.h:2396
@ AT_EnableRule
Definition: parsenodes.h:2399
@ AT_NoForceRowSecurity
Definition: parsenodes.h:2411
@ AT_DetachPartition
Definition: parsenodes.h:2414
@ AT_SetStatistics
Definition: parsenodes.h:2362
@ AT_AttachPartition
Definition: parsenodes.h:2413
@ AT_AddConstraint
Definition: parsenodes.h:2370
@ AT_DropInherit
Definition: parsenodes.h:2404
@ AT_EnableAlwaysTrig
Definition: parsenodes.h:2392
@ AT_SetLogged
Definition: parsenodes.h:2383
@ AT_SetStorage
Definition: parsenodes.h:2365
@ AT_DisableRule
Definition: parsenodes.h:2402
@ AT_DisableRowSecurity
Definition: parsenodes.h:2409
@ AT_SetRelOptions
Definition: parsenodes.h:2388
@ AT_ChangeOwner
Definition: parsenodes.h:2380
@ AT_EnableTrigUser
Definition: parsenodes.h:2397
@ AT_SetExpression
Definition: parsenodes.h:2359
@ AT_ReAddConstraint
Definition: parsenodes.h:2371
@ AT_SetTableSpace
Definition: parsenodes.h:2387
@ AT_GenericOptions
Definition: parsenodes.h:2412
@ AT_ColumnDefault
Definition: parsenodes.h:2355
@ AT_CookedColumnDefault
Definition: parsenodes.h:2356
@ AT_AlterConstraint
Definition: parsenodes.h:2373
@ AT_EnableTrigAll
Definition: parsenodes.h:2395
@ AT_SplitPartition
Definition: parsenodes.h:2416
@ AT_DropCluster
Definition: parsenodes.h:2382
@ AT_ValidateConstraint
Definition: parsenodes.h:2374
@ AT_AddColumn
Definition: parsenodes.h:2353
#define lfirst(lc)
Definition: pg_list.h:172
#define NIL
Definition: pg_list.h:68
uintptr_t Datum
Definition: postgres.h:64
char * psprintf(const char *fmt,...)
Definition: psprintf.c:46
AlterTableType subtype
Definition: parsenodes.h:2434
ObjectAddress address
CollectedCommandType type
union CollectedCommand::@118 d
struct CollectedCommand::@118::@120 alterTable
TupleDesc setDesc
Definition: execnodes.h:340
Tuplestorestate * setResult
Definition: execnodes.h:339
PG_MODULE_MAGIC
Datum get_altertable_subcmdinfo(PG_FUNCTION_ARGS)
Datum get_command_type(PG_FUNCTION_ARGS)
PG_FUNCTION_INFO_V1(get_command_type)
Datum get_command_tag(PG_FUNCTION_ARGS)
void tuplestore_putvalues(Tuplestorestate *state, TupleDesc tdesc, const Datum *values, const bool *isnull)
Definition: tuplestore.c:750
static const char * CreateCommandName(Node *parsetree)
Definition: utility.h:103
text * cstring_to_text(const char *s)
Definition: varlena.c:184
const char * type