mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-14 17:02:56 +00:00
5589efb6de
## Problem We currently have this code duplicated across different PG versions. Moving this to an extension would reduce duplication and simplify maintenance. ## Summary of changes Moving the LastWrittenLSN code from PG versions to the Neon extension and linking it with hooks. Related Postgres PR: https://github.com/neondatabase/postgres/pull/590 Closes: https://github.com/neondatabase/neon/issues/10973 --------- Co-authored-by: Tristan Partin <tristan@neon.tech>
50 lines
1.5 KiB
Diff
50 lines
1.5 KiB
Diff
diff --git a/src/ruminsert.c b/src/ruminsert.c
|
|
index 255e616..7a2240f 100644
|
|
--- a/src/ruminsert.c
|
|
+++ b/src/ruminsert.c
|
|
@@ -628,6 +628,10 @@ rumbuild(Relation heap, Relation index, struct IndexInfo *indexInfo)
|
|
elog(ERROR, "index \"%s\" already contains data",
|
|
RelationGetRelationName(index));
|
|
|
|
+#ifdef NEON_SMGR
|
|
+ smgr_start_unlogged_build(index->rd_smgr);
|
|
+#endif
|
|
+
|
|
initRumState(&buildstate.rumstate, index);
|
|
buildstate.rumstate.isBuild = true;
|
|
buildstate.indtuples = 0;
|
|
@@ -693,6 +697,10 @@ rumbuild(Relation heap, Relation index, struct IndexInfo *indexInfo)
|
|
buildstate.buildStats.nTotalPages = RelationGetNumberOfBlocks(index);
|
|
rumUpdateStats(index, &buildstate.buildStats, buildstate.rumstate.isBuild);
|
|
|
|
+#ifdef NEON_SMGR
|
|
+ smgr_finish_unlogged_build_phase_1(index->rd_smgr);
|
|
+#endif
|
|
+
|
|
/*
|
|
* Write index to xlog
|
|
*/
|
|
@@ -713,6 +721,22 @@ rumbuild(Relation heap, Relation index, struct IndexInfo *indexInfo)
|
|
UnlockReleaseBuffer(buffer);
|
|
}
|
|
|
|
+#ifdef NEON_SMGR
|
|
+ {
|
|
+#if PG_VERSION_NUM >= 160000
|
|
+ RelFileLocator rlocator = RelationGetSmgr(index)->smgr_rlocator.locator;
|
|
+#else
|
|
+ RelFileNode rlocator = RelationGetSmgr(index)->smgr_rnode.node;
|
|
+#endif
|
|
+ if (set_lwlsn_block_range_hook)
|
|
+ set_lwlsn_block_range_hook(XactLastRecEnd, rlocator, MAIN_FORKNUM, 0, RelationGetNumberOfBlocks(index));
|
|
+ if (set_lwlsn_relation_hook)
|
|
+ set_lwlsn_relation_hook(XactLastRecEnd, rlocator, MAIN_FORKNUM);
|
|
+
|
|
+ smgr_end_unlogged_build(index->rd_smgr);
|
|
+ }
|
|
+#endif
|
|
+
|
|
/*
|
|
* Return statistics
|
|
*/
|