/* Prepare cluster_list to be federation centric that will be passed to
* verify_clsuters_exists in federation_functions.c.
*/
static int _verify_fed_clusters(List cluster_list, const char *fed_name,
bool *existing_fed)
{
int rc = SLURM_SUCCESS;
char *tmp_name = NULL;
List tmp_list = list_create(slurmdb_destroy_cluster_rec);
ListIterator itr = list_iterator_create(cluster_list);
while ((tmp_name = list_next(itr))) {
slurmdb_cluster_rec_t *rec =
xmalloc(sizeof(slurmdb_cluster_rec_t));
slurmdb_init_cluster_rec(rec, 0);
rec->name = xstrdup(tmp_name);
list_append(tmp_list, rec);
}
if ((rc = verify_fed_clusters(tmp_list, fed_name, existing_fed)))
goto end_it;
/* have to reconcile lists now, clusters may have been removed from
* tmp_list */
list_iterator_reset(itr);
while ((tmp_name = list_next(itr))) {
if (!list_find_first(tmp_list, _find_cluster_rec_in_list,
tmp_name))
list_delete_item(itr);
}
end_it:
FREE_NULL_LIST(tmp_list);
list_iterator_destroy(itr);
return rc;
}
extern int sacctmgr_modify_federation(int argc, char **argv)
{
int rc = SLURM_SUCCESS;
int i=0;
int cond_set = 0, prev_set = 0, rec_set = 0, set = 0;
List ret_list = NULL;
slurmdb_federation_cond_t *federation_cond =
xmalloc(sizeof(slurmdb_federation_cond_t));
slurmdb_federation_rec_t *federation =
xmalloc(sizeof(slurmdb_federation_rec_t));
slurmdb_init_federation_cond(federation_cond, 0);
slurmdb_init_federation_rec(federation, 0);
for (i=0; i<argc; i++) {
int command_len = strlen(argv[i]);
if (!strncasecmp(argv[i], "Where", MAX(command_len, 5))) {
i++;
prev_set = _set_cond(&i, argc, argv,
federation_cond, NULL);
cond_set |= prev_set;
} else if (!strncasecmp(argv[i], "Set", MAX(command_len, 3))) {
i++;
prev_set = _set_rec(&i, argc, argv, NULL, federation);
rec_set |= prev_set;
} else {
prev_set = _set_cond(&i, argc, argv,
federation_cond, NULL);
cond_set |= prev_set;
}
}
if (exit_code) {
rc = SLURM_ERROR;
goto end_it;
} else if (!rec_set) {
exit_code=1;
fprintf(stderr, " You didn't give me anything to set\n");
rc = SLURM_ERROR;
goto end_it;
} else if (!cond_set) {
if (!commit_check("You didn't set any conditions "
"with 'WHERE'.\n"
"Are you sure you want to continue?")) {
printf("Aborted\n");
rc = SLURM_SUCCESS;
goto end_it;
}
} else if (verify_federations_exist(
federation_cond->federation_list)) {
rc = SLURM_ERROR;
goto end_it;
}
if (federation->cluster_list) {
bool existing_feds = false;
char *mod_fed = NULL;
slurmdb_cluster_rec_t *tmp_c = NULL;
List cluster_list = federation->cluster_list;
if (federation_cond->federation_list &&
(list_count(federation_cond->federation_list) > 1)) {
fprintf(stderr, " Can't assign clusters to "
"multiple federations.\n");
rc = SLURM_ERROR;
goto end_it;
}
/* Add all clusters that need to be removed if clearing all
* clusters or add clusters that will be removed if setting
* clusters to specific set. */
mod_fed = list_peek(federation_cond->federation_list);
if ((!list_count(cluster_list) ||
((tmp_c = list_peek(cluster_list)) &&
*tmp_c->name != '-' && *tmp_c->name != '+')) &&
((rc = _add_clusters_to_remove(cluster_list, mod_fed)) ||
(rc = _change_assigns_to_adds(cluster_list)))) {
goto end_it;
} else if ((rc = verify_fed_clusters(cluster_list, mod_fed,
&existing_feds))) {
goto end_it;
} else if (!list_count(cluster_list)) {
printf("Nothing to change\n");
rc = SLURM_ERROR;
goto end_it;
} else if (existing_feds) {
char *warning = "\nAre you sure you want to continue?";
if (!commit_check(warning)) {
rc = SLURM_ERROR;
goto end_it;
}
}
}
printf(" Setting\n");
sacctmgr_print_federation(federation);
notice_thread_init();
ret_list = acct_storage_g_modify_federations(db_conn, my_uid,
federation_cond,
federation);
if (ret_list && list_count(ret_list)) {
char *object = NULL;
ListIterator itr = list_iterator_create(ret_list);
printf(" Modified federation...\n");
while((object = list_next(itr))) {
printf(" %s\n", object);
}
list_iterator_destroy(itr);
set = 1;
} else if (ret_list) {
printf(" Nothing modified\n");
rc = SLURM_ERROR;
} else {
exit_code=1;
fprintf(stderr, " Error with request: %s\n",
slurm_strerror(errno));
rc = SLURM_ERROR;
}
FREE_NULL_LIST(ret_list);
notice_thread_fini();
if (set) {
if (commit_check("Would you like to commit changes?"))
acct_storage_g_commit(db_conn, 1);
else {
printf(" Changes Discarded\n");
acct_storage_g_commit(db_conn, 0);
}
}
end_it:
slurmdb_destroy_federation_cond(federation_cond);
slurmdb_destroy_federation_rec(federation);
return rc;
}
extern int sacctmgr_add_federation(int argc, char **argv)
{
int rc = SLURM_SUCCESS;
int i = 0, limit_set = 0;
slurmdb_federation_rec_t *start_fed =
xmalloc(sizeof(slurmdb_federation_rec_t));
List name_list = list_create(slurm_destroy_char);
List federation_list;
ListIterator itr = NULL;
char *name = NULL;
slurmdb_init_federation_rec(start_fed, 0);
for (i=0; i<argc; i++) {
int command_len = strlen(argv[i]);
if (!strncasecmp(argv[i], "Where", MAX(command_len, 5))
|| !strncasecmp(argv[i], "Set", MAX(command_len, 3)))
i++;
limit_set += _set_rec(&i, argc, argv, name_list, start_fed);
}
if (exit_code) {
FREE_NULL_LIST(name_list);
slurmdb_destroy_federation_rec(start_fed);
return SLURM_ERROR;
} else if (!list_count(name_list)) {
slurmdb_destroy_federation_rec(start_fed);
FREE_NULL_LIST(name_list);
fprintf(stderr, " Need name of federation to add.\n");
return SLURM_ERROR;
} else if (_remove_existing_feds(name_list)) {
FREE_NULL_LIST(name_list);
slurmdb_destroy_federation_rec(start_fed);
return SLURM_ERROR;
}
if ((list_count(name_list) > 1) &&
start_fed && start_fed->cluster_list &&
list_count(start_fed->cluster_list)) {
slurmdb_destroy_federation_rec(start_fed);
FREE_NULL_LIST(name_list);
fprintf(stderr, " Can't assign clusters to multiple "
"federations.\n");
return SLURM_ERROR;
}
if (start_fed && start_fed->cluster_list &&
list_count(start_fed->cluster_list)) {
bool existing_feds = false;
if (list_count(name_list) > 1){
slurmdb_destroy_federation_rec(start_fed);
FREE_NULL_LIST(name_list);
fprintf(stderr, " Can't assign clusters to "
"multiple federations.\n");
return SLURM_ERROR;
}
/* ensure that clusters exist in db */
/* and if the clusters are already assigned to another fed. */
if (verify_fed_clusters(start_fed->cluster_list, NULL,
&existing_feds)) {
FREE_NULL_LIST(name_list);
slurmdb_destroy_federation_rec(start_fed);
return SLURM_ERROR;
} else if (existing_feds) {
char *warning = "\nAre you sure you want to continue?";
if (!commit_check(warning)) {
FREE_NULL_LIST(name_list);
slurmdb_destroy_federation_rec(start_fed);
return SLURM_ERROR;
}
}
}
printf(" Adding Federation(s)\n");
federation_list = list_create(slurmdb_destroy_federation_rec);
itr = list_iterator_create(name_list);
while((name = list_next(itr))) {
slurmdb_federation_rec_t *fed = NULL;
if (!name[0]) {
fprintf(stderr, " Skipping blank fed name.\n");
continue;
}
fed = xmalloc(sizeof(slurmdb_federation_rec_t));
slurmdb_init_federation_rec(fed, 0);
list_append(federation_list, fed);
slurmdb_copy_federation_rec(fed, start_fed);
fed->name = xstrdup(name);
printf(" %s\n", fed->name);
}
list_iterator_destroy(itr);
FREE_NULL_LIST(name_list);
if (limit_set) {
printf(" Settings\n");
sacctmgr_print_federation(start_fed);
}
slurmdb_destroy_federation_rec(start_fed);
if (!list_count(federation_list)) {
printf(" Nothing new added.\n");
rc = SLURM_ERROR;
goto end_it;
}
notice_thread_init();
rc = acct_storage_g_add_federations(db_conn, my_uid,
federation_list);
notice_thread_fini();
if (rc == SLURM_SUCCESS) {
if (commit_check("Would you like to commit changes?")) {
acct_storage_g_commit(db_conn, 1);
} else {
printf(" Changes Discarded\n");
acct_storage_g_commit(db_conn, 0);
}
} else {
exit_code = 1;
fprintf(stderr, " Problem adding federation(s): %s\n",
slurm_strerror(rc));
rc = SLURM_ERROR;
}
end_it:
FREE_NULL_LIST(federation_list);
return rc;
}
