/*
* Init gtm master -----------------------------------------------------------------
*/
static cmd_t *
prepare_initGtmMaster(void)
{
cmd_t *cmdInitGtmMaster, *cmdGtmConf, *cmdGxid;
char date[MAXTOKEN+1];
FILE *f;
char **fileList = NULL;
/* Kill current gtm, bild work directory and run initgtm */
cmdInitGtmMaster = initCmd(sval(VAR_gtmMasterServer));
snprintf(newCommand(cmdInitGtmMaster), MAXLINE,
"killall -u %s -9 gtm; rm -rf %s; mkdir -p %s;initgtm -Z gtm -D %s",
sval(VAR_pgxcUser),
sval(VAR_gtmMasterDir), sval(VAR_gtmMasterDir), sval(VAR_gtmMasterDir));
/* Then prepare gtm.conf file */
/* Prepare local Stdin */
appendCmdEl(cmdInitGtmMaster, (cmdGtmConf = initCmd(sval(VAR_gtmMasterServer))));
if ((f = prepareLocalStdin(newFilename(cmdGtmConf->localStdin), MAXPATH, NULL)) == NULL)
{
cleanCmd(cmdInitGtmMaster);
return(NULL);
}
fprintf(f,
"#===============================================\n"
"# Added at initialization, %s\n"
"listen_addresses = '*'\n",
timeStampString(date, MAXTOKEN));
if (!is_none(sval(VAR_gtmExtraConfig)))
AddMember(fileList, sval(VAR_gtmExtraConfig));
if (!is_none(sval(VAR_gtmMasterSpecificExtraConfig)))
AddMember(fileList, sval(VAR_gtmMasterSpecificExtraConfig));
appendFiles(f, fileList);
CleanArray(fileList);
fprintf(f,
"port = %s\n"
"nodename = '%s'\n"
"startup = ACT\n"
"# End of addition\n",
sval(VAR_gtmMasterPort), sval(VAR_gtmName));
fclose(f);
/* other options */
snprintf(newCommand(cmdGtmConf), MAXLINE,
"cat >> %s/gtm.conf",
sval(VAR_gtmMasterDir));
/* Setup GTM with appropriate GXID value */
appendCmdEl(cmdGtmConf, (cmdGxid = initCmd(sval(VAR_gtmMasterServer))));
snprintf(newCommand(cmdGxid), MAXLINE,
"(gtm -x 2000 -D %s &); sleep 1; gtm_ctl stop -Z gtm -D %s",
sval(VAR_gtmMasterDir), sval(VAR_gtmMasterDir));
return cmdInitGtmMaster;
}
/*
* Assumes Gtm Slave is configured.
* Caller should check this.
*/
static cmd_t *
prepare_initGtmSlave(void)
{
char date[MAXTOKEN+1];
cmd_t *cmdInitGtm, *cmdGtmConf;
FILE *f;
char **fileList = NULL;
if (!isVarYes(VAR_gtmSlave) || (sval(VAR_gtmSlaveServer) == NULL) || is_none(sval(VAR_gtmSlaveServer)))
{
elog(ERROR, "ERROR: GTM slave is not configured.\n");
return(NULL);
}
/* Kill current gtm, build work directory and run initgtm */
cmdInitGtm = initCmd(sval(VAR_gtmSlaveServer));
snprintf(newCommand(cmdInitGtm), MAXLINE,
"killall -u %s -9 gtm; rm -rf %s; mkdir -p %s; initgtm -Z gtm -D %s",
sval(VAR_pgxcUser),
sval(VAR_gtmSlaveDir), sval(VAR_gtmSlaveDir), sval(VAR_gtmSlaveDir));
/* Prepare gtm.conf file */
/* Prepare local Stdin */
appendCmdEl(cmdInitGtm, (cmdGtmConf = initCmd(sval(VAR_gtmSlaveServer))));
snprintf(newCommand(cmdGtmConf), MAXLINE,
"cat >> %s/gtm.conf",
sval(VAR_gtmSlaveDir));
if ((f = prepareLocalStdin(newFilename(cmdGtmConf->localStdin), MAXPATH, NULL)) == NULL)
{
cleanCmd(cmdInitGtm);
return(NULL);
}
fprintf(f,
"#===============================================\n"
"# Added at initialization, %s\n"
"listen_addresses = '*'\n",
timeStampString(date, MAXPATH+1));
if (!is_none(sval(VAR_gtmExtraConfig)))
AddMember(fileList, sval(VAR_gtmExtraConfig));
if (!is_none(sval(VAR_gtmMasterSpecificExtraConfig)))
AddMember(fileList, sval(VAR_gtmMasterSpecificExtraConfig));
appendFiles(f, fileList);
CleanArray(fileList);
fprintf(f,
"port = %s\n"
"nodename = '%s'\n"
"startup = STANDBY\n"
"active_host = '%s'\n"
"active_port = %d\n"
"# End of addition\n",
sval(VAR_gtmSlavePort), sval(VAR_gtmName),
sval(VAR_gtmMasterServer), atoi(sval(VAR_gtmMasterPort)));
fclose(f);
return (cmdInitGtm);
}
void ARGUMENTS_LIST::BuildArray()
{
if (m_argv) CleanArray();
m_argv = new char*[Argc()+1];
list<string>::const_iterator it = m_argvStrList.begin();
int i=0;
for (; it != m_argvStrList.end(); ++it, ++i)
{
m_argv[i] = new char[it->size()+1];
strcpy(m_argv[i], it->c_str());
}
m_argv[i] = 0;
}
void ARGUMENTS_LIST::Add(const string& argsStr)
{
unsigned int i=0;
while (i<argsStr.size())
{
while (argsStr[i] == ' ') i++;
unsigned int j=i;
while((argsStr[j] != ' ') && (j < argsStr.size())) j++;
string arg = argsStr.substr(i, j-i);
m_argvStrList.push_back(arg);
i = j;
}
CleanArray();
}
global func CleanArray4K(array &a) {return CleanArray(a);}
/*
* Does not check if node name is valid.
*/
static cmd_t *
prepare_initGtmProxy(char *nodeName)
{
cmd_t *cmdInitGtm, *cmdGtmProxyConf;
int idx;
FILE *f;
char timestamp[MAXTOKEN+1];
char **fileList = NULL;
if ((idx = gtmProxyIdx(nodeName)) < 0)
{
elog(ERROR, "ERROR: Specified name %s is not GTM Proxy configuration.\n", nodeName);
return NULL;
}
/* Build directory and run initgtm */
cmdInitGtm = initCmd(aval(VAR_gtmProxyServers)[idx]);
snprintf(newCommand(cmdInitGtm), MAXLINE,
"killall -u %s -9 gtm_proxy;"
"rm -rf %s;"
"mkdir -p %s;"
"initgtm -Z gtm_proxy -D %s",
sval(VAR_pgxcUser),
aval(VAR_gtmProxyDirs)[idx],
aval(VAR_gtmProxyDirs)[idx],
aval(VAR_gtmProxyDirs)[idx]);
/* Configure gtm_proxy.conf */
appendCmdEl(cmdInitGtm, (cmdGtmProxyConf = initCmd(aval(VAR_gtmProxyServers)[idx])));
if ((f = prepareLocalStdin(newFilename(cmdGtmProxyConf->localStdin), MAXPATH, NULL)) == NULL)
{
cleanCmd(cmdInitGtm);
return NULL;
}
fprintf(f,
"#===========================\n"
"# Added at initialization, %s\n"
"listen_addresses = '*'\n"
"worker_threads = 1\n"
"gtm_connect_retry_interval = 1\n",
timeStampString(timestamp, MAXTOKEN));;
if (!is_none(sval(VAR_gtmPxyExtraConfig)))
AddMember(fileList, sval(VAR_gtmPxyExtraConfig));
if (!is_none(aval(VAR_gtmPxySpecificExtraConfig)[idx]))
AddMember(fileList, aval(VAR_gtmPxySpecificExtraConfig)[idx]);
appendFiles(f, fileList);
CleanArray(fileList);
fprintf(f,
"nodename = '%s'\n"
"port = %s\n"
"gtm_host = '%s'\n"
"gtm_port = %s\n"
"# End of addition\n",
aval(VAR_gtmProxyNames)[idx],
aval(VAR_gtmProxyPorts)[idx],
sval(VAR_gtmMasterServer),
sval(VAR_gtmMasterPort));
fclose(f);
snprintf(newCommand(cmdGtmProxyConf), MAXLINE,
"cat >> %s/gtm_proxy.conf", aval(VAR_gtmProxyDirs)[idx]);
return(cmdInitGtm);
}
int
remove_gtmProxy(char *name, bool clean_opt)
{
FILE *f;
int idx;
/* Check if gtm proxy exists */
if ((idx = gtmProxyIdx(name)) < 0)
{
elog(ERROR, "ERROR: %s is not a gtm proxy.\n", name);
return 1;
}
/* Check if it is in use */
if (ifExists(VAR_coordMasterServers, aval(VAR_gtmProxyServers)[idx]) ||
ifExists(VAR_coordSlaveServers, aval(VAR_gtmProxyServers)[idx]) ||
ifExists(VAR_datanodeMasterServers, aval(VAR_gtmProxyServers)[idx]) ||
ifExists(VAR_datanodeSlaveServers, aval(VAR_gtmProxyServers)[idx]))
{
elog(ERROR, "ERROR: GTM Proxy %s is in use\n", name);
return 1;
}
elog(NOTICE, "NOTICE: removing gtm_proxy %s\n", name);
/* Clean */
if (clean_opt)
{
char **nodelist = NULL;
elog(NOTICE, "NOTICE: cleaning target resources.\n");
AddMember(nodelist, name);
clean_gtm_proxy(nodelist);
CleanArray(nodelist);
}
/* Reconfigure */
var_assign(&aval(VAR_gtmProxyNames)[idx], Strdup("none"));
var_assign(&aval(VAR_gtmProxyServers)[idx], Strdup("none"));
var_assign(&aval(VAR_gtmProxyPorts)[idx], Strdup("-1"));
var_assign(&aval(VAR_gtmProxyDirs)[idx], Strdup("none"));
handle_no_slaves();
makeServerList();
/* Update configuration file and backup it */
if ((f = fopen(pgxc_ctl_config_path, "a")) == NULL)
{
/* Should it be panic? */
elog(ERROR, "ERROR: cannot open configuration file \"%s\", %s\n", pgxc_ctl_config_path, strerror(errno));
return 1;
}
fprintf(f,
"#===================================================\n"
"# pgxc configuration file updated due to GTM proxy addition\n"
"# %s\n"
"%s=%s\n" /* gtmProxy */
"%s=( %s )\n" /* gtmProxyNames */
"%s=( %s )\n" /* gtmProxyServers */
"%s=( %s )\n" /* gtmProxyPorts */
"%s=( %s )\n" /* gtmProxyDirs */
"#----End of reconfiguration -------------------------\n",
timeStampString(date, MAXTOKEN+1),
VAR_gtmProxy, sval(VAR_gtmProxy),
VAR_gtmProxyNames, listValue(VAR_gtmProxyNames),
VAR_gtmProxyServers, listValue(VAR_gtmProxyServers),
VAR_gtmProxyPorts, listValue(VAR_gtmProxyPorts),
VAR_gtmProxyDirs, listValue(VAR_gtmProxyDirs));
fclose(f);
backup_configuration();
elog(NOTICE, "Done.\n");
return 0;
}
/*
* Add gtm proxy: to be used after all the configuration is done.
*
* This function only maintains internal configuration, updte configuration file,
* and make backup if configured. You should run init and start it separately.
*/
int
add_gtmProxy(char *name, char *host, int port, char *dir)
{
char port_s[MAXTOKEN+1];
char date[MAXTOKEN+1];
FILE *f;
char **nodelist = NULL;
int rc;
if (is_none(host))
{
elog(ERROR, "ERROR: Cannot add gtm proxy with the name \"none\".\n");
return 1;
}
if (is_none(dir))
{
elog(ERROR, "ERROR: Cannot add gtm proxy with the directory \"none\".\n");
return 1;
}
if (checkSpecificResourceConflict(name, host, port, dir, TRUE))
{
elog(ERROR, "ERROR: New specified name:%s, host:%s, port:%d and dir:\"%s\" conflicts with existing node.\n",
name, host, port, dir);
return 1;
}
if (!isVarYes(VAR_gtmProxy))
{
assign_sval(VAR_gtmProxy, Strdup("y"));
reset_var(VAR_gtmProxyNames);
}
add_val(find_var(VAR_gtmProxyNames), Strdup(name));
add_val(find_var(VAR_gtmProxyServers), Strdup(host));
snprintf(port_s, MAXTOKEN, "%d", port);
add_val(find_var(VAR_gtmProxyPorts), Strdup(port_s));
add_val(find_var(VAR_gtmProxyDirs), Strdup(dir));
add_val(find_var(VAR_gtmPxySpecificExtraConfig), Strdup("none"));
makeServerList();
if ((f = fopen(pgxc_ctl_config_path, "a")) == NULL)
{
/* Should it be panic? */
elog(ERROR, "ERROR: cannot open configuration file \"%s\", %s\n", pgxc_ctl_config_path, strerror(errno));
return 1;
}
fprintf(f,
"#===================================================\n"
"# pgxc configuration file updated due to GTM proxy (%s) addition\n"
"# %s\n",
name,
timeStampString(date, MAXTOKEN+1));
fprintSval(f, VAR_gtmProxy);
fprintAval(f, VAR_gtmProxyNames);
fprintAval(f, VAR_gtmProxyServers);
fprintAval(f, VAR_gtmProxyPorts);
fprintAval(f, VAR_gtmProxyDirs);
fprintAval(f, VAR_gtmPxySpecificExtraConfig);
fprintf(f, "%s", "#----End of reconfiguration -------------------------\n");
fclose(f);
AddMember(nodelist, name);
init_gtm_proxy(nodelist);
rc = start_gtm_proxy(nodelist);
CleanArray(nodelist);
return rc;
}
ARGUMENTS_LIST::~ARGUMENTS_LIST()
{
CleanArray();
}
