src/load.c

   1 /**
   2  * collectd - src/load.c
   3  * Copyright (C) 2005,2006  Florian octo Forster
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License as published by the
   7  * Free Software Foundation; only version 2 of the License is applicable.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  17  *
  18  * Authors:
  19  *   Florian octo Forster <octo at verplant.org>
  20  **/
  21
  22 #include "collectd.h"
  23 #include "common.h"
  24 #include "plugin.h"
  25
  26 #define MODULE_NAME "load"
  27
  28 #if defined(HAVE_GETLOADAVG) || defined(KERNEL_LINUX) || defined(HAVE_LIBSTATGRAB)
  29 # define LOAD_HAVE_READ 1
  30 #else
  31 # define LOAD_HAVE_READ 0
  32 #endif
  33
  34 #ifdef HAVE_SYS_LOADAVG_H
  35 #include <sys/loadavg.h>
  36 #endif
  37
  38 #ifdef HAVE_GETLOADAVG
  39 #if !defined(LOADAVG_1MIN) || !defined(LOADAVG_5MIN) || !defined(LOADAVG_15MIN)
  40 #define LOADAVG_1MIN  0
  41 #define LOADAVG_5MIN  1
  42 #define LOADAVG_15MIN 2
  43 #endif
  44 #endif /* defined(HAVE_GETLOADAVG) */
  45
  46 static data_source_t dsrc[3] =
  47 {
  48         {"shortterm", DS_TYPE_GAUGE, 0.0, 100.0},
  49         {"midterm",   DS_TYPE_GAUGE, 0.0, 100.0},
  50         {"longterm",  DS_TYPE_GAUGE, 0.0, 100.0}
  51 };
  52
  53 static data_set_t ds =
  54 {
  55         "load", 3, dsrc
  56 };
  57
  58 #if LOAD_HAVE_READ
  59 static void load_submit (gauge_t snum, gauge_t mnum, gauge_t lnum)
  60 {
  61         value_t values[3];
  62         value_list_t vl = VALUE_LIST_INIT;
  63
  64         values[0].gauge = snum;
  65         values[1].gauge = mnum;
  66         values[2].gauge = lnum;
  67
  68         vl.values = values;
  69         vl.values_len = STATIC_ARRAY_SIZE (values);
  70         vl.time = time (NULL);
  71         strcpy (vl.host, hostname_g);
  72         strcpy (vl.plugin, "load");
  73
  74         plugin_dispatch_values ("load", &vl);
  75 }
  76
  77 static int load_read (void)
  78 {
  79 #if defined(HAVE_GETLOADAVG)
  80         double load[3];
  81
  82         if (getloadavg (load, 3) == 3)
  83                 load_submit (load[LOADAVG_1MIN], load[LOADAVG_5MIN], load[LOADAVG_15MIN]);
  84         else
  85                 syslog (LOG_WARNING, "load: getloadavg failed: %s", strerror (errno));
  86 /* #endif HAVE_GETLOADAVG */
  87
  88 #elif defined(KERNEL_LINUX)
  89         gauge_t snum, mnum, lnum;
  90         FILE *loadavg;
  91         char buffer[16];
  92
  93         char *fields[8];
  94         int numfields;
  95
  96         if ((loadavg = fopen ("/proc/loadavg", "r")) == NULL)
  97         {
  98                 syslog (LOG_WARNING, "load: fopen: %s", strerror (errno));
  99                 return;
 100         }
 101
 102         if (fgets (buffer, 16, loadavg) == NULL)
 103         {
 104                 syslog (LOG_WARNING, "load: fgets: %s", strerror (errno));
 105                 fclose (loadavg);
 106                 return;
 107         }
 108
 109         if (fclose (loadavg))
 110                 syslog (LOG_WARNING, "load: fclose: %s", strerror (errno));
 111
 112         numfields = strsplit (buffer, fields, 8);
 113
 114         if (numfields < 3)
 115                 return;
 116
 117         snum = atof (fields[0]);
 118         mnum = atof (fields[1]);
 119         lnum = atof (fields[2]);
 120
 121         load_submit (snum, mnum, lnum);
 122 /* #endif KERNEL_LINUX */
 123
 124 #elif defined(HAVE_LIBSTATGRAB)
 125         gauge_t snum, mnum, lnum;
 126         sg_load_stats *ls;
 127
 128         if ((ls = sg_get_load_stats ()) == NULL)
 129                 return;
 130
 131         snum = ls->min1;
 132         mnum = ls->min5;
 133         lnum = ls->min15;
 134
 135         load_submit (snum, mnum, lnum);
 136 #endif /* HAVE_LIBSTATGRAB */
 137
 138         return (0);
 139 }
 140 #endif /* LOAD_HAVE_READ */
 141
 142 void module_register (void)
 143 {
 144         plugin_register_data_set (&ds);
 145 #if LOAD_HAVE_READ
 146         plugin_register_read ("load", load_read);
 147 #endif
 148 }
 149
 150 #undef MODULE_NAME