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; }
extern int sacctmgr_list_account(int argc, char **argv) { int rc = SLURM_SUCCESS; slurmdb_account_cond_t *acct_cond = xmalloc(sizeof(slurmdb_account_cond_t)); List acct_list; int i=0, cond_set=0, prev_set=0; ListIterator itr = NULL; ListIterator itr2 = NULL; slurmdb_account_rec_t *acct = NULL; slurmdb_assoc_rec_t *assoc = 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 */ acct_cond->with_assocs = with_assoc_flag; 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++; prev_set = _set_cond(&i, argc, argv, acct_cond, format_list); cond_set |= prev_set; } if (exit_code) { slurmdb_destroy_account_cond(acct_cond); FREE_NULL_LIST(format_list); return SLURM_ERROR; } else if (!list_count(format_list)) { slurm_addto_char_list(format_list, "Acc,Des,O"); if (acct_cond->with_assocs) slurm_addto_char_list(format_list, "Cl,ParentN,U,Share,GrpJ,GrpN," "GrpCPUs,GrpMEM,GrpS,GrpWall,GrpCPUMins," "MaxJ,MaxN,MaxCPUs,MaxS,MaxW," "MaxCPUMins,QOS,DefaultQOS"); if (acct_cond->with_coords) slurm_addto_char_list(format_list, "Coord"); } if (!acct_cond->with_assocs && cond_set > 1) { if (!commit_check("You requested options that are only valid " "when querying with the withassoc option.\n" "Are you sure you want to continue?")) { printf("Aborted\n"); FREE_NULL_LIST(format_list); slurmdb_destroy_account_cond(acct_cond); return SLURM_SUCCESS; } } print_fields_list = sacctmgr_process_format_list(format_list); FREE_NULL_LIST(format_list); if (exit_code) { slurmdb_destroy_account_cond(acct_cond); FREE_NULL_LIST(print_fields_list); return SLURM_ERROR; } acct_list = slurmdb_accounts_get(db_conn, acct_cond); slurmdb_destroy_account_cond(acct_cond); if (!acct_list) { exit_code=1; fprintf(stderr, " Problem with query.\n"); FREE_NULL_LIST(print_fields_list); return SLURM_ERROR; } itr = list_iterator_create(acct_list); itr2 = list_iterator_create(print_fields_list); print_fields_header(print_fields_list); field_count = list_count(print_fields_list); while((acct = list_next(itr))) { if (acct->assoc_list) { ListIterator itr3 = list_iterator_create(acct->assoc_list); while((assoc = list_next(itr3))) { int curr_inx = 1; while((field = list_next(itr2))) { switch(field->type) { case PRINT_ACCT: field->print_routine( field, acct->name, (curr_inx == field_count)); break; case PRINT_COORDS: field->print_routine( field, acct->coordinators, (curr_inx == field_count)); break; case PRINT_DESC: field->print_routine( field, acct->description, (curr_inx == field_count)); break; case PRINT_ORG: field->print_routine( field, acct->organization, (curr_inx == field_count)); break; default: sacctmgr_print_assoc_rec( assoc, field, NULL, (curr_inx == field_count)); break; } curr_inx++; } list_iterator_reset(itr2); printf("\n"); } list_iterator_destroy(itr3); } else { int curr_inx = 1; while((field = list_next(itr2))) { switch(field->type) { case PRINT_QOS: field->print_routine( field, NULL, NULL, (curr_inx == field_count)); break; case PRINT_ACCT: field->print_routine( field, acct->name, (curr_inx == field_count)); break; case PRINT_COORDS: field->print_routine( field, acct->coordinators, (curr_inx == field_count)); break; case PRINT_DESC: field->print_routine( field, acct->description, (curr_inx == field_count)); break; case PRINT_ORG: field->print_routine( field, acct->organization, (curr_inx == field_count)); break; default: field->print_routine( field, NULL, (curr_inx == field_count)); break; } curr_inx++; } list_iterator_reset(itr2); printf("\n"); } } list_iterator_destroy(itr2); list_iterator_destroy(itr); FREE_NULL_LIST(acct_list); FREE_NULL_LIST(print_fields_list); return rc; }