github/repmgr
1
0
Fork 0
mirror of https://github.com/EnterpriseDB/repmgr.git synced 2026-03-28 17:36:30 +00:00
Code Issues Packages Projects Releases 1 Wiki Activity

repmgr: improve handling application name during standby clone

Browse Source 
Addresses issues noticed while investigating GitHub #238
This commit is contained in:
Ian Barwick
2016-09-26 09:50:56 +09:00
parent 7c1776655b
commit b17593ff4d
1 changed files with 73 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

84
repmgr.c
View File

@@ -146,9 +146,10 @@ static bool copy_file(const char *old_filename, const char *new_filename);
static bool read_backup_label(const char *local_data_directory, struct BackupLabel *out_backup_label);
static void initialize_conninfo_params(t_conninfo_param_list *param_list, bool set_defaults);
static void copy_conninfo_params(t_conninfo_param_list *dest_list, t_conninfo_param_list *source_list);
static void param_set(t_conninfo_param_list *param_list, const char *param, const char *value);
static char *param_get(t_conninfo_param_list *param_list, const char *param);
static bool parse_conninfo_string(const char *conninfo_str, t_conninfo_param_list *param_list, char *errmsg);
static bool parse_conninfo_string(const char *conninfo_str, t_conninfo_param_list *param_list, char *errmsg, bool ignore_application_name);
static void conn_to_param_list(PGconn *conn, t_conninfo_param_list *param_list);
static void parse_pg_basebackup_options(const char *pg_basebackup_options, t_basebackup_options *backup_options);
@@ -1852,25 +1853,33 @@ do_standby_clone(void)
* line. As it's possible the standby will be cloned from a node different
* to its intended upstream, we'll later attempt to fetch the
* upstream node record and overwrite the values set here with
* those from the upstream node record.
* those from the upstream node record (excluding that record's
* application_name)
*/
initialize_conninfo_params(&recovery_conninfo, true);
if (strlen(runtime_options.host))
copy_conninfo_params(&recovery_conninfo, &source_conninfo);
/* Set the default application name to this node's name */
param_set(&recovery_conninfo, "application_name", options.node_name);
/*
* If application_name is set in repmgr.conf's conninfo parameter, use
* this value (if the source host was provided as a conninfo string, any
* application_name values set there will be overridden; we assume the only
* reason to pass an application_name via the command line is in the
* rare corner case where a user wishes to clone a server without
* providing repmgr.conf)
*/
if (strlen(options.conninfo))
{
param_set(&recovery_conninfo, "host", runtime_options.host);
char application_name[MAXLEN] = "";
get_conninfo_value(options.conninfo, "application_name", application_name);
if (strlen(application_name))
{
param_set(&recovery_conninfo, "application_name", application_name);
}
if (strlen(runtime_options.masterport))
{
param_set(&recovery_conninfo, "port", runtime_options.masterport);
}
if (strlen(runtime_options.dbname))
{
param_set(&recovery_conninfo, "dbname", runtime_options.dbname);
}
if (strlen(runtime_options.username))
{
param_set(&recovery_conninfo, "user", runtime_options.username);
}
/* Sanity-check barman connection and installation */
@@ -2008,7 +2017,7 @@ do_standby_clone(void)
/*
* Copy the source connection so that we have some default values,
* particularly stuff like passwords extracted from PGPASSFILE;
* these will be overridden from the upstream conninfo, if provided
* these will be overridden from the upstream conninfo, if provided.
*/
conn_to_param_list(source_conn, &recovery_conninfo);
@@ -2057,7 +2066,8 @@ do_standby_clone(void)
initialize_conninfo_params(&barman_conninfo, false);
parse_success = parse_conninfo_string(barman_conninfo_str, &barman_conninfo, errmsg);
/* parse_conninfo_string() here will remove the upstream's `application_name`, if set */
parse_success = parse_conninfo_string(barman_conninfo_str, &barman_conninfo, errmsg, true);
if (parse_success == false)
{
@@ -2135,7 +2145,9 @@ do_standby_clone(void)
log_verbose(LOG_DEBUG, "parsing upstream conninfo string \"%s\"\n", recovery_conninfo_str);
parse_success = parse_conninfo_string(recovery_conninfo_str, &recovery_conninfo, errmsg);
/* parse_conninfo_string() here will remove the upstream's `application_name`, if set */
parse_success = parse_conninfo_string(recovery_conninfo_str, &recovery_conninfo, errmsg, true);
if (parse_success == false)
{
log_err(_("Unable to parse conninfo string \"%s\" for upstream node:\n%s\n"),
@@ -3638,7 +3650,7 @@ do_standby_follow(void)
initialize_conninfo_params(&local_node_conninfo, false);
parse_success = parse_conninfo_string(local_node_record.conninfo_str, &local_node_conninfo, errmsg);
parse_success = parse_conninfo_string(local_node_record.conninfo_str, &local_node_conninfo, errmsg, false);
if (parse_success == false)
{
@@ -4046,8 +4058,6 @@ do_standby_switchover(void)
exit(ERR_BAD_CONFIG);
}
log_verbose(LOG_INFO, _("remote configuration file \"%s\" found on remote server\n"),
runtime_options.remote_config_file);
@@ -5621,7 +5631,12 @@ write_primary_conninfo(char *line, t_conninfo_param_list *param_list)
/* `application_name` not provided - default to repmgr node name */
if (application_name_provided == false)
{
if (strlen(options.node_name))
appendPQExpBuffer(&conninfo_buf, " application_name=%s", options.node_name);
else
appendPQExpBuffer(&conninfo_buf, " application_name=repmgr");
}
maxlen_snprintf(line, "primary_conninfo = '%s'\n", conninfo_buf.data);
@@ -7223,7 +7238,7 @@ initialize_conninfo_params(t_conninfo_param_list *param_list, bool set_defaults)
param_list->keywords = pg_malloc0(sizeof(char *) * (param_list->size + 1));
param_list->values = pg_malloc0(sizeof(char *) * (param_list->size + 1));
for (c = 0; c <= param_list->size; c++)
for (c = 0; c < param_list->size; c++)
{
param_list->keywords[c] = NULL;
param_list->values[c] = NULL;
@@ -7244,6 +7259,19 @@ initialize_conninfo_params(t_conninfo_param_list *param_list, bool set_defaults)
}
static void
copy_conninfo_params(t_conninfo_param_list *dest_list, t_conninfo_param_list *source_list)
{
int c;
for (c = 0; c < source_list->size && source_list->keywords[c] != NULL; c++)
{
if (source_list->values[c] != NULL && source_list->values[c][0] != '\0')
{
param_set(dest_list, source_list->keywords[c], source_list->values[c]);
}
}
}
static void
param_set(t_conninfo_param_list *param_list, const char *param, const char *value)
{
@@ -7253,7 +7281,7 @@ param_set(t_conninfo_param_list *param_list, const char *param, const char *valu
/*
* Scan array to see if the parameter is already set - if not, replace it
*/
for (c = 0; c <= param_list->size && param_list->keywords[c] != NULL; c++)
for (c = 0; c < param_list->size && param_list->keywords[c] != NULL; c++)
{
if (strcmp(param_list->keywords[c], param) == 0)
{
@@ -7293,11 +7321,11 @@ param_get(t_conninfo_param_list *param_list, const char *param)
{
int c;
for (c = 0; c <= param_list->size && param_list->keywords[c] != NULL; c++)
for (c = 0; c < param_list->size && param_list->keywords[c] != NULL; c++)
{
if (strcmp(param_list->keywords[c], param) == 0)
{
if (param_list->values[c] != NULL)
if (param_list->values[c] != NULL && param_list->values[c][0] != '\0')
return param_list->values[c];
else
return NULL;
@@ -7314,7 +7342,7 @@ param_get(t_conninfo_param_list *param_list, const char *param)
* See conn_to_param_list() to do the same for a PQconn
*/
static bool
parse_conninfo_string(const char *conninfo_str, t_conninfo_param_list *param_list, char *errmsg)
parse_conninfo_string(const char *conninfo_str, t_conninfo_param_list *param_list, char *errmsg, bool ignore_application_name)
{
PQconninfoOption *connOptions;
PQconninfoOption *option;
@@ -7331,6 +7359,10 @@ parse_conninfo_string(const char *conninfo_str, t_conninfo_param_list *param_lis
(option->val != NULL && option->val[0] == '\0'))
continue;
/* Ignore application_name */
if (ignore_application_name == true && strcmp(option->keyword, "application_name") == 0)
continue;
param_set(param_list, option->keyword, option->val);
}