heap ptrack with xlog

Author: alubennikova

src/backend/access/heap/heapam.c

@@ -53,6 +53,7 @@
 #include "access/xlog.h"
 #include "access/xloginsert.h"
 #include "access/xlogutils.h"
+#include "access/ptrack.h"
 #include "catalog/catalog.h"
 #include "catalog/namespace.h"
 #include "commands/vacuum.h"
@@ -2425,6 +2426,7 @@ heap_insert(Relation relation, HeapTuple tup, CommandId cid,
 	HeapTupleCopyEpochFromPage(heaptup, BufferGetPage(buffer));
 
 	/* NO EREPORT(ERROR) from here till changes are logged */
+	ptrack_add_block(relation, BufferGetBlockNumber(buffer));
 	START_CRIT_SECTION();
 
 	RelationPutHeapTuple(relation, buffer, heaptup,
@@ -2799,6 +2801,7 @@ freeze_single_heap_page(Relation relation, Buffer buffer)
 	{
 		int			i;
 
+		ptrack_add_block(relation, BufferGetBlockNumber(buffer));
 		START_CRIT_SECTION();
 
 		MarkBufferDirty(buffer);
@@ -3147,6 +3150,7 @@ heap_multi_insert(Relation relation, HeapTuple *tuples, int ntuples,
 		heap_page_prepare_for_xid(relation, buffer, xid, false);
 
 		/* NO EREPORT(ERROR) from here till changes are logged */
+		ptrack_add_block(relation, BufferGetBlockNumber(buffer));
 		START_CRIT_SECTION();
 
 		/*
@@ -3675,6 +3679,7 @@ heap_delete(Relation relation, ItemPointer tid,
 		(new_infomask & HEAP_XMAX_IS_MULTI) ? true : false);
 	HeapTupleCopyEpochFromPage(&tp, page);
 
+	ptrack_add_block(relation, BufferGetBlockNumber(buffer));
 	START_CRIT_SECTION();
 
 	/*
@@ -4423,6 +4428,7 @@ heap_update(Relation relation, ItemPointer otid, HeapTuple newtup,
 			(infomask_lock_old_tuple & HEAP_XMAX_IS_MULTI) ? true : false);
 		HeapTupleCopyEpochFromPage(&oldtup, page);
 
+		ptrack_add_block(relation, BufferGetBlockNumber(buffer));
 		START_CRIT_SECTION();
 
 		/* Clear obsolete visibility flags ... */
@@ -4600,6 +4606,9 @@ heap_update(Relation relation, ItemPointer otid, HeapTuple newtup,
 	HeapTupleCopyEpochFromPage(&oldtup, page);
 
 	/* NO EREPORT(ERROR) from here till changes are logged */
+	ptrack_add_block(relation, BufferGetBlockNumber(buffer));
+	if (newbuf != buffer)
+		ptrack_add_block(relation, BufferGetBlockNumber(newbuf));
 	START_CRIT_SECTION();
 
 	/*
@@ -5610,6 +5619,7 @@ heap_lock_tuple(Relation relation, HeapTuple tuple,
 		(new_infomask & HEAP_XMAX_IS_MULTI) ? true : false);
 	HeapTupleCopyEpochFromPage(tuple, page);
 
+	ptrack_add_block(relation, BufferGetBlockNumber(*buffer));
 	START_CRIT_SECTION();
 
 	/*
@@ -6355,6 +6365,7 @@ heap_lock_updated_tuple_rec(Relation rel, ItemPointer tid, TransactionId xid,
 			(new_infomask & HEAP_XMAX_IS_MULTI) ? true : false);
 		HeapTupleCopyEpochFromPage(&mytup, BufferGetPage(buf));
 
+		ptrack_add_block(rel, BufferGetBlockNumber(buf));
 		START_CRIT_SECTION();
 
 		/* ... and set them */
@@ -6509,6 +6520,7 @@ heap_finish_speculative(Relation relation, HeapTuple tuple)
 					 "invalid speculative token constant");
 
 	/* NO EREPORT(ERROR) from here till changes are logged */
+	ptrack_add_block(relation, BufferGetBlockNumber(buffer));
 	START_CRIT_SECTION();
 
 	Assert(HeapTupleHeaderIsSpeculative(tuple->t_data));
@@ -6623,6 +6635,7 @@ heap_abort_speculative(Relation relation, HeapTuple tuple)
 	 * do anything special with infomask bits.
 	 */
 
+	ptrack_add_block(relation, BufferGetBlockNumber(buffer));
 	START_CRIT_SECTION();
 
 	/*
@@ -6756,6 +6769,7 @@ heap_inplace_update(Relation relation, HeapTuple tuple)
 		elog(ERROR, "wrong tuple length");
 
 	/* NO EREPORT(ERROR) from here till changes are logged */
+	ptrack_add_block(relation, BufferGetBlockNumber(buffer));
 	START_CRIT_SECTION();
 
 	memcpy((char *) htup + htup->t_hoff,
@@ -8463,6 +8477,7 @@ heap_xlog_clean(XLogReaderState *record)
 	XLogRedoAction action;
 
 	XLogRecGetBlockTag(record, 0, &rnode, NULL, &blkno);
+	ptrack_add_block_redo(rnode, blkno);
 
 	/*
 	 * We're about to remove tuples. In Hot Standby mode, ensure that there's
@@ -8555,6 +8570,7 @@ heap_xlog_visible(XLogReaderState *record)
 	XLogRedoAction action;
 
 	XLogRecGetBlockTag(record, 1, &rnode, NULL, &blkno);
+	ptrack_add_block_redo(rnode, blkno);
 
 	/*
 	 * If there are any Hot Standby transactions running that have an xmin
@@ -8665,6 +8681,11 @@ heap_xlog_freeze_page(XLogReaderState *record)
 	TransactionId cutoff_xid = xlrec->cutoff_xid;
 	Buffer		buffer;
 	int			ntup;
+	RelFileNode rnode;
+	BlockNumber blkno;
+
+	XLogRecGetBlockTag(record, 0, &rnode, NULL, &blkno);
+	ptrack_add_block_redo(rnode, blkno);
 
 	/*
 	 * In Hot Standby mode, ensure that there's no queries running which still
@@ -8750,6 +8771,7 @@ heap_xlog_delete(XLogReaderState *record)
 	ItemPointerData target_tid;
 
 	XLogRecGetBlockTag(record, 0, &target_node, NULL, &blkno);
+	ptrack_add_block_redo(target_node, blkno);
 	ItemPointerSetBlockNumber(&target_tid, blkno);
 	ItemPointerSetOffsetNumber(&target_tid, xlrec->offnum);
 
@@ -8840,6 +8862,7 @@ heap_xlog_insert(XLogReaderState *record)
 	xlrec = (xl_heap_insert *) rec_data;
 
 	XLogRecGetBlockTag(record, 0, &target_node, NULL, &blkno);
+	ptrack_add_block_redo(target_node, blkno);
 	ItemPointerSetBlockNumber(&target_tid, blkno);
 	ItemPointerSetOffsetNumber(&target_tid, xlrec->offnum);
 
@@ -8974,6 +8997,7 @@ heap_xlog_multi_insert(XLogReaderState *record)
 	xlrec = (xl_heap_multi_insert *) rec_data;
 
 	XLogRecGetBlockTag(record, 0, &rnode, NULL, &blkno);
+	ptrack_add_block_redo(rnode, blkno);
 
 	/*
 	 * The visibility map may need to be fixed even if the heap page is
@@ -9136,8 +9160,10 @@ heap_xlog_update(XLogReaderState *record, bool hot_update)
 	oldtup.t_len = 0;
 
 	XLogRecGetBlockTag(record, 0, &rnode, NULL, &newblk);
+	ptrack_add_block_redo(rnode, newblk);
 	if (XLogRecGetBlockTag(record, 1, NULL, NULL, &oldblk))
 	{
+		ptrack_add_block_redo(rnode, oldblk);
 		/* HOT updates are never done across pages */
 		Assert(!hot_update);
 	}
@@ -9388,6 +9414,11 @@ heap_xlog_confirm(XLogReaderState *record)
 	OffsetNumber offnum;
 	ItemId		lp = NULL;
 	HeapTupleHeader htup;
+	RelFileNode rnode;
+	BlockNumber blkno;
+
+	XLogRecGetBlockTag(record, 0, &rnode, NULL, &blkno);
+	ptrack_add_block_redo(rnode, blkno);
 
 	if (XLogReadBufferForRedo(record, 0, &buffer) == BLK_NEEDS_REDO)
 	{
@@ -9424,6 +9455,11 @@ heap_xlog_lock(XLogReaderState *record)
 	OffsetNumber offnum;
 	ItemId		lp = NULL;
 	HeapTupleHeader htup;
+	RelFileNode rnode;
+	BlockNumber blkno;
+
+	XLogRecGetBlockTag(record, 0, &rnode, NULL, &blkno);
+	ptrack_add_block_redo(rnode, blkno);
 
 	/*
 	 * The visibility map may need to be fixed even if the heap page is
@@ -9495,6 +9531,11 @@ heap_xlog_lock_updated(XLogReaderState *record)
 	OffsetNumber offnum;
 	ItemId		lp = NULL;
 	HeapTupleHeader htup;
+	RelFileNode rnode;
+	BlockNumber blkno;
+
+	XLogRecGetBlockTag(record, 0, &rnode, NULL, &blkno);
+	ptrack_add_block_redo(rnode, blkno);
 
 	xlrec = (xl_heap_lock_updated *) XLogRecGetData(record);
 
@@ -9557,6 +9598,11 @@ heap_xlog_inplace(XLogReaderState *record)
 	HeapTupleHeader htup;
 	uint32		oldlen;
 	Size		newlen;
+	RelFileNode rnode;
+	BlockNumber blkno;
+
+	XLogRecGetBlockTag(record, 0, &rnode, NULL, &blkno);
+	ptrack_add_block_redo(rnode, blkno);
 
 	if (XLogReadBufferForRedo(record, 0, &buffer) == BLK_NEEDS_REDO)
 	{
@@ -9597,6 +9643,7 @@ heap_xlog_epoch_shift(XLogReaderState *record)
 	RelFileNode target_node;
 
 	XLogRecGetBlockTag(record, 0, &target_node, NULL, &blkno);
+	ptrack_add_block_redo(target_node, blkno);
 
 	if (XLogReadBufferForRedo(record, 0, &buffer) == BLK_NEEDS_REDO)
 	{