Ensure cleanup in case of early errors in streaming base backups

mhagander · mhagander · commit 433cacfc0f1b · 2014-05-28T13:00:09.000+02:00
Move the code that sends the initial status information as well as the
calculation of paths inside the ENSURE_ERROR_CLEANUP block. If this code
failed, we would "leak" a counter of number of concurrent backups, thereby
making the system always believe it was in backup mode. This could happen
if the sending failed (which it probably never did given that the small
amount of data to send would never cause a flush). It is very low risk, but
all operations after do_pg_start_backup should be protected.
diff --git a/src/backend/replication/basebackup.c b/src/backend/replication/basebackup.c
@@ -109,7 +109,6 @@ perform_base_backup(basebackup_options *opt, DIR *tblspcdir)
 
 	startptr = do_pg_start_backup(opt->label, opt->fastcheckpoint, &starttli,
 								  &labelfile);
-	SendXlogRecPtrResult(startptr, starttli);
 
 	PG_ENSURE_ERROR_CLEANUP(base_backup_cleanup, (Datum) 0);
 	{
@@ -118,6 +117,8 @@ perform_base_backup(basebackup_options *opt, DIR *tblspcdir)
 		struct dirent *de;
 		tablespaceinfo *ti;
 
+		SendXlogRecPtrResult(startptr, starttli);
+
 		/* Collect information about all tablespaces */
 		while ((de = ReadDir(tblspcdir, "pg_tblspc")) != NULL)
 		{

Original file line number	Diff line number	Diff line change
`@@ -109,7 +109,6 @@ perform_base_backup(basebackup_options opt, DIR tblspcdir)`
`109`	`109`
`110`	`110`	`startptr = do_pg_start_backup(opt->label, opt->fastcheckpoint, &starttli,`
`111`	`111`	`&labelfile);`
`112`		`- SendXlogRecPtrResult(startptr, starttli);`
`113`	`112`
`114`	`113`	`PG_ENSURE_ERROR_CLEANUP(base_backup_cleanup, (Datum) 0);`
`115`	`114`	`{`
`@@ -118,6 +117,8 @@ perform_base_backup(basebackup_options opt, DIR tblspcdir)`
`118`	`117`	`struct dirent *de;`
`119`	`118`	`tablespaceinfo *ti;`
`120`	`119`
	`120`	`+ SendXlogRecPtrResult(startptr, starttli);`
	`121`	`+`
`121`	`122`	`/* Collect information about all tablespaces */`
`122`	`123`	`while ((de = ReadDir(tblspcdir, "pg_tblspc")) != NULL)`
`123`	`124`	`{`