blob: 2ad5b4c508c602dc490aec2aef7838d50b1aed46 [file] [log] [blame]
/*****************************************************************************\
* cluster_report_functions.c - Interface to functions dealing with cluster
* reports.
******************************************************************************
* Copyright (C) 2010 Lawrence Livermore National Security.
* Produced at Lawrence Livermore National Laboratory (cf, DISCLAIMER).
* Written by Danny Auble da@llnl.gov, et. al.
* 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.
\*****************************************************************************/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#include "slurm/slurm.h"
#include "slurm/slurm_errno.h"
#include "src/common/slurmdb_defs.h"
#include "src/common/slurm_accounting_storage.h"
#include "src/common/xstring.h"
typedef enum {
CLUSTER_REPORT_UA,
CLUSTER_REPORT_AU,
CLUSTER_REPORT_UW,
CLUSTER_REPORT_WU
} cluster_report_t;
static void _process_ua(List user_list, slurmdb_association_rec_t *assoc)
{
ListIterator itr = NULL;
slurmdb_report_user_rec_t *slurmdb_report_user = NULL;
slurmdb_accounting_rec_t *accting = NULL;
/* make sure we add all associations to this
user rec because we could have some in
partitions which would create another
record otherwise
*/
itr = list_iterator_create(user_list);
while((slurmdb_report_user = list_next(itr))) {
if (!strcmp(slurmdb_report_user->name, assoc->user)
&& !strcmp(slurmdb_report_user->acct, assoc->acct))
break;
}
list_iterator_destroy(itr);
if (!slurmdb_report_user) {
struct passwd *passwd_ptr = NULL;
uid_t uid = NO_VAL;
passwd_ptr = getpwnam(assoc->user);
if (passwd_ptr)
uid = passwd_ptr->pw_uid;
/* In this report we are using the slurmdb_report user
structure to store the information we want
since it is already available and will do
pretty much what we want.
*/
slurmdb_report_user =
xmalloc(sizeof(slurmdb_report_user_rec_t));
slurmdb_report_user->name = xstrdup(assoc->user);
slurmdb_report_user->uid = uid;
slurmdb_report_user->acct = xstrdup(assoc->acct);
list_append(user_list, slurmdb_report_user);
}
/* get the amount of time this assoc used
during the time we are looking at */
itr = list_iterator_create(assoc->accounting_list);
while((accting = list_next(itr))) {
slurmdb_report_user->cpu_secs +=
(uint64_t)accting->alloc_secs;
slurmdb_report_user->consumed_energy +=
(uint64_t)accting->consumed_energy;
}
list_iterator_destroy(itr);
}
static void _process_au(List assoc_list, slurmdb_association_rec_t *assoc)
{
slurmdb_report_assoc_rec_t *slurmdb_report_assoc =
xmalloc(sizeof(slurmdb_report_assoc_rec_t));
ListIterator itr = NULL;
slurmdb_accounting_rec_t *accting = NULL;
list_append(assoc_list, slurmdb_report_assoc);
slurmdb_report_assoc->acct = xstrdup(assoc->acct);
slurmdb_report_assoc->cluster = xstrdup(assoc->cluster);
slurmdb_report_assoc->parent_acct = xstrdup(assoc->parent_acct);
slurmdb_report_assoc->user = xstrdup(assoc->user);
/* get the amount of time this assoc used
during the time we are looking at */
itr = list_iterator_create(assoc->accounting_list);
while((accting = list_next(itr))) {
slurmdb_report_assoc->cpu_secs +=
(uint64_t)accting->alloc_secs;
slurmdb_report_assoc->consumed_energy +=
(uint64_t)accting->consumed_energy;
}
list_iterator_destroy(itr);
}
static void _process_uw(List user_list, slurmdb_wckey_rec_t *wckey)
{
ListIterator itr = NULL;
slurmdb_report_user_rec_t *slurmdb_report_user = NULL;
slurmdb_accounting_rec_t *accting = NULL;
struct passwd *passwd_ptr = NULL;
uid_t uid = NO_VAL;
passwd_ptr = getpwnam(wckey->user);
if (passwd_ptr)
uid = passwd_ptr->pw_uid;
/* In this report we are using the slurmdb_report user
structure to store the information we want
since it is already available and will do
pretty much what we want.
*/
slurmdb_report_user =
xmalloc(sizeof(slurmdb_report_user_rec_t));
slurmdb_report_user->name = xstrdup(wckey->user);
slurmdb_report_user->uid = uid;
slurmdb_report_user->acct = xstrdup(wckey->name);
list_append(user_list, slurmdb_report_user);
/* get the amount of time this wckey used
during the time we are looking at */
itr = list_iterator_create(wckey->accounting_list);
while((accting = list_next(itr))) {
slurmdb_report_user->cpu_secs +=
(uint64_t)accting->alloc_secs;
slurmdb_report_user->consumed_energy +=
(uint64_t)accting->consumed_energy;
}
list_iterator_destroy(itr);
}
static void _process_wu(List assoc_list, slurmdb_wckey_rec_t *wckey)
{
slurmdb_report_assoc_rec_t *slurmdb_report_assoc = NULL,
*parent_assoc = NULL;
ListIterator itr = NULL;
slurmdb_accounting_rec_t *accting = NULL;
/* find the parent */
itr = list_iterator_create(assoc_list);
while((parent_assoc = list_next(itr))) {
if (!parent_assoc->user
&& !strcmp(parent_assoc->acct, wckey->name))
break;
}
list_iterator_destroy(itr);
if (!parent_assoc) {
parent_assoc = xmalloc(sizeof(slurmdb_report_assoc_rec_t));
list_append(assoc_list,
parent_assoc);
parent_assoc->acct = xstrdup(wckey->name);
}
/* now add one for the user */
slurmdb_report_assoc = xmalloc(sizeof(slurmdb_report_assoc_rec_t));
list_append(assoc_list, slurmdb_report_assoc);
slurmdb_report_assoc->acct = xstrdup(wckey->name);
slurmdb_report_assoc->user = xstrdup(wckey->user);
/* get the amount of time this wckey used
during the time we are looking at */
itr = list_iterator_create(wckey->accounting_list);
while((accting = list_next(itr))) {
slurmdb_report_assoc->cpu_secs +=
(uint64_t)accting->alloc_secs;
parent_assoc->cpu_secs +=
(uint64_t)accting->alloc_secs;
slurmdb_report_assoc->consumed_energy +=
(uint64_t)accting->consumed_energy;
parent_assoc->consumed_energy +=
(uint64_t)accting->consumed_energy;
}
list_iterator_destroy(itr);
}
static void _process_assoc_type(
ListIterator itr,
slurmdb_report_cluster_rec_t *slurmdb_report_cluster,
char *cluster_name,
cluster_report_t type)
{
slurmdb_association_rec_t *assoc = NULL;
/* now add the associations of interest here by user */
while((assoc = list_next(itr))) {
if (!assoc->accounting_list
|| !list_count(assoc->accounting_list)
|| ((type == CLUSTER_REPORT_UA) && !assoc->user)) {
list_delete_item(itr);
continue;
}
if (strcmp(cluster_name, assoc->cluster))
continue;
if (type == CLUSTER_REPORT_UA)
_process_ua(slurmdb_report_cluster->user_list,
assoc);
else if (type == CLUSTER_REPORT_AU)
_process_au(slurmdb_report_cluster->assoc_list,
assoc);
list_delete_item(itr);
}
}
static void _process_wckey_type(
ListIterator itr,
slurmdb_report_cluster_rec_t *slurmdb_report_cluster,
char *cluster_name,
cluster_report_t type)
{
slurmdb_wckey_rec_t *wckey = NULL;
/* now add the wckeyiations of interest here by user */
while((wckey = list_next(itr))) {
if (!wckey->accounting_list
|| !list_count(wckey->accounting_list)
|| ((type == CLUSTER_REPORT_UW) && !wckey->user)) {
list_delete_item(itr);
continue;
}
if (strcmp(cluster_name, wckey->cluster))
continue;
if (type == CLUSTER_REPORT_UW)
_process_uw(slurmdb_report_cluster->user_list,
wckey);
else if (type == CLUSTER_REPORT_WU)
_process_wu(slurmdb_report_cluster->assoc_list,
wckey);
list_delete_item(itr);
}
}
static List _process_util_by_report(void *db_conn, char *calling_name,
void *cond, cluster_report_t type)
{ ListIterator itr = NULL;
ListIterator type_itr = NULL;
slurmdb_cluster_cond_t cluster_cond;
List type_list = NULL;
List cluster_list = NULL;
List first_list = NULL;
slurmdb_cluster_rec_t *cluster = NULL;
slurmdb_report_cluster_rec_t *slurmdb_report_cluster = NULL;
time_t start_time, end_time;
int exit_code = 0;
uid_t my_uid = getuid();
List ret_list = list_create(slurmdb_destroy_report_cluster_rec);
slurmdb_init_cluster_cond(&cluster_cond, 0);
cluster_cond.with_deleted = 1;
cluster_cond.with_usage = 1;
if ((type == CLUSTER_REPORT_UA) || (type == CLUSTER_REPORT_AU)) {
start_time = ((slurmdb_association_cond_t *)cond)->usage_start;
end_time = ((slurmdb_association_cond_t *)cond)->usage_end;
cluster_cond.cluster_list =
((slurmdb_association_cond_t *)cond)->cluster_list;
} else if ((type == CLUSTER_REPORT_UW) || (type == CLUSTER_REPORT_WU)) {
start_time = ((slurmdb_wckey_cond_t *)cond)->usage_start;
end_time = ((slurmdb_wckey_cond_t *)cond)->usage_end;
cluster_cond.cluster_list =
((slurmdb_wckey_cond_t *)cond)->cluster_list;
} else {
error("unknown report type %d", type);
return NULL;
}
/* 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.
*/
slurmdb_report_set_start_end_time(&start_time, &end_time);
cluster_cond.usage_end = end_time;
cluster_cond.usage_start = start_time;
cluster_list = acct_storage_g_get_clusters(
db_conn, my_uid, &cluster_cond);
if (!cluster_list) {
exit_code=1;
fprintf(stderr, "%s: Problem with cluster query.\n",
calling_name);
goto end_it;
}
if ((type == CLUSTER_REPORT_UA) || (type == CLUSTER_REPORT_AU)) {
((slurmdb_association_cond_t *)cond)->usage_start = start_time;
((slurmdb_association_cond_t *)cond)->usage_end = end_time;
type_list = acct_storage_g_get_associations(
db_conn, my_uid, cond);
} else if ((type == CLUSTER_REPORT_UW) || (type == CLUSTER_REPORT_WU)) {
((slurmdb_wckey_cond_t *)cond)->usage_start = start_time;
((slurmdb_wckey_cond_t *)cond)->usage_end = end_time;
type_list = acct_storage_g_get_wckeys(
db_conn, my_uid, cond);
}
if (!type_list) {
exit_code=1;
fprintf(stderr, "%s: Problem with get query.\n", calling_name);
goto end_it;
}
if ((type == CLUSTER_REPORT_UA) || (type == CLUSTER_REPORT_AU)) {
first_list = type_list;
type_list = slurmdb_get_hierarchical_sorted_assoc_list(
first_list);
}
/* set up the structures for easy retrieval later */
itr = list_iterator_create(cluster_list);
type_itr = list_iterator_create(type_list);
while((cluster = list_next(itr))) {
/* check to see if this cluster is around during the
time we are looking at */
if (!cluster->accounting_list
|| !list_count(cluster->accounting_list))
continue;
slurmdb_report_cluster = slurmdb_cluster_rec_2_report(cluster);
list_append(ret_list, slurmdb_report_cluster);
if ((type == CLUSTER_REPORT_UA) || (type == CLUSTER_REPORT_UW))
slurmdb_report_cluster->user_list =
list_create(slurmdb_destroy_report_user_rec);
else if ((type == CLUSTER_REPORT_AU)
|| (type == CLUSTER_REPORT_WU))
slurmdb_report_cluster->assoc_list =
list_create(slurmdb_destroy_report_assoc_rec);
if ((type == CLUSTER_REPORT_UA) || (type == CLUSTER_REPORT_AU))
_process_assoc_type(type_itr, slurmdb_report_cluster,
cluster->name, type);
else if ((type == CLUSTER_REPORT_UW)
|| (type == CLUSTER_REPORT_WU))
_process_wckey_type(type_itr, slurmdb_report_cluster,
cluster->name, type);
list_iterator_reset(type_itr);
}
list_iterator_destroy(type_itr);
list_iterator_destroy(itr);
end_it:
if (type_list) {
list_destroy(type_list);
type_list = NULL;
}
if (first_list) {
list_destroy(first_list);
first_list = NULL;
}
if (cluster_list) {
list_destroy(cluster_list);
cluster_list = NULL;
}
if (exit_code) {
if (ret_list) {
list_destroy(ret_list);
ret_list = NULL;
}
}
return ret_list;
}
extern List slurmdb_report_cluster_account_by_user(void *db_conn,
slurmdb_association_cond_t *assoc_cond)
{
return _process_util_by_report(db_conn,
"slurmdb_report_cluster_account_by_user",
assoc_cond, CLUSTER_REPORT_AU);
}
extern List slurmdb_report_cluster_user_by_account(void *db_conn,
slurmdb_association_cond_t *assoc_cond)
{
return _process_util_by_report(db_conn,
"slurmdb_report_cluster_user_by_account",
assoc_cond, CLUSTER_REPORT_UA);
}
extern List slurmdb_report_cluster_wckey_by_user(void *db_conn,
slurmdb_wckey_cond_t *wckey_cond)
{
return _process_util_by_report(db_conn,
"slurmdb_report_cluster_wckey_by_user",
wckey_cond, CLUSTER_REPORT_WU);
}
extern List slurmdb_report_cluster_user_by_wckey(void *db_conn,
slurmdb_wckey_cond_t *wckey_cond)
{
return _process_util_by_report(db_conn,
"slurmdb_report_cluster_user_by_wckey",
wckey_cond, CLUSTER_REPORT_UW);
}