/* 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; }