PostgreSQL Source Code  git master
dest.c
Go to the documentation of this file.
1 /*-------------------------------------------------------------------------
2  *
3  * dest.c
4  * support for communication destinations
5  *
6  *
7  * Portions Copyright (c) 1996-2023, PostgreSQL Global Development Group
8  * Portions Copyright (c) 1994, Regents of the University of California
9  *
10  * IDENTIFICATION
11  * src/backend/tcop/dest.c
12  *
13  *-------------------------------------------------------------------------
14  */
15 /*
16  * INTERFACE ROUTINES
17  * BeginCommand - initialize the destination at start of command
18  * CreateDestReceiver - create tuple receiver object for destination
19  * EndCommand - clean up the destination at end of command
20  * NullCommand - tell dest that an empty query string was recognized
21  * ReadyForQuery - tell dest that we are ready for a new query
22  *
23  * NOTES
24  * These routines do the appropriate work before and after
25  * tuples are returned by a query to keep the backend and the
26  * "destination" portals synchronized.
27  */
28 
29 #include "postgres.h"
30 
31 #include "access/printsimple.h"
32 #include "access/printtup.h"
33 #include "access/xact.h"
34 #include "commands/copy.h"
35 #include "commands/createas.h"
36 #include "commands/matview.h"
37 #include "executor/functions.h"
38 #include "executor/tqueue.h"
40 #include "libpq/libpq.h"
41 #include "libpq/pqformat.h"
42 #include "utils/portal.h"
43 
44 
45 /* ----------------
46  * dummy DestReceiver functions
47  * ----------------
48  */
49 static bool
51 {
52  return true;
53 }
54 
55 static void
56 donothingStartup(DestReceiver *self, int operation, TupleDesc typeinfo)
57 {
58 }
59 
60 static void
62 {
63  /* this is used for both shutdown and destroy methods */
64 }
65 
66 /* ----------------
67  * static DestReceiver structs for dest types needing no local state
68  * ----------------
69  */
70 static const DestReceiver donothingDR = {
72  DestNone
73 };
74 
75 static const DestReceiver debugtupDR = {
77  DestDebug
78 };
79 
80 static const DestReceiver printsimpleDR = {
83 };
84 
85 static const DestReceiver spi_printtupDR = {
87  DestSPI
88 };
89 
90 /*
91  * Globally available receiver for DestNone.
92  *
93  * It's ok to cast the constness away as any modification of the none receiver
94  * would be a bug (which gets easier to catch this way).
95  */
97 
98 /* ----------------
99  * BeginCommand - initialize the destination at start of command
100  * ----------------
101  */
102 void
104 {
105  /* Nothing to do at present */
106 }
107 
108 /* ----------------
109  * CreateDestReceiver - return appropriate receiver function set for dest
110  * ----------------
111  */
112 DestReceiver *
114 {
115  /*
116  * It's ok to cast the constness away as any modification of the none
117  * receiver would be a bug (which gets easier to catch this way).
118  */
119 
120  switch (dest)
121  {
122  case DestRemote:
123  case DestRemoteExecute:
124  return printtup_create_DR(dest);
125 
126  case DestRemoteSimple:
128 
129  case DestNone:
130  return unconstify(DestReceiver *, &donothingDR);
131 
132  case DestDebug:
133  return unconstify(DestReceiver *, &debugtupDR);
134 
135  case DestSPI:
137 
138  case DestTuplestore:
140 
141  case DestIntoRel:
142  return CreateIntoRelDestReceiver(NULL);
143 
144  case DestCopyOut:
145  return CreateCopyDestReceiver();
146 
147  case DestSQLFunction:
149 
150  case DestTransientRel:
152 
153  case DestTupleQueue:
154  return CreateTupleQueueDestReceiver(NULL);
155  }
156 
157  /* should never get here */
158  pg_unreachable();
159 }
160 
161 /* ----------------
162  * EndCommand - clean up the destination at end of command
163  * ----------------
164  */
165 void
166 EndCommand(const QueryCompletion *qc, CommandDest dest, bool force_undecorated_output)
167 {
168  char completionTag[COMPLETION_TAG_BUFSIZE];
169  Size len;
170 
171  switch (dest)
172  {
173  case DestRemote:
174  case DestRemoteExecute:
175  case DestRemoteSimple:
176 
177  len = BuildQueryCompletionString(completionTag, qc,
178  force_undecorated_output);
179  pq_putmessage('C', completionTag, len + 1);
180 
181  case DestNone:
182  case DestDebug:
183  case DestSPI:
184  case DestTuplestore:
185  case DestIntoRel:
186  case DestCopyOut:
187  case DestSQLFunction:
188  case DestTransientRel:
189  case DestTupleQueue:
190  break;
191  }
192 }
193 
194 /* ----------------
195  * EndReplicationCommand - stripped down version of EndCommand
196  *
197  * For use by replication commands.
198  * ----------------
199  */
200 void
201 EndReplicationCommand(const char *commandTag)
202 {
203  pq_putmessage('C', commandTag, strlen(commandTag) + 1);
204 }
205 
206 /* ----------------
207  * NullCommand - tell dest that an empty query string was recognized
208  *
209  * This ensures that there will be a recognizable end to the response
210  * to an Execute message in the extended query protocol.
211  * ----------------
212  */
213 void
215 {
216  switch (dest)
217  {
218  case DestRemote:
219  case DestRemoteExecute:
220  case DestRemoteSimple:
221 
222  /* Tell the FE that we saw an empty query string */
223  pq_putemptymessage('I');
224  break;
225 
226  case DestNone:
227  case DestDebug:
228  case DestSPI:
229  case DestTuplestore:
230  case DestIntoRel:
231  case DestCopyOut:
232  case DestSQLFunction:
233  case DestTransientRel:
234  case DestTupleQueue:
235  break;
236  }
237 }
238 
239 /* ----------------
240  * ReadyForQuery - tell dest that we are ready for a new query
241  *
242  * The ReadyForQuery message is sent so that the FE can tell when
243  * we are done processing a query string.
244  * In versions 3.0 and up, it also carries a transaction state indicator.
245  *
246  * Note that by flushing the stdio buffer here, we can avoid doing it
247  * most other places and thus reduce the number of separate packets sent.
248  * ----------------
249  */
250 void
252 {
253  switch (dest)
254  {
255  case DestRemote:
256  case DestRemoteExecute:
257  case DestRemoteSimple:
258  {
260 
261  pq_beginmessage(&buf, 'Z');
263  pq_endmessage(&buf);
264  }
265  /* Flush output at end of cycle in any case. */
266  pq_flush();
267  break;
268 
269  case DestNone:
270  case DestDebug:
271  case DestSPI:
272  case DestTuplestore:
273  case DestIntoRel:
274  case DestCopyOut:
275  case DestSQLFunction:
276  case DestTransientRel:
277  case DestTupleQueue:
278  break;
279  }
280 }
#define unconstify(underlying_type, expr)
Definition: c.h:1232
#define pg_unreachable()
Definition: c.h:280
size_t Size
Definition: c.h:589
Size BuildQueryCompletionString(char *buff, const QueryCompletion *qc, bool nameonly)
Definition: cmdtag.c:122
#define COMPLETION_TAG_BUFSIZE
Definition: cmdtag.h:17
CommandTag
Definition: cmdtag.h:23
DestReceiver * CreateCopyDestReceiver(void)
Definition: copyto.c:1275
DestReceiver * CreateIntoRelDestReceiver(IntoClause *intoClause)
Definition: createas.c:440
void EndCommand(const QueryCompletion *qc, CommandDest dest, bool force_undecorated_output)
Definition: dest.c:166
static const DestReceiver spi_printtupDR
Definition: dest.c:85
static void donothingStartup(DestReceiver *self, int operation, TupleDesc typeinfo)
Definition: dest.c:56
static void donothingCleanup(DestReceiver *self)
Definition: dest.c:61
void BeginCommand(CommandTag commandTag, CommandDest dest)
Definition: dest.c:103
static const DestReceiver debugtupDR
Definition: dest.c:75
static const DestReceiver donothingDR
Definition: dest.c:70
void ReadyForQuery(CommandDest dest)
Definition: dest.c:251
static bool donothingReceive(TupleTableSlot *slot, DestReceiver *self)
Definition: dest.c:50
void EndReplicationCommand(const char *commandTag)
Definition: dest.c:201
DestReceiver * None_Receiver
Definition: dest.c:96
static const DestReceiver printsimpleDR
Definition: dest.c:80
DestReceiver * CreateDestReceiver(CommandDest dest)
Definition: dest.c:113
void NullCommand(CommandDest dest)
Definition: dest.c:214
CommandDest
Definition: dest.h:86
@ DestSQLFunction
Definition: dest.h:96
@ DestTupleQueue
Definition: dest.h:98
@ DestTuplestore
Definition: dest.h:93
@ DestRemote
Definition: dest.h:89
@ DestDebug
Definition: dest.h:88
@ DestRemoteSimple
Definition: dest.h:91
@ DestCopyOut
Definition: dest.h:95
@ DestTransientRel
Definition: dest.h:97
@ DestSPI
Definition: dest.h:92
@ DestIntoRel
Definition: dest.h:94
@ DestRemoteExecute
Definition: dest.h:90
@ DestNone
Definition: dest.h:87
DestReceiver * CreateSQLFunctionDestReceiver(void)
Definition: functions.c:2059
#define pq_flush()
Definition: libpq.h:46
#define pq_putmessage(msgtype, s, len)
Definition: libpq.h:49
DestReceiver * CreateTransientRelDestReceiver(Oid transientoid)
Definition: matview.c:436
const void size_t len
static char * buf
Definition: pg_test_fsync.c:67
#define InvalidOid
Definition: postgres_ext.h:36
void pq_putemptymessage(char msgtype)
Definition: pqformat.c:391
void pq_endmessage(StringInfo buf)
Definition: pqformat.c:299
void pq_beginmessage(StringInfo buf, char msgtype)
Definition: pqformat.c:88
static void pq_sendbyte(StringInfo buf, uint8 byt)
Definition: pqformat.h:161
void printsimple_startup(DestReceiver *self, int operation, TupleDesc tupdesc)
Definition: printsimple.c:30
bool printsimple(TupleTableSlot *slot, DestReceiver *self)
Definition: printsimple.c:58
bool debugtup(TupleTableSlot *slot, DestReceiver *self)
Definition: printtup.c:459
void debugStartup(DestReceiver *self, int operation, TupleDesc typeinfo)
Definition: printtup.c:441
DestReceiver * printtup_create_DR(CommandDest dest)
Definition: printtup.c:71
bool spi_printtup(TupleTableSlot *slot, DestReceiver *self)
Definition: spi.c:2161
void spi_dest_startup(DestReceiver *self, int operation, TupleDesc typeinfo)
Definition: spi.c:2113
DestReceiver * CreateTupleQueueDestReceiver(shm_mq_handle *handle)
Definition: tqueue.c:119
DestReceiver * CreateTuplestoreDestReceiver(void)
char TransactionBlockStatusCode(void)
Definition: xact.c:4855