diff --git a/contrib/pg_upgrade/check.c b/contrib/pg_upgrade/check.c index 1f67e602defd362d3a8e66ab6b987d71f3df43ec..0376fcbefc18446d789f1424b21998d8a40bc58b 100644 --- a/contrib/pg_upgrade/check.c +++ b/contrib/pg_upgrade/check.c @@ -459,6 +459,13 @@ void create_script_for_cluster_analyze(char **analyze_script_file_name) { FILE *script = NULL; + char *user_specification = ""; + + if (os_info.user_specified) + { + user_specification = pg_malloc(strlen(os_info.user) + 7); + sprintf(user_specification, "-U \"%s\" ", os_info.user); + } *analyze_script_file_name = pg_malloc(MAXPGPATH); @@ -501,7 +508,8 @@ create_script_for_cluster_analyze(char **analyze_script_file_name) ECHO_QUOTE, ECHO_QUOTE); fprintf(script, "echo %sthis script and run:%s\n", ECHO_QUOTE, ECHO_QUOTE); - fprintf(script, "echo %s \"%s/vacuumdb\" --all %s%s\n", ECHO_QUOTE, new_cluster.bindir, + fprintf(script, "echo %s \"%s/vacuumdb\" %s--all %s%s\n", ECHO_QUOTE, + new_cluster.bindir, user_specification, /* Did we copy the free space files? */ (GET_MAJOR_VERSION(old_cluster.major_version) >= 804) ? "--analyze-only" : "--analyze", ECHO_QUOTE); @@ -522,7 +530,8 @@ create_script_for_cluster_analyze(char **analyze_script_file_name) ECHO_QUOTE, ECHO_QUOTE); fprintf(script, "echo %s--------------------------------------------------%s\n", ECHO_QUOTE, ECHO_QUOTE); - fprintf(script, "\"%s/vacuumdb\" --all --analyze-only\n", new_cluster.bindir); + fprintf(script, "\"%s/vacuumdb\" %s--all --analyze-only\n", + new_cluster.bindir, user_specification); fprintf(script, "echo%s\n", ECHO_BLANK); fprintf(script, "echo %sThe server is now available with minimal optimizer statistics.%s\n", ECHO_QUOTE, ECHO_QUOTE); @@ -543,7 +552,8 @@ create_script_for_cluster_analyze(char **analyze_script_file_name) ECHO_QUOTE, ECHO_QUOTE); fprintf(script, "echo %s---------------------------------------------------%s\n", ECHO_QUOTE, ECHO_QUOTE); - fprintf(script, "\"%s/vacuumdb\" --all --analyze-only\n", new_cluster.bindir); + fprintf(script, "\"%s/vacuumdb\" %s--all --analyze-only\n", + new_cluster.bindir, user_specification); fprintf(script, "echo%s\n\n", ECHO_BLANK); #ifndef WIN32 @@ -556,7 +566,8 @@ create_script_for_cluster_analyze(char **analyze_script_file_name) ECHO_QUOTE, ECHO_QUOTE); fprintf(script, "echo %s-------------------------------------------------------------%s\n", ECHO_QUOTE, ECHO_QUOTE); - fprintf(script, "\"%s/vacuumdb\" --all %s\n", new_cluster.bindir, + fprintf(script, "\"%s/vacuumdb\" %s--all %s\n", new_cluster.bindir, + user_specification, /* Did we copy the free space files? */ (GET_MAJOR_VERSION(old_cluster.major_version) >= 804) ? "--analyze-only" : "--analyze"); @@ -573,6 +584,9 @@ create_script_for_cluster_analyze(char **analyze_script_file_name) *analyze_script_file_name, getErrorText(errno)); #endif + if (os_info.user_specified) + pg_free(user_specification); + check_ok(); } diff --git a/contrib/pg_upgrade/option.c b/contrib/pg_upgrade/option.c index dee58ee28fc48f4af5c1d54245927ac7cc48ed47..2774b1e21144da1a965305bcec8f66fd83a023d7 100644 --- a/contrib/pg_upgrade/option.c +++ b/contrib/pg_upgrade/option.c @@ -172,6 +172,7 @@ parseCommandLine(int argc, char *argv[]) case 'U': pg_free(os_info.user); os_info.user = pg_strdup(optarg); + os_info.user_specified = true; /* * Push the user name into the environment so pre-9.1 diff --git a/contrib/pg_upgrade/pg_upgrade.h b/contrib/pg_upgrade/pg_upgrade.h index 0d44f4e4477cf5bd2b95508d041e0b8924cda876..0b3ad200786aac01bd7fef4ffa8985b3c22401cd 100644 --- a/contrib/pg_upgrade/pg_upgrade.h +++ b/contrib/pg_upgrade/pg_upgrade.h @@ -291,6 +291,7 @@ typedef struct const char *progname; /* complete pathname for this program */ char *exec_path; /* full path to my executable */ char *user; /* username for clusters */ + bool user_specified; /* user specified on command-line */ char **old_tablespaces; /* tablespaces */ int num_old_tablespaces; char **libraries; /* loadable libraries */