Experements with assign ptrack_enable.

stalkerg · stalkerg · commit 9e6024952324 · 2016-04-08T16:03:16.000+03:00
diff --git a/src/backend/access/heap/ptrack.c b/src/backend/access/heap/ptrack.c
@@ -46,7 +46,7 @@ bool ptrack_enable = false;
 static Buffer ptrack_readbuf(RelFileNode rnode, BlockNumber blkno, bool extend);
 static void ptrack_extend(SMgrRelation smgr, BlockNumber nvmblocks);
 static void ptrack_set(BlockNumber heapBlk, Buffer vmBuf);
-void SetPtrackClearLSN(void);
+void SetPtrackClearLSN(bool set_invalid);
 Datum pg_ptrack_test(PG_FUNCTION_ARGS);
 
 void
@@ -263,23 +263,29 @@ void ptrack_clear(void)
 	systable_endscan(scan);
 	heap_close(catalog, AccessShareLock);
 
-	SetPtrackClearLSN();
+	SetPtrackClearLSN(false);
 }
 
 void
-SetPtrackClearLSN(void)
+SetPtrackClearLSN(bool set_invalid)
 {
 	int			fd;
-	XLogRecPtr	ptr = GetXLogInsertRecPtr();
+	XLogRecPtr	ptr;
+	char		path[MAXPGPATH];
+	if (set_invalid)
+		ptr = InvalidXLogRecPtr;
+	else
+		ptr = GetXLogInsertRecPtr();
+	snprintf(path, sizeof(path), "%s/global/ptrack_control", DataDir);
 	//LWLockAcquire(ControlFileLock, LW_EXCLUSIVE);
-	fd = BasicOpenFile("global/ptrack_control",
-					   O_RDWR | O_CREAT | PG_BINARY,
+	fd = BasicOpenFile(path,
+					   O_RDWR | O_CREAT | O_EXCL | PG_BINARY,
 					   S_IRUSR | S_IWUSR);
 	if (fd < 0)
 		ereport(PANIC,
 				(errcode_for_file_access(),
 				 errmsg("could not create ptrack control file \"%s\": %m",
-						"global/ptrack_control")));
+						path)));
 
 	errno = 0;
 	if (write(fd, &ptr, sizeof(XLogRecPtr)) != sizeof(XLogRecPtr))
@@ -460,3 +466,11 @@ pg_ptrack_test(PG_FUNCTION_ARGS)
 	result_array = construct_array(result_elems, 2, 23, 4, true, 'i');
 	PG_RETURN_ARRAYTYPE_P(result_array);
 }
+
+void
+assign_ptrack_enable(bool newval, void *extra)
+{
+	/*elog(WARNING, "assign_ptrack_enable:%s %i", newval ? "true" : "false", IsInitProcessingMode());
+	if(IsInitProcessingMode() && !newval)
+		SetPtrackClearLSN(true);*/
+}
diff --git a/src/backend/utils/misc/guc.c b/src/backend/utils/misc/guc.c
@@ -994,7 +994,7 @@ static struct config_bool ConfigureNamesBool[] =
 		},
 		&ptrack_enable,
 		false,
-		NULL, NULL, NULL
+		NULL, &assign_ptrack_enable, NULL
 	},
 
 	{
diff --git a/src/include/access/ptrack.h b/src/include/access/ptrack.h
@@ -13,9 +13,9 @@ extern bool ptrack_enable;
 
 extern void ptrack_save(void);
 
-extern void 
-ptrack_add_block(BlockNumber block_number, RelFileNode rel);
+extern void ptrack_add_block(BlockNumber block_number, RelFileNode rel);
 
 extern void ptrack_clear(void);
+extern void assign_ptrack_enable(bool newval, void *extra);
 
 #endif   /* VISIBILITYMAP_H */

Original file line number	Diff line number	Diff line change
`@@ -994,7 +994,7 @@ static struct config_bool ConfigureNamesBool[] =`
`994`	`994`	`},`
`995`	`995`	`&ptrack_enable,`
`996`	`996`	`false,`
`997`		`- NULL, NULL, NULL`
	`997`	`+ NULL, &assign_ptrack_enable, NULL`
`998`	`998`	`},`
`999`	`999`
`1000`	`1000`	`{`