Line data Source code
1 : /*-------------------------------------------------------------------------
2 : *
3 : * tuptable.h
4 : * tuple table support stuff
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 : * src/include/executor/tuptable.h
11 : *
12 : *-------------------------------------------------------------------------
13 : */
14 : #ifndef TUPTABLE_H
15 : #define TUPTABLE_H
16 :
17 : #include "access/htup.h"
18 : #include "access/htup_details.h"
19 : #include "access/sysattr.h"
20 : #include "access/tupdesc.h"
21 : #include "storage/buf.h"
22 :
23 : /*----------
24 : * The executor stores tuples in a "tuple table" which is a List of
25 : * independent TupleTableSlots.
26 : *
27 : * There's various different types of tuple table slots, each being able to
28 : * store different types of tuples. Additional types of slots can be added
29 : * without modifying core code. The type of a slot is determined by the
30 : * TupleTableSlotOps* passed to the slot creation routine. The builtin types
31 : * of slots are
32 : *
33 : * 1. physical tuple in a disk buffer page (TTSOpsBufferHeapTuple)
34 : * 2. physical tuple constructed in palloc'ed memory (TTSOpsHeapTuple)
35 : * 3. "minimal" physical tuple constructed in palloc'ed memory
36 : * (TTSOpsMinimalTuple)
37 : * 4. "virtual" tuple consisting of Datum/isnull arrays (TTSOpsVirtual)
38 : *
39 : *
40 : * The first two cases are similar in that they both deal with "materialized"
41 : * tuples, but resource management is different. For a tuple in a disk page
42 : * we need to hold a pin on the buffer until the TupleTableSlot's reference
43 : * to the tuple is dropped; while for a palloc'd tuple we usually want the
44 : * tuple pfree'd when the TupleTableSlot's reference is dropped.
45 : *
46 : * A "minimal" tuple is handled similarly to a palloc'd regular tuple.
47 : * At present, minimal tuples never are stored in buffers, so there is no
48 : * parallel to case 1. Note that a minimal tuple has no "system columns".
49 : *
50 : * A "virtual" tuple is an optimization used to minimize physical data copying
51 : * in a nest of plan nodes. Until materialized pass-by-reference Datums in
52 : * the slot point to storage that is not directly associated with the
53 : * TupleTableSlot; generally they will point to part of a tuple stored in a
54 : * lower plan node's output TupleTableSlot, or to a function result
55 : * constructed in a plan node's per-tuple econtext. It is the responsibility
56 : * of the generating plan node to be sure these resources are not released for
57 : * as long as the virtual tuple needs to be valid or is materialized. Note
58 : * also that a virtual tuple does not have any "system columns".
59 : *
60 : * The Datum/isnull arrays of a TupleTableSlot serve double duty. For virtual
61 : * slots they are the authoritative data. For the other builtin slots,
62 : * the arrays contain data extracted from the tuple. (In this state, any
63 : * pass-by-reference Datums point into the physical tuple.) The extracted
64 : * information is built "lazily", ie, only as needed. This serves to avoid
65 : * repeated extraction of data from the physical tuple.
66 : *
67 : * A TupleTableSlot can also be "empty", indicated by flag TTS_FLAG_EMPTY set
68 : * in tts_flags, holding no valid data. This is the only valid state for a
69 : * freshly-created slot that has not yet had a tuple descriptor assigned to
70 : * it. In this state, TTS_FLAG_SHOULDFREE should not be set in tts_flags and
71 : * tts_nvalid should be set to zero.
72 : *
73 : * The tupleDescriptor is simply referenced, not copied, by the TupleTableSlot
74 : * code. The caller of ExecSetSlotDescriptor() is responsible for providing
75 : * a descriptor that will live as long as the slot does. (Typically, both
76 : * slots and descriptors are in per-query memory and are freed by memory
77 : * context deallocation at query end; so it's not worth providing any extra
78 : * mechanism to do more. However, the slot will increment the tupdesc
79 : * reference count if a reference-counted tupdesc is supplied.)
80 : *
81 : * When TTS_FLAG_SHOULDFREE is set in tts_flags, the physical tuple is "owned"
82 : * by the slot and should be freed when the slot's reference to the tuple is
83 : * dropped.
84 : *
85 : * tts_values/tts_isnull are allocated either when the slot is created (when
86 : * the descriptor is provided), or when a descriptor is assigned to the slot;
87 : * they are of length equal to the descriptor's natts.
88 : *
89 : * The TTS_FLAG_SLOW flag is saved state for
90 : * slot_deform_heap_tuple, and should not be touched by any other code.
91 : *----------
92 : */
93 :
94 : /* true = slot is empty */
95 : #define TTS_FLAG_EMPTY (1 << 1)
96 : #define TTS_EMPTY(slot) (((slot)->tts_flags & TTS_FLAG_EMPTY) != 0)
97 :
98 : /* should pfree tuple "owned" by the slot? */
99 : #define TTS_FLAG_SHOULDFREE (1 << 2)
100 : #define TTS_SHOULDFREE(slot) (((slot)->tts_flags & TTS_FLAG_SHOULDFREE) != 0)
101 :
102 : /* saved state for slot_deform_heap_tuple */
103 : #define TTS_FLAG_SLOW (1 << 3)
104 : #define TTS_SLOW(slot) (((slot)->tts_flags & TTS_FLAG_SLOW) != 0)
105 :
106 : /* fixed tuple descriptor */
107 : #define TTS_FLAG_FIXED (1 << 4)
108 : #define TTS_FIXED(slot) (((slot)->tts_flags & TTS_FLAG_FIXED) != 0)
109 :
110 : struct TupleTableSlotOps;
111 : typedef struct TupleTableSlotOps TupleTableSlotOps;
112 :
113 : /* base tuple table slot type */
114 : typedef struct TupleTableSlot
115 : {
116 : NodeTag type;
117 : #define FIELDNO_TUPLETABLESLOT_FLAGS 1
118 : uint16 tts_flags; /* Boolean states */
119 : #define FIELDNO_TUPLETABLESLOT_NVALID 2
120 : AttrNumber tts_nvalid; /* # of valid values in tts_values */
121 : const TupleTableSlotOps *const tts_ops; /* implementation of slot */
122 : #define FIELDNO_TUPLETABLESLOT_TUPLEDESCRIPTOR 4
123 : TupleDesc tts_tupleDescriptor; /* slot's tuple descriptor */
124 : #define FIELDNO_TUPLETABLESLOT_VALUES 5
125 : Datum *tts_values; /* current per-attribute values */
126 : #define FIELDNO_TUPLETABLESLOT_ISNULL 6
127 : bool *tts_isnull; /* current per-attribute isnull flags */
128 : MemoryContext tts_mcxt; /* slot itself is in this context */
129 : ItemPointerData tts_tid; /* stored tuple's tid */
130 : Oid tts_tableOid; /* table oid of tuple */
131 : } TupleTableSlot;
132 :
133 : /* routines for a TupleTableSlot implementation */
134 : struct TupleTableSlotOps
135 : {
136 : /* Minimum size of the slot */
137 : size_t base_slot_size;
138 :
139 : /* Initialization. */
140 : void (*init) (TupleTableSlot *slot);
141 :
142 : /* Destruction. */
143 : void (*release) (TupleTableSlot *slot);
144 :
145 : /*
146 : * Clear the contents of the slot. Only the contents are expected to be
147 : * cleared and not the tuple descriptor. Typically an implementation of
148 : * this callback should free the memory allocated for the tuple contained
149 : * in the slot.
150 : */
151 : void (*clear) (TupleTableSlot *slot);
152 :
153 : /*
154 : * Fill up first natts entries of tts_values and tts_isnull arrays with
155 : * values from the tuple contained in the slot. The function may be called
156 : * with natts more than the number of attributes available in the tuple,
157 : * in which case it should set tts_nvalid to the number of returned
158 : * columns.
159 : */
160 : void (*getsomeattrs) (TupleTableSlot *slot, int natts);
161 :
162 : /*
163 : * Returns value of the given system attribute as a datum and sets isnull
164 : * to false, if it's not NULL. Throws an error if the slot type does not
165 : * support system attributes.
166 : */
167 : Datum (*getsysattr) (TupleTableSlot *slot, int attnum, bool *isnull);
168 :
169 : /*
170 : * Make the contents of the slot solely depend on the slot, and not on
171 : * underlying resources (like another memory context, buffers, etc).
172 : */
173 : void (*materialize) (TupleTableSlot *slot);
174 :
175 : /*
176 : * Copy the contents of the source slot into the destination slot's own
177 : * context. Invoked using callback of the destination slot.
178 : */
179 : void (*copyslot) (TupleTableSlot *dstslot, TupleTableSlot *srcslot);
180 :
181 : /*
182 : * Return a heap tuple "owned" by the slot. It is slot's responsibility to
183 : * free the memory consumed by the heap tuple. If the slot can not "own" a
184 : * heap tuple, it should not implement this callback and should set it as
185 : * NULL.
186 : */
187 : HeapTuple (*get_heap_tuple) (TupleTableSlot *slot);
188 :
189 : /*
190 : * Return a minimal tuple "owned" by the slot. It is slot's responsibility
191 : * to free the memory consumed by the minimal tuple. If the slot can not
192 : * "own" a minimal tuple, it should not implement this callback and should
193 : * set it as NULL.
194 : */
195 : MinimalTuple (*get_minimal_tuple) (TupleTableSlot *slot);
196 :
197 : /*
198 : * Return a copy of heap tuple representing the contents of the slot. The
199 : * copy needs to be palloc'd in the current memory context. The slot
200 : * itself is expected to remain unaffected. It is *not* expected to have
201 : * meaningful "system columns" in the copy. The copy is not be "owned" by
202 : * the slot i.e. the caller has to take responsibility to free memory
203 : * consumed by the slot.
204 : */
205 : HeapTuple (*copy_heap_tuple) (TupleTableSlot *slot);
206 :
207 : /*
208 : * Return a copy of minimal tuple representing the contents of the slot.
209 : * The copy needs to be palloc'd in the current memory context. The slot
210 : * itself is expected to remain unaffected. It is *not* expected to have
211 : * meaningful "system columns" in the copy. The copy is not be "owned" by
212 : * the slot i.e. the caller has to take responsibility to free memory
213 : * consumed by the slot.
214 : */
215 : MinimalTuple (*copy_minimal_tuple) (TupleTableSlot *slot);
216 : };
217 :
218 : /*
219 : * Predefined TupleTableSlotOps for various types of TupleTableSlotOps. The
220 : * same are used to identify the type of a given slot.
221 : */
222 : extern PGDLLIMPORT const TupleTableSlotOps TTSOpsVirtual;
223 : extern PGDLLIMPORT const TupleTableSlotOps TTSOpsHeapTuple;
224 : extern PGDLLIMPORT const TupleTableSlotOps TTSOpsMinimalTuple;
225 : extern PGDLLIMPORT const TupleTableSlotOps TTSOpsBufferHeapTuple;
226 :
227 : #define TTS_IS_VIRTUAL(slot) ((slot)->tts_ops == &TTSOpsVirtual)
228 : #define TTS_IS_HEAPTUPLE(slot) ((slot)->tts_ops == &TTSOpsHeapTuple)
229 : #define TTS_IS_MINIMALTUPLE(slot) ((slot)->tts_ops == &TTSOpsMinimalTuple)
230 : #define TTS_IS_BUFFERTUPLE(slot) ((slot)->tts_ops == &TTSOpsBufferHeapTuple)
231 :
232 :
233 : /*
234 : * Tuple table slot implementations.
235 : */
236 :
237 : typedef struct VirtualTupleTableSlot
238 : {
239 : pg_node_attr(abstract)
240 :
241 : TupleTableSlot base;
242 :
243 : char *data; /* data for materialized slots */
244 : } VirtualTupleTableSlot;
245 :
246 : typedef struct HeapTupleTableSlot
247 : {
248 : pg_node_attr(abstract)
249 :
250 : TupleTableSlot base;
251 :
252 : #define FIELDNO_HEAPTUPLETABLESLOT_TUPLE 1
253 : HeapTuple tuple; /* physical tuple */
254 : #define FIELDNO_HEAPTUPLETABLESLOT_OFF 2
255 : uint32 off; /* saved state for slot_deform_heap_tuple */
256 : HeapTupleData tupdata; /* optional workspace for storing tuple */
257 : } HeapTupleTableSlot;
258 :
259 : /* heap tuple residing in a buffer */
260 : typedef struct BufferHeapTupleTableSlot
261 : {
262 : pg_node_attr(abstract)
263 :
264 : HeapTupleTableSlot base;
265 :
266 : /*
267 : * If buffer is not InvalidBuffer, then the slot is holding a pin on the
268 : * indicated buffer page; drop the pin when we release the slot's
269 : * reference to that buffer. (TTS_FLAG_SHOULDFREE should not be set in
270 : * such a case, since presumably base.tuple is pointing into the buffer.)
271 : */
272 : Buffer buffer; /* tuple's buffer, or InvalidBuffer */
273 : } BufferHeapTupleTableSlot;
274 :
275 : typedef struct MinimalTupleTableSlot
276 : {
277 : pg_node_attr(abstract)
278 :
279 : TupleTableSlot base;
280 :
281 : /*
282 : * In a minimal slot tuple points at minhdr and the fields of that struct
283 : * are set correctly for access to the minimal tuple; in particular,
284 : * minhdr.t_data points MINIMAL_TUPLE_OFFSET bytes before mintuple. This
285 : * allows column extraction to treat the case identically to regular
286 : * physical tuples.
287 : */
288 : #define FIELDNO_MINIMALTUPLETABLESLOT_TUPLE 1
289 : HeapTuple tuple; /* tuple wrapper */
290 : MinimalTuple mintuple; /* minimal tuple, or NULL if none */
291 : HeapTupleData minhdr; /* workspace for minimal-tuple-only case */
292 : #define FIELDNO_MINIMALTUPLETABLESLOT_OFF 4
293 : uint32 off; /* saved state for slot_deform_heap_tuple */
294 : } MinimalTupleTableSlot;
295 :
296 : /*
297 : * TupIsNull -- is a TupleTableSlot empty?
298 : */
299 : #define TupIsNull(slot) \
300 : ((slot) == NULL || TTS_EMPTY(slot))
301 :
302 : /* in executor/execTuples.c */
303 : extern TupleTableSlot *MakeTupleTableSlot(TupleDesc tupleDesc,
304 : const TupleTableSlotOps *tts_ops);
305 : extern TupleTableSlot *ExecAllocTableSlot(List **tupleTable, TupleDesc desc,
306 : const TupleTableSlotOps *tts_ops);
307 : extern void ExecResetTupleTable(List *tupleTable, bool shouldFree);
308 : extern TupleTableSlot *MakeSingleTupleTableSlot(TupleDesc tupdesc,
309 : const TupleTableSlotOps *tts_ops);
310 : extern void ExecDropSingleTupleTableSlot(TupleTableSlot *slot);
311 : extern void ExecSetSlotDescriptor(TupleTableSlot *slot, TupleDesc tupdesc);
312 : extern TupleTableSlot *ExecStoreHeapTuple(HeapTuple tuple,
313 : TupleTableSlot *slot,
314 : bool shouldFree);
315 : extern void ExecForceStoreHeapTuple(HeapTuple tuple,
316 : TupleTableSlot *slot,
317 : bool shouldFree);
318 : extern TupleTableSlot *ExecStoreBufferHeapTuple(HeapTuple tuple,
319 : TupleTableSlot *slot,
320 : Buffer buffer);
321 : extern TupleTableSlot *ExecStorePinnedBufferHeapTuple(HeapTuple tuple,
322 : TupleTableSlot *slot,
323 : Buffer buffer);
324 : extern TupleTableSlot *ExecStoreMinimalTuple(MinimalTuple mtup,
325 : TupleTableSlot *slot,
326 : bool shouldFree);
327 : extern void ExecForceStoreMinimalTuple(MinimalTuple mtup, TupleTableSlot *slot,
328 : bool shouldFree);
329 : extern TupleTableSlot *ExecStoreVirtualTuple(TupleTableSlot *slot);
330 : extern TupleTableSlot *ExecStoreAllNullTuple(TupleTableSlot *slot);
331 : extern void ExecStoreHeapTupleDatum(Datum data, TupleTableSlot *slot);
332 : extern HeapTuple ExecFetchSlotHeapTuple(TupleTableSlot *slot, bool materialize, bool *shouldFree);
333 : extern MinimalTuple ExecFetchSlotMinimalTuple(TupleTableSlot *slot,
334 : bool *shouldFree);
335 : extern Datum ExecFetchSlotHeapTupleDatum(TupleTableSlot *slot);
336 : extern void slot_getmissingattrs(TupleTableSlot *slot, int startAttNum,
337 : int lastAttNum);
338 : extern void slot_getsomeattrs_int(TupleTableSlot *slot, int attnum);
339 :
340 :
341 : #ifndef FRONTEND
342 :
343 : /*
344 : * This function forces the entries of the slot's Datum/isnull arrays to be
345 : * valid at least up through the attnum'th entry.
346 : */
347 : static inline void
348 189173454 : slot_getsomeattrs(TupleTableSlot *slot, int attnum)
349 : {
350 189173454 : if (slot->tts_nvalid < attnum)
351 150300284 : slot_getsomeattrs_int(slot, attnum);
352 189173454 : }
353 :
354 : /*
355 : * slot_getallattrs
356 : * This function forces all the entries of the slot's Datum/isnull
357 : * arrays to be valid. The caller may then extract data directly
358 : * from those arrays instead of using slot_getattr.
359 : */
360 : static inline void
361 14922868 : slot_getallattrs(TupleTableSlot *slot)
362 : {
363 14922868 : slot_getsomeattrs(slot, slot->tts_tupleDescriptor->natts);
364 14922868 : }
365 :
366 :
367 : /*
368 : * slot_attisnull
369 : *
370 : * Detect whether an attribute of the slot is null, without actually fetching
371 : * it.
372 : */
373 : static inline bool
374 8382854 : slot_attisnull(TupleTableSlot *slot, int attnum)
375 : {
376 : Assert(attnum > 0);
377 :
378 8382854 : if (attnum > slot->tts_nvalid)
379 6520278 : slot_getsomeattrs(slot, attnum);
380 :
381 8382854 : return slot->tts_isnull[attnum - 1];
382 : }
383 :
384 : /*
385 : * slot_getattr - fetch one attribute of the slot's contents.
386 : */
387 : static inline Datum
388 73945246 : slot_getattr(TupleTableSlot *slot, int attnum,
389 : bool *isnull)
390 : {
391 : Assert(attnum > 0);
392 :
393 73945246 : if (attnum > slot->tts_nvalid)
394 48074128 : slot_getsomeattrs(slot, attnum);
395 :
396 73945246 : *isnull = slot->tts_isnull[attnum - 1];
397 :
398 73945246 : return slot->tts_values[attnum - 1];
399 : }
400 :
401 : /*
402 : * slot_getsysattr - fetch a system attribute of the slot's current tuple.
403 : *
404 : * If the slot type does not contain system attributes, this will throw an
405 : * error. Hence before calling this function, callers should make sure that
406 : * the slot type is the one that supports system attributes.
407 : */
408 : static inline Datum
409 6370194 : slot_getsysattr(TupleTableSlot *slot, int attnum, bool *isnull)
410 : {
411 : Assert(attnum < 0); /* caller error */
412 :
413 6370194 : if (attnum == TableOidAttributeNumber)
414 : {
415 2108228 : *isnull = false;
416 2108228 : return ObjectIdGetDatum(slot->tts_tableOid);
417 : }
418 4261966 : else if (attnum == SelfItemPointerAttributeNumber)
419 : {
420 4260694 : *isnull = false;
421 4260694 : return PointerGetDatum(&slot->tts_tid);
422 : }
423 :
424 : /* Fetch the system attribute from the underlying tuple. */
425 1272 : return slot->tts_ops->getsysattr(slot, attnum, isnull);
426 : }
427 :
428 : /*
429 : * ExecClearTuple - clear the slot's contents
430 : */
431 : static inline TupleTableSlot *
432 132483090 : ExecClearTuple(TupleTableSlot *slot)
433 : {
434 132483090 : slot->tts_ops->clear(slot);
435 :
436 132483090 : return slot;
437 : }
438 :
439 : /* ExecMaterializeSlot - force a slot into the "materialized" state.
440 : *
441 : * This causes the slot's tuple to be a local copy not dependent on any
442 : * external storage (i.e. pointing into a Buffer, or having allocations in
443 : * another memory context).
444 : *
445 : * A typical use for this operation is to prepare a computed tuple for being
446 : * stored on disk. The original data may or may not be virtual, but in any
447 : * case we need a private copy for heap_insert to scribble on.
448 : */
449 : static inline void
450 14767480 : ExecMaterializeSlot(TupleTableSlot *slot)
451 : {
452 14767480 : slot->tts_ops->materialize(slot);
453 14767480 : }
454 :
455 : /*
456 : * ExecCopySlotHeapTuple - return HeapTuple allocated in caller's context
457 : */
458 : static inline HeapTuple
459 20116630 : ExecCopySlotHeapTuple(TupleTableSlot *slot)
460 : {
461 : Assert(!TTS_EMPTY(slot));
462 :
463 20116630 : return slot->tts_ops->copy_heap_tuple(slot);
464 : }
465 :
466 : /*
467 : * ExecCopySlotMinimalTuple - return MinimalTuple allocated in caller's context
468 : */
469 : static inline MinimalTuple
470 14979628 : ExecCopySlotMinimalTuple(TupleTableSlot *slot)
471 : {
472 14979628 : return slot->tts_ops->copy_minimal_tuple(slot);
473 : }
474 :
475 : /*
476 : * ExecCopySlot - copy one slot's contents into another.
477 : *
478 : * If a source's system attributes are supposed to be accessed in the target
479 : * slot, the target slot and source slot types need to match.
480 : */
481 : static inline TupleTableSlot *
482 11660194 : ExecCopySlot(TupleTableSlot *dstslot, TupleTableSlot *srcslot)
483 : {
484 : Assert(!TTS_EMPTY(srcslot));
485 : Assert(srcslot != dstslot);
486 :
487 11660194 : dstslot->tts_ops->copyslot(dstslot, srcslot);
488 :
489 11660194 : return dstslot;
490 : }
491 :
492 : #endif /* FRONTEND */
493 :
494 : #endif /* TUPTABLE_H */
|
