src/graph.[ch]: Implement "graph_inst_search_field" and "graph_matches_field".
[collection4.git] / src / graph_def.c
1 #include <stdlib.h>
2 #include <string.h>
3 #include <errno.h>
4
5 #include "graph_def.h"
6 #include "graph.h"
7 #include "graph_config.h"
8 #include "graph_ident.h"
9 #include "common.h"
10 #include "oconfig.h"
11
12 #include <fcgiapp.h>
13 #include <fcgi_stdio.h>
14
15 /*
16  * Data structures
17  */
18 struct graph_def_s
19 {
20   graph_ident_t *select;
21
22   char *ds_name;
23   char *legend;
24   uint32_t color;
25   _Bool stack;
26   _Bool area;
27   char *format;
28
29   graph_def_t *next;
30 };
31
32 /*
33  * Private functions
34  */
35 #define DEF_CONFIG_FIELD(field) \
36 static int def_config_##field (const oconfig_item_t *ci, graph_ident_t *ident) \
37 {                                                                              \
38   char *tmp = NULL;                                                            \
39   int status = graph_config_get_string (ci, &tmp);                             \
40   if (status != 0)                                                             \
41     return (status);                                                           \
42   ident_set_##field (ident, tmp);                                              \
43   free (tmp);                                                                  \
44   return (0);                                                                  \
45 } /* }}} int def_config_field */
46
47 DEF_CONFIG_FIELD (host);
48 DEF_CONFIG_FIELD (plugin);
49 DEF_CONFIG_FIELD (plugin_instance);
50 DEF_CONFIG_FIELD (type);
51 DEF_CONFIG_FIELD (type_instance);
52
53 #undef DEF_CONFIG_FIELD
54
55 static int def_config_color (const oconfig_item_t *ci, uint32_t *ret_color) /* {{{ */
56 {
57   char *tmp;
58   char *endptr;
59   uint32_t color;
60
61   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
62     return (EINVAL);
63
64   tmp = ci->values[0].value.string;
65
66   endptr = NULL;
67   errno = 0;
68   color = (uint32_t) strtoul (tmp, &endptr, /* base = */ 16);
69   if ((errno != 0) || (endptr == tmp) || (color > 0x00ffffff))
70     return (EINVAL);
71
72   *ret_color = color;
73
74   return (0);
75 } /* }}} int def_config_color */
76
77 static graph_def_t *def_config_get_obj (graph_config_t *cfg, /* {{{ */
78     const oconfig_item_t *ci)
79 {
80   graph_ident_t *ident;
81   char *ds_name = NULL;
82   graph_def_t *def;
83   int i;
84
85   ident = graph_get_selector (cfg);
86   if (ident == NULL)
87   {
88     fprintf (stderr, "def_config_get_obj: graph_get_selector failed");
89     return (NULL);
90   }
91
92   for (i = 0; i < ci->children_num; i++)
93   {
94     oconfig_item_t *child;
95
96 #define HANDLE_FIELD(name,field) \
97     else if (strcasecmp (name, child->key) == 0) \
98       def_config_##field (child, ident)
99
100     child = ci->children + i;
101     if (strcasecmp ("DSName", child->key) == 0)
102       graph_config_get_string (child, &ds_name);
103
104     HANDLE_FIELD ("Host", host);
105     HANDLE_FIELD ("Plugin", plugin);
106     HANDLE_FIELD ("PluginInstance", plugin_instance);
107     HANDLE_FIELD ("Type", type);
108     HANDLE_FIELD ("TypeInstance", type_instance);
109
110 #undef HANDLE_FIELD
111   }
112
113   def = def_create (cfg, ident, ds_name);
114   if (def == NULL)
115   {
116     fprintf (stderr, "def_config_get_obj: def_create failed\n");
117     ident_destroy (ident);
118     free (ds_name);
119     return (NULL);
120   }
121
122   ident_destroy (ident);
123   free (ds_name);
124
125   return (def);
126 } /* }}} graph_def_t *def_config_get_obj */
127
128 /*
129  * Public functions
130  */
131 graph_def_t *def_create (graph_config_t *cfg, graph_ident_t *ident, /* {{{ */
132     const char *ds_name)
133 {
134   graph_ident_t *selector;
135   graph_def_t *ret;
136
137   if ((cfg == NULL) || (ident == NULL) || (ds_name == NULL))
138     return (NULL);
139
140   selector = graph_get_selector (cfg);
141   if (selector == NULL)
142     return (NULL);
143
144   ret = malloc (sizeof (*ret));
145   if (ret == NULL)
146   {
147     ident_destroy (selector);
148     return (NULL);
149   }
150   memset (ret, 0, sizeof (*ret));
151   ret->legend = NULL;
152   ret->format = NULL;
153
154   ret->ds_name = strdup (ds_name);
155   if (ret->ds_name == NULL)
156   {
157     ident_destroy (selector);
158     free (ret);
159     return (NULL);
160   }
161
162   ret->color = get_random_color ();
163   ret->next = NULL;
164
165   ret->select = ident_copy_with_selector (selector, ident,
166       IDENT_FLAG_REPLACE_ALL);
167   if (ret->select == NULL)
168   {
169     ident_destroy (selector);
170     free (ret->ds_name);
171     free (ret);
172     return (NULL);
173   }
174
175   ident_destroy (selector);
176   return (ret);
177 } /* }}} graph_def_t *def_create */
178
179 void def_destroy (graph_def_t *def) /* {{{ */
180 {
181   graph_def_t *next;
182
183   if (def == NULL)
184     return;
185
186   next = def->next;
187
188   ident_destroy (def->select);
189
190   free (def->ds_name);
191   free (def->legend);
192   free (def->format);
193
194   free (def);
195
196   def_destroy (next);
197 } /* }}} void def_destroy */
198
199 int def_config (graph_config_t *cfg, const oconfig_item_t *ci) /* {{{ */
200 {
201   graph_def_t *def;
202   int i;
203
204   def = def_config_get_obj (cfg, ci);
205   if (def == NULL)
206     return (EINVAL);
207
208   for (i = 0; i < ci->children_num; i++)
209   {
210     oconfig_item_t *child;
211
212     child = ci->children + i;
213     if (strcasecmp ("Legend", child->key) == 0)
214       graph_config_get_string (child, &def->legend);
215     else if (strcasecmp ("Color", child->key) == 0)
216       def_config_color (child, &def->color);
217     else if (strcasecmp ("Stack", child->key) == 0)
218       graph_config_get_bool (child, &def->stack);
219     else if (strcasecmp ("Area", child->key) == 0)
220       graph_config_get_bool (child, &def->area);
221     else if (strcasecmp ("Format", child->key) == 0)
222       graph_config_get_string (child, &def->format);
223   }
224
225   return (graph_add_def (cfg, def));
226 } /* }}} int def_config */
227
228 int def_append (graph_def_t *head, graph_def_t *def) /* {{{ */
229 {
230   graph_def_t *ptr;
231
232   if ((head == NULL) || (def == NULL))
233     return (EINVAL);
234
235   ptr = head;
236   while (ptr->next != NULL)
237     ptr = ptr->next;
238
239   ptr->next = def;
240
241   return (0);
242 } /* }}} int def_append */
243
244 graph_def_t *def_search (graph_def_t *head, graph_ident_t *ident, /* {{{ */
245     const char *ds_name)
246 {
247   graph_def_t *ptr;
248
249   if ((head == NULL) || (ident == NULL) || (ds_name == NULL))
250     return (NULL);
251
252   for (ptr = head; ptr != NULL; ptr = ptr->next)
253   {
254     if (!ident_matches (ptr->select, ident))
255       continue;
256
257     if (strcmp (ptr->ds_name, ds_name) == 0)
258       return (ptr);
259   }
260
261   return (NULL);
262 } /* }}} graph_def_t *def_search */
263
264 _Bool def_matches (graph_def_t *def, graph_ident_t *ident) /* {{{ */
265 {
266   return (ident_matches (def->select, ident));
267 } /* }}} _Bool def_matches */
268
269 int def_foreach (graph_def_t *def, def_callback_t callback, /* {{{ */
270     void *user_data)
271 {
272   graph_def_t *ptr;
273
274   if ((def == NULL) || (callback == NULL))
275     return (EINVAL);
276
277   for (ptr = def; ptr != NULL; ptr = ptr->next)
278   {
279     int status;
280
281     status = (*callback) (ptr, user_data);
282     if (status != 0)
283       return (status);
284   }
285
286   return (0);
287 } /* }}} int def_foreach */
288
289 int def_get_rrdargs (graph_def_t *def, graph_ident_t *ident, /* {{{ */
290     str_array_t *args)
291 {
292   char *file;
293   int index;
294
295   if ((def == NULL) || (ident == NULL) || (args == NULL))
296     return (EINVAL);
297
298   file = ident_to_file (ident);
299   if (file == NULL)
300   {
301     DEBUG ("gl_ident_get_rrdargs: ident_to_file returned NULL.\n");
302     return (-1);
303   }
304
305   DEBUG ("gl_ident_get_rrdargs: file = %s;\n", file);
306
307   index = array_argc (args);
308
309   /* CDEFs */
310   array_append_format (args, "DEF:def_%04i_min=%s:%s:MIN",
311       index, file, def->ds_name);
312   array_append_format (args, "DEF:def_%04i_avg=%s:%s:AVERAGE",
313       index, file, def->ds_name);
314   array_append_format (args, "DEF:def_%04i_max=%s:%s:MAX",
315       index, file, def->ds_name);
316   /* VDEFs */
317   array_append_format (args, "VDEF:vdef_%04i_min=def_%04i_min,MINIMUM",
318       index, index);
319   array_append_format (args, "VDEF:vdef_%04i_avg=def_%04i_avg,AVERAGE",
320       index, index);
321   array_append_format (args, "VDEF:vdef_%04i_max=def_%04i_max,MAXIMUM",
322       index, index);
323   array_append_format (args, "VDEF:vdef_%04i_lst=def_%04i_avg,LAST",
324       index, index);
325
326   /* Graph part */
327   array_append_format (args, "%s:def_%04i_avg#%06"PRIx32":%s%s",
328       def->area ? "AREA" : "LINE1",
329       index, def->color,
330       (def->legend != NULL) ? def->legend : def->ds_name,
331       def->stack ? ":STACK" : "");
332   array_append_format (args, "GPRINT:vdef_%04i_min:%s min,",
333       index, (def->format != NULL) ? def->format : "%6.2lf");
334   array_append_format (args, "GPRINT:vdef_%04i_avg:%s avg,",
335       index, (def->format != NULL) ? def->format : "%6.2lf");
336   array_append_format (args, "GPRINT:vdef_%04i_max:%s max,",
337       index, (def->format != NULL) ? def->format : "%6.2lf");
338   array_append_format (args, "GPRINT:vdef_%04i_lst:%s last\\l",
339       index, (def->format != NULL) ? def->format : "%6.2lf");
340
341   free (file);
342
343   return (0);
344 } /* }}} int def_get_rrdargs */
345
346 /* vim: set sw=2 sts=2 et fdm=marker : */