/*
* slurm_print_federation - prints slurmdb_federation_rec_t (passed as void*
* since slurm.h doesn't know about slurmdb.h).
*/
extern void slurm_print_federation(void *ptr)
{
ListIterator itr;
slurmdb_cluster_rec_t *cluster;
int left_col_size;
char *conn_status[] = {"Disconnected", "Connected"};
char *cluster_name = NULL;
slurmdb_federation_rec_t *fed = (slurmdb_federation_rec_t *)ptr;
xassert(fed);
if (!fed->name)
return;
cluster_name = slurm_get_cluster_name();
left_col_size = strlen("federation:");
printf("%-*s %s\n", left_col_size, "Federation:", fed->name);
list_sort(fed->cluster_list, (ListCmpF)_sort_clusters_by_name);
itr = list_iterator_create(fed->cluster_list);
/* Display local Cluster*/
while ((cluster = list_next(itr))) {
if (!xstrcmp(cluster->name, cluster_name)) {
char *tmp_str =
slurmdb_cluster_fed_states_str(
cluster->fed.state);
printf("%-*s %s:%s:%d ID:%d FedState:%s Weight:%d\n",
left_col_size, "Self:", cluster->name,
cluster->control_host, cluster->control_port,
cluster->fed.id, (tmp_str ? tmp_str : ""),
cluster->fed.weight);
}
}
/* Display siblings */
list_iterator_reset(itr);
while ((cluster = list_next(itr))) {
char *tmp_str = NULL;
int conn_type = 0;
if (!xstrcmp(cluster->name, cluster_name))
continue;
if (cluster->sockfd != -1)
conn_type = 1;
tmp_str = slurmdb_cluster_fed_states_str(cluster->fed.state);
printf("%-*s %s:%s:%d ID:%d FedState:%s Weight:%d "
"PersistConn:%s\n",
left_col_size, "Sibling:", cluster->name,
cluster->control_host, cluster->control_port,
cluster->fed.id, (tmp_str ? tmp_str : ""),
cluster->fed.weight, conn_status[conn_type]);
}
list_iterator_destroy(itr);
xfree(cluster_name);
}
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_federation(int argc, char **argv)
{
int rc = SLURM_SUCCESS;
slurmdb_federation_cond_t *federation_cond =
xmalloc(sizeof(slurmdb_federation_cond_t));
List federation_list;
int i=0;
ListIterator itr = NULL;
ListIterator itr2 = NULL;
slurmdb_federation_rec_t *fed = NULL;
bool print_clusters = false;
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_federation_cond(federation_cond, 0);
federation_cond->federation_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, federation_cond, format_list);
}
if (exit_code) {
slurmdb_destroy_federation_cond(federation_cond);
FREE_NULL_LIST(format_list);
return SLURM_ERROR;
}
if (!list_count(format_list)) {
slurm_addto_char_list(format_list,
"Federation,Cluster,ID%2,"
"Features,FedState");
}
print_fields_list = sacctmgr_process_format_list(format_list);
FREE_NULL_LIST(format_list);
if (exit_code) {
slurmdb_destroy_federation_cond(federation_cond);
FREE_NULL_LIST(print_fields_list);
return SLURM_ERROR;
}
federation_list = acct_storage_g_get_federations(db_conn, my_uid,
federation_cond);
slurmdb_destroy_federation_cond(federation_cond);
if (!federation_list) {
exit_code=1;
fprintf(stderr, " Problem with query.\n");
FREE_NULL_LIST(print_fields_list);
return SLURM_ERROR;
}
itr = list_iterator_create(federation_list);
itr2 = list_iterator_create(print_fields_list);
print_fields_header(print_fields_list);
field_count = list_count(print_fields_list);
/* only print clusters if a cluster field is requested */
while((field = list_next(itr2))) {
switch (field->type) {
case PRINT_CLUSTER:
case PRINT_FEDSTATE:
case PRINT_FEDSTATERAW:
case PRINT_ID:
print_clusters = true;
break;
}
}
list_iterator_reset(itr2);
while ((fed = list_next(itr))) {
int curr_inx = 1;
char *tmp_str = NULL;
uint32_t tmp_uint32 = 0;
slurmdb_cluster_rec_t *tmp_cluster = NULL;
ListIterator itr3 =
list_iterator_create(fed->cluster_list);
if (!tree_display && print_clusters)
tmp_cluster = list_next(itr3);
do {
while((field = list_next(itr2))) {
switch(field->type) {
/* Federation Specific Fields */
case PRINT_FEDERATION:
if (tree_display && tmp_cluster)
tmp_str = NULL;
else
tmp_str = fed->name;
field->print_routine(
field, tmp_str,
(curr_inx == field_count));
break;
case PRINT_FLAGS:
if (tree_display && tmp_cluster)
tmp_str = NULL;
else {
tmp_str =
slurmdb_federation_flags_str(
fed->flags);
}
field->print_routine(
field, tmp_str,
(curr_inx == field_count));
if (tmp_str)
xfree(tmp_str);
break;
/* Cluster Specific Fields */
case PRINT_CLUSTER:
if (!tmp_cluster)
tmp_str = NULL;
else
tmp_str = tmp_cluster->name;
field->print_routine(
field, tmp_str,
(curr_inx == field_count));
break;
case PRINT_FEATURES:
{
List tmp_list = NULL;
if (tmp_cluster)
tmp_list = tmp_cluster->
fed.feature_list;
field->print_routine(
field, tmp_list,
(curr_inx == field_count));
break;
}
case PRINT_FEDSTATE:
if (!tmp_cluster)
tmp_str = NULL;
else {
tmp_str =
slurmdb_cluster_fed_states_str(
tmp_cluster->fed.state);
}
field->print_routine(
field, tmp_str,
(curr_inx == field_count));
break;
case PRINT_FEDSTATERAW:
if (!tmp_cluster)
tmp_uint32 = NO_VAL;
else
tmp_uint32 =
tmp_cluster->fed.state;
field->print_routine(
field, tmp_uint32,
(curr_inx == field_count));
break;
case PRINT_ID:
if (!tmp_cluster)
tmp_uint32 = NO_VAL;
else
tmp_uint32 =
tmp_cluster->fed.id;
field->print_routine(
field, tmp_uint32,
(curr_inx == field_count));
break;
default:
field->print_routine(
field, NULL,
(curr_inx == field_count));
break;
}
curr_inx++;
}
list_iterator_reset(itr2);
printf("\n");
} while(print_clusters && (tmp_cluster = list_next(itr3)));
list_iterator_destroy(itr3);
}
list_iterator_destroy(itr2);
list_iterator_destroy(itr);
FREE_NULL_LIST(federation_list);
FREE_NULL_LIST(print_fields_list);
return rc;
}
/*
* slurm_print_federation - prints slurmdb_federation_rec_t (passed as void*
* since slurm.h doesn't know about slurmdb.h).
*/
extern void slurm_print_federation(void *ptr)
{
ListIterator itr;
slurmdb_cluster_rec_t *cluster;
int left_col_size;
char *cluster_name = NULL;
slurmdb_federation_rec_t *fed = (slurmdb_federation_rec_t *)ptr;
if (!fed || !fed->name)
return;
if (working_cluster_rec)
cluster_name = xstrdup(working_cluster_rec->name);
else
cluster_name = slurm_get_cluster_name();
left_col_size = strlen("federation:");
printf("%-*s %s\n", left_col_size, "Federation:",
fed->name);
list_sort(fed->cluster_list, (ListCmpF)_sort_clusters_by_name);
itr = list_iterator_create(fed->cluster_list);
/* Display local Cluster */
while ((cluster = list_next(itr))) {
char *features, *tmp_str;
if (xstrcmp(cluster->name, cluster_name))
continue;
features = slurm_char_list_to_xstr( cluster->fed.feature_list);
tmp_str = slurmdb_cluster_fed_states_str( cluster->fed.state);
printf("%-*s %s:%s:%d ID:%d FedState:%s Features:%s\n",
left_col_size, "Self:", cluster->name,
cluster->control_host ? cluster->control_host : "",
cluster->control_port,
cluster->fed.id, (tmp_str ? tmp_str : ""),
features ? features : "");
xfree(features);
break;
}
/* Display siblings */
list_iterator_reset(itr);
while ((cluster = list_next(itr))) {
char *tmp_str = NULL;
char *features = NULL;
if (!xstrcmp(cluster->name, cluster_name))
continue;
features = slurm_char_list_to_xstr(cluster->fed.feature_list);
tmp_str = slurmdb_cluster_fed_states_str(cluster->fed.state);
printf("%-*s %s:%s:%d ID:%d FedState:%s Features:%s PersistConnSend/Recv:%s/%s\n",
left_col_size, "Sibling:", cluster->name,
cluster->control_host ? cluster->control_host : "",
cluster->control_port,
cluster->fed.id, (tmp_str ? tmp_str : ""),
features ? features : "",
cluster->fed.send ? "Yes" : "No",
cluster->fed.recv ? "Yes" : "No");
xfree(features);
}
list_iterator_destroy(itr);
xfree(cluster_name);
}
