extern mysql_db_info_t *create_mysql_db_info(slurm_mysql_plugin_type_t type)
{
mysql_db_info_t *db_info = xmalloc(sizeof(mysql_db_info_t));
switch (type) {
case SLURM_MYSQL_PLUGIN_AS:
db_info->port = slurm_get_accounting_storage_port();
if (!db_info->port) {
db_info->port = DEFAULT_MYSQL_PORT;
slurm_set_accounting_storage_port(db_info->port);
}
db_info->host = slurm_get_accounting_storage_host();
db_info->backup = slurm_get_accounting_storage_backup_host();
db_info->user = slurm_get_accounting_storage_user();
db_info->pass = slurm_get_accounting_storage_pass();
break;
case SLURM_MYSQL_PLUGIN_JC:
db_info->port = slurm_get_jobcomp_port();
if (!db_info->port) {
db_info->port = DEFAULT_MYSQL_PORT;
slurm_set_jobcomp_port(db_info->port);
}
db_info->host = slurm_get_jobcomp_host();
db_info->user = slurm_get_jobcomp_user();
db_info->pass = slurm_get_jobcomp_pass();
break;
default:
xfree(db_info);
fatal("Unknown mysql_db_info %d", type);
}
return db_info;
}
static void _load_slurm_config(void)
{
acct_storage_backup_host = slurm_get_accounting_storage_backup_host();
acct_storage_host = slurm_get_accounting_storage_host();
acct_storage_loc = slurm_get_accounting_storage_loc();
acct_storage_pass = slurm_get_accounting_storage_pass();
acct_storage_port = slurm_get_accounting_storage_port();
acct_storage_type = slurm_get_accounting_storage_type();
acct_storage_user = slurm_get_accounting_storage_user();
auth_type = slurm_get_auth_type();
msg_timeout = slurm_get_msg_timeout();
plugin_dir = slurm_get_plugin_dir();
private_data = slurm_get_private_data();
slurm_user_id = slurm_get_slurm_user_id();
track_wckey = slurm_get_track_wckey();
}
static void _layout_conf_dbd(GtkTreeStore *treestore)
{
ListIterator itr = NULL;
GtkTreeIter iter;
config_key_pair_t *key_pair;
int update = 0;
time_t now = time(NULL);
char tmp_str[128], *user_name = NULL;
List dbd_config_list = NULL;
/* first load accounting parms from slurm.conf */
char *acct_storage_backup_host =
slurm_get_accounting_storage_backup_host();
char *acct_storage_host = slurm_get_accounting_storage_host();
char *acct_storage_loc = slurm_get_accounting_storage_loc();
char *acct_storage_pass = slurm_get_accounting_storage_pass();
uint32_t acct_storage_port = slurm_get_accounting_storage_port();
char *acct_storage_type = slurm_get_accounting_storage_type();
char *acct_storage_user = slurm_get_accounting_storage_user();
char *auth_type = slurm_get_auth_type();
uint16_t msg_timeout = slurm_get_msg_timeout();
char *plugin_dir = slurm_get_plugin_dir();
uint16_t private_data = slurm_get_private_data();
uint32_t slurm_user_id = slurm_get_slurm_user_id();
uint16_t track_wckey = slurm_get_track_wckey();
slurm_make_time_str(&now, tmp_str, sizeof(tmp_str));
add_display_treestore_line_with_font(
update, treestore, &iter,
"SLURM Configuration data as of", tmp_str, "bold");
add_display_treestore_line(update, treestore, &iter,
"AccountingStorageBackupHost",
acct_storage_backup_host);
add_display_treestore_line(update, treestore, &iter,
"AccountingStorageHost", acct_storage_host);
add_display_treestore_line(update, treestore, &iter,
"AccountingStorageLoc", acct_storage_loc);
add_display_treestore_line(update, treestore, &iter,
"AccountingStoragePass", acct_storage_pass);
sprintf(tmp_str, "%u", acct_storage_port);
add_display_treestore_line(update, treestore, &iter,
"AccountingStoragePort", tmp_str);
add_display_treestore_line(update, treestore, &iter,
"AccountingStorageType", acct_storage_type);
add_display_treestore_line(update, treestore, &iter,
"AccountingStorageUser", acct_storage_user);
add_display_treestore_line(update, treestore, &iter,
"AuthType", auth_type);
sprintf(tmp_str, "%u sec", msg_timeout);
add_display_treestore_line(update, treestore, &iter,
"MessageTimeout", tmp_str);
add_display_treestore_line(update, treestore, &iter,
"PluginDir", plugin_dir);
private_data_string(private_data, tmp_str, sizeof(tmp_str));
add_display_treestore_line(update, treestore, &iter,
"PrivateData", tmp_str);
user_name = uid_to_string(slurm_user_id);
sprintf(tmp_str, "%s(%u)", user_name, slurm_user_id);
xfree(user_name);
add_display_treestore_line(update, treestore, &iter,
"SlurmUserId", tmp_str);
add_display_treestore_line(update, treestore, &iter,
"SLURM_CONF", default_slurm_config_file);
add_display_treestore_line(update, treestore, &iter,
"SLURM_VERSION", SLURM_VERSION_STRING);
sprintf(tmp_str, "%u", track_wckey);
add_display_treestore_line(update, treestore, &iter,
"TrackWCKey", tmp_str);
xfree(acct_storage_backup_host);
xfree(acct_storage_host);
xfree(acct_storage_loc);
xfree(acct_storage_pass);
xfree(acct_storage_type);
xfree(acct_storage_user);
xfree(auth_type);
xfree(plugin_dir);
/* now load accounting parms from slurmdbd.conf */
/* second load slurmdbd.conf parms */
if (!(dbd_config_list = slurmdb_config_get(NULL)))
return;
add_display_treestore_line_with_font(
update, treestore, &iter,
"\nSlurmDBD Configuration:", NULL, "bold");
itr = list_iterator_create(dbd_config_list);
while ((key_pair = list_next(itr))) {
add_display_treestore_line(update, treestore, &iter,
key_pair->name,
key_pair->value);
}
list_iterator_destroy(itr);
}
/* Open a connection to the Slurm DBD and set slurmdbd_conn */
static void _open_slurmdbd_conn(bool need_db)
{
bool try_backup = true;
int rc;
if (slurmdbd_conn && slurmdbd_conn->fd >= 0) {
debug("Attempt to re-open slurmdbd socket");
/* clear errno (checked after this for errors) */
errno = 0;
return;
}
slurm_persist_conn_close(slurmdbd_conn);
if (!slurmdbd_conn) {
slurmdbd_conn = xmalloc(sizeof(slurm_persist_conn_t));
slurmdbd_conn->flags =
PERSIST_FLAG_DBD | PERSIST_FLAG_RECONNECT;
slurmdbd_conn->persist_type = PERSIST_TYPE_DBD;
if (!slurmdbd_cluster)
slurmdbd_cluster = slurm_get_cluster_name();
slurmdbd_conn->cluster_name = xstrdup(slurmdbd_cluster);
slurmdbd_conn->timeout = (slurm_get_msg_timeout() + 35) * 1000;
slurmdbd_conn->rem_port = slurm_get_accounting_storage_port();
if (!slurmdbd_conn->rem_port) {
slurmdbd_conn->rem_port = SLURMDBD_PORT;
slurm_set_accounting_storage_port(
slurmdbd_conn->rem_port);
}
}
slurmdbd_shutdown = 0;
slurmdbd_conn->shutdown = &slurmdbd_shutdown;
slurmdbd_conn->version = SLURM_PROTOCOL_VERSION;
xfree(slurmdbd_conn->rem_host);
slurmdbd_conn->rem_host = slurm_get_accounting_storage_host();
if (!slurmdbd_conn->rem_host) {
slurmdbd_conn->rem_host = xstrdup(DEFAULT_STORAGE_HOST);
slurm_set_accounting_storage_host(
slurmdbd_conn->rem_host);
}
again:
if (((rc = slurm_persist_conn_open(slurmdbd_conn)) != SLURM_SUCCESS) &&
try_backup) {
xfree(slurmdbd_conn->rem_host);
try_backup = false;
if ((slurmdbd_conn->rem_host =
slurm_get_accounting_storage_backup_host()))
goto again;
}
if (rc == SLURM_SUCCESS) {
/* set the timeout to the timeout to be used for all other
* messages */
slurmdbd_conn->timeout = SLURMDBD_TIMEOUT * 1000;
if (slurmdbd_conn->trigger_callbacks.dbd_resumed)
(slurmdbd_conn->trigger_callbacks.dbd_resumed)();
if (slurmdbd_conn->trigger_callbacks.db_resumed)
(slurmdbd_conn->trigger_callbacks.db_resumed)();
}
if ((!need_db && (rc == ESLURM_DB_CONNECTION)) ||
(rc == SLURM_SUCCESS)) {
debug("slurmdbd: Sent PersistInit msg");
/* clear errno (checked after this for
errors)
*/
errno = 0;
} else {
if ((rc == ESLURM_DB_CONNECTION) &&
slurmdbd_conn->trigger_callbacks.db_fail)
(slurmdbd_conn->trigger_callbacks.db_fail)();
error("slurmdbd: Sending PersistInit msg: %m");
slurm_persist_conn_close(slurmdbd_conn);
}
}
/* Open a connection to the Slurm DBD and set slurmdbd_conn */
static void _open_slurmdbd_conn(bool need_db)
{
char *backup_host = NULL;
int rc;
if (slurmdbd_conn && slurmdbd_conn->fd >= 0) {
debug("Attempt to re-open slurmdbd socket");
/* clear errno (checked after this for errors) */
errno = 0;
return;
}
slurm_persist_conn_close(slurmdbd_conn);
if (!slurmdbd_conn) {
slurmdbd_conn = xmalloc(sizeof(slurm_persist_conn_t));
slurmdbd_conn->flags =
PERSIST_FLAG_DBD | PERSIST_FLAG_RECONNECT;
slurmdbd_conn->persist_type = PERSIST_TYPE_DBD;
if (!slurmdbd_cluster)
slurmdbd_cluster = slurm_get_cluster_name();
slurmdbd_conn->cluster_name = xstrdup(slurmdbd_cluster);
slurmdbd_conn->timeout = (slurm_get_msg_timeout() + 35) * 1000;
slurmdbd_conn->rem_port = slurm_get_accounting_storage_port();
if (!slurmdbd_conn->rem_port) {
slurmdbd_conn->rem_port = SLURMDBD_PORT;
slurm_set_accounting_storage_port(
slurmdbd_conn->rem_port);
}
}
slurmdbd_shutdown = 0;
slurmdbd_conn->shutdown = &slurmdbd_shutdown;
slurmdbd_conn->version = SLURM_PROTOCOL_VERSION;
xfree(slurmdbd_conn->rem_host);
slurmdbd_conn->rem_host = slurm_get_accounting_storage_host();
if (!slurmdbd_conn->rem_host) {
slurmdbd_conn->rem_host = xstrdup(DEFAULT_STORAGE_HOST);
slurm_set_accounting_storage_host(
slurmdbd_conn->rem_host);
}
// See if a backup slurmdbd is configured
backup_host = slurm_get_accounting_storage_backup_host();
again:
// A connection failure is only an error if backup dne or also fails
if (backup_host)
slurmdbd_conn->flags |= PERSIST_FLAG_SUPPRESS_ERR;
else
slurmdbd_conn->flags &= (~PERSIST_FLAG_SUPPRESS_ERR);
if (((rc = slurm_persist_conn_open(slurmdbd_conn)) != SLURM_SUCCESS) &&
backup_host) {
xfree(slurmdbd_conn->rem_host);
// Force the next error to display
slurmdbd_conn->comm_fail_time = 0;
slurmdbd_conn->rem_host = backup_host;
backup_host = NULL;
goto again;
}
xfree(backup_host);
if (rc == SLURM_SUCCESS) {
/* set the timeout to the timeout to be used for all other
* messages */
slurmdbd_conn->timeout = SLURMDBD_TIMEOUT * 1000;
if (slurmdbd_conn->trigger_callbacks.dbd_resumed)
(slurmdbd_conn->trigger_callbacks.dbd_resumed)();
if (slurmdbd_conn->trigger_callbacks.db_resumed)
(slurmdbd_conn->trigger_callbacks.db_resumed)();
}
if ((!need_db && (rc == ESLURM_DB_CONNECTION)) ||
(rc == SLURM_SUCCESS)) {
debug("slurmdbd: Sent PersistInit msg");
/* clear errno (checked after this for
errors)
*/
errno = 0;
} else {
if ((rc == ESLURM_DB_CONNECTION) &&
slurmdbd_conn->trigger_callbacks.db_fail)
(slurmdbd_conn->trigger_callbacks.db_fail)();
error("slurmdbd: Sending PersistInit msg: %m");
slurm_persist_conn_close(slurmdbd_conn);
}
}
