src/load.c

   1 /**
   2  * collectd - src/load.c
   3  * Copyright (C) 2005-2007  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 #ifdef HAVE_SYS_LOADAVG_H
  27 #include <sys/loadavg.h>
  28 #endif
  29
  30 #ifdef HAVE_GETLOADAVG
  31 #if !defined(LOADAVG_1MIN) || !defined(LOADAVG_5MIN) || !defined(LOADAVG_15MIN)
  32 #define LOADAVG_1MIN  0
  33 #define LOADAVG_5MIN  1
  34 #define LOADAVG_15MIN 2
  35 #endif
  36 #endif /* defined(HAVE_GETLOADAVG) */
  37
  38 static void load_submit (gauge_t snum, gauge_t mnum, gauge_t lnum)
  39 {
  40         value_t values[3];
  41         value_list_t vl = VALUE_LIST_INIT;
  42
  43         values[0].gauge = snum;
  44         values[1].gauge = mnum;
  45         values[2].gauge = lnum;
  46
  47         vl.values = values;
  48         vl.values_len = STATIC_ARRAY_SIZE (values);
  49         vl.time = time (NULL);
  50         strcpy (vl.host, hostname_g);
  51         strcpy (vl.plugin, "load");
  52         strcpy (vl.type, "load");
  53
  54         plugin_dispatch_values (&vl);
  55 }
  56
  57 static int load_read (void)
  58 {
  59 #if defined(HAVE_GETLOADAVG)
  60         double load[3];
  61
  62         if (getloadavg (load, 3) == 3)
  63                 load_submit (load[LOADAVG_1MIN], load[LOADAVG_5MIN], load[LOADAVG_15MIN]);
  64         else
  65         {
  66                 char errbuf[1024];
  67                 WARNING ("load: getloadavg failed: %s",
  68                                 sstrerror (errno, errbuf, sizeof (errbuf)));
  69         }
  70 /* #endif HAVE_GETLOADAVG */
  71
  72 #elif defined(KERNEL_LINUX)
  73         gauge_t snum, mnum, lnum;
  74         FILE *loadavg;
  75         char buffer[16];
  76
  77         char *fields[8];
  78         int numfields;
  79
  80         if ((loadavg = fopen ("/proc/loadavg", "r")) == NULL)
  81         {
  82                 char errbuf[1024];
  83                 WARNING ("load: fopen: %s",
  84                                 sstrerror (errno, errbuf, sizeof (errbuf)));
  85                 return (-1);
  86         }
  87
  88         if (fgets (buffer, 16, loadavg) == NULL)
  89         {
  90                 char errbuf[1024];
  91                 WARNING ("load: fgets: %s",
  92                                 sstrerror (errno, errbuf, sizeof (errbuf)));
  93                 fclose (loadavg);
  94                 return (-1);
  95         }
  96
  97         if (fclose (loadavg))
  98         {
  99                 char errbuf[1024];
 100                 WARNING ("load: fclose: %s",
 101                                 sstrerror (errno, errbuf, sizeof (errbuf)));
 102         }
 103
 104         numfields = strsplit (buffer, fields, 8);
 105
 106         if (numfields < 3)
 107                 return (-1);
 108
 109         snum = atof (fields[0]);
 110         mnum = atof (fields[1]);
 111         lnum = atof (fields[2]);
 112
 113         load_submit (snum, mnum, lnum);
 114 /* #endif KERNEL_LINUX */
 115
 116 #elif defined(HAVE_LIBSTATGRAB)
 117         gauge_t snum, mnum, lnum;
 118         sg_load_stats *ls;
 119
 120         if ((ls = sg_get_load_stats ()) == NULL)
 121                 return;
 122
 123         snum = ls->min1;
 124         mnum = ls->min5;
 125         lnum = ls->min15;
 126
 127         load_submit (snum, mnum, lnum);
 128 /* #endif HAVE_LIBSTATGRAB */
 129
 130 #else
 131 # error "No applicable input method."
 132 #endif
 133
 134         return (0);
 135 }
 136
 137 void module_register (void)
 138 {
 139         plugin_register_read ("load", load_read);
 140 } /* void module_register */