bdfe455233ec1e9d53afb6cf2e8c95adea0111da
1 /**
2 * collectd - src/rrdcached.c
3 * Copyright (C) 2008 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 **/
22 #include "collectd.h"
23 #include "plugin.h"
24 #include "common.h"
26 #include <rrd_client.h>
28 /*
29 * Private variables
30 */
31 static const char *config_keys[] =
32 {
33 "DaemonAddress",
34 "DataDir"
35 };
36 static int config_keys_num = STATIC_ARRAY_SIZE (config_keys);
38 static char *datadir = NULL;
39 static char *daemon_address = NULL;
41 static int value_list_to_string (char *buffer, int buffer_len,
42 const data_set_t *ds, const value_list_t *vl)
43 {
44 int offset;
45 int status;
46 int i;
48 assert (0 == strcmp (ds->type, vl->type));
50 memset (buffer, '\0', buffer_len);
52 status = ssnprintf (buffer, buffer_len, "%u", (unsigned int) vl->time);
53 if ((status < 1) || (status >= buffer_len))
54 return (-1);
55 offset = status;
57 for (i = 0; i < ds->ds_num; i++)
58 {
59 if ((ds->ds[i].type != DS_TYPE_COUNTER)
60 && (ds->ds[i].type != DS_TYPE_GAUGE))
61 return (-1);
63 if (ds->ds[i].type == DS_TYPE_COUNTER)
64 {
65 status = ssnprintf (buffer + offset, buffer_len - offset,
66 ",%llu", vl->values[i].counter);
67 }
68 else /* if (ds->ds[i].type == DS_TYPE_GAUGE) */
69 {
70 status = ssnprintf (buffer + offset, buffer_len - offset,
71 ",%lf", vl->values[i].gauge);
72 }
74 if ((status < 1) || (status >= (buffer_len - offset)))
75 return (-1);
77 offset += status;
78 } /* for ds->ds_num */
80 return (0);
81 } /* int value_list_to_string */
83 static int value_list_to_filename (char *buffer, int buffer_len,
84 const data_set_t *ds, const value_list_t *vl)
85 {
86 int offset = 0;
87 int status;
89 assert (0 == strcmp (ds->type, vl->type));
91 if (datadir != NULL)
92 {
93 status = ssnprintf (buffer + offset, buffer_len - offset,
94 "%s/", datadir);
95 if ((status < 1) || (status >= buffer_len - offset))
96 return (-1);
97 offset += status;
98 }
100 status = ssnprintf (buffer + offset, buffer_len - offset,
101 "%s/", vl->host);
102 if ((status < 1) || (status >= buffer_len - offset))
103 return (-1);
104 offset += status;
106 if (strlen (vl->plugin_instance) > 0)
107 status = ssnprintf (buffer + offset, buffer_len - offset,
108 "%s-%s/", vl->plugin, vl->plugin_instance);
109 else
110 status = ssnprintf (buffer + offset, buffer_len - offset,
111 "%s/", vl->plugin);
112 if ((status < 1) || (status >= buffer_len - offset))
113 return (-1);
114 offset += status;
116 if (strlen (vl->type_instance) > 0)
117 status = ssnprintf (buffer + offset, buffer_len - offset,
118 "%s-%s", vl->type, vl->type_instance);
119 else
120 status = ssnprintf (buffer + offset, buffer_len - offset,
121 "%s", vl->type);
122 if ((status < 1) || (status >= buffer_len - offset))
123 return (-1);
124 offset += status;
126 {
127 time_t now;
128 struct tm stm;
130 /* TODO: Find a way to minimize the calls to `localtime_r',
131 * since they are pretty expensive.. */
132 now = time (NULL);
133 if (localtime_r (&now, &stm) == NULL)
134 {
135 ERROR ("rrdcached plugin: localtime_r failed");
136 return (1);
137 }
139 strftime (buffer + offset, buffer_len - offset,
140 "-%Y-%m-%d", &stm);
141 }
143 return (0);
144 } /* int value_list_to_filename */
146 static int rc_config (const char *key, const char *value)
147 {
148 if (strcasecmp ("DataDir", key) == 0)
149 {
150 if (datadir != NULL)
151 free (datadir);
152 datadir = strdup (value);
153 if (datadir != NULL)
154 {
155 int len = strlen (datadir);
156 while ((len > 0) && (datadir[len - 1] == '/'))
157 {
158 len--;
159 datadir[len] = '\0';
160 }
161 if (len <= 0)
162 {
163 free (datadir);
164 datadir = NULL;
165 }
166 }
167 }
168 else if (strcasecmp ("DaemonAddress", key) == 0)
169 {
170 sfree (daemon_address);
171 daemon_address = strdup (value);
172 if (daemon_address == NULL)
173 {
174 ERROR ("rrdcached plugin: strdup failed.");
175 return (1);
176 }
177 }
178 else
179 {
180 return (-1);
181 }
182 return (0);
183 } /* int rc_config */
185 static int rc_write (const data_set_t *ds, const value_list_t *vl)
186 {
187 char filename[512];
188 char values[512];
189 char *values_array[2];
190 int status;
192 if (daemon_address == NULL)
193 {
194 ERROR ("rrdcached plugin: daemon_address == NULL.");
195 plugin_unregister_write ("rrdcached");
196 return (-1);
197 }
199 if (strcmp (ds->type, vl->type) != 0)
200 {
201 ERROR ("rrdcached plugin: DS type does not match value list type");
202 return (-1);
203 }
205 if (value_list_to_filename (filename, sizeof (filename), ds, vl) != 0)
206 {
207 ERROR ("rrdcached plugin: value_list_to_filename failed.");
208 return (-1);
209 }
211 if (value_list_to_string (values, sizeof (values), ds, vl) != 0)
212 {
213 ERROR ("rrdcached plugin: value_list_to_string failed.");
214 return (-1);
215 }
217 values_array[0] = values;
218 values_array[1] = NULL;
220 /* TODO: Check if the file exists. */
222 status = rrdc_connect (daemon_address);
223 if (status != 0)
224 {
225 ERROR ("rrdcached plugin: rrdc_connect (%s) failed with status %i.",
226 daemon_address, status);
227 return (-1);
228 }
230 status = rrdc_update (filename, /* values_num = */ 1, (void *) values_array);
231 if (status != 0)
232 {
233 ERROR ("rrdcached plugin: rrdc_update (%s, [%s], 1) failed with "
234 "status %i.",
235 filename, values_array[0], status);
236 return (-1);
237 }
239 return (0);
240 } /* int rc_write */
242 static int rc_shutdown (void)
243 {
244 rrdc_disconnect ();
245 return (0);
246 } /* int rc_shutdown */
248 void module_register (void)
249 {
250 plugin_register_config ("rrdcached", rc_config,
251 config_keys, config_keys_num);
252 plugin_register_write ("rrdcached", rc_write);
253 plugin_register_shutdown ("rrdcached", rc_shutdown);
254 } /* void module_register */
256 /*
257 * vim: set sw=2 sts=2 et :
258 */