| /*****************************************************************************\ |
| * user_reports.c - functions for generating user reports |
| * from accounting infrastructure. |
| ***************************************************************************** |
| * |
| * Copyright (C) 2008 Lawrence Livermore National Security. |
| * Produced at Lawrence Livermore National Laboratory (cf, DISCLAIMER). |
| * Written by Danny Auble <da@llnl.gov> |
| * CODE-OCEC-09-009. All rights reserved. |
| * |
| * This file is part of SLURM, a resource management program. |
| * For details, see <http://slurm.schedmd.com/>. |
| * Please also read the included file: DISCLAIMER. |
| * |
| * SLURM is free software; you can redistribute it and/or modify it under |
| * the terms of the GNU General Public License as published by the Free |
| * Software Foundation; either version 2 of the License, or (at your option) |
| * any later version. |
| * |
| * In addition, as a special exception, the copyright holders give permission |
| * to link the code of portions of this program with the OpenSSL library under |
| * certain conditions as described in each individual source file, and |
| * distribute linked combinations including the two. You must obey the GNU |
| * General Public License in all respects for all of the code used other than |
| * OpenSSL. If you modify file(s) with this exception, you may extend this |
| * exception to your version of the file(s), but you are not obligated to do |
| * so. If you do not wish to do so, delete this exception statement from your |
| * version. If you delete this exception statement from all source files in |
| * the program, then also delete it here. |
| * |
| * SLURM is distributed in the hope that it will be useful, but WITHOUT ANY |
| * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS |
| * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more |
| * details. |
| * |
| * You should have received a copy of the GNU General Public License along |
| * with SLURM; if not, write to the Free Software Foundation, Inc., |
| * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
| \*****************************************************************************/ |
| |
| #include "user_reports.h" |
| |
| enum { |
| PRINT_USER_ACCT, |
| PRINT_USER_CLUSTER, |
| PRINT_USER_LOGIN, |
| PRINT_USER_PROPER, |
| PRINT_USER_USED, |
| PRINT_USER_ENERGY, |
| }; |
| |
| static List print_fields_list = NULL; /* types are of print_field_t */ |
| static bool group_accts = false; |
| static uint32_t top_limit = 10; |
| |
| static int _set_cond(int *start, int argc, char *argv[], |
| slurmdb_user_cond_t *user_cond, List format_list) |
| { |
| int i; |
| int set = 0; |
| int end = 0; |
| int local_cluster_flag = all_clusters_flag; |
| slurmdb_association_cond_t *assoc_cond = NULL; |
| time_t start_time, end_time; |
| int command_len = 0; |
| |
| if (!user_cond) { |
| error("We need an slurmdb_user_cond to call this"); |
| return SLURM_ERROR; |
| } |
| |
| user_cond->with_deleted = 1; |
| user_cond->with_assocs = 1; |
| if (!user_cond->assoc_cond) { |
| user_cond->assoc_cond = |
| xmalloc(sizeof(slurmdb_association_cond_t)); |
| user_cond->assoc_cond->with_usage = 1; |
| } |
| assoc_cond = user_cond->assoc_cond; |
| |
| if (!assoc_cond->cluster_list) |
| assoc_cond->cluster_list = list_create(slurm_destroy_char); |
| for (i=(*start); i<argc; i++) { |
| end = parse_option_end(argv[i]); |
| if (!end) |
| command_len=strlen(argv[i]); |
| else |
| command_len=end-1; |
| |
| if (!end && !strncasecmp(argv[i], "all_clusters", |
| MAX(command_len, 1))) { |
| local_cluster_flag = 1; |
| continue; |
| } else if (!end && !strncasecmp(argv[i], "group", |
| MAX(command_len, 1))) { |
| group_accts = 1; |
| } else if (!end |
| || !strncasecmp (argv[i], "Users", |
| MAX(command_len, 1))) { |
| if (!assoc_cond->user_list) |
| assoc_cond->user_list = |
| list_create(slurm_destroy_char); |
| slurm_addto_char_list(assoc_cond->user_list, |
| argv[i]+end); |
| set = 1; |
| } else if (!strncasecmp (argv[i], "Accounts", |
| MAX(command_len, 2)) |
| || !strncasecmp(argv[i], "Acct", |
| MAX(command_len, 4))) { |
| if (!assoc_cond->acct_list) |
| assoc_cond->acct_list = |
| list_create(slurm_destroy_char); |
| slurm_addto_char_list(assoc_cond->acct_list, |
| argv[i]+end); |
| set = 1; |
| } else if (!strncasecmp (argv[i], "Clusters", |
| MAX(command_len, 1))) { |
| slurm_addto_char_list(assoc_cond->cluster_list, |
| argv[i]+end); |
| set = 1; |
| } else if (!strncasecmp (argv[i], "End", MAX(command_len, 1))) { |
| assoc_cond->usage_end = parse_time(argv[i]+end, 1); |
| set = 1; |
| } else if (!strncasecmp (argv[i], "Format", |
| MAX(command_len, 1))) { |
| if (format_list) |
| slurm_addto_char_list(format_list, argv[i]+end); |
| } else if (!strncasecmp (argv[i], "Start", |
| MAX(command_len, 1))) { |
| assoc_cond->usage_start = parse_time(argv[i]+end, 1); |
| set = 1; |
| } else if (!strncasecmp (argv[i], "TopCount", |
| MAX(command_len, 1))) { |
| if (get_uint(argv[i]+end, &top_limit, "TopCount") |
| != SLURM_SUCCESS) |
| exit_code = 1; |
| } else { |
| exit_code=1; |
| fprintf(stderr, " Unknown condition: %s\n" |
| "Use keyword set to modify value\n", argv[i]); |
| } |
| } |
| (*start) = i; |
| |
| if (!local_cluster_flag && !list_count(assoc_cond->cluster_list)) { |
| char *temp = slurm_get_cluster_name(); |
| if (temp) |
| list_append(assoc_cond->cluster_list, temp); |
| } |
| |
| /* This needs to be done on some systems to make sure |
| cluster_cond isn't messed. This has happened on some 64 |
| bit machines and this is here to be on the safe side. |
| */ |
| start_time = assoc_cond->usage_start; |
| end_time = assoc_cond->usage_end; |
| slurmdb_report_set_start_end_time(&start_time, &end_time); |
| assoc_cond->usage_start = start_time; |
| assoc_cond->usage_end = end_time; |
| |
| return set; |
| } |
| |
| static int _setup_print_fields_list(List format_list) |
| { |
| ListIterator itr = NULL; |
| print_field_t *field = NULL; |
| char *object = NULL; |
| |
| if (!format_list || !list_count(format_list)) { |
| exit_code=1; |
| fprintf(stderr, |
| " We need a format list to set up the print.\n"); |
| return SLURM_ERROR; |
| } |
| |
| if (!print_fields_list) |
| print_fields_list = list_create(destroy_print_field); |
| |
| itr = list_iterator_create(format_list); |
| while((object = list_next(itr))) { |
| char *tmp_char = NULL; |
| int command_len = 0; |
| int newlen = 0; |
| |
| if ((tmp_char = strstr(object, "\%"))) { |
| newlen = atoi(tmp_char+1); |
| tmp_char[0] = '\0'; |
| } |
| |
| command_len = strlen(object); |
| |
| field = xmalloc(sizeof(print_field_t)); |
| if (!strncasecmp("Accounts", object, MAX(command_len, 1))) { |
| field->type = PRINT_USER_ACCT; |
| field->name = xstrdup("Account"); |
| field->len = 15; |
| field->print_routine = print_fields_str; |
| } else if (!strncasecmp("Cluster", object, |
| MAX(command_len, 1))) { |
| field->type = PRINT_USER_CLUSTER; |
| field->name = xstrdup("Cluster"); |
| field->len = 9; |
| field->print_routine = print_fields_str; |
| } else if (!strncasecmp("Login", object, MAX(command_len, 1))) { |
| field->type = PRINT_USER_LOGIN; |
| field->name = xstrdup("Login"); |
| field->len = 9; |
| field->print_routine = print_fields_str; |
| } else if (!strncasecmp("Proper", object, MAX(command_len, 1))) { |
| field->type = PRINT_USER_PROPER; |
| field->name = xstrdup("Proper Name"); |
| field->len = 15; |
| field->print_routine = print_fields_str; |
| } else if (!strncasecmp("Used", object, MAX(command_len, 1))) { |
| field->type = PRINT_USER_USED; |
| field->name = xstrdup("Used"); |
| if (time_format == SLURMDB_REPORT_TIME_SECS_PER |
| || time_format == SLURMDB_REPORT_TIME_MINS_PER |
| || time_format == SLURMDB_REPORT_TIME_HOURS_PER) |
| field->len = 18; |
| else |
| field->len = 10; |
| field->print_routine = slurmdb_report_print_time; |
| }else if (!strncasecmp("Energy", object, MAX(command_len, 1))) { |
| field->type = PRINT_USER_ENERGY; |
| field->name = xstrdup("Energy"); |
| field->len = 10; |
| field->print_routine = print_fields_int; |
| } else { |
| exit_code=1; |
| fprintf(stderr, " Unknown field '%s'\n", object); |
| xfree(field); |
| continue; |
| } |
| |
| if (newlen) |
| field->len = newlen; |
| |
| list_append(print_fields_list, field); |
| } |
| list_iterator_destroy(itr); |
| |
| return SLURM_SUCCESS; |
| } |
| |
| extern int user_top(int argc, char *argv[]) |
| { |
| int rc = SLURM_SUCCESS; |
| slurmdb_user_cond_t *user_cond = xmalloc(sizeof(slurmdb_user_cond_t)); |
| ListIterator itr = NULL; |
| ListIterator itr2 = NULL; |
| ListIterator itr3 = NULL; |
| ListIterator cluster_itr = NULL; |
| List format_list = list_create(slurm_destroy_char); |
| List slurmdb_report_cluster_list = NULL; |
| char *object = NULL; |
| |
| int i=0; |
| slurmdb_report_user_rec_t *slurmdb_report_user = NULL; |
| slurmdb_report_cluster_rec_t *slurmdb_report_cluster = NULL; |
| print_field_t *field = NULL; |
| int field_count = 0; |
| |
| print_fields_list = list_create(destroy_print_field); |
| |
| _set_cond(&i, argc, argv, user_cond, format_list); |
| |
| if (!list_count(format_list)) |
| slurm_addto_char_list(format_list, "Cl,L,P,A,U,Energy"); |
| |
| _setup_print_fields_list(format_list); |
| list_destroy(format_list); |
| |
| if (!(slurmdb_report_cluster_list = |
| slurmdb_report_user_top_usage(db_conn, user_cond, group_accts))) { |
| exit_code = 1; |
| goto end_it; |
| } |
| |
| if (print_fields_have_header) { |
| char start_char[20]; |
| char end_char[20]; |
| time_t my_start = user_cond->assoc_cond->usage_start; |
| time_t my_end = user_cond->assoc_cond->usage_end-1; |
| |
| slurm_make_time_str(&my_start, start_char, sizeof(start_char)); |
| slurm_make_time_str(&my_end, end_char, sizeof(end_char)); |
| printf("----------------------------------------" |
| "----------------------------------------\n"); |
| printf("Top %u Users %s - %s (%d secs)\n", |
| top_limit, start_char, end_char, |
| (int)(user_cond->assoc_cond->usage_end |
| - user_cond->assoc_cond->usage_start)); |
| |
| switch(time_format) { |
| case SLURMDB_REPORT_TIME_PERCENT: |
| printf("Time reported in %s\n", time_format_string); |
| break; |
| default: |
| printf("Time reported in CPU %s\n", time_format_string); |
| break; |
| } |
| printf("----------------------------------------" |
| "----------------------------------------\n"); |
| } |
| |
| itr2 = list_iterator_create(print_fields_list); |
| print_fields_header(print_fields_list); |
| |
| field_count = list_count(print_fields_list); |
| |
| cluster_itr = list_iterator_create(slurmdb_report_cluster_list); |
| while((slurmdb_report_cluster = list_next(cluster_itr))) { |
| int count = 0; |
| list_sort(slurmdb_report_cluster->user_list, |
| (ListCmpF)sort_user_dec); |
| |
| itr = list_iterator_create(slurmdb_report_cluster->user_list); |
| while((slurmdb_report_user = list_next(itr))) { |
| int curr_inx = 1; |
| while((field = list_next(itr2))) { |
| char *tmp_char = NULL; |
| struct passwd *pwd = NULL; |
| switch(field->type) { |
| case PRINT_USER_ACCT: |
| itr3 = list_iterator_create( |
| slurmdb_report_user->acct_list); |
| while((object = list_next(itr3))) { |
| if (tmp_char) |
| xstrfmtcat(tmp_char, |
| ", %s", |
| object); |
| else |
| xstrcat(tmp_char, |
| object); |
| } |
| list_iterator_destroy(itr3); |
| field->print_routine( |
| field, |
| tmp_char, |
| (curr_inx == field_count)); |
| xfree(tmp_char); |
| break; |
| case PRINT_USER_CLUSTER: |
| field->print_routine( |
| field, |
| slurmdb_report_cluster->name, |
| (curr_inx == field_count)); |
| break; |
| case PRINT_USER_LOGIN: |
| field->print_routine(field, |
| slurmdb_report_user->name, |
| (curr_inx == |
| field_count)); |
| break; |
| case PRINT_USER_PROPER: |
| pwd = getpwnam(slurmdb_report_user->name); |
| if (pwd) { |
| tmp_char = strtok(pwd->pw_gecos, |
| ","); |
| if (!tmp_char) |
| tmp_char = |
| pwd->pw_gecos; |
| } |
| field->print_routine(field, |
| tmp_char, |
| (curr_inx == |
| field_count)); |
| break; |
| case PRINT_USER_USED: |
| field->print_routine( |
| field, |
| slurmdb_report_user->cpu_secs, |
| slurmdb_report_cluster->cpu_secs, |
| (curr_inx == field_count)); |
| break; |
| case PRINT_USER_ENERGY: |
| field->print_routine( |
| field, |
| slurmdb_report_user->consumed_energy, |
| (curr_inx ==field_count)); |
| break; |
| default: |
| field->print_routine( |
| field, NULL, |
| (curr_inx == field_count)); |
| break; |
| } |
| curr_inx++; |
| } |
| list_iterator_reset(itr2); |
| printf("\n"); |
| count++; |
| if (count >= top_limit) |
| break; |
| } |
| list_iterator_destroy(itr); |
| } |
| list_iterator_destroy(cluster_itr); |
| end_it: |
| /* group_accts could be set in the set_cond function and needs |
| * to be cleared here, or anytime _set_cond is called. |
| */ |
| group_accts = 0; |
| slurmdb_destroy_user_cond(user_cond); |
| |
| if (slurmdb_report_cluster_list) { |
| list_destroy(slurmdb_report_cluster_list); |
| slurmdb_report_cluster_list = NULL; |
| } |
| |
| if (print_fields_list) { |
| list_destroy(print_fields_list); |
| print_fields_list = NULL; |
| } |
| |
| return rc; |
| } |
| |