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