/* * add clusters to accounting system * IN: cluster_list List of slurmdb_cluster_rec_t * * RET: SLURM_SUCCESS on success SLURM_ERROR else */ extern int slurmdb_clusters_add(void *db_conn, List cluster_list) { if (db_api_uid == -1) db_api_uid = getuid(); return acct_storage_g_add_clusters(db_conn, db_api_uid, cluster_list); }
extern int sacctmgr_add_cluster(int argc, char **argv) { int rc = SLURM_SUCCESS; int i = 0; slurmdb_cluster_rec_t *cluster = NULL; slurmdb_cluster_rec_t *start_cluster = xmalloc(sizeof(slurmdb_cluster_rec_t)); List name_list = list_create(slurm_destroy_char); List cluster_list = NULL; slurmdb_assoc_rec_t start_assoc; int limit_set = 0; ListIterator itr = NULL, itr_c = NULL; char *name = NULL; slurmdb_init_assoc_rec(&start_assoc, 0); slurmdb_init_cluster_rec(start_cluster, 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_assoc, start_cluster); } if (exit_code) { FREE_NULL_LIST(name_list); slurmdb_destroy_cluster_rec(start_cluster); return SLURM_ERROR; } else if (!list_count(name_list)) { FREE_NULL_LIST(name_list); slurmdb_destroy_cluster_rec(start_cluster); exit_code=1; fprintf(stderr, " Need name of cluster to add.\n"); return SLURM_ERROR; } else { List temp_list = NULL; slurmdb_cluster_cond_t cluster_cond; slurmdb_init_cluster_cond(&cluster_cond, 0); cluster_cond.cluster_list = name_list; cluster_cond.classification = start_cluster->classification; temp_list = acct_storage_g_get_clusters(db_conn, my_uid, &cluster_cond); if (!temp_list) { exit_code=1; fprintf(stderr, " Problem getting clusters from database. " "Contact your admin.\n"); slurmdb_destroy_cluster_rec(start_cluster); return SLURM_ERROR; } itr_c = list_iterator_create(name_list); itr = list_iterator_create(temp_list); while((name = list_next(itr_c))) { slurmdb_cluster_rec_t *cluster_rec = NULL; list_iterator_reset(itr); while((cluster_rec = list_next(itr))) { if (!xstrcasecmp(cluster_rec->name, name)) break; } if (cluster_rec) { printf(" This cluster %s already exists. " "Not adding.\n", name); list_delete_item(itr_c); } } list_iterator_destroy(itr); list_iterator_destroy(itr_c); FREE_NULL_LIST(temp_list); if (!list_count(name_list)) { FREE_NULL_LIST(name_list); slurmdb_destroy_cluster_rec(start_cluster); return SLURM_ERROR; } } if (start_cluster->fed.name) { int rc; List fed_list = list_create(slurm_destroy_char); list_append(fed_list, xstrdup(start_cluster->fed.name)); rc = verify_federations_exist(fed_list); FREE_NULL_LIST(fed_list); if (rc) { slurmdb_destroy_cluster_rec(start_cluster); FREE_NULL_LIST(name_list); return SLURM_ERROR; } } printf(" Adding Cluster(s)\n"); cluster_list = list_create(slurmdb_destroy_cluster_rec); itr = list_iterator_create(name_list); while((name = list_next(itr))) { if (!name[0]) { exit_code=1; fprintf(stderr, " No blank names are " "allowed when adding.\n"); rc = SLURM_ERROR; continue; } cluster = xmalloc(sizeof(slurmdb_cluster_rec_t)); slurmdb_init_cluster_rec(cluster, 0); list_append(cluster_list, cluster); slurmdb_copy_cluster_rec(cluster, start_cluster); cluster->name = xstrdup(name); printf(" Name = %s\n", cluster->name); cluster->root_assoc = xmalloc(sizeof(slurmdb_assoc_rec_t)); slurmdb_init_assoc_rec(cluster->root_assoc, 0); cluster->root_assoc->def_qos_id = start_assoc.def_qos_id; cluster->root_assoc->shares_raw = start_assoc.shares_raw; slurmdb_copy_assoc_rec_limits( cluster->root_assoc, &start_assoc); } list_iterator_destroy(itr); FREE_NULL_LIST(name_list); if (limit_set) printf(" Setting\n"); if (limit_set & CLUS_REC_SET) sacctmgr_print_cluster(start_cluster); if (limit_set & CLUS_ASSOC_SET) { printf(" Default Limits:\n"); sacctmgr_print_assoc_limits(&start_assoc); FREE_NULL_LIST(start_assoc.qos_list); } slurmdb_destroy_cluster_rec(start_cluster); if (!list_count(cluster_list)) { printf(" Nothing new added.\n"); rc = SLURM_ERROR; goto end_it; } /* Since we are creating tables with add cluster that can't be rolled back. So we ask before hand if they are serious about it so we can rollback if needed. */ if (commit_check("Would you like to commit changes?")) { notice_thread_init(); rc = acct_storage_g_add_clusters(db_conn, my_uid, cluster_list); notice_thread_fini(); if (rc == SLURM_SUCCESS) { acct_storage_g_commit(db_conn, 1); } else { exit_code=1; fprintf(stderr, " Problem adding clusters: %s\n", slurm_strerror(rc)); /* this isn't really needed, but just to be safe */ acct_storage_g_commit(db_conn, 0); } } else { printf(" Changes Discarded\n"); /* this isn't really needed, but just to be safe */ acct_storage_g_commit(db_conn, 0); } end_it: FREE_NULL_LIST(cluster_list); return rc; }
/* * add clusters to accounting system * IN: cluster_list List of slurmdb_cluster_rec_t * * RET: SLURM_SUCCESS on success SLURM_ERROR else */ extern int slurmdb_clusters_add(void *db_conn, List cluster_list) { return acct_storage_g_add_clusters(db_conn, getuid(), cluster_list); }