extern int sacctmgr_list_cluster(int argc, char *argv[])
{
int rc = SLURM_SUCCESS;
slurmdb_cluster_cond_t *cluster_cond =
xmalloc(sizeof(slurmdb_cluster_cond_t));
List cluster_list;
int i=0;
ListIterator itr = NULL;
ListIterator itr2 = NULL;
slurmdb_cluster_rec_t *cluster = NULL;
char *tmp_char = NULL;
int field_count = 0;
print_field_t *field = NULL;
List format_list = list_create(slurm_destroy_char);
List print_fields_list; /* types are of print_field_t */
slurmdb_init_cluster_cond(cluster_cond, 0);
cluster_cond->cluster_list = list_create(slurm_destroy_char);
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++;
_set_cond(&i, argc, argv, cluster_cond, format_list);
}
if(exit_code) {
slurmdb_destroy_cluster_cond(cluster_cond);
list_destroy(format_list);
return SLURM_ERROR;
}
if(!list_count(format_list)) {
slurm_addto_char_list(format_list,
"Cl,Controlh,Controlp,RPC");
if(!without_limits)
slurm_addto_char_list(format_list,
"Fa,GrpJ,GrpN,GrpS,MaxJ,MaxN,"
"MaxS,MaxW,QOS,DefaultQOS");
}
cluster_cond->with_deleted = with_deleted;
print_fields_list = sacctmgr_process_format_list(format_list);
list_destroy(format_list);
if(exit_code) {
slurmdb_destroy_cluster_cond(cluster_cond);
list_destroy(print_fields_list);
return SLURM_ERROR;
}
cluster_list = acct_storage_g_get_clusters(db_conn, my_uid,
cluster_cond);
slurmdb_destroy_cluster_cond(cluster_cond);
if(!cluster_list) {
exit_code=1;
fprintf(stderr, " Problem with query.\n");
list_destroy(print_fields_list);
return SLURM_ERROR;
}
itr = list_iterator_create(cluster_list);
itr2 = list_iterator_create(print_fields_list);
print_fields_header(print_fields_list);
field_count = list_count(print_fields_list);
while((cluster = list_next(itr))) {
int curr_inx = 1;
slurmdb_association_rec_t *assoc = cluster->root_assoc;
/* set up the working cluster rec so nodecnt's and node names
* are handled correctly */
working_cluster_rec = cluster;
while((field = list_next(itr2))) {
switch(field->type) {
case PRINT_CLUSTER:
field->print_routine(field,
cluster->name,
(curr_inx == field_count));
break;
case PRINT_CHOST:
field->print_routine(field,
cluster->control_host,
(curr_inx == field_count));
break;
case PRINT_CPORT:
field->print_routine(field,
cluster->control_port,
(curr_inx == field_count));
break;
case PRINT_CLASS:
field->print_routine(field,
get_classification_str(
cluster->
classification),
(curr_inx == field_count));
break;
case PRINT_CPUS:
{
char tmp_char[9];
convert_num_unit((float)cluster->cpu_count,
tmp_char, sizeof(tmp_char),
UNIT_NONE);
field->print_routine(field,
tmp_char,
(curr_inx == field_count));
break;
}
case PRINT_DQOS:
if(!g_qos_list) {
g_qos_list = acct_storage_g_get_qos(
db_conn,
my_uid,
NULL);
}
tmp_char = slurmdb_qos_str(g_qos_list,
assoc->def_qos_id);
field->print_routine(
field,
tmp_char,
(curr_inx == field_count));
break;
case PRINT_FAIRSHARE:
field->print_routine(
field,
assoc->shares_raw,
(curr_inx == field_count));
break;
case PRINT_FLAGS:
{
char *tmp_char = slurmdb_cluster_flags_2_str(
cluster->flags);
field->print_routine(
field,
tmp_char,
(curr_inx == field_count));
xfree(tmp_char);
break;
}
case PRINT_GRPC:
field->print_routine(field,
assoc->grp_cpus,
(curr_inx == field_count));
break;
case PRINT_GRPJ:
field->print_routine(field,
assoc->grp_jobs,
(curr_inx == field_count));
break;
case PRINT_GRPN:
field->print_routine(field,
assoc->grp_nodes,
(curr_inx == field_count));
break;
case PRINT_GRPS:
field->print_routine(field,
assoc->grp_submit_jobs,
(curr_inx == field_count));
break;
case PRINT_MAXCM:
field->print_routine(
field,
assoc->max_cpu_mins_pj,
(curr_inx == field_count));
break;
case PRINT_MAXC:
field->print_routine(field,
assoc->max_cpus_pj,
(curr_inx == field_count));
break;
case PRINT_MAXJ:
field->print_routine(field,
assoc->max_jobs,
(curr_inx == field_count));
break;
case PRINT_MAXN:
field->print_routine(field,
assoc->max_nodes_pj,
(curr_inx == field_count));
break;
case PRINT_MAXS:
field->print_routine(field,
assoc->max_submit_jobs,
(curr_inx == field_count));
break;
case PRINT_MAXW:
field->print_routine(
field,
assoc->max_wall_pj,
(curr_inx == field_count));
break;
case PRINT_NODECNT:
{
hostlist_t hl = hostlist_create(cluster->nodes);
int cnt = 0;
if(hl) {
cnt = hostlist_count(hl);
hostlist_destroy(hl);
}
field->print_routine(
field,
cnt,
(curr_inx == field_count));
break;
}
case PRINT_CLUSTER_NODES:
field->print_routine(
field,
cluster->nodes,
(curr_inx == field_count));
break;
case PRINT_QOS:
if(!g_qos_list)
g_qos_list = acct_storage_g_get_qos(
db_conn, my_uid, NULL);
field->print_routine(field,
g_qos_list,
assoc->qos_list,
(curr_inx == field_count));
break;
case PRINT_QOS_RAW:
field->print_routine(field,
assoc->qos_list,
(curr_inx == field_count));
break;
case PRINT_RPC_VERSION:
field->print_routine(
field,
cluster->rpc_version,
(curr_inx == field_count));
break;
case PRINT_SELECT:
field->print_routine(
field,
cluster->plugin_id_select,
(curr_inx == field_count));
break;
default:
field->print_routine(
field, NULL,
(curr_inx == field_count));
break;
}
curr_inx++;
}
list_iterator_reset(itr2);
printf("\n");
}
/* clear the working cluster rec */
working_cluster_rec = NULL;
list_iterator_destroy(itr2);
list_iterator_destroy(itr);
list_destroy(cluster_list);
list_destroy(print_fields_list);
return rc;
}
extern void sacctmgr_print_association_rec(slurmdb_association_rec_t *assoc,
print_field_t *field, List tree_list,
bool last)
{
char *print_acct = NULL;
char *tmp_char = NULL;
xassert(field);
if(!assoc) {
field->print_routine(field, NULL, last);
return;
}
switch(field->type) {
case PRINT_ACCT:
if (tree_display) {
char *local_acct = NULL;
char *parent_acct = NULL;
if (assoc->user) {
local_acct = xstrdup_printf("|%s", assoc->acct);
parent_acct = assoc->acct;
} else {
local_acct =
xstrdup(assoc->acct);
parent_acct =
assoc->parent_acct;
}
print_acct = slurmdb_tree_name_get(
local_acct, parent_acct, tree_list);
xfree(local_acct);
} else {
print_acct = assoc->acct;
}
field->print_routine(field, print_acct, last);
break;
case PRINT_CLUSTER:
field->print_routine(field, assoc->cluster, last);
break;
case PRINT_DQOS:
if (!g_qos_list)
g_qos_list = acct_storage_g_get_qos(
db_conn, my_uid, NULL);
tmp_char = slurmdb_qos_str(g_qos_list, assoc->def_qos_id);
field->print_routine(field, tmp_char, last);
break;
case PRINT_FAIRSHARE:
if (assoc->shares_raw == SLURMDB_FS_USE_PARENT)
print_fields_str(field, "parent", last);
else
field->print_routine(field, assoc->shares_raw, last);
break;
case PRINT_GRPCM:
field->print_routine(field, assoc->grp_cpu_mins, last);
break;
case PRINT_GRPCRM:
field->print_routine(field, assoc->grp_cpu_run_mins, last);
break;
case PRINT_GRPC:
field->print_routine(field, assoc->grp_cpus, last);
break;
case PRINT_GRPJ:
field->print_routine(field, assoc->grp_jobs, last);
break;
case PRINT_GRPN:
field->print_routine(field, assoc->grp_nodes, last);
break;
case PRINT_GRPS:
field->print_routine(field, assoc->grp_submit_jobs, last);
break;
case PRINT_GRPW:
field->print_routine(field, assoc->grp_wall, last);
break;
case PRINT_ID:
field->print_routine(field, assoc->id, last);
break;
case PRINT_LFT:
field->print_routine(field, assoc->lft, last);
break;
case PRINT_MAXCM:
field->print_routine(field, assoc->max_cpu_mins_pj, last);
break;
case PRINT_MAXCRM:
field->print_routine(field, assoc->max_cpu_run_mins, last);
break;
case PRINT_MAXC:
field->print_routine(field, assoc->max_cpus_pj, last);
break;
case PRINT_MAXJ:
field->print_routine(field, assoc->max_jobs, last);
break;
case PRINT_MAXN:
field->print_routine(field, assoc->max_nodes_pj, last);
break;
case PRINT_MAXS:
field->print_routine(field, assoc->max_submit_jobs, last);
break;
case PRINT_MAXW:
field->print_routine(field, assoc->max_wall_pj, last);
break;
case PRINT_PID:
field->print_routine(field, assoc->parent_id, last);
break;
case PRINT_PNAME:
field->print_routine(field, assoc->parent_acct, last);
break;
case PRINT_PART:
field->print_routine(field, assoc->partition, last);
break;
case PRINT_QOS:
if (!g_qos_list)
g_qos_list = acct_storage_g_get_qos(
db_conn, my_uid, NULL);
field->print_routine(field, g_qos_list, assoc->qos_list, last);
break;
case PRINT_QOS_RAW:
field->print_routine(field, assoc->qos_list, last);
break;
case PRINT_RGT:
field->print_routine(field, assoc->rgt, last);
break;
case PRINT_USER:
field->print_routine(field, assoc->user, last);
break;
default:
field->print_routine(field, NULL, last);
break;
}
}
static bool _isdefault(List qos_list)
{
int rc = 0;
slurmdb_association_cond_t assoc_cond;
slurmdb_association_rec_t *assoc = NULL;
ListIterator itr;
List ret_list = NULL;
char *name = NULL;
if(!qos_list || !list_count(qos_list))
return rc;
/* this needs to happen before any removing takes place so we
can figure out things correctly */
xassert(g_qos_list);
memset(&assoc_cond, 0, sizeof(slurmdb_association_cond_t));
assoc_cond.without_parent_info = 1;
assoc_cond.def_qos_id_list = list_create(slurm_destroy_char);
itr = list_iterator_create(qos_list);
while ((name = list_next(itr))) {
uint32_t id = str_2_slurmdb_qos(g_qos_list, name);
if(id == NO_VAL)
continue;
list_append(assoc_cond.def_qos_id_list,
xstrdup_printf("%u", id));
}
list_iterator_destroy(itr);
ret_list = acct_storage_g_get_associations(
db_conn, my_uid, &assoc_cond);
list_destroy(assoc_cond.def_qos_id_list);
if(!ret_list || !list_count(ret_list))
goto end_it;
fprintf(stderr," Associations listed below have these "
"as their Default QOS.\n");
itr = list_iterator_create(ret_list);
while((assoc = list_next(itr))) {
name = slurmdb_qos_str(g_qos_list, assoc->def_qos_id);
if (!assoc->user) {
// see if this isn't a user
fprintf(stderr,
" DefQOS = %-10s C = %-10s A = %-20s\n",
name, assoc->cluster, assoc->acct);
} else if (assoc->partition) {
// see if there is a partition name
fprintf(stderr,
" DefQOS = %-10s C = %-10s A = %-20s "
"U = %-9s P = %s\n",
name, assoc->cluster, assoc->acct,
assoc->user, assoc->partition);
} else {
fprintf(stderr,
" DefQOS = %-10s C = %-10s A = %-20s "
"U = %-9s\n",
name, assoc->cluster, assoc->acct, assoc->user);
}
}
list_iterator_destroy(itr);
rc = 1;
end_it:
if(ret_list)
list_destroy(ret_list);
return rc;
}
extern bool sacctmgr_check_default_qos(uint32_t qos_id,
slurmdb_association_cond_t *assoc_cond)
{
char *object = NULL;
ListIterator itr;
slurmdb_association_rec_t *assoc;
List no_access_list = NULL;
List assoc_list = NULL;
if (qos_id == NO_VAL)
return true;
assoc_list = acct_storage_g_get_associations(
db_conn, my_uid, assoc_cond);
if (!assoc_list) {
fprintf(stderr, "Couldn't get a list back for checking qos.\n");
return false;
}
if (!g_qos_list)
g_qos_list = acct_storage_g_get_qos(db_conn, my_uid, NULL);
itr = list_iterator_create(assoc_list);
while ((assoc = list_next(itr))) {
char *qos = NULL;
if (assoc->qos_list) {
int check_qos = qos_id;
if (check_qos == -1)
check_qos = assoc->def_qos_id;
if ((int)check_qos > 0) {
ListIterator qos_itr =
list_iterator_create(assoc->qos_list);
while ((qos = list_next(qos_itr))) {
if (qos[0] == '-')
continue;
else if (qos[0] == '+')
qos++;
/* info("looking for %u ?= %u", */
/* check_qos, slurm_atoul(qos)); */
if (check_qos == slurm_atoul(qos))
break;
}
list_iterator_destroy(qos_itr);
} else
qos = "";
}
if (!qos) {
char *name = slurmdb_qos_str(g_qos_list,
assoc->def_qos_id);
if (!assoc->user) {
// see if this isn't a user
object = xstrdup_printf(
" DefQOS = %-10s C = %-10s A = %-20s ",
name, assoc->cluster, assoc->acct);
} else if (assoc->partition) {
// see if there is a partition name
object = xstrdup_printf(
" DefQOS = %-10s C = %-10s A = %-20s "
"U = %-9s P = %s",
name, assoc->cluster, assoc->acct,
assoc->user, assoc->partition);
} else {
object = xstrdup_printf(
" DefQOS = %-10s C = %-10s A = %-20s "
"U = %-9s",
name, assoc->cluster,
assoc->acct, assoc->user);
}
if (!no_access_list)
no_access_list =
list_create(slurm_destroy_char);
list_append(no_access_list, object);
}
}
list_iterator_destroy(itr);
list_destroy(assoc_list);
if (!no_access_list)
return true;
fprintf(stderr,
" These associations don't have access to their "
"default qos.\n");
fprintf(stderr,
" Please give them access before they the default "
"can be set to this.\n");
itr = list_iterator_create(no_access_list);
while ((object = list_next(itr)))
fprintf(stderr, "%s\n", object);
list_iterator_destroy(itr);
list_destroy(no_access_list);
return 0;
}
