Add ginPostingListDecodeOneSegment()

Nikita Glukhov · Nikita Glukhov · commit 14840f3ca8b5 · 2022-07-27T18:20:35.000+03:00
diff --git a/src/backend/access/gin/ginpostinglist.c b/src/backend/access/gin/ginpostinglist.c
@@ -288,6 +288,33 @@ ginPostingListDecode(GinPostingList *plist, int *ndecoded)
 										   ndecoded);
 }
 
+int
+ginPostingListDecodeOneSegment(GinPostingList *segment, ItemPointer result)
+{
+	unsigned char *ptr;
+	unsigned char *endptr;
+	uint64		val;
+	int			ndecoded = 0;
+
+	/* copy the first item */
+	Assert(OffsetNumberIsValid(ItemPointerGetOffsetNumber(&segment->first)));
+	result[ndecoded] = segment->first;
+	ndecoded++;
+
+	val = itemptr_to_uint64(&segment->first);
+	ptr = segment->bytes;
+	endptr = segment->bytes + segment->nbytes;
+
+	while (ptr < endptr)
+	{
+		val += decode_varbyte(&ptr);
+		uint64_to_itemptr(val, &result[ndecoded]);
+		ndecoded++;
+	}
+
+	return ndecoded;
+}
+
 /*
  * Decode multiple posting list segments into an array of item pointers.
  * The number of items is returned in *ndecoded_out. The segments are stored
diff --git a/src/include/access/gin_private.h b/src/include/access/gin_private.h
@@ -472,6 +472,7 @@ extern GinPostingList *ginCompressPostingList(const ItemPointer ipd, int nipd,
 extern int	ginPostingListDecodeAllSegmentsToTbm(GinPostingList *ptr, int totalsize, TIDBitmap *tbm);
 
 extern ItemPointer ginPostingListDecodeAllSegments(GinPostingList *ptr, int len, int *ndecoded);
+extern int ginPostingListDecodeOneSegment(GinPostingList *segment, ItemPointer result);
 extern ItemPointer ginPostingListDecode(GinPostingList *ptr, int *ndecoded);
 extern ItemPointer ginMergeItemPointers(ItemPointerData *a, uint32 na,
 										ItemPointerData *b, uint32 nb,
