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);
FREE_NULL_LIST(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,GrpTRES,GrpS,MaxJ,"
"MaxTRES,MaxS,MaxW,QOS,"
"DefaultQOS");
if (with_fed)
slurm_addto_char_list(format_list,
"Federation,ID,Weight,FedState");
}
cluster_cond->with_deleted = with_deleted;
print_fields_list = sacctmgr_process_format_list(format_list);
FREE_NULL_LIST(format_list);
if (exit_code) {
slurmdb_destroy_cluster_cond(cluster_cond);
FREE_NULL_LIST(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");
FREE_NULL_LIST(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;
/* 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_FEDERATION:
field->print_routine(field, cluster->fed.name,
(curr_inx == field_count));
break;
case PRINT_FEDSTATE:
{
char *tmp_str = slurmdb_cluster_fed_states_str(
cluster->fed.state);
field->print_routine(field, tmp_str,
(curr_inx == field_count));
break;
}
case PRINT_FEDSTATERAW:
field->print_routine(field, cluster->fed.state,
(curr_inx == field_count));
break;
case PRINT_ID:
field->print_routine(field, cluster->fed.id,
(curr_inx == field_count));
break;
case PRINT_WEIGHT:
field->print_routine(field, cluster->fed.weight,
(curr_inx == field_count));
break;
case PRINT_TRES:
sacctmgr_initialize_g_tres_list();
tmp_char = slurmdb_make_tres_string_from_simple(
cluster->tres_str, g_tres_list, NO_VAL,
CONVERT_NUM_UNIT_EXACT);
field->print_routine(field, tmp_char,
(curr_inx == field_count));
xfree(tmp_char);
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_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_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:
sacctmgr_print_assoc_rec(cluster->root_assoc,
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);
FREE_NULL_LIST(cluster_list);
FREE_NULL_LIST(print_fields_list);
return rc;
}
/* sacctmgr_list_reservation()
*/
int sacctmgr_list_reservation(int argc, char **argv)
{
List reservation_list;
ListIterator itr;
ListIterator itr2;
List format_list;
List print_fields_list;
slurmdb_reservation_cond_t *reservation_cond =
xmalloc(sizeof(slurmdb_reservation_cond_t));
slurmdb_reservation_rec_t *reservation;
int field_count, i;
print_field_t *field;
char *tmp_char;
/* If we don't have any arguments make sure we set up the
* time correctly for just the past day. */
if (argc == 0) {
struct tm start_tm;
reservation_cond->time_start = time(NULL);
if (!slurm_localtime_r(&reservation_cond->time_start,
&start_tm)) {
fprintf(stderr,
" Couldn't get localtime from %ld",
(long)reservation_cond->time_start);
slurmdb_destroy_reservation_cond(reservation_cond);
exit_code = 1;
return 0;
}
start_tm.tm_sec = 0;
start_tm.tm_min = 0;
start_tm.tm_hour = 0;
start_tm.tm_mday--;
reservation_cond->time_start = slurm_mktime(&start_tm);
}
format_list = list_create(slurm_destroy_char);
for (i=0; i<argc; i++) {
int command_len = strlen(argv[i]);
if (!xstrncasecmp(argv[i], "Where", MAX(command_len, 5))
|| !xstrncasecmp(argv[i], "Set", MAX(command_len, 3)))
i++;
_set_cond(&i, argc, argv, reservation_cond, format_list);
}
if (reservation_cond->nodes && !reservation_cond->cluster_list) {
char *cluster_name = slurm_get_cluster_name();
char *warning = xstrdup_printf(
"If requesting nodes you must also request the cluster.\nWould you like to use the local cluster of '%s'?",
cluster_name);
if (!commit_check(warning)) {
exit_code = 1;
} else {
reservation_cond->cluster_list =
list_create(slurm_destroy_char);
list_append(reservation_cond->cluster_list,
cluster_name);
cluster_name = NULL;
}
xfree(warning);
xfree(cluster_name);
}
if (exit_code) {
slurmdb_destroy_reservation_cond(reservation_cond);
FREE_NULL_LIST(format_list);
return SLURM_ERROR;
}
if (!list_count(format_list)) {
/* Append to the format list the fields
* we want to print, these are the data structure
* members of the type returned by slurmdbd
*/
slurm_addto_char_list(format_list,
"Cluster,Name%15,TRES%30,"
"TimeStart,TimeEnd,Unused");
}
reservation_list = slurmdb_reservations_get(
db_conn, reservation_cond);
slurmdb_destroy_reservation_cond(reservation_cond);
if (!reservation_list) {
exit_code=1;
fprintf(stderr, " Problem with query.\n");
FREE_NULL_LIST(format_list);
return SLURM_ERROR;
}
/* Process the format list creating a list of
* print field_t structures
*/
print_fields_list = sacctmgr_process_format_list(format_list);
FREE_NULL_LIST(format_list);
itr = list_iterator_create(reservation_list);
itr2 = list_iterator_create(print_fields_list);
print_fields_header(print_fields_list);
field_count = list_count(print_fields_list);
/* For each reservation prints the data structure members
*/
while ((reservation = list_next(itr))) {
while ((field = list_next(itr2))) {
switch (field->type) {
case PRINT_ASSOC_NAME:
field->print_routine(
field,
reservation->assocs,
field_count);
break;
case PRINT_CLUSTER:
field->print_routine(
field,
reservation->cluster,
field_count);
break;
case PRINT_FLAGS:
tmp_char = reservation_flags_string(
reservation->flags);
field->print_routine(
field,
tmp_char,
field_count);
xfree(tmp_char);
break;
case PRINT_ID:
field->print_routine(field,
reservation->id,
field_count);
break;
case PRINT_NAME:
field->print_routine(field,
reservation->name,
field_count);
break;
case PRINT_NODENAME:
field->print_routine(
field,
reservation->nodes,
field_count);
break;
case PRINT_NODEINX:
field->print_routine(
field,
reservation->node_inx,
field_count);
break;
case PRINT_TIMEEND:
field->print_routine(
field,
reservation->time_end,
field_count);
break;
case PRINT_TIMESTART:
field->print_routine(
field,
reservation->time_start,
field_count);
break;
case PRINT_TRES:
sacctmgr_initialize_g_tres_list();
tmp_char = slurmdb_make_tres_string_from_simple(
reservation->tres_str, g_tres_list,
NO_VAL, CONVERT_NUM_UNIT_EXACT,
0, NULL);
field->print_routine(field,
tmp_char,
field_count);
xfree(tmp_char);
break;
case PRINT_UNUSED:
field->print_routine(
field,
reservation->unused_wall,
field_count);
break;
}
}
list_iterator_reset(itr2);
printf("\n");
}
list_iterator_destroy(itr);
list_iterator_destroy(itr2);
FREE_NULL_LIST(reservation_list);
FREE_NULL_LIST(print_fields_list);
return 0;
}
