2 * collectd - src/utils_db_query.c
3 * Copyright (C) 2008,2009 Florian octo Forster
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.
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.
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
19 * Florian octo Forster <octo at verplant.org>
25 #include "configfile.h"
26 #include "utils_db_query.h"
31 struct udb_result_s; /* {{{ */
32 typedef struct udb_result_s udb_result_t;
36 char *instance_prefix;
44 size_t legacy_position;
45 /* When in legacy mode:
46 * - type/ds hold the format of the data
47 * - instance_prefix is used as type-instance if non-NULL
48 * - legacy_position holds the index of the column to use as value.
54 struct udb_query_s /* {{{ */
62 unsigned int min_version;
63 unsigned int max_version;
65 udb_result_t *results;
68 struct udb_result_preparation_area_s /* {{{ */
71 size_t *instances_pos;
73 char **instances_buffer;
76 struct udb_result_preparation_area_s *next;
78 typedef struct udb_result_preparation_area_s udb_result_preparation_area_t;
80 struct udb_query_preparation_area_s /* {{{ */
89 udb_result_preparation_area_t *result_prep_areas;
93 * Config Private functions
95 static int udb_config_set_string (char **ret_string, /* {{{ */
100 if ((ci->values_num != 1)
101 || (ci->values[0].type != OCONFIG_TYPE_STRING))
103 WARNING ("db query utils: The `%s' config option "
104 "needs exactly one string argument.", ci->key);
108 string = strdup (ci->values[0].value.string);
111 ERROR ("db query utils: strdup failed.");
115 if (*ret_string != NULL)
117 *ret_string = string;
120 } /* }}} int udb_config_set_string */
122 static int udb_config_add_string (char ***ret_array, /* {{{ */
123 size_t *ret_array_len, oconfig_item_t *ci)
129 if (ci->values_num < 1)
131 WARNING ("db query utils: The `%s' config option "
132 "needs at least one argument.", ci->key);
136 for (i = 0; i < ci->values_num; i++)
138 if (ci->values[i].type != OCONFIG_TYPE_STRING)
140 WARNING ("db query utils: Argument %i to the `%s' option "
141 "is not a string.", i + 1, ci->key);
146 array_len = *ret_array_len;
147 array = (char **) realloc (*ret_array,
148 sizeof (char *) * (array_len + ci->values_num));
151 ERROR ("db query utils: realloc failed.");
156 for (i = 0; i < ci->values_num; i++)
158 array[array_len] = strdup (ci->values[i].value.string);
159 if (array[array_len] == NULL)
161 ERROR ("db query utils: strdup failed.");
162 *ret_array_len = array_len;
168 *ret_array_len = array_len;
170 } /* }}} int udb_config_add_string */
172 static int udb_config_set_uint (unsigned int *ret_value, /* {{{ */
177 if ((ci->values_num != 1)
178 || (ci->values[0].type != OCONFIG_TYPE_NUMBER))
180 WARNING ("db query utils: The `%s' config option "
181 "needs exactly one numeric argument.", ci->key);
185 tmp = ci->values[0].value.number;
186 if ((tmp < 0.0) || (tmp > ((double) UINT_MAX)))
189 *ret_value = (unsigned int) (tmp + .5);
191 } /* }}} int udb_config_set_uint */
194 * Legacy result private functions
196 static void udb_legacy_result_finish_result (const udb_result_t const *r, /* {{{ */
197 udb_result_preparation_area_t *prep_area)
199 if ((r == NULL) || (prep_area))
202 assert (r->legacy_mode == 1);
204 prep_area->ds = NULL;
205 } /* }}} void udb_legacy_result_finish_result */
207 static int udb_legacy_result_handle_result (udb_result_t *r, /* {{{ */
208 udb_query_preparation_area_t *q_area,
209 udb_result_preparation_area_t *r_area,
210 const udb_query_t const *q, char **column_values)
212 value_list_t vl = VALUE_LIST_INIT;
216 assert (r->legacy_mode == 1);
217 assert (r_area->ds != NULL);
218 assert (r_area->ds->ds_num == 1);
223 value_str = column_values[r->legacy_position];
224 if (0 != parse_value (value_str, &vl.values[0], r_area->ds->ds[0].type))
226 ERROR ("db query utils: udb_legacy_result_handle_result: "
227 "Parsing `%s' as %s failed.", value_str,
228 DS_TYPE_TO_STRING (r_area->ds->ds[0].type));
233 if (q_area->interval > 0)
234 vl.interval = q_area->interval;
236 sstrncpy (vl.host, q_area->host, sizeof (vl.host));
237 sstrncpy (vl.plugin, q_area->plugin, sizeof (vl.plugin));
238 sstrncpy (vl.plugin_instance, q_area->db_name, sizeof (vl.type_instance));
239 sstrncpy (vl.type, r->type, sizeof (vl.type));
241 if (r->instance_prefix != NULL)
242 sstrncpy (vl.type_instance, r->instance_prefix,
243 sizeof (vl.type_instance));
245 plugin_dispatch_values (&vl);
248 } /* }}} int udb_legacy_result_handle_result */
250 static int udb_legacy_result_prepare_result (const udb_result_t const *r, /* {{{ */
251 udb_result_preparation_area_t *prep_area,
252 char **column_names, size_t column_num)
257 assert (r->legacy_mode == 1);
259 /* Make sure previous preparations are cleaned up. */
260 udb_legacy_result_finish_result (r, prep_area);
262 if (r->legacy_position >= column_num)
264 ERROR ("db query utils: The legacy configuration specified (at least) "
265 "%zu `Column's, but the query returned only %zu columns!",
266 r->legacy_position + 1, column_num);
270 /* Read `ds' and check number of values {{{ */
271 prep_area->ds = plugin_get_ds (r->type);
272 if (prep_area->ds == NULL)
274 ERROR ("db query utils: udb_result_prepare_result: Type `%s' is not "
275 "known by the daemon. See types.db(5) for details.",
280 if (prep_area->ds->ds_num != 1)
282 ERROR ("db query utils: udb_result_prepare_result: The type `%s' "
283 "requires exactly %i values, but the legacy configuration "
284 "requires exactly one!",
286 prep_area->ds->ds_num);
292 } /* }}} int udb_legacy_result_prepare_result */
294 static int udb_legacy_result_create (const char *query_name, /* {{{ */
295 udb_result_t **r_head, oconfig_item_t *ci, size_t position)
299 if ((ci->values_num < 1) || (ci->values_num > 2)
300 || (ci->values[0].type != OCONFIG_TYPE_STRING)
301 || ((ci->values_num == 2)
302 && (ci->values[1].type != OCONFIG_TYPE_STRING)))
304 WARNING ("db query utils: The `Column' block needs either one or two "
305 "string arguments.");
309 r = (udb_result_t *) malloc (sizeof (*r));
312 ERROR ("db query utils: malloc failed.");
315 memset (r, 0, sizeof (*r));
318 r->legacy_position = position;
320 r->type = strdup (ci->values[0].value.string);
323 ERROR ("db query utils: strdup failed.");
328 r->instance_prefix = NULL;
329 if (ci->values_num == 2)
331 r->instance_prefix = strdup (ci->values[1].value.string);
332 if (r->instance_prefix == NULL)
334 ERROR ("db query utils: strdup failed.");
341 /* If all went well, add this result to the list of results. */
351 while (last->next != NULL)
358 } /* }}} int udb_legacy_result_create */
361 * Result private functions
363 static int udb_result_submit (udb_result_t *r, /* {{{ */
364 udb_result_preparation_area_t *r_area,
365 const udb_query_t const *q, udb_query_preparation_area_t *q_area)
367 value_list_t vl = VALUE_LIST_INIT;
371 assert (r->legacy_mode == 0);
372 assert (r_area->ds != NULL);
373 assert (((size_t) r_area->ds->ds_num) == r->values_num);
375 vl.values = (value_t *) calloc (r_area->ds->ds_num, sizeof (value_t));
376 if (vl.values == NULL)
378 ERROR ("db query utils: malloc failed.");
381 vl.values_len = r_area->ds->ds_num;
383 for (i = 0; i < r->values_num; i++)
385 char *value_str = r_area->values_buffer[i];
387 if (0 != parse_value (value_str, &vl.values[i], r_area->ds->ds[i].type))
389 ERROR ("db query utils: udb_result_submit: Parsing `%s' as %s failed.",
390 value_str, DS_TYPE_TO_STRING (r_area->ds->ds[i].type));
396 if (q_area->interval > 0)
397 vl.interval = q_area->interval;
399 sstrncpy (vl.host, q_area->host, sizeof (vl.host));
400 sstrncpy (vl.plugin, q_area->plugin, sizeof (vl.plugin));
401 sstrncpy (vl.plugin_instance, q_area->db_name, sizeof (vl.type_instance));
402 sstrncpy (vl.type, r->type, sizeof (vl.type));
404 /* Set vl.type_instance {{{ */
405 if (r->instances_num <= 0)
407 if (r->instance_prefix == NULL)
408 vl.type_instance[0] = 0;
410 sstrncpy (vl.type_instance, r->instance_prefix,
411 sizeof (vl.type_instance));
413 else /* if ((r->instances_num > 0) */
415 if (r->instance_prefix == NULL)
417 strjoin (vl.type_instance, sizeof (vl.type_instance),
418 r_area->instances_buffer, r->instances_num, "-");
422 char tmp[DATA_MAX_NAME_LEN];
424 strjoin (tmp, sizeof (tmp), r_area->instances_buffer,
425 r->instances_num, "-");
426 tmp[sizeof (tmp) - 1] = 0;
428 snprintf (vl.type_instance, sizeof (vl.type_instance), "%s-%s",
429 r->instance_prefix, tmp);
432 vl.type_instance[sizeof (vl.type_instance) - 1] = 0;
435 plugin_dispatch_values (&vl);
439 } /* }}} void udb_result_submit */
441 static void udb_result_finish_result (const udb_result_t const *r, /* {{{ */
442 udb_result_preparation_area_t *prep_area)
444 if ((r == NULL) || (prep_area == NULL))
447 if (r->legacy_mode == 1)
449 udb_legacy_result_finish_result (r, prep_area);
453 assert (r->legacy_mode == 0);
455 prep_area->ds = NULL;
456 sfree (prep_area->instances_pos);
457 sfree (prep_area->values_pos);
458 sfree (prep_area->instances_buffer);
459 sfree (prep_area->values_buffer);
460 } /* }}} void udb_result_finish_result */
462 static int udb_result_handle_result (udb_result_t *r, /* {{{ */
463 udb_query_preparation_area_t *q_area,
464 udb_result_preparation_area_t *r_area,
465 const udb_query_t const *q, char **column_values)
469 assert (r && q_area && r_area);
471 if (r->legacy_mode == 1)
472 return (udb_legacy_result_handle_result (r, q_area, r_area,
475 assert (r->legacy_mode == 0);
477 for (i = 0; i < r->instances_num; i++)
478 r_area->instances_buffer[i] = column_values[r_area->instances_pos[i]];
480 for (i = 0; i < r->values_num; i++)
481 r_area->values_buffer[i] = column_values[r_area->values_pos[i]];
483 return udb_result_submit (r, r_area, q, q_area);
484 } /* }}} int udb_result_handle_result */
486 static int udb_result_prepare_result (const udb_result_t const *r, /* {{{ */
487 udb_result_preparation_area_t *prep_area,
488 char **column_names, size_t column_num)
492 if ((r == NULL) || (prep_area == NULL))
495 if (r->legacy_mode == 1)
496 return (udb_legacy_result_prepare_result (r, prep_area,
497 column_names, column_num));
499 assert (r->legacy_mode == 0);
501 #define BAIL_OUT(status) \
502 prep_area->ds = NULL; \
503 sfree (prep_area->instances_pos); \
504 sfree (prep_area->values_pos); \
505 sfree (prep_area->instances_buffer); \
506 sfree (prep_area->values_buffer); \
509 /* Make sure previous preparations are cleaned up. */
510 udb_result_finish_result (r, prep_area);
511 prep_area->instances_pos = NULL;
512 prep_area->values_pos = NULL;
514 /* Read `ds' and check number of values {{{ */
515 prep_area->ds = plugin_get_ds (r->type);
516 if (prep_area->ds == NULL)
518 ERROR ("db query utils: udb_result_prepare_result: Type `%s' is not "
519 "known by the daemon. See types.db(5) for details.",
524 if (((size_t) prep_area->ds->ds_num) != r->values_num)
526 ERROR ("db query utils: udb_result_prepare_result: The type `%s' "
527 "requires exactly %i value%s, but the configuration specifies %zu.",
529 prep_area->ds->ds_num, (prep_area->ds->ds_num == 1) ? "" : "s",
535 /* Allocate r->instances_pos, r->values_pos, r->instances_buffer, and
536 * r->values_buffer {{{ */
537 if (r->instances_num > 0)
539 prep_area->instances_pos
540 = (size_t *) calloc (r->instances_num, sizeof (size_t));
541 if (prep_area->instances_pos == NULL)
543 ERROR ("db query utils: udb_result_prepare_result: malloc failed.");
547 prep_area->instances_buffer
548 = (char **) calloc (r->instances_num, sizeof (char *));
549 if (prep_area->instances_buffer == NULL)
551 ERROR ("db query utils: udb_result_prepare_result: malloc failed.");
554 } /* if (r->instances_num > 0) */
556 prep_area->values_pos
557 = (size_t *) calloc (r->values_num, sizeof (size_t));
558 if (prep_area->values_pos == NULL)
560 ERROR ("db query utils: udb_result_prepare_result: malloc failed.");
564 prep_area->values_buffer
565 = (char **) calloc (r->values_num, sizeof (char *));
566 if (prep_area->values_buffer == NULL)
568 ERROR ("db query utils: udb_result_prepare_result: malloc failed.");
573 /* Determine the position of the instance columns {{{ */
574 for (i = 0; i < r->instances_num; i++)
578 for (j = 0; j < column_num; j++)
580 if (strcasecmp (r->instances[i], column_names[j]) == 0)
582 prep_area->instances_pos[i] = j;
589 ERROR ("db query utils: udb_result_prepare_result: "
590 "Column `%s' could not be found.",
594 } /* }}} for (i = 0; i < r->instances_num; i++) */
596 /* Determine the position of the value columns {{{ */
597 for (i = 0; i < r->values_num; i++)
601 for (j = 0; j < column_num; j++)
603 if (strcasecmp (r->values[i], column_names[j]) == 0)
605 prep_area->values_pos[i] = j;
612 ERROR ("db query utils: udb_result_prepare_result: "
613 "Column `%s' could not be found.",
617 } /* }}} for (i = 0; i < r->values_num; i++) */
621 } /* }}} int udb_result_prepare_result */
623 static void udb_result_free (udb_result_t *r) /* {{{ */
632 for (i = 0; i < r->instances_num; i++)
633 sfree (r->instances[i]);
634 sfree (r->instances);
636 for (i = 0; i < r->values_num; i++)
637 sfree (r->values[i]);
640 udb_result_free (r->next);
643 } /* }}} void udb_result_free */
645 static int udb_result_create (const char *query_name, /* {{{ */
646 udb_result_t **r_head, oconfig_item_t *ci)
652 if (ci->values_num != 0)
654 WARNING ("db query utils: The `Result' block doesn't accept "
655 "any arguments. Ignoring %i argument%s.",
656 ci->values_num, (ci->values_num == 1) ? "" : "s");
659 r = (udb_result_t *) malloc (sizeof (*r));
662 ERROR ("db query utils: malloc failed.");
665 memset (r, 0, sizeof (*r));
667 r->instance_prefix = NULL;
672 /* Fill the `udb_result_t' structure.. */
674 for (i = 0; i < ci->children_num; i++)
676 oconfig_item_t *child = ci->children + i;
678 if (strcasecmp ("Type", child->key) == 0)
679 status = udb_config_set_string (&r->type, child);
680 else if (strcasecmp ("InstancePrefix", child->key) == 0)
681 status = udb_config_set_string (&r->instance_prefix, child);
682 else if (strcasecmp ("InstancesFrom", child->key) == 0)
683 status = udb_config_add_string (&r->instances, &r->instances_num, child);
684 else if (strcasecmp ("ValuesFrom", child->key) == 0)
685 status = udb_config_add_string (&r->values, &r->values_num, child);
688 WARNING ("db query utils: Query `%s': Option `%s' not allowed here.",
689 query_name, child->key);
697 /* Check that all necessary options have been given. */
702 WARNING ("db query utils: `Type' not given for "
703 "result in query `%s'", query_name);
706 if (r->values == NULL)
708 WARNING ("db query utils: `ValuesFrom' not given for "
709 "result in query `%s'", query_name);
714 } /* while (status == 0) */
722 /* If all went well, add this result to the list of results. */
732 while (last->next != NULL)
739 } /* }}} int udb_result_create */
742 * Query private functions
744 void udb_query_free_one (udb_query_t *q) /* {{{ */
750 sfree (q->statement);
752 udb_result_free (q->results);
755 } /* }}} void udb_query_free_one */
758 * Query public functions
760 int udb_query_create (udb_query_t ***ret_query_list, /* {{{ */
761 size_t *ret_query_list_len, oconfig_item_t *ci,
762 udb_query_create_callback_t cb, int legacy_mode)
764 udb_query_t **query_list;
765 size_t query_list_len;
771 size_t legacy_position;
773 if ((ret_query_list == NULL) || (ret_query_list_len == NULL))
775 query_list = *ret_query_list;
776 query_list_len = *ret_query_list_len;
778 if ((ci->values_num != 1)
779 || (ci->values[0].type != OCONFIG_TYPE_STRING))
781 WARNING ("db query utils: The `Query' block "
782 "needs exactly one string argument.");
786 q = (udb_query_t *) malloc (sizeof (*q));
789 ERROR ("db query utils: malloc failed.");
792 memset (q, 0, sizeof (*q));
793 q->legacy_mode = legacy_mode;
795 q->max_version = UINT_MAX;
799 status = udb_config_set_string (&q->name, ci);
806 /* Fill the `udb_query_t' structure.. */
807 for (i = 0; i < ci->children_num; i++)
809 oconfig_item_t *child = ci->children + i;
811 if (strcasecmp ("Statement", child->key) == 0)
812 status = udb_config_set_string (&q->statement, child);
813 else if (strcasecmp ("Result", child->key) == 0)
814 status = udb_result_create (q->name, &q->results, child);
815 else if (strcasecmp ("MinVersion", child->key) == 0)
816 status = udb_config_set_uint (&q->min_version, child);
817 else if (strcasecmp ("MaxVersion", child->key) == 0)
818 status = udb_config_set_uint (&q->max_version, child);
820 /* PostgreSQL compatibility code */
821 else if ((strcasecmp ("Query", child->key) == 0)
822 && (q->legacy_mode == 1))
824 WARNING ("db query utils: Query `%s': The `Query' option is "
825 "deprecated. Please use `Statement' instead.",
827 status = udb_config_set_string (&q->statement, child);
829 else if ((strcasecmp ("Column", child->key) == 0)
830 && (q->legacy_mode == 1))
832 WARNING ("db query utils: Query `%s': The `Column' option is "
833 "deprecated. Please use the new syntax instead.",
835 status = udb_legacy_result_create (q->name, &q->results, child,
839 else if ((strcasecmp ("MinPGVersion", child->key) == 0)
840 && (q->legacy_mode == 1))
842 WARNING ("db query utils: Query `%s': The `MinPGVersion' option is "
843 "deprecated. Please use `MinVersion' instead.",
845 status = udb_config_set_uint (&q->min_version, child);
847 else if ((strcasecmp ("MaxPGVersion", child->key) == 0)
848 && (q->legacy_mode == 1))
850 WARNING ("db query utils: Query `%s': The `MaxPGVersion' option is "
851 "deprecated. Please use `MaxVersion' instead.",
853 status = udb_config_set_uint (&q->max_version, child);
856 /* Call custom callbacks */
859 status = (*cb) (q, child);
862 WARNING ("db query utils: The configuration callback failed "
863 "to handle `%s'.", child->key);
868 WARNING ("db query utils: Query `%s': Option `%s' not allowed here.",
869 q->name, child->key);
877 /* Check that all necessary options have been given. */
880 if (q->statement == NULL)
882 WARNING ("db query utils: Query `%s': No `Statement' given.", q->name);
885 if (q->results == NULL)
887 WARNING ("db query utils: Query `%s': No (valid) `Result' block given.",
891 } /* if (status == 0) */
893 /* If all went well, add this query to the list of queries within the
894 * database structure. */
899 temp = (udb_query_t **) realloc (query_list,
900 sizeof (*query_list) * (query_list_len + 1));
903 ERROR ("db query utils: realloc failed");
909 query_list[query_list_len] = q;
916 udb_query_free_one (q);
920 *ret_query_list = query_list;
921 *ret_query_list_len = query_list_len;
924 } /* }}} int udb_query_create */
926 void udb_query_free (udb_query_t **query_list, size_t query_list_len) /* {{{ */
930 if (query_list == NULL)
933 for (i = 0; i < query_list_len; i++)
934 udb_query_free_one (query_list[i]);
937 } /* }}} void udb_query_free */
939 int udb_query_pick_from_list_by_name (const char *name, /* {{{ */
940 udb_query_t **src_list, size_t src_list_len,
941 udb_query_t ***dst_list, size_t *dst_list_len)
946 if ((name == NULL) || (src_list == NULL) || (dst_list == NULL)
947 || (dst_list_len == NULL))
949 ERROR ("db query utils: udb_query_pick_from_list_by_name: "
950 "Invalid argument.");
955 for (i = 0; i < src_list_len; i++)
957 udb_query_t **tmp_list;
960 if (strcasecmp (name, src_list[i]->name) != 0)
963 tmp_list_len = *dst_list_len;
964 tmp_list = (udb_query_t **) realloc (*dst_list, (tmp_list_len + 1)
965 * sizeof (udb_query_t *));
966 if (tmp_list == NULL)
968 ERROR ("db query utils: realloc failed.");
972 tmp_list[tmp_list_len] = src_list[i];
975 *dst_list = tmp_list;
976 *dst_list_len = tmp_list_len;
979 } /* for (i = 0; i < src_list_len; i++) */
983 ERROR ("db query utils: Cannot find query `%s'. Make sure the <Query> "
984 "block is above the database definition!",
990 DEBUG ("db query utils: Added %i versions of query `%s'.",
995 } /* }}} int udb_query_pick_from_list_by_name */
997 int udb_query_pick_from_list (oconfig_item_t *ci, /* {{{ */
998 udb_query_t **src_list, size_t src_list_len,
999 udb_query_t ***dst_list, size_t *dst_list_len)
1003 if ((ci == NULL) || (src_list == NULL) || (dst_list == NULL)
1004 || (dst_list_len == NULL))
1006 ERROR ("db query utils: udb_query_pick_from_list: "
1007 "Invalid argument.");
1011 if ((ci->values_num != 1)
1012 || (ci->values[0].type != OCONFIG_TYPE_STRING))
1014 ERROR ("db query utils: The `%s' config option "
1015 "needs exactly one string argument.", ci->key);
1018 name = ci->values[0].value.string;
1020 return (udb_query_pick_from_list_by_name (name,
1021 src_list, src_list_len,
1022 dst_list, dst_list_len));
1023 } /* }}} int udb_query_pick_from_list */
1025 const char *udb_query_get_name (udb_query_t *q) /* {{{ */
1031 } /* }}} const char *udb_query_get_name */
1033 const char *udb_query_get_statement (udb_query_t *q) /* {{{ */
1038 return (q->statement);
1039 } /* }}} const char *udb_query_get_statement */
1041 void udb_query_set_user_data (udb_query_t *q, void *user_data) /* {{{ */
1046 q->user_data = user_data;
1047 } /* }}} void udb_query_set_user_data */
1049 void *udb_query_get_user_data (udb_query_t *q) /* {{{ */
1054 return (q->user_data);
1055 } /* }}} void *udb_query_get_user_data */
1057 int udb_query_check_version (udb_query_t *q, unsigned int version) /* {{{ */
1062 if ((version < q->min_version) || (version > q->max_version))
1066 } /* }}} int udb_query_check_version */
1068 void udb_query_finish_result (const udb_query_t const *q, /* {{{ */
1069 udb_query_preparation_area_t *prep_area)
1071 udb_result_preparation_area_t *r_area;
1074 if ((q == NULL) || (prep_area == NULL))
1077 prep_area->column_num = 0;
1078 sfree (prep_area->host);
1079 sfree (prep_area->plugin);
1080 sfree (prep_area->db_name);
1082 prep_area->interval = -1;
1084 for (r = q->results, r_area = prep_area->result_prep_areas;
1085 r != NULL; r = r->next, r_area = r_area->next)
1087 /* this may happen during error conditions of the caller */
1090 udb_result_finish_result (r, r_area);
1092 } /* }}} void udb_query_finish_result */
1094 int udb_query_handle_result (const udb_query_t const *q, /* {{{ */
1095 udb_query_preparation_area_t *prep_area, char **column_values)
1097 udb_result_preparation_area_t *r_area;
1102 if ((q == NULL) || (prep_area == NULL))
1105 if ((prep_area->column_num < 1) || (prep_area->host == NULL)
1106 || (prep_area->plugin == NULL) || (prep_area->db_name == NULL))
1108 ERROR ("db query utils: Query `%s': Query is not prepared; "
1109 "can't handle result.", q->name);
1113 #if defined(COLLECT_DEBUG) && COLLECT_DEBUG /* {{{ */
1118 for (i = 0; i < prep_area->column_num; i++)
1120 DEBUG ("db query utils: udb_query_handle_result (%s, %s): "
1121 "column[%zu] = %s;",
1122 prep_area->db_name, q->name, i, column_values[i]);
1128 for (r = q->results, r_area = prep_area->result_prep_areas;
1129 r != NULL; r = r->next, r_area = r_area->next)
1131 status = udb_result_handle_result (r, prep_area, r_area,
1139 ERROR ("db query utils: udb_query_handle_result (%s, %s): "
1140 "All results failed.", prep_area->db_name, q->name);
1145 } /* }}} int udb_query_handle_result */
1147 int udb_query_prepare_result (const udb_query_t const *q, /* {{{ */
1148 udb_query_preparation_area_t *prep_area,
1149 const char *host, const char *plugin, const char *db_name,
1150 char **column_names, size_t column_num, int interval)
1152 udb_result_preparation_area_t *r_area;
1156 if ((q == NULL) || (prep_area == NULL))
1159 udb_query_finish_result (q, prep_area);
1161 prep_area->column_num = column_num;
1162 prep_area->host = strdup (host);
1163 prep_area->plugin = strdup (plugin);
1164 prep_area->db_name = strdup (db_name);
1166 prep_area->interval = interval;
1168 if ((prep_area->host == NULL) || (prep_area->plugin == NULL)
1169 || (prep_area->db_name == NULL))
1171 ERROR ("db query utils: Query `%s': Prepare failed: Out of memory.", q->name);
1172 udb_query_finish_result (q, prep_area);
1176 #if defined(COLLECT_DEBUG) && COLLECT_DEBUG
1181 for (i = 0; i < column_num; i++)
1183 DEBUG ("db query utils: udb_query_prepare_result: "
1184 "query = %s; column[%zu] = %s;",
1185 q->name, i, column_names[i]);
1190 for (r = q->results, r_area = prep_area->result_prep_areas;
1191 r != NULL; r = r->next, r_area = r_area->next)
1195 ERROR ("db query utils: Query `%s': Invalid number of result "
1196 "preparation areas.", q->name);
1197 udb_query_finish_result (q, prep_area);
1201 status = udb_result_prepare_result (r, r_area, column_names, column_num);
1204 udb_query_finish_result (q, prep_area);
1210 } /* }}} int udb_query_prepare_result */
1212 udb_query_preparation_area_t *
1213 udb_query_allocate_preparation_area (udb_query_t *q) /* {{{ */
1215 udb_query_preparation_area_t *q_area;
1216 udb_result_preparation_area_t **next_r_area;
1219 q_area = (udb_query_preparation_area_t *)malloc (sizeof (*q_area));
1223 memset (q_area, 0, sizeof (*q_area));
1225 next_r_area = &q_area->result_prep_areas;
1226 for (r = q->results; r != NULL; r = r->next)
1228 udb_result_preparation_area_t *r_area;
1230 r_area = (udb_result_preparation_area_t *)malloc (sizeof (*r_area));
1233 for (r_area = q_area->result_prep_areas;
1234 r_area != NULL; r_area = r_area->next)
1242 memset (r_area, 0, sizeof (*r_area));
1244 *next_r_area = r_area;
1245 next_r_area = &r_area->next;
1249 } /* }}} udb_query_preparation_area_t *udb_query_allocate_preparation_area */
1252 udb_query_delete_preparation_area (udb_query_preparation_area_t *q_area) /* {{{ */
1254 udb_result_preparation_area_t *r_area;
1259 r_area = q_area->result_prep_areas;
1260 while (r_area != NULL)
1262 udb_result_preparation_area_t *area = r_area;
1264 r_area = r_area->next;
1266 sfree (area->instances_pos);
1267 sfree (area->values_pos);
1268 sfree (area->instances_buffer);
1269 sfree (area->values_buffer);
1273 sfree (q_area->host);
1274 sfree (q_area->plugin);
1275 sfree (q_area->db_name);
1278 } /* }}} void udb_query_delete_preparation_area */
1280 /* vim: set sw=2 sts=2 et fdm=marker : */