mirror of
https://github.com/EnterpriseDB/repmgr.git
synced 2026-03-23 07:06:30 +00:00
Add repmgr client command.
This commit is contained in:
postgres
320
repmgr.c
Normal file
320
repmgr.c
Normal file
@@ -0,0 +1,320 @@
|
||||
/*
|
||||
* repmgr.c
|
||||
* Copyright (c) 2ndQuadrant, 2010
|
||||
*
|
||||
* Command interpreter for the repmgr
|
||||
* This module execute isome tasks based on commands and then exit
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
#include <time.h>
|
||||
|
||||
#include "repmgr.h"
|
||||
|
||||
#define RECOVERY_FILE "recovery.conf"
|
||||
#define RECOVERY_DONE_FILE "recovery.done"
|
||||
|
||||
|
||||
void checkNodeConfiguration(char *conninfo);
|
||||
void getPrimaryConnection(void);
|
||||
|
||||
|
||||
void help(void);
|
||||
void do_standby_clone(char *conninfo);
|
||||
|
||||
|
||||
int
|
||||
main(int argc, char **argv)
|
||||
{
|
||||
char myClusterName[MAXLEN];
|
||||
int myLocalId = -1;
|
||||
char myConninfo[MAXLEN];
|
||||
|
||||
if (argc != 2)
|
||||
help();
|
||||
|
||||
/*
|
||||
* Read the configuration file: repmgr.conf
|
||||
*/
|
||||
parse_config(myClusterName, &myLocalId, myConninfo);
|
||||
if (myLocalId == -1)
|
||||
{
|
||||
fprintf(stderr, "Node information is missing. "
|
||||
"Check the configuration file.");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
/* XXX should we check the master pre requisites? */
|
||||
|
||||
|
||||
/* Check what is the action we need to execute */
|
||||
/* XXX Probably we can do this better but it works for now */
|
||||
if (strcasecmp(argv[1], "STANDBY") == 0)
|
||||
{
|
||||
if (strcasecmp(argv[2], "CLONE") == 0)
|
||||
do_standby_clone(myConninfo);
|
||||
else if (strcasecmp(argv[2], "PROMOTE") == 0)
|
||||
do_standby_promote();
|
||||
else if (strcasecmp(argv[2], "FOLLOW") == 0)
|
||||
do_standby_follow();
|
||||
else
|
||||
help();
|
||||
}
|
||||
else
|
||||
help();
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
do_standby_clone(char *conninfo)
|
||||
{
|
||||
PGconn *conn;
|
||||
PGresult *res;
|
||||
char sqlquery[8192];
|
||||
char script[8192];
|
||||
|
||||
int r;
|
||||
char data_dir_full_path[MAXLEN];
|
||||
char *current_dir;
|
||||
char data_dir[MAXLEN];
|
||||
char recovery_file_path[MAXLEN];
|
||||
FILE *recovery_file;
|
||||
|
||||
char line[MAXLEN];
|
||||
|
||||
/* inform the master we will start a backup */
|
||||
conn = establishDBConnection(conninfo, true);
|
||||
|
||||
fprintf(stderr, "Starting backup...");
|
||||
|
||||
/* Get the data directory full path and the last subdirectory */
|
||||
sprintf(sqlquery, "SELECT setting, "
|
||||
" TRIM(SUBSTRING(setting FROM '.[^/]*$'), '/') "
|
||||
" FROM pg_settings WHERE name = 'data_directory'");
|
||||
res = PQexec(conn, sqlquery);
|
||||
if (PQresultStatus(res) != PGRES_TUPLES_OK)
|
||||
{
|
||||
fprintf(stderr, "Can't get info about data directory: %s", PQerrorMessage(conn));
|
||||
PQclear(res);
|
||||
PQfinish(conn);
|
||||
return;
|
||||
}
|
||||
strcpy(data_dir_full_path, PQgetvalue(res, 0, 0));
|
||||
strcpy(data_dir, PQgetvalue(res, 0, 1));
|
||||
PQclear(res);
|
||||
|
||||
sprintf(sqlquery, "SELECT pg_start_backup('repmgr_standby_clone_%ld')", time(NULL));
|
||||
res = PQexec(conn, sqlquery);
|
||||
if (PQresultStatus(res) != PGRES_TUPLES_OK)
|
||||
{
|
||||
fprintf(stderr, "Can't start backup: %s", PQerrorMessage(conn));
|
||||
PQclear(res);
|
||||
PQfinish(conn);
|
||||
return;
|
||||
}
|
||||
PQclear(res);
|
||||
PQfinish(conn);
|
||||
|
||||
/* rsync data directory to current location */
|
||||
sprintf(script, "rsync -r %s .", data_dir_full_path);
|
||||
r = system(script);
|
||||
if (r != 0)
|
||||
{
|
||||
fprintf(stderr, "Can't rsync data directory");
|
||||
return;
|
||||
}
|
||||
|
||||
/* inform the master that we have finished the backup */
|
||||
conn = establishDBConnection(conninfo, true);
|
||||
|
||||
fprintf(stderr, "Finishing backup...");
|
||||
|
||||
sprintf(sqlquery, "SELECT pg_stop_backup()");
|
||||
res = PQexec(conn, sqlquery);
|
||||
if (PQresultStatus(res) != PGRES_TUPLES_OK)
|
||||
{
|
||||
fprintf(stderr, "Can't stop backup: %s", PQerrorMessage(conn));
|
||||
PQclear(res);
|
||||
PQfinish(conn);
|
||||
return;
|
||||
}
|
||||
PQclear(res);
|
||||
PQfinish(conn);
|
||||
|
||||
/* Finally, write the recovery.conf file */
|
||||
getcwd(current_dir, MAXLEN);
|
||||
strcpy(recovery_file_path, current_dir);
|
||||
strcat(recovery_file_path, "/");
|
||||
strcat(recovery_file_path, data_dir);
|
||||
strcat(recovery_file_path, "/");
|
||||
strcat(recovery_file_path, RECOVERY_FILE);
|
||||
free(current_dir);
|
||||
|
||||
recovery_file = fopen(recovery_file_path, "w");
|
||||
if (recovery_file == NULL)
|
||||
{
|
||||
fprintf(stderr, "could not create recovery.conf file, it could be necesary to create it manually");
|
||||
return;
|
||||
}
|
||||
|
||||
strcpy(line, "standby_mode = on\n");
|
||||
if (fputs(line, recovery_file) == EOF)
|
||||
{
|
||||
fprintf(stderr, "recovery file could not be written, it could be necesary to create it manually");
|
||||
fclose(recovery_file);
|
||||
return;
|
||||
}
|
||||
|
||||
strcpy(line, "primary_conninfo = ");
|
||||
strcat(line, conninfo);
|
||||
if (fputs(line, recovery_file) == EOF)
|
||||
{
|
||||
fprintf(stderr, "recovery file could not be written, it could be necesary to create it manually");
|
||||
fclose(recovery_file);
|
||||
return;
|
||||
}
|
||||
|
||||
/*FreeFile(recovery_file);*/
|
||||
fclose(recovery_file);
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
do_standby_promote(char *conninfo)
|
||||
{
|
||||
PGconn *conn;
|
||||
PGresult *res;
|
||||
char sqlquery[8192];
|
||||
char script[8192];
|
||||
|
||||
char data_dir[MAXLEN];
|
||||
char recovery_file_path[MAXLEN];
|
||||
char recovery_done_path[MAXLEN];
|
||||
|
||||
/* inform the master we will start a backup */
|
||||
conn = establishDBConnection(conninfo, true);
|
||||
|
||||
fprintf(stderr, "Promoting standby...");
|
||||
|
||||
/* Get the data directory full path and the last subdirectory */
|
||||
sprintf(sqlquery, "SELECT setting "
|
||||
" FROM pg_settings WHERE name = 'data_directory'");
|
||||
res = PQexec(conn, sqlquery);
|
||||
if (PQresultStatus(res) != PGRES_TUPLES_OK)
|
||||
{
|
||||
fprintf(stderr, "Can't get info about data directory: %s", PQerrorMessage(conn));
|
||||
PQclear(res);
|
||||
PQfinish(conn);
|
||||
return;
|
||||
}
|
||||
strcpy(data_dir, PQgetvalue(res, 0, 0));
|
||||
PQclear(res);
|
||||
PQfinish(conn);
|
||||
|
||||
strcpy(recovery_file_path, data_dir);
|
||||
strcat(recovery_file_path, "/");
|
||||
strcpy(recovery_file_path, RECOVERY_FILE);
|
||||
|
||||
strcpy(recovery_done_path, data_dir);
|
||||
strcat(recovery_done_path, "/");
|
||||
strcpy(recovery_done_path, RECOVERY_DONE);
|
||||
rename(recovery_file_path, recovery_done_path);
|
||||
|
||||
sprintf(script, "pg_ctl -D %s restart", data_dir);
|
||||
r = system(script);
|
||||
if (r != 0)
|
||||
{
|
||||
fprintf(stderr, "Can't restart service");
|
||||
return;
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
do_standby_follow(char *conninfo)
|
||||
{
|
||||
PGconn *conn;
|
||||
PGresult *res;
|
||||
char sqlquery[8192];
|
||||
|
||||
char data_dir[MAXLEN];
|
||||
char recovery_file_path[MAXLEN];
|
||||
FILE *recovery_file;
|
||||
|
||||
char line[MAXLEN];
|
||||
|
||||
/* inform the master we will start a backup */
|
||||
conn = establishDBConnection(conninfo, true);
|
||||
|
||||
fprintf(stderr, "Changing standby's primary...");
|
||||
|
||||
/* Get the data directory full path and the last subdirectory */
|
||||
sprintf(sqlquery, "SELECT setting "
|
||||
" FROM pg_settings WHERE name = 'data_directory'");
|
||||
res = PQexec(conn, sqlquery);
|
||||
if (PQresultStatus(res) != PGRES_TUPLES_OK)
|
||||
{
|
||||
fprintf(stderr, "Can't get info about data directory: %s", PQerrorMessage(conn));
|
||||
PQclear(res);
|
||||
PQfinish(conn);
|
||||
return;
|
||||
}
|
||||
strcpy(data_dir, PQgetvalue(res, 0, 0));
|
||||
PQclear(res);
|
||||
PQfinish(conn);
|
||||
|
||||
/* Finally, write the recovery.conf file */
|
||||
strcpy(recovery_file_path, data_dir);
|
||||
strcat(recovery_file_path, "/");
|
||||
strcat(recovery_file_path, RECOVERY_FILE);
|
||||
|
||||
recovery_file = fopen(recovery_file_path, "w");
|
||||
if (recovery_file == NULL)
|
||||
{
|
||||
fprintf(stderr, "could not create recovery.conf file, it could be necesary to create it manually");
|
||||
return;
|
||||
}
|
||||
|
||||
strcpy(line, "standby_mode = on\n");
|
||||
if (fputs(line, recovery_file) == EOF)
|
||||
{
|
||||
fprintf(stderr, "recovery file could not be written, it could be necesary to create it manually");
|
||||
fclose(recovery_file);
|
||||
return;
|
||||
}
|
||||
|
||||
strcpy(line, "primary_conninfo = ");
|
||||
strcat(line, conninfo);
|
||||
if (fputs(line, recovery_file) == EOF)
|
||||
{
|
||||
fprintf(stderr, "recovery file could not be written, it could be necesary to create it manually");
|
||||
fclose(recovery_file);
|
||||
return;
|
||||
}
|
||||
|
||||
/*FreeFile(recovery_file);*/
|
||||
fclose(recovery_file);
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
help(void)
|
||||
{
|
||||
fprintf(stderr, "repmgr: command program that performs tasks and then exits.\n"
|
||||
"COMMANDS:\n"
|
||||
"standby clone - allows creation of a new standby\n"
|
||||
"standby promote - allows manual promotion of a specific standby into a "
|
||||
"new master in the event of a failover\n"
|
||||
"standby follow - allows the standby to re-point itself to a new master");
|
||||
exit(1);
|
||||
}
|
||||
Reference in New Issue
Block a user