1 /**
2 * collectd - src/utils_db_query.c
3 * Copyright (C) 2008,2009 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 "common.h"
24 #include "plugin.h"
25 #include "configfile.h"
26 #include "utils_db_query.h"
28 /*
29 * Data types
30 */
31 struct udb_result_s; /* {{{ */
32 typedef struct udb_result_s udb_result_t;
33 struct udb_result_s
34 {
35 char *type;
36 char *instance_prefix;
37 char **instances;
38 size_t instances_num;
39 char **values;
40 size_t values_num;
42 /* Preparation area */
43 const data_set_t *ds;
44 size_t *instances_pos;
45 size_t *values_pos;
46 char **instances_buffer;
47 char **values_buffer;
49 /* Legacy data */
50 int legacy_mode;
51 size_t legacy_position;
52 /* When in legacy mode:
53 * - type/ds hold the format of the data
54 * - instance_prefix is used as type-instance if non-NULL
55 * - legacy_position holds the index of the column to use as value.
56 */
58 udb_result_t *next;
59 }; /* }}} */
61 struct udb_query_s /* {{{ */
62 {
63 char *name;
64 char *statement;
65 void *user_data;
67 int legacy_mode;
69 unsigned int min_version;
70 unsigned int max_version;
72 /* Preparation area */
73 size_t column_num;
74 char *host;
75 char *plugin;
76 char *db_name;
78 udb_result_t *results;
79 }; /* }}} */
81 /*
82 * Config Private functions
83 */
84 static int udb_config_set_string (char **ret_string, /* {{{ */
85 oconfig_item_t *ci)
86 {
87 char *string;
89 if ((ci->values_num != 1)
90 || (ci->values[0].type != OCONFIG_TYPE_STRING))
91 {
92 WARNING ("db query utils: The `%s' config option "
93 "needs exactly one string argument.", ci->key);
94 return (-1);
95 }
97 string = strdup (ci->values[0].value.string);
98 if (string == NULL)
99 {
100 ERROR ("db query utils: strdup failed.");
101 return (-1);
102 }
104 if (*ret_string != NULL)
105 free (*ret_string);
106 *ret_string = string;
108 return (0);
109 } /* }}} int udb_config_set_string */
111 static int udb_config_add_string (char ***ret_array, /* {{{ */
112 size_t *ret_array_len, oconfig_item_t *ci)
113 {
114 char **array;
115 size_t array_len;
116 int i;
118 if (ci->values_num < 1)
119 {
120 WARNING ("db query utils: The `%s' config option "
121 "needs at least one argument.", ci->key);
122 return (-1);
123 }
125 for (i = 0; i < ci->values_num; i++)
126 {
127 if (ci->values[i].type != OCONFIG_TYPE_STRING)
128 {
129 WARNING ("db query utils: Argument %i to the `%s' option "
130 "is not a string.", i + 1, ci->key);
131 return (-1);
132 }
133 }
135 array_len = *ret_array_len;
136 array = (char **) realloc (*ret_array,
137 sizeof (char *) * (array_len + ci->values_num));
138 if (array == NULL)
139 {
140 ERROR ("db query utils: realloc failed.");
141 return (-1);
142 }
143 *ret_array = array;
145 for (i = 0; i < ci->values_num; i++)
146 {
147 array[array_len] = strdup (ci->values[i].value.string);
148 if (array[array_len] == NULL)
149 {
150 ERROR ("db query utils: strdup failed.");
151 *ret_array_len = array_len;
152 return (-1);
153 }
154 array_len++;
155 }
157 *ret_array_len = array_len;
158 return (0);
159 } /* }}} int udb_config_add_string */
161 static int udb_config_set_uint (unsigned int *ret_value, /* {{{ */
162 oconfig_item_t *ci)
163 {
164 double tmp;
166 if ((ci->values_num != 1)
167 || (ci->values[0].type != OCONFIG_TYPE_NUMBER))
168 {
169 WARNING ("db query utils: The `%s' config option "
170 "needs exactly one numeric argument.", ci->key);
171 return (-1);
172 }
174 tmp = ci->values[0].value.number;
175 if ((tmp < 0.0) || (tmp > ((double) UINT_MAX)))
176 return (-ERANGE);
178 *ret_value = (unsigned int) (tmp + .5);
179 return (0);
180 } /* }}} int udb_config_set_uint */
182 /*
183 * Legacy result private functions
184 */
185 static void udb_legacy_result_finish_result (udb_result_t *r) /* {{{ */
186 {
187 if (r == NULL)
188 return;
190 assert (r->legacy_mode == 1);
192 r->ds = NULL;
193 } /* }}} void udb_legacy_result_finish_result */
195 static int udb_legacy_result_handle_result (udb_result_t *r, /* {{{ */
196 udb_query_t *q, char **column_values)
197 {
198 value_list_t vl = VALUE_LIST_INIT;
199 value_t value;
200 char *endptr;
202 assert (r->legacy_mode == 1);
203 assert (r->ds != NULL);
204 assert (r->ds->ds_num == 1);
206 vl.values = &value;
207 vl.values_len = 1;
209 endptr = NULL;
210 errno = 0;
211 if (r->ds->ds[0].type == DS_TYPE_COUNTER)
212 vl.values[0].counter = (counter_t) strtoll (column_values[r->legacy_position],
213 &endptr, /* base = */ 0);
214 else if (r->ds->ds[0].type == DS_TYPE_GAUGE)
215 vl.values[0].gauge = (gauge_t) strtod (column_values[r->legacy_position],
216 &endptr);
217 else
218 errno = EINVAL;
220 if ((endptr == column_values[r->legacy_position]) || (errno != 0))
221 {
222 WARNING ("db query utils: udb_result_submit: Parsing `%s' as %s failed.",
223 column_values[r->legacy_position],
224 (r->ds->ds[0].type == DS_TYPE_COUNTER) ? "counter" : "gauge");
225 vl.values[0].gauge = NAN;
226 }
228 sstrncpy (vl.host, q->host, sizeof (vl.host));
229 sstrncpy (vl.plugin, q->plugin, sizeof (vl.plugin));
230 sstrncpy (vl.plugin_instance, q->db_name, sizeof (vl.type_instance));
231 sstrncpy (vl.type, r->type, sizeof (vl.type));
233 if (r->instance_prefix != NULL)
234 sstrncpy (vl.type_instance, r->instance_prefix,
235 sizeof (vl.type_instance));
237 plugin_dispatch_values (&vl);
239 return (0);
240 } /* }}} int udb_legacy_result_handle_result */
242 static int udb_legacy_result_prepare_result (udb_result_t *r, /* {{{ */
243 char **column_names, size_t column_num)
244 {
245 if (r == NULL)
246 return (-EINVAL);
248 assert (r->legacy_mode == 1);
250 /* Make sure previous preparations are cleaned up. */
251 udb_legacy_result_finish_result (r);
253 if (r->legacy_position >= column_num)
254 {
255 ERROR ("db query utils: The legacy configuration specified (at least) "
256 "%zu `Column's, but the query returned only %zu columns!",
257 r->legacy_position + 1, column_num);
258 return (-ENOENT);
259 }
261 /* Read `ds' and check number of values {{{ */
262 r->ds = plugin_get_ds (r->type);
263 if (r->ds == NULL)
264 {
265 ERROR ("db query utils: udb_result_prepare_result: Type `%s' is not "
266 "known by the daemon. See types.db(5) for details.",
267 r->type);
268 return (-1);
269 }
271 if (r->ds->ds_num != 1)
272 {
273 ERROR ("db query utils: udb_result_prepare_result: The type `%s' "
274 "requires exactly %i values, but the legacy configuration "
275 "requires exactly one!",
276 r->type,
277 r->ds->ds_num);
278 return (-1);
279 }
280 /* }}} */
282 return (0);
283 } /* }}} int udb_legacy_result_prepare_result */
285 static int udb_legacy_result_create (const char *query_name, /* {{{ */
286 udb_result_t **r_head, oconfig_item_t *ci, size_t position)
287 {
288 udb_result_t *r;
290 if ((ci->values_num < 1) || (ci->values_num > 2)
291 || (ci->values[0].type != OCONFIG_TYPE_STRING)
292 || ((ci->values_num == 2)
293 && (ci->values[1].type != OCONFIG_TYPE_STRING)))
294 {
295 WARNING ("db query utils: The `Column' block needs either one or two "
296 "string arguments.");
297 return (-1);
298 }
300 r = (udb_result_t *) malloc (sizeof (*r));
301 if (r == NULL)
302 {
303 ERROR ("db query utils: malloc failed.");
304 return (-1);
305 }
306 memset (r, 0, sizeof (*r));
308 r->legacy_mode = 1;
309 r->legacy_position = position;
311 r->type = strdup (ci->values[0].value.string);
312 if (r->type == NULL)
313 {
314 ERROR ("db query utils: strdup failed.");
315 free (r);
316 return (-1);
317 }
319 r->instance_prefix = NULL;
320 if (ci->values_num == 2)
321 {
322 r->instance_prefix = strdup (ci->values[1].value.string);
323 if (r->instance_prefix == NULL)
324 {
325 ERROR ("db query utils: strdup failed.");
326 free (r->type);
327 free (r);
328 return (-1);
329 }
330 }
332 /* If all went well, add this result to the list of results. */
333 if (*r_head == NULL)
334 {
335 *r_head = r;
336 }
337 else
338 {
339 udb_result_t *last;
341 last = *r_head;
342 while (last->next != NULL)
343 last = last->next;
345 last->next = r;
346 }
348 return (0);
349 } /* }}} int udb_legacy_result_create */
351 /*
352 * Result private functions
353 */
354 static void udb_result_submit (udb_result_t *r, udb_query_t *q) /* {{{ */
355 {
356 value_list_t vl = VALUE_LIST_INIT;
357 size_t i;
359 assert (r != NULL);
360 assert (r->legacy_mode == 0);
361 assert (r->ds != NULL);
362 assert (((size_t) r->ds->ds_num) == r->values_num);
364 DEBUG ("db query utils: udb_result_submit: r->instance_prefix = %s;",
365 (r->instance_prefix == NULL) ? "NULL" : r->instance_prefix);
366 for (i = 0; i < r->instances_num; i++)
367 {
368 DEBUG ("db query utils: udb_result_submit: r->instances_buffer[%zu] = %s;",
369 i, r->instances_buffer[i]);
370 }
372 vl.values = (value_t *) calloc (r->ds->ds_num, sizeof (value_t));
373 if (vl.values == NULL)
374 {
375 ERROR ("db query utils: malloc failed.");
376 return;
377 }
378 vl.values_len = r->ds->ds_num;
380 for (i = 0; i < r->values_num; i++)
381 {
382 char *endptr;
384 endptr = NULL;
385 errno = 0;
386 if (r->ds->ds[i].type == DS_TYPE_COUNTER)
387 vl.values[i].counter = (counter_t) strtoll (r->values_buffer[i],
388 &endptr, /* base = */ 0);
389 else if (r->ds->ds[i].type == DS_TYPE_GAUGE)
390 vl.values[i].gauge = (gauge_t) strtod (r->values_buffer[i], &endptr);
391 else
392 errno = EINVAL;
394 if ((endptr == r->values_buffer[i]) || (errno != 0))
395 {
396 WARNING ("db query utils: udb_result_submit: Parsing `%s' as %s failed.",
397 r->values_buffer[i],
398 (r->ds->ds[i].type == DS_TYPE_COUNTER) ? "counter" : "gauge");
399 vl.values[i].gauge = NAN;
400 }
401 }
403 sstrncpy (vl.host, q->host, sizeof (vl.host));
404 sstrncpy (vl.plugin, q->plugin, sizeof (vl.plugin));
405 sstrncpy (vl.plugin_instance, q->db_name, sizeof (vl.type_instance));
406 sstrncpy (vl.type, r->type, sizeof (vl.type));
408 if (r->instance_prefix == NULL)
409 {
410 strjoin (vl.type_instance, sizeof (vl.type_instance),
411 r->instances_buffer, r->instances_num, "-");
412 }
413 else
414 {
415 char tmp[DATA_MAX_NAME_LEN];
417 strjoin (tmp, sizeof (tmp), r->instances_buffer, r->instances_num, "-");
418 tmp[sizeof (tmp) - 1] = 0;
420 snprintf (vl.type_instance, sizeof (vl.type_instance), "%s-%s",
421 r->instance_prefix, tmp);
422 }
423 vl.type_instance[sizeof (vl.type_instance) - 1] = 0;
425 plugin_dispatch_values (&vl);
427 sfree (vl.values);
428 } /* }}} void udb_result_submit */
430 static void udb_result_finish_result (udb_result_t *r) /* {{{ */
431 {
432 if (r == NULL)
433 return;
435 if (r->legacy_mode == 1)
436 {
437 udb_legacy_result_finish_result (r);
438 return;
439 }
441 assert (r->legacy_mode == 0);
443 r->ds = NULL;
444 sfree (r->instances_pos);
445 sfree (r->values_pos);
446 sfree (r->instances_buffer);
447 sfree (r->values_buffer);
448 } /* }}} void udb_result_finish_result */
450 static int udb_result_handle_result (udb_result_t *r, /* {{{ */
451 udb_query_t *q, char **column_values)
452 {
453 size_t i;
455 if (r->legacy_mode == 1)
456 return (udb_legacy_result_handle_result (r, q, column_values));
458 assert (r->legacy_mode == 0);
460 for (i = 0; i < r->instances_num; i++)
461 r->instances_buffer[i] = column_values[r->instances_pos[i]];
463 for (i = 0; i < r->values_num; i++)
464 r->values_buffer[i] = column_values[r->values_pos[i]];
466 udb_result_submit (r, q);
468 return (0);
469 } /* }}} int udb_result_handle_result */
471 static int udb_result_prepare_result (udb_result_t *r, /* {{{ */
472 char **column_names, size_t column_num)
473 {
474 size_t i;
476 if (r == NULL)
477 return (-EINVAL);
479 if (r->legacy_mode == 1)
480 return (udb_legacy_result_prepare_result (r, column_names, column_num));
482 assert (r->legacy_mode == 0);
484 #define BAIL_OUT(status) \
485 r->ds = NULL; \
486 sfree (r->instances_pos); \
487 sfree (r->values_pos); \
488 sfree (r->instances_buffer); \
489 sfree (r->values_buffer); \
490 return (status)
492 /* Make sure previous preparations are cleaned up. */
493 udb_result_finish_result (r);
494 r->instances_pos = NULL;
495 r->values_pos = NULL;
497 /* Read `ds' and check number of values {{{ */
498 r->ds = plugin_get_ds (r->type);
499 if (r->ds == NULL)
500 {
501 ERROR ("db query utils: udb_result_prepare_result: Type `%s' is not "
502 "known by the daemon. See types.db(5) for details.",
503 r->type);
504 BAIL_OUT (-1);
505 }
507 if (((size_t) r->ds->ds_num) != r->values_num)
508 {
509 ERROR ("db query utils: udb_result_prepare_result: The type `%s' "
510 "requires exactly %i value%s, but the configuration specifies %zu.",
511 r->type,
512 r->ds->ds_num, (r->ds->ds_num == 1) ? "" : "s",
513 r->values_num);
514 BAIL_OUT (-1);
515 }
516 /* }}} */
518 /* Allocate r->instances_pos, r->values_pos, r->instances_buffer, and
519 * r->values_buffer {{{ */
520 r->instances_pos = (size_t *) calloc (r->instances_num, sizeof (size_t));
521 if (r->instances_pos == NULL)
522 {
523 ERROR ("db query utils: udb_result_prepare_result: malloc failed.");
524 BAIL_OUT (-ENOMEM);
525 }
527 r->values_pos = (size_t *) calloc (r->values_num, sizeof (size_t));
528 if (r->values_pos == NULL)
529 {
530 ERROR ("db query utils: udb_result_prepare_result: malloc failed.");
531 BAIL_OUT (-ENOMEM);
532 }
534 r->instances_buffer = (char **) calloc (r->instances_num, sizeof (char *));
535 if (r->instances_buffer == NULL)
536 {
537 ERROR ("db query utils: udb_result_prepare_result: malloc failed.");
538 BAIL_OUT (-ENOMEM);
539 }
541 r->values_buffer = (char **) calloc (r->values_num, sizeof (char *));
542 if (r->values_buffer == NULL)
543 {
544 ERROR ("db query utils: udb_result_prepare_result: malloc failed.");
545 BAIL_OUT (-ENOMEM);
546 }
547 /* }}} */
549 /* Determine the position of the instance columns {{{ */
550 for (i = 0; i < r->instances_num; i++)
551 {
552 size_t j;
554 for (j = 0; j < column_num; j++)
555 {
556 if (strcasecmp (r->instances[i], column_names[j]) == 0)
557 {
558 r->instances_pos[i] = j;
559 break;
560 }
561 }
563 if (j >= column_num)
564 {
565 ERROR ("db query utils: udb_result_prepare_result: "
566 "Column `%s' could not be found.",
567 r->instances[i]);
568 BAIL_OUT (-ENOENT);
569 }
570 } /* }}} for (i = 0; i < r->instances_num; i++) */
572 /* Determine the position of the value columns {{{ */
573 for (i = 0; i < r->values_num; i++)
574 {
575 size_t j;
577 for (j = 0; j < column_num; j++)
578 {
579 if (strcasecmp (r->values[i], column_names[j]) == 0)
580 {
581 r->values_pos[i] = j;
582 break;
583 }
584 }
586 if (j >= column_num)
587 {
588 ERROR ("db query utils: udb_result_prepare_result: "
589 "Column `%s' could not be found.",
590 r->values[i]);
591 BAIL_OUT (-ENOENT);
592 }
593 } /* }}} for (i = 0; i < r->values_num; i++) */
595 #undef BAIL_OUT
596 return (0);
597 } /* }}} int udb_result_prepare_result */
599 static void udb_result_free (udb_result_t *r) /* {{{ */
600 {
601 size_t i;
603 if (r == NULL)
604 return;
606 sfree (r->type);
608 for (i = 0; i < r->instances_num; i++)
609 sfree (r->instances[i]);
610 sfree (r->instances);
612 for (i = 0; i < r->values_num; i++)
613 sfree (r->values[i]);
614 sfree (r->values);
616 udb_result_free (r->next);
618 sfree (r);
619 } /* }}} void udb_result_free */
621 static int udb_result_create (const char *query_name, /* {{{ */
622 udb_result_t **r_head, oconfig_item_t *ci)
623 {
624 udb_result_t *r;
625 int status;
626 int i;
628 if (ci->values_num != 0)
629 {
630 WARNING ("db query utils: The `Result' block doesn't accept "
631 "any arguments. Ignoring %i argument%s.",
632 ci->values_num, (ci->values_num == 1) ? "" : "s");
633 }
635 r = (udb_result_t *) malloc (sizeof (*r));
636 if (r == NULL)
637 {
638 ERROR ("db query utils: malloc failed.");
639 return (-1);
640 }
641 memset (r, 0, sizeof (*r));
642 r->type = NULL;
643 r->instance_prefix = NULL;
644 r->instances = NULL;
645 r->values = NULL;
646 r->next = NULL;
648 /* Fill the `udb_result_t' structure.. */
649 status = 0;
650 for (i = 0; i < ci->children_num; i++)
651 {
652 oconfig_item_t *child = ci->children + i;
654 if (strcasecmp ("Type", child->key) == 0)
655 status = udb_config_set_string (&r->type, child);
656 else if (strcasecmp ("InstancePrefix", child->key) == 0)
657 status = udb_config_set_string (&r->instance_prefix, child);
658 else if (strcasecmp ("InstancesFrom", child->key) == 0)
659 status = udb_config_add_string (&r->instances, &r->instances_num, child);
660 else if (strcasecmp ("ValuesFrom", child->key) == 0)
661 status = udb_config_add_string (&r->values, &r->values_num, child);
662 else
663 {
664 WARNING ("db query utils: Query `%s': Option `%s' not allowed here.",
665 query_name, child->key);
666 status = -1;
667 }
669 if (status != 0)
670 break;
671 }
673 /* Check that all necessary options have been given. */
674 while (status == 0)
675 {
676 if (r->type == NULL)
677 {
678 WARNING ("db query utils: `Type' not given for "
679 "result in query `%s'", query_name);
680 status = -1;
681 }
682 if (r->instances == NULL)
683 {
684 WARNING ("db query utils: `InstancesFrom' not given for "
685 "result in query `%s'", query_name);
686 status = -1;
687 }
688 if (r->values == NULL)
689 {
690 WARNING ("db query utils: `ValuesFrom' not given for "
691 "result in query `%s'", query_name);
692 status = -1;
693 }
695 break;
696 } /* while (status == 0) */
698 if (status != 0)
699 {
700 udb_result_free (r);
701 return (-1);
702 }
704 /* If all went well, add this result to the list of results. */
705 if (*r_head == NULL)
706 {
707 *r_head = r;
708 }
709 else
710 {
711 udb_result_t *last;
713 last = *r_head;
714 while (last->next != NULL)
715 last = last->next;
717 last->next = r;
718 }
720 return (0);
721 } /* }}} int udb_result_create */
723 /*
724 * Query private functions
725 */
726 void udb_query_free_one (udb_query_t *q) /* {{{ */
727 {
728 if (q == NULL)
729 return;
731 sfree (q->name);
732 sfree (q->statement);
734 udb_result_free (q->results);
736 sfree (q);
737 } /* }}} void udb_query_free_one */
739 /*
740 * Query public functions
741 */
742 int udb_query_create (udb_query_t ***ret_query_list, /* {{{ */
743 size_t *ret_query_list_len, oconfig_item_t *ci,
744 udb_query_create_callback_t cb, int legacy_mode)
745 {
746 udb_query_t **query_list;
747 size_t query_list_len;
749 udb_query_t *q;
750 int status;
751 int i;
753 size_t legacy_position;
755 if ((ret_query_list == NULL) || (ret_query_list_len == NULL))
756 return (-EINVAL);
757 query_list = *ret_query_list;
758 query_list_len = *ret_query_list_len;
760 if ((ci->values_num != 1)
761 || (ci->values[0].type != OCONFIG_TYPE_STRING))
762 {
763 WARNING ("db query utils: The `Query' block "
764 "needs exactly one string argument.");
765 return (-1);
766 }
768 q = (udb_query_t *) malloc (sizeof (*q));
769 if (q == NULL)
770 {
771 ERROR ("db query utils: malloc failed.");
772 return (-1);
773 }
774 memset (q, 0, sizeof (*q));
775 q->legacy_mode = legacy_mode;
776 q->min_version = 0;
777 q->max_version = UINT_MAX;
779 legacy_position = 0;
781 status = udb_config_set_string (&q->name, ci);
782 if (status != 0)
783 {
784 sfree (q);
785 return (status);
786 }
788 /* Fill the `udb_query_t' structure.. */
789 for (i = 0; i < ci->children_num; i++)
790 {
791 oconfig_item_t *child = ci->children + i;
793 if (strcasecmp ("Statement", child->key) == 0)
794 status = udb_config_set_string (&q->statement, child);
795 else if (strcasecmp ("Result", child->key) == 0)
796 status = udb_result_create (q->name, &q->results, child);
797 else if (strcasecmp ("MinVersion", child->key) == 0)
798 status = udb_config_set_uint (&q->min_version, child);
799 else if (strcasecmp ("MaxVersion", child->key) == 0)
800 status = udb_config_set_uint (&q->max_version, child);
802 /* PostgreSQL compatibility code */
803 else if ((strcasecmp ("Query", child->key) == 0)
804 && (q->legacy_mode == 1))
805 {
806 WARNING ("db query utils: Query `%s': The `Query' option is "
807 "deprecated. Please use `Statement' instead.",
808 q->name);
809 status = udb_config_set_string (&q->statement, child);
810 }
811 else if ((strcasecmp ("Column", child->key) == 0)
812 && (q->legacy_mode == 1))
813 {
814 WARNING ("db query utils: Query `%s': The `Column' option is "
815 "deprecated. Please use the new syntax instead.",
816 q->name);
817 status = udb_legacy_result_create (q->name, &q->results, child,
818 legacy_position);
819 legacy_position++;
820 }
821 else if ((strcasecmp ("MinPGVersion", child->key) == 0)
822 && (q->legacy_mode == 1))
823 {
824 WARNING ("db query utils: Query `%s': The `MinPGVersion' option is "
825 "deprecated. Please use `MinVersion' instead.",
826 q->name);
827 status = udb_config_set_uint (&q->min_version, child);
828 }
829 else if ((strcasecmp ("MaxPGVersion", child->key) == 0)
830 && (q->legacy_mode == 1))
831 {
832 WARNING ("db query utils: Query `%s': The `MaxPGVersion' option is "
833 "deprecated. Please use `MaxVersion' instead.",
834 q->name);
835 status = udb_config_set_uint (&q->max_version, child);
836 }
838 /* Call custom callbacks */
839 else if (cb != NULL)
840 {
841 status = (*cb) (q, child);
842 if (status != 0)
843 {
844 WARNING ("db query utils: The configuration callback failed "
845 "to handle `%s'.", child->key);
846 }
847 }
848 else
849 {
850 WARNING ("db query utils: Query `%s': Option `%s' not allowed here.",
851 q->name, child->key);
852 status = -1;
853 }
855 if (status != 0)
856 break;
857 }
859 /* Check that all necessary options have been given. */
860 if (status == 0)
861 {
862 if (q->statement == NULL)
863 {
864 WARNING ("db query utils: Query `%s': No `Statement' given.", q->name);
865 status = -1;
866 }
867 if (q->results == NULL)
868 {
869 WARNING ("db query utils: Query `%s': No (valid) `Result' block given.",
870 q->name);
871 status = -1;
872 }
873 } /* if (status == 0) */
875 /* If all went well, add this query to the list of queries within the
876 * database structure. */
877 if (status == 0)
878 {
879 udb_query_t **temp;
881 temp = (udb_query_t **) realloc (query_list,
882 sizeof (*query_list) * (query_list_len + 1));
883 if (temp == NULL)
884 {
885 ERROR ("db query utils: realloc failed");
886 status = -1;
887 }
888 else
889 {
890 query_list = temp;
891 query_list[query_list_len] = q;
892 query_list_len++;
893 }
894 }
896 if (status != 0)
897 {
898 udb_query_free_one (q);
899 return (-1);
900 }
902 *ret_query_list = query_list;
903 *ret_query_list_len = query_list_len;
905 return (0);
906 } /* }}} int udb_query_create */
908 void udb_query_free (udb_query_t **query_list, size_t query_list_len) /* {{{ */
909 {
910 size_t i;
912 if (query_list == NULL)
913 return;
915 for (i = 0; i < query_list_len; i++)
916 udb_query_free_one (query_list[i]);
918 sfree (query_list);
919 } /* }}} void udb_query_free */
921 int udb_query_pick_from_list_by_name (const char *name, /* {{{ */
922 udb_query_t **src_list, size_t src_list_len,
923 udb_query_t ***dst_list, size_t *dst_list_len)
924 {
925 size_t i;
926 int num_added;
928 if ((name == NULL) || (src_list == NULL) || (dst_list == NULL)
929 || (dst_list_len == NULL))
930 {
931 ERROR ("db query utils: udb_query_pick_from_list_by_name: "
932 "Invalid argument.");
933 return (-EINVAL);
934 }
936 num_added = 0;
937 for (i = 0; i < src_list_len; i++)
938 {
939 udb_query_t **tmp_list;
940 size_t tmp_list_len;
942 if (strcasecmp (name, src_list[i]->name) != 0)
943 continue;
945 tmp_list_len = *dst_list_len;
946 tmp_list = (udb_query_t **) realloc (*dst_list, (tmp_list_len + 1)
947 * sizeof (udb_query_t *));
948 if (tmp_list == NULL)
949 {
950 ERROR ("db query utils: realloc failed.");
951 return (-ENOMEM);
952 }
954 tmp_list[tmp_list_len] = src_list[i];
955 tmp_list_len++;
957 *dst_list = tmp_list;
958 *dst_list_len = tmp_list_len;
960 num_added++;
961 } /* for (i = 0; i < src_list_len; i++) */
963 if (num_added <= 0)
964 {
965 ERROR ("db query utils: Cannot find query `%s'. Make sure the <Query> "
966 "block is above the database definition!",
967 name);
968 return (-ENOENT);
969 }
970 else
971 {
972 DEBUG ("db query utils: Added %i versions of query `%s'.",
973 num_added, name);
974 }
976 return (0);
977 } /* }}} int udb_query_pick_from_list_by_name */
979 int udb_query_pick_from_list (oconfig_item_t *ci, /* {{{ */
980 udb_query_t **src_list, size_t src_list_len,
981 udb_query_t ***dst_list, size_t *dst_list_len)
982 {
983 const char *name;
985 if ((ci == NULL) || (src_list == NULL) || (dst_list == NULL)
986 || (dst_list_len == NULL))
987 {
988 ERROR ("db query utils: udb_query_pick_from_list: "
989 "Invalid argument.");
990 return (-EINVAL);
991 }
993 if ((ci->values_num != 1)
994 || (ci->values[0].type != OCONFIG_TYPE_STRING))
995 {
996 ERROR ("db query utils: The `%s' config option "
997 "needs exactly one string argument.", ci->key);
998 return (-1);
999 }
1000 name = ci->values[0].value.string;
1002 return (udb_query_pick_from_list_by_name (name,
1003 src_list, src_list_len,
1004 dst_list, dst_list_len));
1005 } /* }}} int udb_query_pick_from_list */
1007 const char *udb_query_get_name (udb_query_t *q) /* {{{ */
1008 {
1009 if (q == NULL)
1010 return (NULL);
1012 return (q->name);
1013 } /* }}} const char *udb_query_get_name */
1015 const char *udb_query_get_statement (udb_query_t *q) /* {{{ */
1016 {
1017 if (q == NULL)
1018 return (NULL);
1020 return (q->statement);
1021 } /* }}} const char *udb_query_get_statement */
1023 void udb_query_set_user_data (udb_query_t *q, void *user_data) /* {{{ */
1024 {
1025 if (q == NULL)
1026 return;
1028 q->user_data = user_data;
1029 } /* }}} void udb_query_set_user_data */
1031 void *udb_query_get_user_data (udb_query_t *q) /* {{{ */
1032 {
1033 if (q == NULL)
1034 return (NULL);
1036 return (q->user_data);
1037 } /* }}} void *udb_query_get_user_data */
1039 int udb_query_check_version (udb_query_t *q, unsigned int version) /* {{{ */
1040 {
1041 if (q == NULL)
1042 return (-EINVAL);
1044 if ((version < q->min_version) || (version > q->max_version))
1045 return (0);
1047 return (1);
1048 } /* }}} int udb_query_check_version */
1050 void udb_query_finish_result (udb_query_t *q) /* {{{ */
1051 {
1052 udb_result_t *r;
1054 if (q == NULL)
1055 return;
1057 q->column_num = 0;
1058 sfree (q->host);
1059 sfree (q->plugin);
1060 sfree (q->db_name);
1062 for (r = q->results; r != NULL; r = r->next)
1063 udb_result_finish_result (r);
1064 } /* }}} void udb_query_finish_result */
1066 int udb_query_handle_result (udb_query_t *q, char **column_values) /* {{{ */
1067 {
1068 udb_result_t *r;
1069 int success;
1070 int status;
1072 if (q == NULL)
1073 return (-EINVAL);
1075 if ((q->column_num < 1) || (q->host == NULL) || (q->plugin == NULL)
1076 || (q->db_name == NULL))
1077 {
1078 ERROR ("db query utils: Query `%s': Query is not prepared; "
1079 "can't handle result.", q->name);
1080 return (-EINVAL);
1081 }
1083 #if defined(COLLECT_DEBUG) && COLLECT_DEBUG /* {{{ */
1084 do
1085 {
1086 size_t i;
1088 for (i = 0; i < q->column_num; i++)
1089 {
1090 DEBUG ("db query utils: udb_query_handle_result (%s, %s): "
1091 "column[%zu] = %s;",
1092 q->db_name, q->name, i, column_values[i]);
1093 }
1094 } while (0);
1095 #endif /* }}} */
1097 success = 0;
1098 for (r = q->results; r != NULL; r = r->next)
1099 {
1100 status = udb_result_handle_result (r, q, column_values);
1101 if (status == 0)
1102 success++;
1103 }
1105 if (success == 0)
1106 {
1107 ERROR ("db query utils: udb_query_handle_result (%s, %s): "
1108 "All results failed.", q->db_name, q->name);
1109 return (-1);
1110 }
1112 return (0);
1113 } /* }}} int udb_query_handle_result */
1115 int udb_query_prepare_result (udb_query_t *q, /* {{{ */
1116 const char *host, const char *plugin, const char *db_name,
1117 char **column_names, size_t column_num)
1118 {
1119 udb_result_t *r;
1120 int status;
1122 if (q == NULL)
1123 return (-EINVAL);
1125 udb_query_finish_result (q);
1127 q->column_num = column_num;
1128 q->host = strdup (host);
1129 q->plugin = strdup (plugin);
1130 q->db_name = strdup (db_name);
1132 if ((q->host == NULL) || (q->plugin == NULL) || (q->db_name == NULL))
1133 {
1134 ERROR ("db query utils: Query `%s': Prepare failed: Out of memory.", q->name);
1135 udb_query_finish_result (q);
1136 return (-ENOMEM);
1137 }
1139 #if defined(COLLECT_DEBUG) && COLLECT_DEBUG
1140 do
1141 {
1142 size_t i;
1144 for (i = 0; i < column_num; i++)
1145 {
1146 DEBUG ("db query utils: udb_query_prepare_result: "
1147 "query = %s; column[%zu] = %s;",
1148 q->name, i, column_names[i]);
1149 }
1150 } while (0);
1151 #endif
1153 for (r = q->results; r != NULL; r = r->next)
1154 {
1155 status = udb_result_prepare_result (r, column_names, column_num);
1156 if (status != 0)
1157 {
1158 udb_query_finish_result (q);
1159 return (status);
1160 }
1161 }
1163 return (0);
1164 } /* }}} int udb_query_prepare_result */
1166 /* vim: set sw=2 sts=2 et fdm=marker : */