[Issue #149] Allow to set primary_slot_name during restore, option "-S,--slot" can now be used with "restore" command

Author: gsmolk

src/pg_probackup.c

@@ -653,12 +653,16 @@ main(int argc, char *argv[])
 		if (force)
 			no_validate = true;
 
+		if (replication_slot != NULL)
+			restore_as_replica = true;
+
 		/* keep all params in one structure */
 		restore_params = pgut_new(pgRestoreParams);
 		restore_params->is_restore = (backup_subcmd == RESTORE_CMD);
 		restore_params->force = force;
 		restore_params->no_validate = no_validate;
 		restore_params->restore_as_replica = restore_as_replica;
+		restore_params->primary_slot_name = replication_slot;
 		restore_params->skip_block_validation = skip_block_validation;
 		restore_params->skip_external_dirs = skip_external_dirs;
 		restore_params->partial_db_list = NULL;

src/pg_probackup.h

@@ -392,6 +392,7 @@ typedef struct pgRestoreParams
 	bool	skip_external_dirs;
 	bool	skip_block_validation; //Start using it
 	const char *restore_command;
+	const char *primary_slot_name;
 
 	/* options for partial restore */
 	PartialRestoreType partial_restore_type;

src/restore.c

@@ -937,6 +937,7 @@ create_recovery_conf(time_t backup_id,
 	/* construct restore_command */
 	if (pitr_requested)
 	{
+		fio_fprintf(fp, "\n## recovery settings\n");
 		/* If restore_command is provided, use it. Otherwise construct it from scratch. */
 		if (restore_command_provided)
 			sprintf(restore_command_guc, "%s", instance_config.restore_command);
@@ -1012,21 +1013,25 @@ create_recovery_conf(time_t backup_id,
 			fio_fprintf(fp, "recovery_target_action = '%s'\n", "pause");
 	}
 
+	if (pitr_requested)
+	{
+		elog(LOG, "Setting restore_command to '%s'", restore_command_guc);
+		fio_fprintf(fp, "restore_command = '%s'\n", restore_command_guc);
+	}
+
 	if (params->restore_as_replica)
 	{
+		fio_fprintf(fp, "\n## standby settings\n");
 	/* standby_mode was removed in PG12 */
 #if PG_VERSION_NUM < 120000
 		fio_fprintf(fp, "standby_mode = 'on'\n");
 #endif
 
 		if (backup->primary_conninfo)
 			fio_fprintf(fp, "primary_conninfo = '%s'\n", backup->primary_conninfo);
-	}
 
-	if (pitr_requested)
-	{
-		elog(LOG, "Setting restore_command to '%s'", restore_command_guc);
-		fio_fprintf(fp, "restore_command = '%s'\n", restore_command_guc);
+		if (params->primary_slot_name != NULL)
+			fio_fprintf(fp, "primary_slot_name = '%s'\n", params->primary_slot_name);
 	}
 
 	if (fio_fflush(fp) != 0 ||