fix build with librabbitmq 0.6.0
[collectd.git] / src / load.c
1 /**
2  * collectd - src/load.c
3  * Copyright (C) 2005-2008  Florian octo Forster
4  * Copyright (C) 2009       Manuel Sanmartin
5  *
6  * This program is free software; you can redistribute it and/or modify it
7  * under the terms of the GNU General Public License as published by the
8  * Free Software Foundation; only version 2 of the License is applicable.
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  *   Manuel Sanmartin
22  **/
23
24 #define _BSD_SOURCE
25
26 #include "collectd.h"
27 #include "common.h"
28 #include "plugin.h"
29
30 #ifdef HAVE_SYS_LOADAVG_H
31 #include <sys/loadavg.h>
32 #endif
33
34 #if HAVE_STATGRAB_H
35 # include <statgrab.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 #ifdef HAVE_PERFSTAT
47 # include <sys/proc.h> /* AIX 5 */
48 # include <sys/protosw.h>
49 # include <libperfstat.h>
50 #endif /* HAVE_PERFSTAT */
51
52 static void load_submit (gauge_t snum, gauge_t mnum, gauge_t lnum)
53 {
54         value_t values[3];
55         value_list_t vl = VALUE_LIST_INIT;
56
57         values[0].gauge = snum;
58         values[1].gauge = mnum;
59         values[2].gauge = lnum;
60
61         vl.values = values;
62         vl.values_len = STATIC_ARRAY_SIZE (values);
63         sstrncpy (vl.host, hostname_g, sizeof (vl.host));
64         sstrncpy (vl.plugin, "load", sizeof (vl.plugin));
65         sstrncpy (vl.type, "load", sizeof (vl.type));
66
67         plugin_dispatch_values (&vl);
68 }
69
70 static int load_read (void)
71 {
72 #if defined(HAVE_GETLOADAVG)
73         double load[3];
74
75         if (getloadavg (load, 3) == 3)
76                 load_submit (load[LOADAVG_1MIN], load[LOADAVG_5MIN], load[LOADAVG_15MIN]);
77         else
78         {
79                 char errbuf[1024];
80                 WARNING ("load: getloadavg failed: %s",
81                                 sstrerror (errno, errbuf, sizeof (errbuf)));
82         }
83 /* #endif HAVE_GETLOADAVG */
84
85 #elif defined(KERNEL_LINUX)
86         gauge_t snum, mnum, lnum;
87         FILE *loadavg;
88         char buffer[16];
89
90         char *fields[8];
91         int numfields;
92
93         if ((loadavg = fopen ("/proc/loadavg", "r")) == NULL)
94         {
95                 char errbuf[1024];
96                 WARNING ("load: fopen: %s",
97                                 sstrerror (errno, errbuf, sizeof (errbuf)));
98                 return (-1);
99         }
100
101         if (fgets (buffer, 16, loadavg) == NULL)
102         {
103                 char errbuf[1024];
104                 WARNING ("load: fgets: %s",
105                                 sstrerror (errno, errbuf, sizeof (errbuf)));
106                 fclose (loadavg);
107                 return (-1);
108         }
109
110         if (fclose (loadavg))
111         {
112                 char errbuf[1024];
113                 WARNING ("load: fclose: %s",
114                                 sstrerror (errno, errbuf, sizeof (errbuf)));
115         }
116
117         numfields = strsplit (buffer, fields, 8);
118
119         if (numfields < 3)
120                 return (-1);
121
122         snum = atof (fields[0]);
123         mnum = atof (fields[1]);
124         lnum = atof (fields[2]);
125
126         load_submit (snum, mnum, lnum);
127 /* #endif KERNEL_LINUX */
128
129 #elif HAVE_LIBSTATGRAB
130         gauge_t snum, mnum, lnum;
131         sg_load_stats *ls;
132
133         if ((ls = sg_get_load_stats ()) == NULL)
134                 return;
135
136         snum = ls->min1;
137         mnum = ls->min5;
138         lnum = ls->min15;
139
140         load_submit (snum, mnum, lnum);
141 /* #endif HAVE_LIBSTATGRAB */
142
143 #elif HAVE_PERFSTAT
144         gauge_t snum, mnum, lnum;
145         perfstat_cpu_total_t cputotal;
146
147         if (perfstat_cpu_total(NULL,  &cputotal, sizeof(perfstat_cpu_total_t), 1) < 0)
148         {
149                 char errbuf[1024];
150                 WARNING ("load: perfstat_cpu : %s",
151                                 sstrerror (errno, errbuf, sizeof (errbuf)));
152                 return (-1);
153         }
154
155         snum = (float)cputotal.loadavg[0]/(float)(1<<SBITS);
156         mnum = (float)cputotal.loadavg[1]/(float)(1<<SBITS);
157         lnum = (float)cputotal.loadavg[2]/(float)(1<<SBITS);
158
159         load_submit (snum, mnum, lnum);
160 /* #endif HAVE_PERFSTAT */
161
162 #else
163 # error "No applicable input method."
164 #endif
165
166         return (0);
167 }
168
169 void module_register (void)
170 {
171         plugin_register_read ("load", load_read);
172 } /* void module_register */