| /*------------------------------------------------------------------------- |
| * |
| * heaptoast.c |
| * Heap-specific definitions for external and compressed storage |
| * of variable size attributes. |
| * |
| * Copyright (c) 2000-2023, PostgreSQL Global Development Group |
| * |
| * |
| * IDENTIFICATION |
| * src/backend/access/heap/heaptoast.c |
| * |
| * |
| * INTERFACE ROUTINES |
| * heap_toast_insert_or_update - |
| * Try to make a given tuple fit into one page by compressing |
| * or moving off attributes |
| * |
| * heap_toast_delete - |
| * Reclaim toast storage when a tuple is deleted |
| * |
| *------------------------------------------------------------------------- |
| */ |
| |
| #include "postgres.h" |
| |
| #include "access/detoast.h" |
| #include "access/genam.h" |
| #include "access/heapam.h" |
| #include "access/heaptoast.h" |
| #include "access/toast_helper.h" |
| #include "access/toast_internals.h" |
| #include "utils/fmgroids.h" |
| |
| |
| static void * |
| heap_toast_insert_or_update_generic(Relation rel, void *newtup, void *oldtup, |
| MemTupleBinding *pbind, int options, int toast_tuple_target, bool ismemtuple); |
| |
| /* ---------- |
| * heap_toast_delete - |
| * |
| * Cascaded delete toast-entries on DELETE |
| * ---------- |
| */ |
| void |
| heap_toast_delete(Relation rel, HeapTuple oldtup, bool is_speculative) |
| { |
| TupleDesc tupleDesc; |
| Datum toast_values[MaxHeapAttributeNumber]; |
| bool toast_isnull[MaxHeapAttributeNumber]; |
| |
| /* |
| * We should only ever be called for tuples of plain relations or |
| * materialized views --- recursing on a toast rel is bad news. |
| */ |
| Assert(rel->rd_rel->relkind == RELKIND_RELATION || |
| rel->rd_rel->relkind == RELKIND_MATVIEW); |
| |
| /* |
| * Get the tuple descriptor and break down the tuple into fields. |
| * |
| * NOTE: it's debatable whether to use heap_deform_tuple() here or just |
| * heap_getattr() only the varlena columns. The latter could win if there |
| * are few varlena columns and many non-varlena ones. However, |
| * heap_deform_tuple costs only O(N) while the heap_getattr way would cost |
| * O(N^2) if there are many varlena columns, so it seems better to err on |
| * the side of linear cost. (We won't even be here unless there's at |
| * least one varlena column, by the way.) |
| */ |
| tupleDesc = rel->rd_att; |
| |
| Assert(tupleDesc->natts <= MaxHeapAttributeNumber); |
| heap_deform_tuple(oldtup, tupleDesc, toast_values, toast_isnull); |
| |
| /* Do the real work. */ |
| toast_delete_external(rel, toast_values, toast_isnull, is_speculative); |
| } |
| |
| |
| /* ---------- |
| * heap_toast_insert_or_update - |
| * |
| * Delete no-longer-used toast-entries and create new ones to |
| * make the new tuple fit on INSERT or UPDATE |
| * |
| * Inputs: |
| * newtup: the candidate new tuple to be inserted |
| * oldtup: the old row version for UPDATE, or NULL for INSERT |
| * options: options to be passed to heap_insert() for toast rows |
| * Result: |
| * either newtup if no toasting is needed, or a palloc'd modified tuple |
| * that is what should actually get stored |
| * |
| * NOTE: neither newtup nor oldtup will be modified. This is a change |
| * from the pre-8.1 API of this routine. |
| * ---------- |
| */ |
| |
| HeapTuple |
| heap_toast_insert_or_update(Relation rel, HeapTuple newtup, HeapTuple oldtup, int options) |
| { |
| return (HeapTuple) heap_toast_insert_or_update_generic(rel, newtup, oldtup, NULL, options, TOAST_TUPLE_TARGET, false); |
| } |
| |
| MemTuple memtup_toast_insert_or_update(Relation rel, MemTuple newtup, MemTuple oldtup, |
| MemTupleBinding *pbind, int toast_tuple_target, int options) |
| { |
| return (MemTuple) heap_toast_insert_or_update_generic(rel, newtup, oldtup, pbind, toast_tuple_target, options, true); |
| } |
| |
| static void * |
| heap_toast_insert_or_update_generic(Relation rel, void *newtup, void *oldtup, |
| MemTupleBinding *pbind, int options, int toast_tuple_target, bool ismemtuple) |
| { |
| void *result_gtuple; |
| TupleDesc tupleDesc; |
| int numAttrs; |
| |
| Size maxDataLen; |
| Size hoff; |
| |
| bool toast_isnull[MaxHeapAttributeNumber]; |
| bool toast_oldisnull[MaxHeapAttributeNumber]; |
| Datum toast_values[MaxHeapAttributeNumber]; |
| Datum toast_oldvalues[MaxHeapAttributeNumber]; |
| ToastAttrInfo toast_attr[MaxHeapAttributeNumber]; |
| ToastTupleContext ttc; |
| |
| AssertImply(ismemtuple, pbind); |
| AssertImply(!ismemtuple, !pbind); |
| |
| /* |
| * Ignore the INSERT_SPECULATIVE option. Speculative insertions/super |
| * deletions just normally insert/delete the toast values. It seems |
| * easiest to deal with that here, instead on, potentially, multiple |
| * callers. |
| */ |
| options &= ~HEAP_INSERT_SPECULATIVE; |
| |
| /* |
| * We should only ever be called for tuples of plain relations or |
| * materialized views --- recursing on a toast rel is bad news. |
| */ |
| Assert(rel->rd_rel->relkind == RELKIND_RELATION || |
| rel->rd_rel->relkind == RELKIND_MATVIEW); |
| |
| /* |
| * Get the tuple descriptor and break down the tuple(s) into fields. |
| */ |
| tupleDesc = rel->rd_att; |
| numAttrs = tupleDesc->natts; |
| |
| Assert(numAttrs <= MaxHeapAttributeNumber); |
| |
| if (ismemtuple) |
| memtuple_deform((MemTuple) newtup, pbind, toast_values, toast_isnull); |
| else |
| heap_deform_tuple(newtup, tupleDesc, toast_values, toast_isnull); |
| |
| if (oldtup != NULL) |
| { |
| if (ismemtuple) |
| memtuple_deform((MemTuple) oldtup, pbind, toast_oldvalues, toast_oldisnull); |
| else |
| heap_deform_tuple(oldtup, tupleDesc, toast_oldvalues, toast_oldisnull); |
| } |
| |
| /* ---------- |
| * Prepare for toasting |
| * ---------- |
| */ |
| ttc.ttc_rel = rel; |
| ttc.ttc_values = toast_values; |
| ttc.ttc_isnull = toast_isnull; |
| if (oldtup == NULL) |
| { |
| ttc.ttc_oldvalues = NULL; |
| ttc.ttc_oldisnull = NULL; |
| } |
| else |
| { |
| ttc.ttc_oldvalues = toast_oldvalues; |
| ttc.ttc_oldisnull = toast_oldisnull; |
| } |
| ttc.ttc_attr = toast_attr; |
| toast_tuple_init(&ttc); |
| |
| /* ---------- |
| * Compress and/or save external until data fits into target length |
| * |
| * 1: Inline compress attributes with attstorage EXTENDED, and store very |
| * large attributes with attstorage EXTENDED or EXTERNAL external |
| * immediately |
| * 2: Store attributes with attstorage EXTENDED or EXTERNAL external |
| * 3: Inline compress attributes with attstorage MAIN |
| * 4: Store attributes with attstorage MAIN external |
| * ---------- |
| */ |
| |
| if (!ismemtuple) |
| { |
| /* compute header overhead --- this should match heap_form_tuple() */ |
| hoff = SizeofHeapTupleHeader; |
| if ((ttc.ttc_flags & TOAST_HAS_NULLS) != 0) |
| hoff += BITMAPLEN(numAttrs); |
| hoff = MAXALIGN(hoff); |
| /* now convert to a limit on the tuple data size */ |
| maxDataLen = RelationGetToastTupleTarget(rel, TOAST_TUPLE_TARGET) - hoff; |
| } |
| else |
| { |
| maxDataLen = toast_tuple_target; |
| hoff = -1; /* keep compiler quiet about using 'hoff' uninitialized */ |
| } |
| |
| /* |
| * Look for attributes with attstorage EXTENDED to compress. Also find |
| * large attributes with attstorage EXTENDED or EXTERNAL, and store them |
| * external. |
| */ |
| while (heap_compute_data_size(tupleDesc, |
| toast_values, toast_isnull) > maxDataLen) |
| { |
| int biggest_attno; |
| |
| biggest_attno = toast_tuple_find_biggest_attribute(&ttc, true, false); |
| if (biggest_attno < 0) |
| break; |
| |
| /* |
| * Attempt to compress it inline, if it has attstorage EXTENDED |
| */ |
| if (TupleDescAttr(tupleDesc, biggest_attno)->attstorage == TYPSTORAGE_EXTENDED) |
| toast_tuple_try_compression(&ttc, biggest_attno); |
| else |
| { |
| /* |
| * has attstorage EXTERNAL, ignore on subsequent compression |
| * passes |
| */ |
| toast_attr[biggest_attno].tai_colflags |= TOASTCOL_INCOMPRESSIBLE; |
| } |
| |
| /* |
| * If this value is by itself more than maxDataLen (after compression |
| * if any), push it out to the toast table immediately, if possible. |
| * This avoids uselessly compressing other fields in the common case |
| * where we have one long field and several short ones. |
| * |
| * XXX maybe the threshold should be less than maxDataLen? |
| */ |
| if (toast_attr[biggest_attno].tai_size > maxDataLen && |
| rel->rd_rel->reltoastrelid != InvalidOid) |
| toast_tuple_externalize(&ttc, biggest_attno, options); |
| } |
| |
| /* |
| * Second we look for attributes of attstorage EXTENDED or EXTERNAL that |
| * are still inline, and make them external. But skip this if there's no |
| * toast table to push them to. |
| */ |
| while (heap_compute_data_size(tupleDesc, |
| toast_values, toast_isnull) > maxDataLen && |
| rel->rd_rel->reltoastrelid != InvalidOid) |
| { |
| int biggest_attno; |
| |
| biggest_attno = toast_tuple_find_biggest_attribute(&ttc, false, false); |
| if (biggest_attno < 0) |
| break; |
| toast_tuple_externalize(&ttc, biggest_attno, options); |
| } |
| |
| /* |
| * Round 3 - this time we take attributes with storage MAIN into |
| * compression |
| */ |
| while (heap_compute_data_size(tupleDesc, |
| toast_values, toast_isnull) > maxDataLen) |
| { |
| int biggest_attno; |
| |
| biggest_attno = toast_tuple_find_biggest_attribute(&ttc, true, true); |
| if (biggest_attno < 0) |
| break; |
| |
| toast_tuple_try_compression(&ttc, biggest_attno); |
| } |
| |
| /* |
| * Finally we store attributes of type MAIN externally. At this point we |
| * increase the target tuple size, so that MAIN attributes aren't stored |
| * externally unless really necessary. |
| */ |
| /* |
| * FIXME: Should we do something like this with memtuples on |
| * AO tables too? Currently we do not increase the target tuple size for AO |
| * table, so there are occasions when columns of type 'm' will be stored |
| * out-of-line but they could otherwise be accommodated in-block |
| * c.f. upstream Postgres commit ca7c8168de76459380577eda56a3ed09b4f6195c |
| */ |
| if (!ismemtuple) |
| maxDataLen = TOAST_TUPLE_TARGET_MAIN - hoff; |
| |
| while (heap_compute_data_size(tupleDesc, |
| toast_values, toast_isnull) > maxDataLen && |
| rel->rd_rel->reltoastrelid != InvalidOid) |
| { |
| int biggest_attno; |
| |
| biggest_attno = toast_tuple_find_biggest_attribute(&ttc, false, true); |
| if (biggest_attno < 0) |
| break; |
| |
| toast_tuple_externalize(&ttc, biggest_attno, options); |
| } |
| |
| /* |
| * In the case we toasted any values, we need to build a new heap tuple |
| * with the changed values. |
| */ |
| if ((ttc.ttc_flags & TOAST_NEEDS_CHANGE) != 0) |
| { |
| if (ismemtuple) |
| { |
| result_gtuple = (void *) memtuple_form(pbind, toast_values, toast_isnull); |
| } |
| else |
| { |
| HeapTupleHeader olddata = ((HeapTuple) newtup)->t_data; |
| HeapTupleHeader new_data; |
| int32 new_header_len; |
| int32 new_data_len; |
| int32 new_tuple_len; |
| HeapTuple result_tuple; |
| |
| /* |
| * Calculate the new size of the tuple. |
| * |
| * Note: we used to assume here that the old tuple's t_hoff must equal |
| * the new_header_len value, but that was incorrect. The old tuple |
| * might have a smaller-than-current natts, if there's been an ALTER |
| * TABLE ADD COLUMN since it was stored; and that would lead to a |
| * different conclusion about the size of the null bitmap, or even |
| * whether there needs to be one at all. |
| */ |
| new_header_len = SizeofHeapTupleHeader; |
| if ((ttc.ttc_flags & TOAST_HAS_NULLS) != 0) |
| new_header_len += BITMAPLEN(numAttrs); |
| new_header_len = MAXALIGN(new_header_len); |
| new_data_len = heap_compute_data_size(tupleDesc, |
| toast_values, toast_isnull); |
| new_tuple_len = new_header_len + new_data_len; |
| |
| /* |
| * Allocate and zero the space needed, and fill HeapTupleData fields. |
| */ |
| result_tuple = (HeapTuple) palloc0(HEAPTUPLESIZE + new_tuple_len); |
| result_tuple->t_len = new_tuple_len; |
| result_tuple->t_self = ((HeapTuple) newtup)->t_self; |
| result_tuple->t_tableOid = ((HeapTuple) newtup)->t_tableOid; |
| new_data = (HeapTupleHeader) ((char *) result_tuple + HEAPTUPLESIZE); |
| result_tuple->t_data = new_data; |
| |
| /* |
| * Copy the existing tuple header, but adjust natts and t_hoff. |
| */ |
| memcpy(new_data, olddata, SizeofHeapTupleHeader); |
| HeapTupleHeaderSetNatts(new_data, numAttrs); |
| new_data->t_hoff = new_header_len; |
| |
| /* Copy over the data, and fill the null bitmap if needed */ |
| heap_fill_tuple(tupleDesc, |
| toast_values, |
| toast_isnull, |
| (char *) new_data + new_header_len, |
| new_data_len, |
| &(new_data->t_infomask), |
| ((ttc.ttc_flags & TOAST_HAS_NULLS) != 0) ? |
| new_data->t_bits : NULL); |
| result_gtuple = (void *) result_tuple; |
| } |
| } |
| else |
| result_gtuple = newtup; |
| |
| toast_tuple_cleanup(&ttc); |
| |
| return result_gtuple; |
| } |
| |
| |
| /* ---------- |
| * toast_flatten_tuple - |
| * |
| * "Flatten" a tuple to contain no out-of-line toasted fields. |
| * (This does not eliminate compressed or short-header datums.) |
| * |
| * Note: we expect the caller already checked HeapTupleHasExternal(tup), |
| * so there is no need for a short-circuit path. |
| * ---------- |
| */ |
| HeapTuple |
| toast_flatten_tuple(HeapTuple tup, TupleDesc tupleDesc) |
| { |
| HeapTuple new_tuple; |
| int numAttrs = tupleDesc->natts; |
| int i; |
| Datum toast_values[MaxTupleAttributeNumber]; |
| bool toast_isnull[MaxTupleAttributeNumber]; |
| bool toast_free[MaxTupleAttributeNumber]; |
| |
| /* |
| * Break down the tuple into fields. |
| */ |
| Assert(numAttrs <= MaxTupleAttributeNumber); |
| heap_deform_tuple(tup, tupleDesc, toast_values, toast_isnull); |
| |
| memset(toast_free, 0, numAttrs * sizeof(bool)); |
| |
| for (i = 0; i < numAttrs; i++) |
| { |
| /* |
| * Look at non-null varlena attributes |
| */ |
| if (!toast_isnull[i] && TupleDescAttr(tupleDesc, i)->attlen == -1) |
| { |
| struct varlena *new_value; |
| |
| new_value = (struct varlena *) DatumGetPointer(toast_values[i]); |
| if (VARATT_IS_EXTERNAL(new_value)) |
| { |
| new_value = detoast_external_attr(new_value); |
| toast_values[i] = PointerGetDatum(new_value); |
| toast_free[i] = true; |
| } |
| } |
| } |
| |
| /* |
| * Form the reconfigured tuple. |
| */ |
| new_tuple = heap_form_tuple(tupleDesc, toast_values, toast_isnull); |
| |
| /* |
| * Be sure to copy the tuple's identity fields. We also make a point of |
| * copying visibility info, just in case anybody looks at those fields in |
| * a syscache entry. |
| */ |
| new_tuple->t_self = tup->t_self; |
| new_tuple->t_tableOid = tup->t_tableOid; |
| |
| new_tuple->t_data->t_choice = tup->t_data->t_choice; |
| new_tuple->t_data->t_ctid = tup->t_data->t_ctid; |
| new_tuple->t_data->t_infomask &= ~HEAP_XACT_MASK; |
| new_tuple->t_data->t_infomask |= |
| tup->t_data->t_infomask & HEAP_XACT_MASK; |
| new_tuple->t_data->t_infomask2 &= ~HEAP2_XACT_MASK; |
| new_tuple->t_data->t_infomask2 |= |
| tup->t_data->t_infomask2 & HEAP2_XACT_MASK; |
| |
| /* |
| * Free allocated temp values |
| */ |
| for (i = 0; i < numAttrs; i++) |
| if (toast_free[i]) |
| pfree(DatumGetPointer(toast_values[i])); |
| |
| return new_tuple; |
| } |
| |
| |
| /* ---------- |
| * toast_flatten_tuple_to_datum - |
| * |
| * "Flatten" a tuple containing out-of-line toasted fields into a Datum. |
| * The result is always palloc'd in the current memory context. |
| * |
| * We have a general rule that Datums of container types (rows, arrays, |
| * ranges, etc) must not contain any external TOAST pointers. Without |
| * this rule, we'd have to look inside each Datum when preparing a tuple |
| * for storage, which would be expensive and would fail to extend cleanly |
| * to new sorts of container types. |
| * |
| * However, we don't want to say that tuples represented as HeapTuples |
| * can't contain toasted fields, so instead this routine should be called |
| * when such a HeapTuple is being converted into a Datum. |
| * |
| * While we're at it, we decompress any compressed fields too. This is not |
| * necessary for correctness, but reflects an expectation that compression |
| * will be more effective if applied to the whole tuple not individual |
| * fields. We are not so concerned about that that we want to deconstruct |
| * and reconstruct tuples just to get rid of compressed fields, however. |
| * So callers typically won't call this unless they see that the tuple has |
| * at least one external field. |
| * |
| * On the other hand, in-line short-header varlena fields are left alone. |
| * If we "untoasted" them here, they'd just get changed back to short-header |
| * format anyway within heap_fill_tuple. |
| * ---------- |
| */ |
| Datum |
| toast_flatten_tuple_to_datum(HeapTupleHeader tup, |
| uint32 tup_len, |
| TupleDesc tupleDesc) |
| { |
| HeapTupleHeader new_data; |
| int32 new_header_len; |
| int32 new_data_len; |
| int32 new_tuple_len; |
| HeapTupleData tmptup; |
| int numAttrs = tupleDesc->natts; |
| int i; |
| bool has_nulls = false; |
| Datum toast_values[MaxTupleAttributeNumber]; |
| bool toast_isnull[MaxTupleAttributeNumber]; |
| bool toast_free[MaxTupleAttributeNumber]; |
| |
| /* Build a temporary HeapTuple control structure */ |
| tmptup.t_len = tup_len; |
| ItemPointerSetInvalid(&(tmptup.t_self)); |
| tmptup.t_tableOid = InvalidOid; |
| tmptup.t_data = tup; |
| |
| /* |
| * Break down the tuple into fields. |
| */ |
| Assert(numAttrs <= MaxTupleAttributeNumber); |
| heap_deform_tuple(&tmptup, tupleDesc, toast_values, toast_isnull); |
| |
| memset(toast_free, 0, numAttrs * sizeof(bool)); |
| |
| for (i = 0; i < numAttrs; i++) |
| { |
| /* |
| * Look at non-null varlena attributes |
| */ |
| if (toast_isnull[i]) |
| has_nulls = true; |
| else if (TupleDescAttr(tupleDesc, i)->attlen == -1) |
| { |
| struct varlena *new_value; |
| |
| new_value = (struct varlena *) DatumGetPointer(toast_values[i]); |
| if (VARATT_IS_EXTERNAL(new_value) || |
| VARATT_IS_COMPRESSED(new_value)) |
| { |
| new_value = detoast_attr(new_value); |
| toast_values[i] = PointerGetDatum(new_value); |
| toast_free[i] = true; |
| } |
| } |
| } |
| |
| /* |
| * Calculate the new size of the tuple. |
| * |
| * This should match the reconstruction code in |
| * heap_toast_insert_or_update. |
| */ |
| new_header_len = SizeofHeapTupleHeader; |
| if (has_nulls) |
| new_header_len += BITMAPLEN(numAttrs); |
| new_header_len = MAXALIGN(new_header_len); |
| new_data_len = heap_compute_data_size(tupleDesc, |
| toast_values, toast_isnull); |
| new_tuple_len = new_header_len + new_data_len; |
| |
| new_data = (HeapTupleHeader) palloc0(new_tuple_len); |
| |
| /* |
| * Copy the existing tuple header, but adjust natts and t_hoff. |
| */ |
| memcpy(new_data, tup, SizeofHeapTupleHeader); |
| HeapTupleHeaderSetNatts(new_data, numAttrs); |
| new_data->t_hoff = new_header_len; |
| |
| /* Set the composite-Datum header fields correctly */ |
| HeapTupleHeaderSetDatumLength(new_data, new_tuple_len); |
| HeapTupleHeaderSetTypeId(new_data, tupleDesc->tdtypeid); |
| HeapTupleHeaderSetTypMod(new_data, tupleDesc->tdtypmod); |
| |
| /* Copy over the data, and fill the null bitmap if needed */ |
| heap_fill_tuple(tupleDesc, |
| toast_values, |
| toast_isnull, |
| (char *) new_data + new_header_len, |
| new_data_len, |
| &(new_data->t_infomask), |
| has_nulls ? new_data->t_bits : NULL); |
| |
| /* |
| * Free allocated temp values |
| */ |
| for (i = 0; i < numAttrs; i++) |
| if (toast_free[i]) |
| pfree(DatumGetPointer(toast_values[i])); |
| |
| return PointerGetDatum(new_data); |
| } |
| |
| |
| /* ---------- |
| * toast_build_flattened_tuple - |
| * |
| * Build a tuple containing no out-of-line toasted fields. |
| * (This does not eliminate compressed or short-header datums.) |
| * |
| * This is essentially just like heap_form_tuple, except that it will |
| * expand any external-data pointers beforehand. |
| * |
| * It's not very clear whether it would be preferable to decompress |
| * in-line compressed datums while at it. For now, we don't. |
| * ---------- |
| */ |
| HeapTuple |
| toast_build_flattened_tuple(TupleDesc tupleDesc, |
| Datum *values, |
| bool *isnull) |
| { |
| HeapTuple new_tuple; |
| int numAttrs = tupleDesc->natts; |
| int num_to_free; |
| int i; |
| Datum new_values[MaxTupleAttributeNumber]; |
| Pointer freeable_values[MaxTupleAttributeNumber]; |
| |
| /* |
| * We can pass the caller's isnull array directly to heap_form_tuple, but |
| * we potentially need to modify the values array. |
| */ |
| Assert(numAttrs <= MaxTupleAttributeNumber); |
| memcpy(new_values, values, numAttrs * sizeof(Datum)); |
| |
| num_to_free = 0; |
| for (i = 0; i < numAttrs; i++) |
| { |
| /* |
| * Look at non-null varlena attributes |
| */ |
| if (!isnull[i] && TupleDescAttr(tupleDesc, i)->attlen == -1) |
| { |
| struct varlena *new_value; |
| |
| new_value = (struct varlena *) DatumGetPointer(new_values[i]); |
| if (VARATT_IS_EXTERNAL(new_value)) |
| { |
| new_value = detoast_external_attr(new_value); |
| new_values[i] = PointerGetDatum(new_value); |
| freeable_values[num_to_free++] = (Pointer) new_value; |
| } |
| } |
| } |
| |
| /* |
| * Form the reconfigured tuple. |
| */ |
| new_tuple = heap_form_tuple(tupleDesc, new_values, isnull); |
| |
| /* |
| * Free allocated temp values |
| */ |
| for (i = 0; i < num_to_free; i++) |
| pfree(freeable_values[i]); |
| |
| return new_tuple; |
| } |
| |
| /* |
| * Fetch a TOAST slice from a heap table. |
| * |
| * toastrel is the relation from which chunks are to be fetched. |
| * valueid identifies the TOAST value from which chunks are being fetched. |
| * attrsize is the total size of the TOAST value. |
| * sliceoffset is the byte offset within the TOAST value from which to fetch. |
| * slicelength is the number of bytes to be fetched from the TOAST value. |
| * result is the varlena into which the results should be written. |
| */ |
| void |
| heap_fetch_toast_slice(Relation toastrel, Oid valueid, int32 attrsize, |
| int32 sliceoffset, int32 slicelength, |
| struct varlena *result) |
| { |
| Relation *toastidxs; |
| ScanKeyData toastkey[3]; |
| TupleDesc toasttupDesc = toastrel->rd_att; |
| int nscankeys; |
| SysScanDesc toastscan; |
| HeapTuple ttup; |
| int32 expectedchunk; |
| int32 totalchunks = ((attrsize - 1) / TOAST_MAX_CHUNK_SIZE) + 1; |
| int startchunk; |
| int endchunk; |
| int num_indexes; |
| int validIndex; |
| SnapshotData SnapshotToast; |
| |
| /* Look for the valid index of toast relation */ |
| validIndex = toast_open_indexes(toastrel, |
| AccessShareLock, |
| &toastidxs, |
| &num_indexes); |
| |
| startchunk = sliceoffset / TOAST_MAX_CHUNK_SIZE; |
| endchunk = (sliceoffset + slicelength - 1) / TOAST_MAX_CHUNK_SIZE; |
| Assert(endchunk <= totalchunks); |
| |
| /* Set up a scan key to fetch from the index. */ |
| ScanKeyInit(&toastkey[0], |
| (AttrNumber) 1, |
| BTEqualStrategyNumber, F_OIDEQ, |
| ObjectIdGetDatum(valueid)); |
| |
| /* |
| * No additional condition if fetching all chunks. Otherwise, use an |
| * equality condition for one chunk, and a range condition otherwise. |
| */ |
| if (startchunk == 0 && endchunk == totalchunks - 1) |
| nscankeys = 1; |
| else if (startchunk == endchunk) |
| { |
| ScanKeyInit(&toastkey[1], |
| (AttrNumber) 2, |
| BTEqualStrategyNumber, F_INT4EQ, |
| Int32GetDatum(startchunk)); |
| nscankeys = 2; |
| } |
| else |
| { |
| ScanKeyInit(&toastkey[1], |
| (AttrNumber) 2, |
| BTGreaterEqualStrategyNumber, F_INT4GE, |
| Int32GetDatum(startchunk)); |
| ScanKeyInit(&toastkey[2], |
| (AttrNumber) 2, |
| BTLessEqualStrategyNumber, F_INT4LE, |
| Int32GetDatum(endchunk)); |
| nscankeys = 3; |
| } |
| |
| /* Prepare for scan */ |
| init_toast_snapshot(&SnapshotToast); |
| toastscan = systable_beginscan_ordered(toastrel, toastidxs[validIndex], |
| &SnapshotToast, nscankeys, toastkey); |
| |
| /* |
| * Read the chunks by index |
| * |
| * The index is on (valueid, chunkidx) so they will come in order |
| */ |
| expectedchunk = startchunk; |
| while ((ttup = systable_getnext_ordered(toastscan, ForwardScanDirection)) != NULL) |
| { |
| int32 curchunk; |
| Pointer chunk; |
| bool isnull; |
| char *chunkdata; |
| int32 chunksize; |
| int32 expected_size; |
| int32 chcpystrt; |
| int32 chcpyend; |
| |
| /* |
| * Have a chunk, extract the sequence number and the data |
| */ |
| curchunk = DatumGetInt32(fastgetattr(ttup, 2, toasttupDesc, &isnull)); |
| Assert(!isnull); |
| chunk = DatumGetPointer(fastgetattr(ttup, 3, toasttupDesc, &isnull)); |
| Assert(!isnull); |
| if (!VARATT_IS_EXTENDED(chunk)) |
| { |
| chunksize = VARSIZE(chunk) - VARHDRSZ; |
| chunkdata = VARDATA(chunk); |
| } |
| else if (VARATT_IS_SHORT(chunk)) |
| { |
| /* could happen due to heap_form_tuple doing its thing */ |
| chunksize = VARSIZE_SHORT(chunk) - VARHDRSZ_SHORT; |
| chunkdata = VARDATA_SHORT(chunk); |
| } |
| else |
| { |
| /* should never happen */ |
| elog(ERROR, "found toasted toast chunk for toast value %u in %s", |
| valueid, RelationGetRelationName(toastrel)); |
| chunksize = 0; /* keep compiler quiet */ |
| chunkdata = NULL; |
| } |
| |
| /* |
| * Some checks on the data we've found |
| */ |
| if (curchunk != expectedchunk) |
| ereport(ERROR, |
| (errcode(ERRCODE_DATA_CORRUPTED), |
| errmsg_internal("unexpected chunk number %d (expected %d) for toast value %u in %s", |
| curchunk, expectedchunk, valueid, |
| RelationGetRelationName(toastrel)))); |
| if (curchunk > endchunk) |
| ereport(ERROR, |
| (errcode(ERRCODE_DATA_CORRUPTED), |
| errmsg_internal("unexpected chunk number %d (out of range %d..%d) for toast value %u in %s", |
| curchunk, |
| startchunk, endchunk, valueid, |
| RelationGetRelationName(toastrel)))); |
| expected_size = curchunk < totalchunks - 1 ? TOAST_MAX_CHUNK_SIZE |
| : attrsize - ((totalchunks - 1) * TOAST_MAX_CHUNK_SIZE); |
| if (chunksize != expected_size) |
| ereport(ERROR, |
| (errcode(ERRCODE_DATA_CORRUPTED), |
| errmsg_internal("unexpected chunk size %d (expected %d) in chunk %d of %d for toast value %u in %s", |
| chunksize, expected_size, |
| curchunk, totalchunks, valueid, |
| RelationGetRelationName(toastrel)))); |
| |
| /* |
| * Copy the data into proper place in our result |
| */ |
| chcpystrt = 0; |
| chcpyend = chunksize - 1; |
| if (curchunk == startchunk) |
| chcpystrt = sliceoffset % TOAST_MAX_CHUNK_SIZE; |
| if (curchunk == endchunk) |
| chcpyend = (sliceoffset + slicelength - 1) % TOAST_MAX_CHUNK_SIZE; |
| |
| memcpy(VARDATA(result) + |
| (curchunk * TOAST_MAX_CHUNK_SIZE - sliceoffset) + chcpystrt, |
| chunkdata + chcpystrt, |
| (chcpyend - chcpystrt) + 1); |
| |
| expectedchunk++; |
| } |
| |
| /* |
| * Final checks that we successfully fetched the datum |
| */ |
| if (expectedchunk != (endchunk + 1)) |
| ereport(ERROR, |
| (errcode(ERRCODE_DATA_CORRUPTED), |
| errmsg_internal("missing chunk number %d for toast value %u in %s", |
| expectedchunk, valueid, |
| RelationGetRelationName(toastrel)))); |
| |
| /* End scan and close indexes. */ |
| systable_endscan_ordered(toastscan); |
| toast_close_indexes(toastidxs, num_indexes, AccessShareLock); |
| } |