extern int sacctmgr_add_cluster(int argc, char *argv[])
{
int rc = SLURM_SUCCESS;
int i = 0;
slurmdb_cluster_rec_t *cluster = NULL;
List name_list = list_create(slurm_destroy_char);
List cluster_list = NULL;
slurmdb_association_rec_t start_assoc;
int limit_set = 0;
ListIterator itr = NULL, itr_c = NULL;
char *name = NULL;
uint16_t class = 0;
slurmdb_init_association_rec(&start_assoc, 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, &class);
}
if(exit_code) {
list_destroy(name_list);
return SLURM_ERROR;
} else if(!list_count(name_list)) {
list_destroy(name_list);
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 = class;
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");
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(!strcasecmp(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);
list_destroy(temp_list);
if(!list_count(name_list)) {
list_destroy(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);
cluster->flags = NO_VAL;
cluster->name = xstrdup(name);
cluster->classification = class;
cluster->root_assoc =
xmalloc(sizeof(slurmdb_association_rec_t));
slurmdb_init_association_rec(cluster->root_assoc, 0);
printf(" Name = %s\n", cluster->name);
if(cluster->classification)
printf(" Classification= %s\n",
get_classification_str(cluster->classification));
cluster->root_assoc->def_qos_id = start_assoc.def_qos_id;
cluster->root_assoc->shares_raw = start_assoc.shares_raw;
cluster->root_assoc->grp_cpus = start_assoc.grp_cpus;
cluster->root_assoc->grp_jobs = start_assoc.grp_jobs;
cluster->root_assoc->grp_nodes = start_assoc.grp_nodes;
cluster->root_assoc->grp_submit_jobs =
start_assoc.grp_submit_jobs;
cluster->root_assoc->max_cpu_mins_pj =
start_assoc.max_cpu_mins_pj;
cluster->root_assoc->max_cpus_pj = start_assoc.max_cpus_pj;
cluster->root_assoc->max_jobs = start_assoc.max_jobs;
cluster->root_assoc->max_nodes_pj = start_assoc.max_nodes_pj;
cluster->root_assoc->max_submit_jobs =
start_assoc.max_submit_jobs;
cluster->root_assoc->max_wall_pj = start_assoc.max_wall_pj;
cluster->root_assoc->qos_list =
copy_char_list(start_assoc.qos_list);
}
list_iterator_destroy(itr);
list_destroy(name_list);
if(limit_set) {
printf(" Default Limits\n");
sacctmgr_print_assoc_limits(&start_assoc);
if(start_assoc.qos_list)
list_destroy(start_assoc.qos_list);
}
if(!list_count(cluster_list)) {
printf(" Nothing new added.\n");
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:
list_destroy(cluster_list);
return rc;
}
extern int sacctmgr_add_account(int argc, char *argv[])
{
int rc = SLURM_SUCCESS;
int i=0;
ListIterator itr = NULL, itr_c = NULL;
slurmdb_account_rec_t *acct = NULL;
slurmdb_association_rec_t *assoc = NULL;
slurmdb_association_cond_t assoc_cond;
List name_list = list_create(slurm_destroy_char);
List cluster_list = list_create(slurm_destroy_char);
char *cluster = NULL;
char *name = NULL;
List acct_list = NULL;
List assoc_list = NULL;
List local_assoc_list = NULL;
List local_account_list = NULL;
char *acct_str = NULL;
char *assoc_str = NULL;
int limit_set = 0;
slurmdb_account_rec_t *start_acct =
xmalloc(sizeof(slurmdb_account_rec_t));
slurmdb_association_rec_t *start_assoc =
xmalloc(sizeof(slurmdb_association_rec_t));
slurmdb_init_association_rec(start_assoc, 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, cluster_list,
start_acct, start_assoc);
}
if (exit_code)
return SLURM_ERROR;
if (!name_list || !list_count(name_list)) {
list_destroy(name_list);
list_destroy(cluster_list);
slurmdb_destroy_association_rec(start_assoc);
slurmdb_destroy_account_rec(start_acct);
exit_code=1;
fprintf(stderr, " Need name of account to add.\n");
return SLURM_SUCCESS;
} else {
slurmdb_account_cond_t account_cond;
memset(&account_cond, 0, sizeof(slurmdb_account_cond_t));
memset(&assoc_cond, 0, sizeof(slurmdb_association_cond_t));
assoc_cond.acct_list = name_list;
account_cond.assoc_cond = &assoc_cond;
local_account_list = acct_storage_g_get_accounts(
db_conn, my_uid, &account_cond);
}
if (!local_account_list) {
exit_code=1;
fprintf(stderr, " Problem getting accounts from database. "
"Contact your admin.\n");
list_destroy(name_list);
list_destroy(cluster_list);
slurmdb_destroy_association_rec(start_assoc);
slurmdb_destroy_account_rec(start_acct);
return SLURM_ERROR;
}
if (!start_assoc->parent_acct)
start_assoc->parent_acct = xstrdup("root");
if (!cluster_list || !list_count(cluster_list)) {
slurmdb_cluster_rec_t *cluster_rec = NULL;
List tmp_list =
acct_storage_g_get_clusters(db_conn, my_uid, NULL);
if (!tmp_list) {
exit_code=1;
fprintf(stderr,
" Problem getting clusters from database. "
"Contact your admin.\n");
list_destroy(name_list);
list_destroy(cluster_list);
slurmdb_destroy_association_rec(start_assoc);
slurmdb_destroy_account_rec(start_acct);
list_destroy(local_account_list);
return SLURM_ERROR;
}
if (!list_count(tmp_list)) {
exit_code=1;
fprintf(stderr,
" Can't add accounts, no cluster "
"defined yet.\n"
" Please contact your administrator.\n");
list_destroy(name_list);
list_destroy(cluster_list);
slurmdb_destroy_association_rec(start_assoc);
slurmdb_destroy_account_rec(start_acct);
list_destroy(local_account_list);
return SLURM_ERROR;
}
if (!cluster_list)
list_create(slurm_destroy_char);
else
list_flush(cluster_list);
itr_c = list_iterator_create(tmp_list);
while((cluster_rec = list_next(itr_c))) {
list_append(cluster_list, xstrdup(cluster_rec->name));
}
list_iterator_destroy(itr_c);
list_destroy(tmp_list);
} else {
List temp_list = NULL;
slurmdb_cluster_cond_t cluster_cond;
slurmdb_init_cluster_cond(&cluster_cond, 0);
cluster_cond.cluster_list = cluster_list;
temp_list = acct_storage_g_get_clusters(db_conn, my_uid,
&cluster_cond);
itr_c = list_iterator_create(cluster_list);
itr = list_iterator_create(temp_list);
while((cluster = list_next(itr_c))) {
slurmdb_cluster_rec_t *cluster_rec = NULL;
list_iterator_reset(itr);
while((cluster_rec = list_next(itr))) {
if (!strcasecmp(cluster_rec->name, cluster))
break;
}
if (!cluster_rec) {
exit_code=1;
fprintf(stderr, " This cluster '%s' "
"doesn't exist.\n"
" Contact your admin "
"to add it to accounting.\n",
cluster);
list_delete_item(itr_c);
}
}
list_iterator_destroy(itr);
list_iterator_destroy(itr_c);
list_destroy(temp_list);
if (!list_count(cluster_list)) {
slurmdb_destroy_association_rec(start_assoc);
slurmdb_destroy_account_rec(start_acct);
list_destroy(local_account_list);
return SLURM_ERROR;
}
}
acct_list = list_create(slurmdb_destroy_account_rec);
assoc_list = list_create(slurmdb_destroy_association_rec);
memset(&assoc_cond, 0, sizeof(slurmdb_association_cond_t));
assoc_cond.acct_list = list_create(NULL);
itr = list_iterator_create(name_list);
while((name = list_next(itr)))
list_append(assoc_cond.acct_list, name);
list_iterator_destroy(itr);
list_append(assoc_cond.acct_list, start_assoc->parent_acct);
assoc_cond.cluster_list = cluster_list;
local_assoc_list = acct_storage_g_get_associations(
db_conn, my_uid, &assoc_cond);
list_destroy(assoc_cond.acct_list);
if (!local_assoc_list) {
exit_code=1;
fprintf(stderr, " Problem getting associations from database. "
"Contact your admin.\n");
list_destroy(name_list);
list_destroy(cluster_list);
slurmdb_destroy_association_rec(start_assoc);
slurmdb_destroy_account_rec(start_acct);
list_destroy(local_account_list);
return SLURM_ERROR;
}
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;
}
acct = NULL;
if (!sacctmgr_find_account_from_list(local_account_list, name)) {
acct = xmalloc(sizeof(slurmdb_account_rec_t));
acct->assoc_list =
list_create(slurmdb_destroy_association_rec);
acct->name = xstrdup(name);
if (start_acct->description)
acct->description =
xstrdup(start_acct->description);
else
acct->description = xstrdup(name);
if (start_acct->organization)
acct->organization =
xstrdup(start_acct->organization);
else if (strcmp(start_assoc->parent_acct, "root"))
acct->organization =
xstrdup(start_assoc->parent_acct);
else
acct->organization = xstrdup(name);
xstrfmtcat(acct_str, " %s\n", name);
list_append(acct_list, acct);
}
itr_c = list_iterator_create(cluster_list);
while((cluster = list_next(itr_c))) {
if (sacctmgr_find_account_base_assoc_from_list(
local_assoc_list, name, cluster)) {
//printf(" already have this assoc\n");
continue;
}
if (!sacctmgr_find_account_base_assoc_from_list(
local_assoc_list, start_assoc->parent_acct,
cluster)) {
exit_code=1;
fprintf(stderr, " Parent account '%s' "
"doesn't exist on "
"cluster %s\n"
" Contact your admin "
"to add this account.\n",
start_assoc->parent_acct, cluster);
continue;
}
assoc = xmalloc(sizeof(slurmdb_association_rec_t));
slurmdb_init_association_rec(assoc, 0);
assoc->acct = xstrdup(name);
assoc->cluster = xstrdup(cluster);
assoc->def_qos_id = start_assoc->def_qos_id;
assoc->parent_acct = xstrdup(start_assoc->parent_acct);
assoc->shares_raw = start_assoc->shares_raw;
assoc->grp_cpu_mins = start_assoc->grp_cpu_mins;
assoc->grp_cpu_run_mins = start_assoc->grp_cpu_run_mins;
assoc->grp_cpus = start_assoc->grp_cpus;
assoc->grp_jobs = start_assoc->grp_jobs;
assoc->grp_mem = start_assoc->grp_mem;
assoc->grp_nodes = start_assoc->grp_nodes;
assoc->grp_submit_jobs = start_assoc->grp_submit_jobs;
assoc->grp_wall = start_assoc->grp_wall;
assoc->max_cpu_mins_pj = start_assoc->max_cpu_mins_pj;
assoc->max_cpus_pj = start_assoc->max_cpus_pj;
assoc->max_jobs = start_assoc->max_jobs;
assoc->max_nodes_pj = start_assoc->max_nodes_pj;
assoc->max_submit_jobs = start_assoc->max_submit_jobs;
assoc->max_wall_pj = start_assoc->max_wall_pj;
assoc->qos_list = copy_char_list(start_assoc->qos_list);
if (acct)
list_append(acct->assoc_list, assoc);
else
list_append(assoc_list, assoc);
xstrfmtcat(assoc_str,
" A = %-10.10s"
" C = %-10.10s\n",
assoc->acct,
assoc->cluster);
}
list_iterator_destroy(itr_c);
}
list_iterator_destroy(itr);
list_destroy(local_account_list);
list_destroy(local_assoc_list);
if (!list_count(acct_list) && !list_count(assoc_list)) {
printf(" Nothing new added.\n");
rc = SLURM_ERROR;
goto end_it;
} else if (!assoc_str) {
exit_code=1;
fprintf(stderr, " No associations created.\n");
goto end_it;
}
if (acct_str) {
printf(" Adding Account(s)\n%s", acct_str);
printf(" Settings\n");
if (start_acct->description)
printf(" Description = %s\n",
start_acct->description);
else
printf(" Description = %s\n", "Account Name");
if (start_acct->organization)
printf(" Organization = %s\n",
start_acct->organization);
else
printf(" Organization = %s\n",
"Parent/Account Name");
xfree(acct_str);
}
if (assoc_str) {
printf(" Associations\n%s", assoc_str);
xfree(assoc_str);
}
if (limit_set) {
printf(" Settings\n");
sacctmgr_print_assoc_limits(start_assoc);
}
notice_thread_init();
if (list_count(acct_list))
rc = acct_storage_g_add_accounts(db_conn, my_uid, acct_list);
if (rc == SLURM_SUCCESS) {
if (list_count(assoc_list))
rc = acct_storage_g_add_associations(db_conn, my_uid,
assoc_list);
} else {
exit_code=1;
fprintf(stderr, " Problem adding accounts: %s\n",
slurm_strerror(rc));
rc = SLURM_ERROR;
notice_thread_fini();
goto end_it;
}
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,
" error: Problem adding account associations: %s\n",
slurm_strerror(rc));
rc = SLURM_ERROR;
}
end_it:
list_destroy(name_list);
list_destroy(cluster_list);
list_destroy(acct_list);
list_destroy(assoc_list);
slurmdb_destroy_association_rec(start_assoc);
slurmdb_destroy_account_rec(start_acct);
return rc;
}
extern int sacctmgr_add_qos(int argc, char *argv[])
{
int rc = SLURM_SUCCESS;
int i=0, limit_set=0;
ListIterator itr = NULL;
slurmdb_qos_rec_t *qos = NULL;
slurmdb_qos_rec_t *start_qos = xmalloc(sizeof(slurmdb_qos_rec_t));
List name_list = list_create(slurm_destroy_char);
char *description = NULL;
char *name = NULL;
List qos_list = NULL;
char *qos_str = NULL;
slurmdb_init_qos_rec(start_qos, 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_qos);
}
if(exit_code) {
list_destroy(name_list);
xfree(description);
return SLURM_ERROR;
} else if(!list_count(name_list)) {
list_destroy(name_list);
slurmdb_destroy_qos_rec(start_qos);
exit_code=1;
fprintf(stderr, " Need name of qos to add.\n");
return SLURM_SUCCESS;
}
if(!g_qos_list) {
g_qos_list = acct_storage_g_get_qos(db_conn, my_uid, NULL);
if(!g_qos_list) {
exit_code=1;
fprintf(stderr, " Problem getting qos's "
"from database. "
"Contact your admin.\n");
list_destroy(name_list);
xfree(description);
return SLURM_ERROR;
}
}
qos_list = list_create(slurmdb_destroy_qos_rec);
itr = list_iterator_create(name_list);
while((name = list_next(itr))) {
qos = NULL;
if(!sacctmgr_find_qos_from_list(g_qos_list, name)) {
qos = xmalloc(sizeof(slurmdb_qos_rec_t));
slurmdb_init_qos_rec(qos, 0);
qos->name = xstrdup(name);
if(start_qos->description)
qos->description =
xstrdup(start_qos->description);
else
qos->description = xstrdup(name);
qos->grace_time = start_qos->grace_time;
qos->grp_cpu_mins = start_qos->grp_cpu_mins;
qos->grp_cpus = start_qos->grp_cpus;
qos->grp_jobs = start_qos->grp_jobs;
qos->grp_mem = start_qos->grp_mem;
qos->grp_nodes = start_qos->grp_nodes;
qos->grp_submit_jobs = start_qos->grp_submit_jobs;
qos->grp_wall = start_qos->grp_wall;
qos->max_cpu_mins_pj = start_qos->max_cpu_mins_pj;
qos->max_cpu_run_mins_pu =
start_qos->max_cpu_run_mins_pu;
qos->max_cpus_pj = start_qos->max_cpus_pj;
qos->max_cpus_pu = start_qos->max_cpus_pu;
qos->max_jobs_pu = start_qos->max_jobs_pu;
qos->max_nodes_pj = start_qos->max_nodes_pj;
qos->max_nodes_pu = start_qos->max_nodes_pu;
qos->max_submit_jobs_pu = start_qos->max_submit_jobs_pu;
qos->max_wall_pj = start_qos->max_wall_pj;
qos->preempt_list =
copy_char_list(start_qos->preempt_list);
qos->preempt_mode = start_qos->preempt_mode;
qos->priority = start_qos->priority;
qos->usage_factor = start_qos->usage_factor;
qos->usage_thres = start_qos->usage_thres;
xstrfmtcat(qos_str, " %s\n", name);
list_append(qos_list, qos);
}
}
list_iterator_destroy(itr);
list_destroy(name_list);
if (g_qos_list) {
list_destroy(g_qos_list);
g_qos_list = NULL;
}
if(!list_count(qos_list)) {
printf(" Nothing new added.\n");
goto end_it;
}
if(qos_str) {
printf(" Adding QOS(s)\n%s", qos_str);
printf(" Settings\n");
if(description)
printf(" Description = %s\n", description);
else
printf(" Description = %s\n", "QOS Name");
sacctmgr_print_qos_limits(start_qos);
xfree(qos_str);
}
notice_thread_init();
if(list_count(qos_list))
rc = acct_storage_g_add_qos(db_conn, my_uid, qos_list);
else
goto end_it;
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 QOS: %s\n",
slurm_strerror(rc));
rc = SLURM_ERROR;
}
end_it:
list_destroy(qos_list);
xfree(description);
return rc;
}