pg_probackup/delete.c

/*-------------------------------------------------------------------------
 *
 * delete.c: delete backup files.
 *
 * Copyright (c) 2009-2013, NIPPON TELEGRAPH AND TELEPHONE CORPORATION
 *
 *-------------------------------------------------------------------------
 */

#include "pg_probackup.h"

#include <dirent.h>
#include <time.h>
#include <unistd.h>

static int pgBackupDeleteFiles(pgBackup *backup);
static void delete_walfiles(XLogRecPtr oldest_lsn, TimeLineID oldest_tli,
							bool delete_all);

int
do_delete(time_t backup_id)
{
	int			i;
	int			b_index;
	int			ret;
	parray		*backup_list;
	pgBackup	*last_backup = NULL;

	/* DATE are always required */
	if (backup_id == 0)
		elog(ERROR, "required backup ID not specified");

	/* Lock backup catalog */
	ret = catalog_lock(false);
	if (ret == -1)
		elog(ERROR, "can't lock backup catalog.");
	else if (ret == 1)
		elog(ERROR,
			 "another pg_probackup is running, stop delete.");

	/* Get complete list of backups */
	backup_list = catalog_get_backup_list(0);
	if (!backup_list)
		elog(ERROR, "No backup list found, can't process any more.");

	/* Find backup to be deleted */
	for (i = 0; i < parray_num(backup_list); i++)
	{
		last_backup = (pgBackup *) parray_get(backup_list, i);
		if (last_backup->start_time == backup_id)
			goto found_backup;
	}

	elog(ERROR, "no backup found, cannot delete.");

found_backup:
	b_index = i;
	/* check for interrupt */
	if (interrupted)
		elog(ERROR, "interrupted during delete backup");

	/* just do it */
	pgBackupDeleteFiles(last_backup);

	if (last_backup->status == BACKUP_STATUS_ERROR)
		return 0;

	/* remove all increments after removed backup */
	for (i = b_index - 1; i >= 0; i--)
	{
		pgBackup *backup = (pgBackup *) parray_get(backup_list, i);
		if (backup->backup_mode >= BACKUP_MODE_FULL)
			break;
		if ((backup->status == BACKUP_STATUS_OK || backup->status == BACKUP_STATUS_CORRUPT) &&
			(backup->backup_mode == BACKUP_MODE_DIFF_PAGE || backup->backup_mode == BACKUP_MODE_DIFF_PTRACK)
		)
			pgBackupDeleteFiles(backup);
	}

	/* release catalog lock */
	catalog_unlock();

	/* cleanup */
	parray_walk(backup_list, pgBackupFree);
	parray_free(backup_list);

	if (delete_wal)
		do_deletewal(backup_id, false);

	return 0;
}

/*
 * Delete in archive WAL segments that are not needed anymore. The oldest
 * segment to be kept is the first segment that the oldest full backup
 * found around needs to keep.
 */
int
do_deletewal(time_t backup_id, bool strict)
{
	size_t		i;
	int			ret;
	parray		*backup_list;
	XLogRecPtr	oldest_lsn = InvalidXLogRecPtr;
	TimeLineID	oldest_tli;
	bool		backup_found = false;

	/* Lock backup catalog */
	ret = catalog_lock(false);
	if (ret == -1)
		elog(ERROR, "can't lock backup catalog.");
	else if (ret == 1)
		elog(ERROR,
			 "another pg_probackup is running, stop delete.");

	/* Find oldest LSN, used by backups */
	backup_list = catalog_get_backup_list(0);
	for (i = 0; i < parray_num(backup_list); i++)
	{
		pgBackup   *last_backup = (pgBackup *) parray_get(backup_list, i);

		if (last_backup->status == BACKUP_STATUS_OK)
		{
			oldest_lsn = last_backup->start_lsn;
			oldest_tli = last_backup->tli;

			if (strict && backup_id != 0 && backup_id >= last_backup->start_time)
			{
				backup_found = true;
				break;
			}
		}
	}

	if (strict && backup_id != 0 && backup_found == false)
		elog(ERROR, "not found backup for deletwal command");

	catalog_unlock();
	parray_walk(backup_list, pgBackupFree);
	parray_free(backup_list);

	delete_walfiles(oldest_lsn, oldest_tli, false);

	return 0;
}

/*
 * Remove backups by retention policy. Retention policy is configured by
 * retention_redundancy and retention_window variables.
 */
int
do_retention_purge(void)
{
	parray	   *backup_list;
	uint32		backup_num;
	size_t		i;
	time_t		days_threshold = time(NULL) - (retention_window * 60 * 60 * 24);
	XLogRecPtr	oldest_lsn = InvalidXLogRecPtr;
	TimeLineID	oldest_tli;
	int			ret;

	if (retention_redundancy > 0)
		elog(LOG, "REDUNDANCY=%u", retention_redundancy);
	if (retention_window > 0)
		elog(LOG, "WINDOW=%u", retention_window);

	if (retention_redundancy == 0 && retention_window == 0)
		elog(ERROR, "retention policy is not set");

	/* Lock backup catalog */
	ret = catalog_lock(false);
	if (ret == 1)
		elog(ERROR,
			 "cannot lock backup catalog, another pg_probackup is running");

	/* Get a complete list of backups. */
	backup_list = catalog_get_backup_list(0);
	if (parray_num(backup_list) == 0)
	{
		elog(INFO, "backup list is empty");
		elog(INFO, "exit");
		catalog_unlock();
		return 0;
	}

	/* Find target backups to be deleted */
	backup_num = 0;
	for (i = 0; i < parray_num(backup_list); i++)
	{
		pgBackup   *backup = (pgBackup *) parray_get(backup_list, i);
		uint32		backup_num_evaluate = backup_num;

		/*
		 * When a validate full backup was found, we can delete the
		 * backup that is older than it using the number of generations.
		 */
		if (backup->backup_mode == BACKUP_MODE_FULL)
			backup_num++;

		/* Evaluate if this backup is eligible for removal */
		if (backup_num_evaluate + 1 <= retention_redundancy ||
			(retention_window > 0 && backup->start_time >= days_threshold))
		{
			/* Save LSN and Timeline to remove unnecessary WAL segments */
			oldest_lsn = backup->start_lsn;
			oldest_tli = backup->tli;
			continue;
		}

		/* Delete backup and update status to DELETED */
		pgBackupDeleteFiles(backup);
	}

	/* Purge WAL files */
	delete_walfiles(oldest_lsn, oldest_tli, true);

	/* Cleanup */
	parray_walk(backup_list, pgBackupFree);
	parray_free(backup_list);

	catalog_unlock();

	elog(INFO, "purging is finished");

	return 0;
}

/*
 * Delete backup files of the backup and update the status of the backup to
 * BACKUP_STATUS_DELETED.
 */
static int
pgBackupDeleteFiles(pgBackup *backup)
{
	size_t		i;
	char		path[MAXPGPATH];
	char		timestamp[20];
	parray	   *files;

	/*
	 * If the backup was deleted already, there is nothing to do.
	 */
	if (backup->status == BACKUP_STATUS_DELETED)
		return 0;

	time2iso(timestamp, lengthof(timestamp), backup->start_time);

	elog(INFO, "delete: %s %s", base36enc(backup->start_time), timestamp);

	/*
	 * Update STATUS to BACKUP_STATUS_DELETING in preparation for the case which
	 * the error occurs before deleting all backup files.
	 */
	if (!check)
	{
		backup->status = BACKUP_STATUS_DELETING;
		pgBackupWriteIni(backup);
	}

	/* list files to be deleted */
	files = parray_new();
	pgBackupGetPath(backup, path, lengthof(path), NULL);
	dir_list_file(files, path, NULL, true, true);

	/* delete leaf node first */
	parray_qsort(files, pgFileComparePathDesc);
	for (i = 0; i < parray_num(files); i++)
	{
		pgFile *file = (pgFile *) parray_get(files, i);

		/* print progress */
		elog(LOG, "delete file(%zd/%lu) \"%s\"", i + 1,
				(unsigned long) parray_num(files), file->path);

		/* skip actual deletion in check mode */
		if (!check)
		{
			if (remove(file->path))
			{
				elog(WARNING, "can't remove \"%s\": %s", file->path,
					strerror(errno));
				parray_walk(files, pgFileFree);
				parray_free(files);

				return 1;
			}
		}
	}

	parray_walk(files, pgFileFree);
	parray_free(files);

	return 0;
}

/*
 * Delete WAL segments up to oldest_lsn.
 *
 * If oldest_lsn is invalid function exists. But if delete_all is true then
 * WAL segements will be deleted anyway.
 */
static void
delete_walfiles(XLogRecPtr oldest_lsn, TimeLineID oldest_tli, bool delete_all)
{
	XLogSegNo   targetSegNo;
	char		oldestSegmentNeeded[MAXFNAMELEN];
	DIR		   *arcdir;
	struct dirent *arcde;
	char		wal_file[MAXPGPATH];
	char		max_wal_file[MAXPGPATH];
	char		min_wal_file[MAXPGPATH];
	int			rc;

	if (XLogRecPtrIsInvalid(oldest_lsn) && !delete_all)
		return;

	max_wal_file[0] = '\0';
	min_wal_file[0] = '\0';

	if (!XLogRecPtrIsInvalid(oldest_lsn))
	{
		XLByteToSeg(oldest_lsn, targetSegNo);
		XLogFileName(oldestSegmentNeeded, oldest_tli, targetSegNo);

		elog(LOG, "removing WAL segments older than %s", oldestSegmentNeeded);
	}
	else
		elog(LOG, "removing all WAL segments");

	/*
	 * Now it is time to do the actual work and to remove all the segments
	 * not needed anymore.
	 */
	if ((arcdir = opendir(arclog_path)) != NULL)
	{
		while (errno = 0, (arcde = readdir(arcdir)) != NULL)
		{
			/*
			 * We ignore the timeline part of the XLOG segment identifiers in
			 * deciding whether a segment is still needed.  This ensures that
			 * we won't prematurely remove a segment from a parent timeline.
			 * We could probably be a little more proactive about removing
			 * segments of non-parent timelines, but that would be a whole lot
			 * more complicated.
			 *
			 * We use the alphanumeric sorting property of the filenames to
			 * decide which ones are earlier than the exclusiveCleanupFileName
			 * file. Note that this means files are not removed in the order
			 * they were originally written, in case this worries you.
			 */
			if (IsXLogFileName(arcde->d_name) ||
				IsPartialXLogFileName(arcde->d_name) ||
				IsBackupHistoryFileName(arcde->d_name))
			{
				if (XLogRecPtrIsInvalid(oldest_lsn) ||
					strncmp(arcde->d_name, oldestSegmentNeeded,
							XLOG_FNAME_LEN) < 0)
				{
					/*
					 * Use the original file name again now, including any
					 * extension that might have been chopped off before testing
					 * the sequence.
					 */
					snprintf(wal_file, MAXPGPATH, "%s/%s",
							 arclog_path, arcde->d_name);

					rc = unlink(wal_file);
					if (rc != 0)
					{
						elog(WARNING, "could not remove file \"%s\": %s",
							 wal_file, strerror(errno));
						break;
					}
					if (verbose)
						elog(LOG, "removed WAL segment \"%s\"", wal_file);

					if (max_wal_file[0] == '\0' ||
						strcmp(max_wal_file + 8, arcde->d_name + 8) < 0)
						strcpy(max_wal_file, arcde->d_name);

					if (min_wal_file[0] == '\0' ||
						strcmp(min_wal_file + 8, arcde->d_name + 8) > 0)
						strcpy(min_wal_file, arcde->d_name);
				}
			}
		}

		if (!verbose && min_wal_file[0] != '\0')
			elog(INFO, "removed min WAL segment \"%s\"", min_wal_file);
		if (!verbose && max_wal_file[0] != '\0')
			elog(INFO, "removed max WAL segment \"%s\"", max_wal_file);

		if (errno)
			elog(WARNING, "could not read archive location \"%s\": %s",
				 arclog_path, strerror(errno));
		if (closedir(arcdir))
			elog(WARNING, "could not close archive location \"%s\": %s",
				 arclog_path, strerror(errno));
	}
	else
		elog(WARNING, "could not open archive location \"%s\": %s",
			 arclog_path, strerror(errno));
}