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 /* {{{ */
87 udb_result_preparation_area_t *result_prep_areas;
91 * Config Private functions
93 static int udb_config_set_string (char **ret_string, /* {{{ */
98 if ((ci->values_num != 1)
99 || (ci->values[0].type != OCONFIG_TYPE_STRING))
101 WARNING ("db query utils: The `%s' config option "
102 "needs exactly one string argument.", ci->key);
106 string = strdup (ci->values[0].value.string);
109 ERROR ("db query utils: strdup failed.");
113 if (*ret_string != NULL)
115 *ret_string = string;
118 } /* }}} int udb_config_set_string */
120 static int udb_config_add_string (char ***ret_array, /* {{{ */
121 size_t *ret_array_len, oconfig_item_t *ci)
127 if (ci->values_num < 1)
129 WARNING ("db query utils: The `%s' config option "
130 "needs at least one argument.", ci->key);
134 for (i = 0; i < ci->values_num; i++)
136 if (ci->values[i].type != OCONFIG_TYPE_STRING)
138 WARNING ("db query utils: Argument %i to the `%s' option "
139 "is not a string.", i + 1, ci->key);
144 array_len = *ret_array_len;
145 array = (char **) realloc (*ret_array,
146 sizeof (char *) * (array_len + ci->values_num));
149 ERROR ("db query utils: realloc failed.");
154 for (i = 0; i < ci->values_num; i++)
156 array[array_len] = strdup (ci->values[i].value.string);
157 if (array[array_len] == NULL)
159 ERROR ("db query utils: strdup failed.");
160 *ret_array_len = array_len;
166 *ret_array_len = array_len;
168 } /* }}} int udb_config_add_string */
170 static int udb_config_set_uint (unsigned int *ret_value, /* {{{ */
175 if ((ci->values_num != 1)
176 || (ci->values[0].type != OCONFIG_TYPE_NUMBER))
178 WARNING ("db query utils: The `%s' config option "
179 "needs exactly one numeric argument.", ci->key);
183 tmp = ci->values[0].value.number;
184 if ((tmp < 0.0) || (tmp > ((double) UINT_MAX)))
187 *ret_value = (unsigned int) (tmp + .5);
189 } /* }}} int udb_config_set_uint */
192 * Legacy result private functions
194 static void udb_legacy_result_finish_result (const udb_result_t const *r, /* {{{ */
195 udb_result_preparation_area_t *prep_area)
197 if ((r == NULL) || (prep_area))
200 assert (r->legacy_mode == 1);
202 prep_area->ds = NULL;
203 } /* }}} void udb_legacy_result_finish_result */
205 static int udb_legacy_result_handle_result (udb_result_t *r, /* {{{ */
206 udb_query_preparation_area_t *q_area,
207 udb_result_preparation_area_t *r_area,
208 const udb_query_t const *q, char **column_values)
210 value_list_t vl = VALUE_LIST_INIT;
214 assert (r->legacy_mode == 1);
215 assert (r_area->ds != NULL);
216 assert (r_area->ds->ds_num == 1);
221 value_str = column_values[r->legacy_position];
222 if (0 != parse_value (value_str, &vl.values[0], r_area->ds->ds[0].type))
224 ERROR ("db query utils: udb_legacy_result_handle_result: "
225 "Parsing `%s' as %s failed.", value_str,
226 DS_TYPE_TO_STRING (r_area->ds->ds[0].type));
231 sstrncpy (vl.host, q_area->host, sizeof (vl.host));
232 sstrncpy (vl.plugin, q_area->plugin, sizeof (vl.plugin));
233 sstrncpy (vl.plugin_instance, q_area->db_name, sizeof (vl.type_instance));
234 sstrncpy (vl.type, r->type, sizeof (vl.type));
236 if (r->instance_prefix != NULL)
237 sstrncpy (vl.type_instance, r->instance_prefix,
238 sizeof (vl.type_instance));
240 plugin_dispatch_values (&vl);
243 } /* }}} int udb_legacy_result_handle_result */
245 static int udb_legacy_result_prepare_result (const udb_result_t const *r, /* {{{ */
246 udb_result_preparation_area_t *prep_area,
247 char **column_names, size_t column_num)
252 assert (r->legacy_mode == 1);
254 /* Make sure previous preparations are cleaned up. */
255 udb_legacy_result_finish_result (r, prep_area);
257 if (r->legacy_position >= column_num)
259 ERROR ("db query utils: The legacy configuration specified (at least) "
260 "%zu `Column's, but the query returned only %zu columns!",
261 r->legacy_position + 1, column_num);
265 /* Read `ds' and check number of values {{{ */
266 prep_area->ds = plugin_get_ds (r->type);
267 if (prep_area->ds == NULL)
269 ERROR ("db query utils: udb_result_prepare_result: Type `%s' is not "
270 "known by the daemon. See types.db(5) for details.",
275 if (prep_area->ds->ds_num != 1)
277 ERROR ("db query utils: udb_result_prepare_result: The type `%s' "
278 "requires exactly %i values, but the legacy configuration "
279 "requires exactly one!",
281 prep_area->ds->ds_num);
287 } /* }}} int udb_legacy_result_prepare_result */
289 static int udb_legacy_result_create (const char *query_name, /* {{{ */
290 udb_result_t **r_head, oconfig_item_t *ci, size_t position)
294 if ((ci->values_num < 1) || (ci->values_num > 2)
295 || (ci->values[0].type != OCONFIG_TYPE_STRING)
296 || ((ci->values_num == 2)
297 && (ci->values[1].type != OCONFIG_TYPE_STRING)))
299 WARNING ("db query utils: The `Column' block needs either one or two "
300 "string arguments.");
304 r = (udb_result_t *) malloc (sizeof (*r));
307 ERROR ("db query utils: malloc failed.");
310 memset (r, 0, sizeof (*r));
313 r->legacy_position = position;
315 r->type = strdup (ci->values[0].value.string);
318 ERROR ("db query utils: strdup failed.");
323 r->instance_prefix = NULL;
324 if (ci->values_num == 2)
326 r->instance_prefix = strdup (ci->values[1].value.string);
327 if (r->instance_prefix == NULL)
329 ERROR ("db query utils: strdup failed.");
336 /* If all went well, add this result to the list of results. */
346 while (last->next != NULL)
353 } /* }}} int udb_legacy_result_create */
356 * Result private functions
358 static int udb_result_submit (udb_result_t *r, /* {{{ */
359 udb_result_preparation_area_t *r_area,
360 const udb_query_t const *q, udb_query_preparation_area_t *q_area)
362 value_list_t vl = VALUE_LIST_INIT;
366 assert (r->legacy_mode == 0);
367 assert (r_area->ds != NULL);
368 assert (((size_t) r_area->ds->ds_num) == r->values_num);
370 vl.values = (value_t *) calloc (r_area->ds->ds_num, sizeof (value_t));
371 if (vl.values == NULL)
373 ERROR ("db query utils: malloc failed.");
376 vl.values_len = r_area->ds->ds_num;
378 for (i = 0; i < r->values_num; i++)
380 char *value_str = r_area->values_buffer[i];
382 if (0 != parse_value (value_str, &vl.values[i], r_area->ds->ds[i].type))
384 ERROR ("db query utils: udb_result_submit: Parsing `%s' as %s failed.",
385 value_str, DS_TYPE_TO_STRING (r_area->ds->ds[i].type));
391 sstrncpy (vl.host, q_area->host, sizeof (vl.host));
392 sstrncpy (vl.plugin, q_area->plugin, sizeof (vl.plugin));
393 sstrncpy (vl.plugin_instance, q_area->db_name, sizeof (vl.type_instance));
394 sstrncpy (vl.type, r->type, sizeof (vl.type));
396 /* Set vl.type_instance {{{ */
397 if (r->instances_num <= 0)
399 if (r->instance_prefix == NULL)
400 vl.type_instance[0] = 0;
402 sstrncpy (vl.type_instance, r->instance_prefix,
403 sizeof (vl.type_instance));
405 else /* if ((r->instances_num > 0) */
407 if (r->instance_prefix == NULL)
409 strjoin (vl.type_instance, sizeof (vl.type_instance),
410 r_area->instances_buffer, r->instances_num, "-");
414 char tmp[DATA_MAX_NAME_LEN];
416 strjoin (tmp, sizeof (tmp), r_area->instances_buffer,
417 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);
424 vl.type_instance[sizeof (vl.type_instance) - 1] = 0;
427 plugin_dispatch_values (&vl);
431 } /* }}} void udb_result_submit */
433 static void udb_result_finish_result (const udb_result_t const *r, /* {{{ */
434 udb_result_preparation_area_t *prep_area)
436 if ((r == NULL) || (prep_area == NULL))
439 if (r->legacy_mode == 1)
441 udb_legacy_result_finish_result (r, prep_area);
445 assert (r->legacy_mode == 0);
447 prep_area->ds = NULL;
448 sfree (prep_area->instances_pos);
449 sfree (prep_area->values_pos);
450 sfree (prep_area->instances_buffer);
451 sfree (prep_area->values_buffer);
452 } /* }}} void udb_result_finish_result */
454 static int udb_result_handle_result (udb_result_t *r, /* {{{ */
455 udb_query_preparation_area_t *q_area,
456 udb_result_preparation_area_t *r_area,
457 const udb_query_t const *q, char **column_values)
461 assert (r && q_area && r_area);
463 if (r->legacy_mode == 1)
464 return (udb_legacy_result_handle_result (r, q_area, r_area,
467 assert (r->legacy_mode == 0);
469 for (i = 0; i < r->instances_num; i++)
470 r_area->instances_buffer[i] = column_values[r_area->instances_pos[i]];
472 for (i = 0; i < r->values_num; i++)
473 r_area->values_buffer[i] = column_values[r_area->values_pos[i]];
475 return udb_result_submit (r, r_area, q, q_area);
476 } /* }}} int udb_result_handle_result */
478 static int udb_result_prepare_result (const udb_result_t const *r, /* {{{ */
479 udb_result_preparation_area_t *prep_area,
480 char **column_names, size_t column_num)
484 if ((r == NULL) || (prep_area == NULL))
487 if (r->legacy_mode == 1)
488 return (udb_legacy_result_prepare_result (r, prep_area,
489 column_names, column_num));
491 assert (r->legacy_mode == 0);
493 #define BAIL_OUT(status) \
494 prep_area->ds = NULL; \
495 sfree (prep_area->instances_pos); \
496 sfree (prep_area->values_pos); \
497 sfree (prep_area->instances_buffer); \
498 sfree (prep_area->values_buffer); \
501 /* Make sure previous preparations are cleaned up. */
502 udb_result_finish_result (r, prep_area);
503 prep_area->instances_pos = NULL;
504 prep_area->values_pos = NULL;
506 /* Read `ds' and check number of values {{{ */
507 prep_area->ds = plugin_get_ds (r->type);
508 if (prep_area->ds == NULL)
510 ERROR ("db query utils: udb_result_prepare_result: Type `%s' is not "
511 "known by the daemon. See types.db(5) for details.",
516 if (((size_t) prep_area->ds->ds_num) != r->values_num)
518 ERROR ("db query utils: udb_result_prepare_result: The type `%s' "
519 "requires exactly %i value%s, but the configuration specifies %zu.",
521 prep_area->ds->ds_num, (prep_area->ds->ds_num == 1) ? "" : "s",
527 /* Allocate r->instances_pos, r->values_pos, r->instances_buffer, and
528 * r->values_buffer {{{ */
529 if (r->instances_num > 0)
531 prep_area->instances_pos
532 = (size_t *) calloc (r->instances_num, sizeof (size_t));
533 if (prep_area->instances_pos == NULL)
535 ERROR ("db query utils: udb_result_prepare_result: malloc failed.");
539 prep_area->instances_buffer
540 = (char **) calloc (r->instances_num, sizeof (char *));
541 if (prep_area->instances_buffer == NULL)
543 ERROR ("db query utils: udb_result_prepare_result: malloc failed.");
546 } /* if (r->instances_num > 0) */
548 prep_area->values_pos
549 = (size_t *) calloc (r->values_num, sizeof (size_t));
550 if (prep_area->values_pos == NULL)
552 ERROR ("db query utils: udb_result_prepare_result: malloc failed.");
556 prep_area->values_buffer
557 = (char **) calloc (r->values_num, sizeof (char *));
558 if (prep_area->values_buffer == NULL)
560 ERROR ("db query utils: udb_result_prepare_result: malloc failed.");
565 /* Determine the position of the instance columns {{{ */
566 for (i = 0; i < r->instances_num; i++)
570 for (j = 0; j < column_num; j++)
572 if (strcasecmp (r->instances[i], column_names[j]) == 0)
574 prep_area->instances_pos[i] = j;
581 ERROR ("db query utils: udb_result_prepare_result: "
582 "Column `%s' could not be found.",
586 } /* }}} for (i = 0; i < r->instances_num; i++) */
588 /* Determine the position of the value columns {{{ */
589 for (i = 0; i < r->values_num; i++)
593 for (j = 0; j < column_num; j++)
595 if (strcasecmp (r->values[i], column_names[j]) == 0)
597 prep_area->values_pos[i] = j;
604 ERROR ("db query utils: udb_result_prepare_result: "
605 "Column `%s' could not be found.",
609 } /* }}} for (i = 0; i < r->values_num; i++) */
613 } /* }}} int udb_result_prepare_result */
615 static void udb_result_free (udb_result_t *r) /* {{{ */
624 for (i = 0; i < r->instances_num; i++)
625 sfree (r->instances[i]);
626 sfree (r->instances);
628 for (i = 0; i < r->values_num; i++)
629 sfree (r->values[i]);
632 udb_result_free (r->next);
635 } /* }}} void udb_result_free */
637 static int udb_result_create (const char *query_name, /* {{{ */
638 udb_result_t **r_head, oconfig_item_t *ci)
644 if (ci->values_num != 0)
646 WARNING ("db query utils: The `Result' block doesn't accept "
647 "any arguments. Ignoring %i argument%s.",
648 ci->values_num, (ci->values_num == 1) ? "" : "s");
651 r = (udb_result_t *) malloc (sizeof (*r));
654 ERROR ("db query utils: malloc failed.");
657 memset (r, 0, sizeof (*r));
659 r->instance_prefix = NULL;
664 /* Fill the `udb_result_t' structure.. */
666 for (i = 0; i < ci->children_num; i++)
668 oconfig_item_t *child = ci->children + i;
670 if (strcasecmp ("Type", child->key) == 0)
671 status = udb_config_set_string (&r->type, child);
672 else if (strcasecmp ("InstancePrefix", child->key) == 0)
673 status = udb_config_set_string (&r->instance_prefix, child);
674 else if (strcasecmp ("InstancesFrom", child->key) == 0)
675 status = udb_config_add_string (&r->instances, &r->instances_num, child);
676 else if (strcasecmp ("ValuesFrom", child->key) == 0)
677 status = udb_config_add_string (&r->values, &r->values_num, child);
680 WARNING ("db query utils: Query `%s': Option `%s' not allowed here.",
681 query_name, child->key);
689 /* Check that all necessary options have been given. */
694 WARNING ("db query utils: `Type' not given for "
695 "result in query `%s'", query_name);
698 if (r->values == NULL)
700 WARNING ("db query utils: `ValuesFrom' not given for "
701 "result in query `%s'", query_name);
706 } /* while (status == 0) */
714 /* If all went well, add this result to the list of results. */
724 while (last->next != NULL)
731 } /* }}} int udb_result_create */
734 * Query private functions
736 void udb_query_free_one (udb_query_t *q) /* {{{ */
742 sfree (q->statement);
744 udb_result_free (q->results);
747 } /* }}} void udb_query_free_one */
750 * Query public functions
752 int udb_query_create (udb_query_t ***ret_query_list, /* {{{ */
753 size_t *ret_query_list_len, oconfig_item_t *ci,
754 udb_query_create_callback_t cb, int legacy_mode)
756 udb_query_t **query_list;
757 size_t query_list_len;
763 size_t legacy_position;
765 if ((ret_query_list == NULL) || (ret_query_list_len == NULL))
767 query_list = *ret_query_list;
768 query_list_len = *ret_query_list_len;
770 if ((ci->values_num != 1)
771 || (ci->values[0].type != OCONFIG_TYPE_STRING))
773 WARNING ("db query utils: The `Query' block "
774 "needs exactly one string argument.");
778 q = (udb_query_t *) malloc (sizeof (*q));
781 ERROR ("db query utils: malloc failed.");
784 memset (q, 0, sizeof (*q));
785 q->legacy_mode = legacy_mode;
787 q->max_version = UINT_MAX;
791 status = udb_config_set_string (&q->name, ci);
798 /* Fill the `udb_query_t' structure.. */
799 for (i = 0; i < ci->children_num; i++)
801 oconfig_item_t *child = ci->children + i;
803 if (strcasecmp ("Statement", child->key) == 0)
804 status = udb_config_set_string (&q->statement, child);
805 else if (strcasecmp ("Result", child->key) == 0)
806 status = udb_result_create (q->name, &q->results, child);
807 else if (strcasecmp ("MinVersion", child->key) == 0)
808 status = udb_config_set_uint (&q->min_version, child);
809 else if (strcasecmp ("MaxVersion", child->key) == 0)
810 status = udb_config_set_uint (&q->max_version, child);
812 /* PostgreSQL compatibility code */
813 else if ((strcasecmp ("Query", child->key) == 0)
814 && (q->legacy_mode == 1))
816 WARNING ("db query utils: Query `%s': The `Query' option is "
817 "deprecated. Please use `Statement' instead.",
819 status = udb_config_set_string (&q->statement, child);
821 else if ((strcasecmp ("Column", child->key) == 0)
822 && (q->legacy_mode == 1))
824 WARNING ("db query utils: Query `%s': The `Column' option is "
825 "deprecated. Please use the new syntax instead.",
827 status = udb_legacy_result_create (q->name, &q->results, child,
831 else if ((strcasecmp ("MinPGVersion", child->key) == 0)
832 && (q->legacy_mode == 1))
834 WARNING ("db query utils: Query `%s': The `MinPGVersion' option is "
835 "deprecated. Please use `MinVersion' instead.",
837 status = udb_config_set_uint (&q->min_version, child);
839 else if ((strcasecmp ("MaxPGVersion", child->key) == 0)
840 && (q->legacy_mode == 1))
842 WARNING ("db query utils: Query `%s': The `MaxPGVersion' option is "
843 "deprecated. Please use `MaxVersion' instead.",
845 status = udb_config_set_uint (&q->max_version, child);
848 /* Call custom callbacks */
851 status = (*cb) (q, child);
854 WARNING ("db query utils: The configuration callback failed "
855 "to handle `%s'.", child->key);
860 WARNING ("db query utils: Query `%s': Option `%s' not allowed here.",
861 q->name, child->key);
869 /* Check that all necessary options have been given. */
872 if (q->statement == NULL)
874 WARNING ("db query utils: Query `%s': No `Statement' given.", q->name);
877 if (q->results == NULL)
879 WARNING ("db query utils: Query `%s': No (valid) `Result' block given.",
883 } /* if (status == 0) */
885 /* If all went well, add this query to the list of queries within the
886 * database structure. */
891 temp = (udb_query_t **) realloc (query_list,
892 sizeof (*query_list) * (query_list_len + 1));
895 ERROR ("db query utils: realloc failed");
901 query_list[query_list_len] = q;
908 udb_query_free_one (q);
912 *ret_query_list = query_list;
913 *ret_query_list_len = query_list_len;
916 } /* }}} int udb_query_create */
918 void udb_query_free (udb_query_t **query_list, size_t query_list_len) /* {{{ */
922 if (query_list == NULL)
925 for (i = 0; i < query_list_len; i++)
926 udb_query_free_one (query_list[i]);
929 } /* }}} void udb_query_free */
931 int udb_query_pick_from_list_by_name (const char *name, /* {{{ */
932 udb_query_t **src_list, size_t src_list_len,
933 udb_query_t ***dst_list, size_t *dst_list_len)
938 if ((name == NULL) || (src_list == NULL) || (dst_list == NULL)
939 || (dst_list_len == NULL))
941 ERROR ("db query utils: udb_query_pick_from_list_by_name: "
942 "Invalid argument.");
947 for (i = 0; i < src_list_len; i++)
949 udb_query_t **tmp_list;
952 if (strcasecmp (name, src_list[i]->name) != 0)
955 tmp_list_len = *dst_list_len;
956 tmp_list = (udb_query_t **) realloc (*dst_list, (tmp_list_len + 1)
957 * sizeof (udb_query_t *));
958 if (tmp_list == NULL)
960 ERROR ("db query utils: realloc failed.");
964 tmp_list[tmp_list_len] = src_list[i];
967 *dst_list = tmp_list;
968 *dst_list_len = tmp_list_len;
971 } /* for (i = 0; i < src_list_len; i++) */
975 ERROR ("db query utils: Cannot find query `%s'. Make sure the <Query> "
976 "block is above the database definition!",
982 DEBUG ("db query utils: Added %i versions of query `%s'.",
987 } /* }}} int udb_query_pick_from_list_by_name */
989 int udb_query_pick_from_list (oconfig_item_t *ci, /* {{{ */
990 udb_query_t **src_list, size_t src_list_len,
991 udb_query_t ***dst_list, size_t *dst_list_len)
995 if ((ci == NULL) || (src_list == NULL) || (dst_list == NULL)
996 || (dst_list_len == NULL))
998 ERROR ("db query utils: udb_query_pick_from_list: "
999 "Invalid argument.");
1003 if ((ci->values_num != 1)
1004 || (ci->values[0].type != OCONFIG_TYPE_STRING))
1006 ERROR ("db query utils: The `%s' config option "
1007 "needs exactly one string argument.", ci->key);
1010 name = ci->values[0].value.string;
1012 return (udb_query_pick_from_list_by_name (name,
1013 src_list, src_list_len,
1014 dst_list, dst_list_len));
1015 } /* }}} int udb_query_pick_from_list */
1017 const char *udb_query_get_name (udb_query_t *q) /* {{{ */
1023 } /* }}} const char *udb_query_get_name */
1025 const char *udb_query_get_statement (udb_query_t *q) /* {{{ */
1030 return (q->statement);
1031 } /* }}} const char *udb_query_get_statement */
1033 void udb_query_set_user_data (udb_query_t *q, void *user_data) /* {{{ */
1038 q->user_data = user_data;
1039 } /* }}} void udb_query_set_user_data */
1041 void *udb_query_get_user_data (udb_query_t *q) /* {{{ */
1046 return (q->user_data);
1047 } /* }}} void *udb_query_get_user_data */
1049 int udb_query_check_version (udb_query_t *q, unsigned int version) /* {{{ */
1054 if ((version < q->min_version) || (version > q->max_version))
1058 } /* }}} int udb_query_check_version */
1060 void udb_query_finish_result (const udb_query_t const *q, /* {{{ */
1061 udb_query_preparation_area_t *prep_area)
1063 udb_result_preparation_area_t *r_area;
1066 if ((q == NULL) || (prep_area == NULL))
1069 prep_area->column_num = 0;
1070 sfree (prep_area->host);
1071 sfree (prep_area->plugin);
1072 sfree (prep_area->db_name);
1074 for (r = q->results, r_area = prep_area->result_prep_areas;
1075 r != NULL; r = r->next, r_area = r_area->next)
1077 /* this may happen during error conditions of the caller */
1080 udb_result_finish_result (r, r_area);
1082 } /* }}} void udb_query_finish_result */
1084 int udb_query_handle_result (const udb_query_t const *q, /* {{{ */
1085 udb_query_preparation_area_t *prep_area, char **column_values)
1087 udb_result_preparation_area_t *r_area;
1092 if ((q == NULL) || (prep_area == NULL))
1095 if ((prep_area->column_num < 1) || (prep_area->host == NULL)
1096 || (prep_area->plugin == NULL) || (prep_area->db_name == NULL))
1098 ERROR ("db query utils: Query `%s': Query is not prepared; "
1099 "can't handle result.", q->name);
1103 #if defined(COLLECT_DEBUG) && COLLECT_DEBUG /* {{{ */
1108 for (i = 0; i < prep_area->column_num; i++)
1110 DEBUG ("db query utils: udb_query_handle_result (%s, %s): "
1111 "column[%zu] = %s;",
1112 prep_area->db_name, q->name, i, column_values[i]);
1118 for (r = q->results, r_area = prep_area->result_prep_areas;
1119 r != NULL; r = r->next, r_area = r_area->next)
1121 status = udb_result_handle_result (r, prep_area, r_area,
1129 ERROR ("db query utils: udb_query_handle_result (%s, %s): "
1130 "All results failed.", prep_area->db_name, q->name);
1135 } /* }}} int udb_query_handle_result */
1137 int udb_query_prepare_result (const udb_query_t const *q, /* {{{ */
1138 udb_query_preparation_area_t *prep_area,
1139 const char *host, const char *plugin, const char *db_name,
1140 char **column_names, size_t column_num)
1142 udb_result_preparation_area_t *r_area;
1146 if ((q == NULL) || (prep_area == NULL))
1149 udb_query_finish_result (q, prep_area);
1151 prep_area->column_num = column_num;
1152 prep_area->host = strdup (host);
1153 prep_area->plugin = strdup (plugin);
1154 prep_area->db_name = strdup (db_name);
1156 if ((prep_area->host == NULL) || (prep_area->plugin == NULL)
1157 || (prep_area->db_name == NULL))
1159 ERROR ("db query utils: Query `%s': Prepare failed: Out of memory.", q->name);
1160 udb_query_finish_result (q, prep_area);
1164 #if defined(COLLECT_DEBUG) && COLLECT_DEBUG
1169 for (i = 0; i < column_num; i++)
1171 DEBUG ("db query utils: udb_query_prepare_result: "
1172 "query = %s; column[%zu] = %s;",
1173 q->name, i, column_names[i]);
1178 for (r = q->results, r_area = prep_area->result_prep_areas;
1179 r != NULL; r = r->next, r_area = r_area->next)
1183 ERROR ("db query utils: Query `%s': Invalid number of result "
1184 "preparation areas.", q->name);
1185 udb_query_finish_result (q, prep_area);
1189 status = udb_result_prepare_result (r, r_area, column_names, column_num);
1192 udb_query_finish_result (q, prep_area);
1198 } /* }}} int udb_query_prepare_result */
1200 udb_query_preparation_area_t *
1201 udb_query_allocate_preparation_area (udb_query_t *q) /* {{{ */
1203 udb_query_preparation_area_t *q_area;
1204 udb_result_preparation_area_t **next_r_area;
1207 q_area = (udb_query_preparation_area_t *)malloc (sizeof (*q_area));
1211 memset (q_area, 0, sizeof (*q_area));
1213 next_r_area = &q_area->result_prep_areas;
1214 for (r = q->results; r != NULL; r = r->next)
1216 udb_result_preparation_area_t *r_area;
1218 r_area = (udb_result_preparation_area_t *)malloc (sizeof (*r_area));
1221 for (r_area = q_area->result_prep_areas;
1222 r_area != NULL; r_area = r_area->next)
1230 memset (r_area, 0, sizeof (*r_area));
1232 *next_r_area = r_area;
1233 next_r_area = &r_area->next;
1237 } /* }}} udb_query_preparation_area_t *udb_query_allocate_preparation_area */
1240 udb_query_delete_preparation_area (udb_query_preparation_area_t *q_area) /* {{{ */
1242 udb_result_preparation_area_t *r_area;
1247 r_area = q_area->result_prep_areas;
1248 while (r_area != NULL)
1250 udb_result_preparation_area_t *area = r_area;
1252 r_area = r_area->next;
1254 sfree (area->instances_pos);
1255 sfree (area->values_pos);
1256 sfree (area->instances_buffer);
1257 sfree (area->values_buffer);
1261 sfree (q_area->host);
1262 sfree (q_area->plugin);
1263 sfree (q_area->db_name);
1266 } /* }}} void udb_query_delete_preparation_area */
1268 /* vim: set sw=2 sts=2 et fdm=marker : */