1 /**
2 * collectd - src/utils_cache.c
3 * Copyright (C) 2007,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 * Author:
19 * Florian octo Forster <octo at verplant.org>
20 **/
22 #include "collectd.h"
23 #include "common.h"
24 #include "plugin.h"
25 #include "utils_avltree.h"
26 #include "utils_cache.h"
27 #include "utils_threshold.h"
28 #include "meta_data.h"
30 #include <assert.h>
31 #include <pthread.h>
33 typedef struct cache_entry_s
34 {
35 char name[6 * DATA_MAX_NAME_LEN];
36 int values_num;
37 gauge_t *values_gauge;
38 value_t *values_raw;
39 /* Time contained in the package
40 * (for calculating rates) */
41 time_t last_time;
42 /* Time according to the local clock
43 * (for purging old entries) */
44 time_t last_update;
45 /* Interval in which the data is collected
46 * (for purding old entries) */
47 int interval;
48 int state;
49 int hits;
51 /*
52 * +-----+-----+-----+-----+-----+-----+-----+-----+-----+----
53 * ! 0 ! 1 ! 2 ! 3 ! 4 ! 5 ! 6 ! 7 ! 8 ! ...
54 * +-----+-----+-----+-----+-----+-----+-----+-----+-----+----
55 * ! ds0 ! ds1 ! ds2 ! ds0 ! ds1 ! ds2 ! ds0 ! ds1 ! ds2 ! ...
56 * +-----+-----+-----+-----+-----+-----+-----+-----+-----+----
57 * ! t = 0 ! t = 1 ! t = 2 ! ...
58 * +-----------------+-----------------+-----------------+----
59 */
60 gauge_t *history;
61 size_t history_index; /* points to the next position to write to. */
62 size_t history_length;
64 meta_data_t *meta;
65 } cache_entry_t;
67 static c_avl_tree_t *cache_tree = NULL;
68 static pthread_mutex_t cache_lock = PTHREAD_MUTEX_INITIALIZER;
70 static int cache_compare (const cache_entry_t *a, const cache_entry_t *b)
71 {
72 assert ((a != NULL) && (b != NULL));
73 return (strcmp (a->name, b->name));
74 } /* int cache_compare */
76 static cache_entry_t *cache_alloc (int values_num)
77 {
78 cache_entry_t *ce;
80 ce = (cache_entry_t *) malloc (sizeof (cache_entry_t));
81 if (ce == NULL)
82 {
83 ERROR ("utils_cache: cache_alloc: malloc failed.");
84 return (NULL);
85 }
86 memset (ce, '\0', sizeof (cache_entry_t));
87 ce->values_num = values_num;
89 ce->values_gauge = calloc (values_num, sizeof (*ce->values_gauge));
90 ce->values_raw = calloc (values_num, sizeof (*ce->values_raw));
91 if ((ce->values_gauge == NULL) || (ce->values_raw == NULL))
92 {
93 sfree (ce->values_gauge);
94 sfree (ce->values_raw);
95 sfree (ce);
96 ERROR ("utils_cache: cache_alloc: calloc failed.");
97 return (NULL);
98 }
100 ce->history = NULL;
101 ce->history_length = 0;
102 ce->meta = NULL;
104 return (ce);
105 } /* cache_entry_t *cache_alloc */
107 static void cache_free (cache_entry_t *ce)
108 {
109 if (ce == NULL)
110 return;
112 sfree (ce->values_gauge);
113 sfree (ce->values_raw);
114 sfree (ce->history);
115 if (ce->meta != NULL)
116 {
117 meta_data_destroy (ce->meta);
118 ce->meta = NULL;
119 }
120 sfree (ce);
121 } /* void cache_free */
123 static int uc_send_notification (const char *name)
124 {
125 cache_entry_t *ce = NULL;
126 int status;
128 char *name_copy;
129 char *host;
130 char *plugin;
131 char *plugin_instance;
132 char *type;
133 char *type_instance;
135 notification_t n;
137 name_copy = strdup (name);
138 if (name_copy == NULL)
139 {
140 ERROR ("uc_send_notification: strdup failed.");
141 return (-1);
142 }
144 status = parse_identifier (name_copy, &host,
145 &plugin, &plugin_instance,
146 &type, &type_instance);
147 if (status != 0)
148 {
149 ERROR ("uc_send_notification: Cannot parse name `%s'", name);
150 return (-1);
151 }
153 /* Copy the associative members */
154 notification_init (&n, NOTIF_FAILURE, /* host = */ NULL,
155 host, plugin, plugin_instance, type, type_instance);
157 sfree (name_copy);
158 name_copy = host = plugin = plugin_instance = type = type_instance = NULL;
160 pthread_mutex_lock (&cache_lock);
162 /*
163 * Set the time _after_ getting the lock because we don't know how long
164 * acquiring the lock takes and we will use this time later to decide
165 * whether or not the state is OKAY.
166 */
167 n.time = time (NULL);
169 status = c_avl_get (cache_tree, name, (void *) &ce);
170 if (status != 0)
171 {
172 pthread_mutex_unlock (&cache_lock);
173 sfree (name_copy);
174 return (-1);
175 }
177 /* Check if the entry has been updated in the meantime */
178 if ((n.time - ce->last_update) < (2 * ce->interval))
179 {
180 ce->state = STATE_OKAY;
181 pthread_mutex_unlock (&cache_lock);
182 sfree (name_copy);
183 return (-1);
184 }
186 ssnprintf (n.message, sizeof (n.message),
187 "%s has not been updated for %i seconds.", name,
188 (int) (n.time - ce->last_update));
190 pthread_mutex_unlock (&cache_lock);
192 plugin_dispatch_notification (&n);
194 return (0);
195 } /* int uc_send_notification */
197 static void uc_check_range (const data_set_t *ds, cache_entry_t *ce)
198 {
199 int i;
201 for (i = 0; i < ds->ds_num; i++)
202 {
203 if (isnan (ce->values_gauge[i]))
204 continue;
205 else if (ce->values_gauge[i] < ds->ds[i].min)
206 ce->values_gauge[i] = NAN;
207 else if (ce->values_gauge[i] > ds->ds[i].max)
208 ce->values_gauge[i] = NAN;
209 }
210 } /* void uc_check_range */
212 static int uc_insert (const data_set_t *ds, const value_list_t *vl,
213 const char *key)
214 {
215 int i;
216 char *key_copy;
217 cache_entry_t *ce;
219 /* `cache_lock' has been locked by `uc_update' */
221 key_copy = strdup (key);
222 if (key_copy == NULL)
223 {
224 ERROR ("uc_insert: strdup failed.");
225 return (-1);
226 }
228 ce = cache_alloc (ds->ds_num);
229 if (ce == NULL)
230 {
231 sfree (key_copy);
232 ERROR ("uc_insert: cache_alloc (%i) failed.", ds->ds_num);
233 return (-1);
234 }
236 sstrncpy (ce->name, key, sizeof (ce->name));
238 for (i = 0; i < ds->ds_num; i++)
239 {
240 switch (ds->ds[i].type)
241 {
242 case DS_TYPE_COUNTER:
243 ce->values_gauge[i] = NAN;
244 ce->values_raw[i].counter = vl->values[i].counter;
245 break;
247 case DS_TYPE_GAUGE:
248 ce->values_gauge[i] = vl->values[i].gauge;
249 ce->values_raw[i].gauge = vl->values[i].gauge;
250 break;
252 case DS_TYPE_DERIVE:
253 ce->values_gauge[i] = NAN;
254 ce->values_raw[i].derive = vl->values[i].derive;
255 break;
257 case DS_TYPE_ABSOLUTE:
258 ce->values_gauge[i] = NAN;
259 if (vl->interval > 0)
260 ce->values_gauge[i] = ((double) vl->values[i].absolute)
261 / ((double) vl->interval);
262 ce->values_raw[i].absolute = vl->values[i].absolute;
263 break;
265 default:
266 /* This shouldn't happen. */
267 ERROR ("uc_insert: Don't know how to handle data source type %i.",
268 ds->ds[i].type);
269 return (-1);
270 } /* switch (ds->ds[i].type) */
271 } /* for (i) */
273 /* Prune invalid gauge data */
274 uc_check_range (ds, ce);
276 ce->last_time = vl->time;
277 ce->last_update = time (NULL);
278 ce->interval = vl->interval;
279 ce->state = STATE_OKAY;
281 if (c_avl_insert (cache_tree, key_copy, ce) != 0)
282 {
283 sfree (key_copy);
284 ERROR ("uc_insert: c_avl_insert failed.");
285 return (-1);
286 }
288 DEBUG ("uc_insert: Added %s to the cache.", key);
289 return (0);
290 } /* int uc_insert */
292 int uc_init (void)
293 {
294 if (cache_tree == NULL)
295 cache_tree = c_avl_create ((int (*) (const void *, const void *))
296 cache_compare);
298 return (0);
299 } /* int uc_init */
301 int uc_check_timeout (void)
302 {
303 time_t now;
304 cache_entry_t *ce;
306 char **keys = NULL;
307 int keys_len = 0;
309 char *key;
310 c_avl_iterator_t *iter;
311 int i;
313 pthread_mutex_lock (&cache_lock);
315 now = time (NULL);
317 /* Build a list of entries to be flushed */
318 iter = c_avl_get_iterator (cache_tree);
319 while (c_avl_iterator_next (iter, (void *) &key, (void *) &ce) == 0)
320 {
321 /* If entry has not been updated, add to `keys' array */
322 if ((now - ce->last_update) >= (timeout_g * ce->interval))
323 {
324 char **tmp;
326 tmp = (char **) realloc ((void *) keys,
327 (keys_len + 1) * sizeof (char *));
328 if (tmp == NULL)
329 {
330 ERROR ("uc_check_timeout: realloc failed.");
331 c_avl_iterator_destroy (iter);
332 sfree (keys);
333 pthread_mutex_unlock (&cache_lock);
334 return (-1);
335 }
337 keys = tmp;
338 keys[keys_len] = strdup (key);
339 if (keys[keys_len] == NULL)
340 {
341 ERROR ("uc_check_timeout: strdup failed.");
342 continue;
343 }
344 keys_len++;
345 }
346 } /* while (c_avl_iterator_next) */
348 ce = NULL;
350 for (i = 0; i < keys_len; i++)
351 {
352 int status;
354 status = ut_check_interesting (keys[i]);
356 if (status < 0)
357 {
358 ERROR ("uc_check_timeout: ut_check_interesting failed.");
359 sfree (keys[i]);
360 continue;
361 }
362 else if (status == 0) /* ``service'' is uninteresting */
363 {
364 DEBUG ("uc_check_timeout: %s is missing but ``uninteresting''",
365 keys[i]);
366 ce = NULL;
367 status = c_avl_remove (cache_tree, keys[i],
368 (void *) &key, (void *) &ce);
369 if (status != 0)
370 {
371 ERROR ("uc_check_timeout: c_avl_remove (%s) failed.", keys[i]);
372 }
373 sfree (keys[i]);
374 sfree (key);
375 if (ce != NULL)
376 cache_free (ce);
377 continue;
378 }
380 /* If we get here, the value is ``interesting''. Query the record from the
381 * cache and update the state field. */
382 if (c_avl_get (cache_tree, keys[i], (void *) &ce) != 0)
383 {
384 ERROR ("uc_check_timeout: cannot get data for %s from cache", keys[i]);
385 /* Do not free `keys[i]' so a notification is sent further down. */
386 continue;
387 }
388 assert (ce != NULL);
390 if (status == 2) /* persist */
391 {
392 DEBUG ("uc_check_timeout: %s is missing, sending notification.",
393 keys[i]);
394 ce->state = STATE_MISSING;
395 /* Do not free `keys[i]' so a notification is sent further down. */
396 }
397 else if (status == 1) /* do not persist */
398 {
399 if (ce->state == STATE_MISSING)
400 {
401 DEBUG ("uc_check_timeout: %s is missing but "
402 "notification has already been sent.",
403 keys[i]);
404 /* Set `keys[i]' to NULL to no notification is sent. */
405 sfree (keys[i]);
406 }
407 else /* (ce->state != STATE_MISSING) */
408 {
409 DEBUG ("uc_check_timeout: %s is missing, sending one notification.",
410 keys[i]);
411 ce->state = STATE_MISSING;
412 /* Do not free `keys[i]' so a notification is sent further down. */
413 }
414 }
415 else
416 {
417 WARNING ("uc_check_timeout: ut_check_interesting (%s) returned "
418 "invalid status %i.",
419 keys[i], status);
420 sfree (keys[i]);
421 }
423 /* Make really sure the next iteration doesn't work with this pointer.
424 * There have been too many bugs in the past.. :/ -- octo */
425 ce = NULL;
426 } /* for (keys[i]) */
428 c_avl_iterator_destroy (iter);
430 pthread_mutex_unlock (&cache_lock);
432 for (i = 0; i < keys_len; i++)
433 {
434 if (keys[i] == NULL)
435 continue;
437 uc_send_notification (keys[i]);
438 sfree (keys[i]);
439 }
441 sfree (keys);
443 return (0);
444 } /* int uc_check_timeout */
446 int uc_update (const data_set_t *ds, const value_list_t *vl)
447 {
448 char name[6 * DATA_MAX_NAME_LEN];
449 cache_entry_t *ce = NULL;
450 int send_okay_notification = 0;
451 time_t update_delay = 0;
452 notification_t n;
453 int status;
454 int i;
456 if (FORMAT_VL (name, sizeof (name), vl) != 0)
457 {
458 ERROR ("uc_update: FORMAT_VL failed.");
459 return (-1);
460 }
462 pthread_mutex_lock (&cache_lock);
464 status = c_avl_get (cache_tree, name, (void *) &ce);
465 if (status != 0) /* entry does not yet exist */
466 {
467 status = uc_insert (ds, vl, name);
468 pthread_mutex_unlock (&cache_lock);
469 return (status);
470 }
472 assert (ce != NULL);
473 assert (ce->values_num == ds->ds_num);
475 if (ce->last_time >= vl->time)
476 {
477 pthread_mutex_unlock (&cache_lock);
478 NOTICE ("uc_update: Value too old: name = %s; value time = %u; "
479 "last cache update = %u;",
480 name, (unsigned int) vl->time, (unsigned int) ce->last_time);
481 return (-1);
482 }
484 /* Send a notification (after the lock has been released) if we switch the
485 * state from something else to `okay'. */
486 if (ce->state == STATE_MISSING)
487 {
488 send_okay_notification = 1;
489 ce->state = STATE_OKAY;
490 update_delay = time (NULL) - ce->last_update;
491 }
493 for (i = 0; i < ds->ds_num; i++)
494 {
495 switch (ds->ds[i].type)
496 {
497 case DS_TYPE_COUNTER:
498 {
499 counter_t diff;
501 /* check if the counter has wrapped around */
502 if (vl->values[i].counter < ce->values_raw[i].counter)
503 {
504 if (ce->values_raw[i].counter <= 4294967295U)
505 diff = (4294967295U - ce->values_raw[i].counter)
506 + vl->values[i].counter;
507 else
508 diff = (18446744073709551615ULL - ce->values_raw[i].counter)
509 + vl->values[i].counter;
510 }
511 else /* counter has NOT wrapped around */
512 {
513 diff = vl->values[i].counter - ce->values_raw[i].counter;
514 }
516 ce->values_gauge[i] = ((double) diff)
517 / ((double) (vl->time - ce->last_time));
518 ce->values_raw[i].counter = vl->values[i].counter;
519 }
520 break;
522 case DS_TYPE_GAUGE:
523 ce->values_raw[i].gauge = vl->values[i].gauge;
524 ce->values_gauge[i] = vl->values[i].gauge;
525 break;
527 case DS_TYPE_DERIVE:
528 {
529 derive_t diff;
531 diff = vl->values[i].derive - ce->values_raw[i].derive;
533 ce->values_gauge[i] = ((double) diff)
534 / ((double) (vl->time - ce->last_time));
535 ce->values_raw[i].derive = vl->values[i].derive;
536 }
537 break;
539 case DS_TYPE_ABSOLUTE:
540 ce->values_gauge[i] = ((double) vl->values[i].absolute)
541 / ((double) (vl->time - ce->last_time));
542 ce->values_raw[i].absolute = vl->values[i].absolute;
543 break;
545 default:
546 /* This shouldn't happen. */
547 pthread_mutex_unlock (&cache_lock);
548 ERROR ("uc_update: Don't know how to handle data source type %i.",
549 ds->ds[i].type);
550 return (-1);
551 } /* switch (ds->ds[i].type) */
553 DEBUG ("uc_update: %s: ds[%i] = %lf", name, i, ce->values_gauge[i]);
554 } /* for (i) */
556 /* Update the history if it exists. */
557 if (ce->history != NULL)
558 {
559 assert (ce->history_index < ce->history_length);
560 for (i = 0; i < ce->values_num; i++)
561 {
562 size_t hist_idx = (ce->values_num * ce->history_index) + i;
563 ce->history[hist_idx] = ce->values_gauge[i];
564 }
566 assert (ce->history_length > 0);
567 ce->history_index = (ce->history_index + 1) % ce->history_length;
568 }
570 /* Prune invalid gauge data */
571 uc_check_range (ds, ce);
573 ce->last_time = vl->time;
574 ce->last_update = time (NULL);
575 ce->interval = vl->interval;
577 pthread_mutex_unlock (&cache_lock);
579 if (send_okay_notification == 0)
580 return (0);
582 /* Do not send okay notifications for uninteresting values, i. e. values for
583 * which no threshold is configured. */
584 status = ut_check_interesting (name);
585 if (status <= 0)
586 return (0);
588 /* Initialize the notification */
589 memset (&n, '\0', sizeof (n));
590 NOTIFICATION_INIT_VL (&n, vl, ds);
592 n.severity = NOTIF_OKAY;
593 n.time = vl->time;
595 ssnprintf (n.message, sizeof (n.message),
596 "Received a value for %s. It was missing for %u seconds.",
597 name, (unsigned int) update_delay);
599 plugin_dispatch_notification (&n);
601 return (0);
602 } /* int uc_update */
604 int uc_get_rate_by_name (const char *name, gauge_t **ret_values, size_t *ret_values_num)
605 {
606 gauge_t *ret = NULL;
607 size_t ret_num = 0;
608 cache_entry_t *ce = NULL;
609 int status = 0;
611 pthread_mutex_lock (&cache_lock);
613 if (c_avl_get (cache_tree, name, (void *) &ce) == 0)
614 {
615 assert (ce != NULL);
617 /* remove missing values from getval */
618 if (ce->state == STATE_MISSING)
619 {
620 status = -1;
621 }
622 else
623 {
624 ret_num = ce->values_num;
625 ret = (gauge_t *) malloc (ret_num * sizeof (gauge_t));
626 if (ret == NULL)
627 {
628 ERROR ("utils_cache: uc_get_rate_by_name: malloc failed.");
629 status = -1;
630 }
631 else
632 {
633 memcpy (ret, ce->values_gauge, ret_num * sizeof (gauge_t));
634 }
635 }
636 }
637 else
638 {
639 DEBUG ("utils_cache: uc_get_rate_by_name: No such value: %s", name);
640 status = -1;
641 }
643 pthread_mutex_unlock (&cache_lock);
645 if (status == 0)
646 {
647 *ret_values = ret;
648 *ret_values_num = ret_num;
649 }
651 return (status);
652 } /* gauge_t *uc_get_rate_by_name */
654 gauge_t *uc_get_rate (const data_set_t *ds, const value_list_t *vl)
655 {
656 char name[6 * DATA_MAX_NAME_LEN];
657 gauge_t *ret = NULL;
658 size_t ret_num = 0;
659 int status;
661 if (FORMAT_VL (name, sizeof (name), vl) != 0)
662 {
663 ERROR ("utils_cache: uc_get_rate: FORMAT_VL failed.");
664 return (NULL);
665 }
667 status = uc_get_rate_by_name (name, &ret, &ret_num);
668 if (status != 0)
669 return (NULL);
671 /* This is important - the caller has no other way of knowing how many
672 * values are returned. */
673 if (ret_num != (size_t) ds->ds_num)
674 {
675 ERROR ("utils_cache: uc_get_rate: ds[%s] has %i values, "
676 "but uc_get_rate_by_name returned %zu.",
677 ds->type, ds->ds_num, ret_num);
678 sfree (ret);
679 return (NULL);
680 }
682 return (ret);
683 } /* gauge_t *uc_get_rate */
685 int uc_get_names (char ***ret_names, time_t **ret_times, size_t *ret_number)
686 {
687 c_avl_iterator_t *iter;
688 char *key;
689 cache_entry_t *value;
691 char **names = NULL;
692 time_t *times = NULL;
693 size_t number = 0;
695 int status = 0;
697 if ((ret_names == NULL) || (ret_number == NULL))
698 return (-1);
700 pthread_mutex_lock (&cache_lock);
702 iter = c_avl_get_iterator (cache_tree);
703 while (c_avl_iterator_next (iter, (void *) &key, (void *) &value) == 0)
704 {
705 char **temp;
707 /* remove missing values when list values */
708 if (value->state == STATE_MISSING)
709 continue;
711 if (ret_times != NULL)
712 {
713 time_t *tmp_times;
715 tmp_times = (time_t *) realloc (times, sizeof (time_t) * (number + 1));
716 if (tmp_times == NULL)
717 {
718 status = -1;
719 break;
720 }
721 times = tmp_times;
722 times[number] = value->last_time;
723 }
725 temp = (char **) realloc (names, sizeof (char *) * (number + 1));
726 if (temp == NULL)
727 {
728 status = -1;
729 break;
730 }
731 names = temp;
732 names[number] = strdup (key);
733 if (names[number] == NULL)
734 {
735 status = -1;
736 break;
737 }
738 number++;
739 } /* while (c_avl_iterator_next) */
741 c_avl_iterator_destroy (iter);
742 pthread_mutex_unlock (&cache_lock);
744 if (status != 0)
745 {
746 size_t i;
748 for (i = 0; i < number; i++)
749 {
750 sfree (names[i]);
751 }
752 sfree (names);
754 return (-1);
755 }
757 *ret_names = names;
758 if (ret_times != NULL)
759 *ret_times = times;
760 *ret_number = number;
762 return (0);
763 } /* int uc_get_names */
765 int uc_get_state (const data_set_t *ds, const value_list_t *vl)
766 {
767 char name[6 * DATA_MAX_NAME_LEN];
768 cache_entry_t *ce = NULL;
769 int ret = STATE_ERROR;
771 if (FORMAT_VL (name, sizeof (name), vl) != 0)
772 {
773 ERROR ("uc_get_state: FORMAT_VL failed.");
774 return (STATE_ERROR);
775 }
777 pthread_mutex_lock (&cache_lock);
779 if (c_avl_get (cache_tree, name, (void *) &ce) == 0)
780 {
781 assert (ce != NULL);
782 ret = ce->state;
783 }
785 pthread_mutex_unlock (&cache_lock);
787 return (ret);
788 } /* int uc_get_state */
790 int uc_set_state (const data_set_t *ds, const value_list_t *vl, int state)
791 {
792 char name[6 * DATA_MAX_NAME_LEN];
793 cache_entry_t *ce = NULL;
794 int ret = -1;
796 if (FORMAT_VL (name, sizeof (name), vl) != 0)
797 {
798 ERROR ("uc_get_state: FORMAT_VL failed.");
799 return (STATE_ERROR);
800 }
802 pthread_mutex_lock (&cache_lock);
804 if (c_avl_get (cache_tree, name, (void *) &ce) == 0)
805 {
806 assert (ce != NULL);
807 ret = ce->state;
808 ce->state = state;
809 }
811 pthread_mutex_unlock (&cache_lock);
813 return (ret);
814 } /* int uc_set_state */
816 int uc_get_history_by_name (const char *name,
817 gauge_t *ret_history, size_t num_steps, size_t num_ds)
818 {
819 cache_entry_t *ce = NULL;
820 size_t i;
821 int status = 0;
823 pthread_mutex_lock (&cache_lock);
825 status = c_avl_get (cache_tree, name, (void *) &ce);
826 if (status != 0)
827 {
828 pthread_mutex_unlock (&cache_lock);
829 return (-ENOENT);
830 }
832 if (((size_t) ce->values_num) != num_ds)
833 {
834 pthread_mutex_unlock (&cache_lock);
835 return (-EINVAL);
836 }
838 /* Check if there are enough values available. If not, increase the buffer
839 * size. */
840 if (ce->history_length < num_steps)
841 {
842 gauge_t *tmp;
843 size_t i;
845 tmp = realloc (ce->history, sizeof (*ce->history)
846 * num_steps * ce->values_num);
847 if (tmp == NULL)
848 {
849 pthread_mutex_unlock (&cache_lock);
850 return (-ENOMEM);
851 }
853 for (i = ce->history_length * ce->values_num;
854 i < (num_steps * ce->values_num);
855 i++)
856 tmp[i] = NAN;
858 ce->history = tmp;
859 ce->history_length = num_steps;
860 } /* if (ce->history_length < num_steps) */
862 /* Copy the values to the output buffer. */
863 for (i = 0; i < num_steps; i++)
864 {
865 size_t src_index;
866 size_t dst_index;
868 if (i < ce->history_index)
869 src_index = ce->history_index - (i + 1);
870 else
871 src_index = ce->history_length + ce->history_index - (i + 1);
872 src_index = src_index * num_ds;
874 dst_index = i * num_ds;
876 memcpy (ret_history + dst_index, ce->history + src_index,
877 sizeof (*ret_history) * num_ds);
878 }
880 pthread_mutex_unlock (&cache_lock);
882 return (0);
883 } /* int uc_get_history_by_name */
885 int uc_get_history (const data_set_t *ds, const value_list_t *vl,
886 gauge_t *ret_history, size_t num_steps, size_t num_ds)
887 {
888 char name[6 * DATA_MAX_NAME_LEN];
890 if (FORMAT_VL (name, sizeof (name), vl) != 0)
891 {
892 ERROR ("utils_cache: uc_get_history: FORMAT_VL failed.");
893 return (-1);
894 }
896 return (uc_get_history_by_name (name, ret_history, num_steps, num_ds));
897 } /* int uc_get_history */
899 int uc_get_hits (const data_set_t *ds, const value_list_t *vl)
900 {
901 char name[6 * DATA_MAX_NAME_LEN];
902 cache_entry_t *ce = NULL;
903 int ret = STATE_ERROR;
905 if (FORMAT_VL (name, sizeof (name), vl) != 0)
906 {
907 ERROR ("uc_get_state: FORMAT_VL failed.");
908 return (STATE_ERROR);
909 }
911 pthread_mutex_lock (&cache_lock);
913 if (c_avl_get (cache_tree, name, (void *) &ce) == 0)
914 {
915 assert (ce != NULL);
916 ret = ce->hits;
917 }
919 pthread_mutex_unlock (&cache_lock);
921 return (ret);
922 } /* int uc_get_hits */
924 int uc_set_hits (const data_set_t *ds, const value_list_t *vl, int hits)
925 {
926 char name[6 * DATA_MAX_NAME_LEN];
927 cache_entry_t *ce = NULL;
928 int ret = -1;
930 if (FORMAT_VL (name, sizeof (name), vl) != 0)
931 {
932 ERROR ("uc_get_state: FORMAT_VL failed.");
933 return (STATE_ERROR);
934 }
936 pthread_mutex_lock (&cache_lock);
938 if (c_avl_get (cache_tree, name, (void *) &ce) == 0)
939 {
940 assert (ce != NULL);
941 ret = ce->hits;
942 ce->hits = hits;
943 }
945 pthread_mutex_unlock (&cache_lock);
947 return (ret);
948 } /* int uc_set_hits */
950 int uc_inc_hits (const data_set_t *ds, const value_list_t *vl, int step)
951 {
952 char name[6 * DATA_MAX_NAME_LEN];
953 cache_entry_t *ce = NULL;
954 int ret = -1;
956 if (FORMAT_VL (name, sizeof (name), vl) != 0)
957 {
958 ERROR ("uc_get_state: FORMAT_VL failed.");
959 return (STATE_ERROR);
960 }
962 pthread_mutex_lock (&cache_lock);
964 if (c_avl_get (cache_tree, name, (void *) &ce) == 0)
965 {
966 assert (ce != NULL);
967 ret = ce->hits;
968 ce->hits = ret + step;
969 }
971 pthread_mutex_unlock (&cache_lock);
973 return (ret);
974 } /* int uc_inc_hits */
976 /*
977 * Meta data interface
978 */
979 /* XXX: This function will acquire `cache_lock' but will not free it! */
980 static meta_data_t *uc_get_meta (const value_list_t *vl) /* {{{ */
981 {
982 char name[6 * DATA_MAX_NAME_LEN];
983 cache_entry_t *ce = NULL;
984 int status;
986 status = FORMAT_VL (name, sizeof (name), vl);
987 if (status != 0)
988 {
989 ERROR ("utils_cache: uc_get_meta: FORMAT_VL failed.");
990 return (NULL);
991 }
993 pthread_mutex_lock (&cache_lock);
995 status = c_avl_get (cache_tree, name, (void *) &ce);
996 if (status != 0)
997 {
998 pthread_mutex_unlock (&cache_lock);
999 return (NULL);
1000 }
1001 assert (ce != NULL);
1003 if (ce->meta == NULL)
1004 ce->meta = meta_data_create ();
1006 if (ce->meta == NULL)
1007 pthread_mutex_unlock (&cache_lock);
1009 return (ce->meta);
1010 } /* }}} meta_data_t *uc_get_meta */
1012 /* Sorry about this preprocessor magic, but it really makes this file much
1013 * shorter.. */
1014 #define UC_WRAP(wrap_function) { \
1015 meta_data_t *meta; \
1016 int status; \
1017 meta = uc_get_meta (vl); \
1018 if (meta == NULL) return (-1); \
1019 status = wrap_function (meta, key); \
1020 pthread_mutex_unlock (&cache_lock); \
1021 return (status); \
1022 }
1023 int uc_meta_data_exists (const value_list_t *vl, const char *key)
1024 UC_WRAP (meta_data_exists)
1026 int uc_meta_data_delete (const value_list_t *vl, const char *key)
1027 UC_WRAP (meta_data_delete)
1028 #undef UC_WRAP
1030 /* We need a new version of this macro because the following functions take
1031 * two argumetns. */
1032 #define UC_WRAP(wrap_function) { \
1033 meta_data_t *meta; \
1034 int status; \
1035 meta = uc_get_meta (vl); \
1036 if (meta == NULL) return (-1); \
1037 status = wrap_function (meta, key, value); \
1038 pthread_mutex_unlock (&cache_lock); \
1039 return (status); \
1040 }
1041 int uc_meta_data_add_string (const value_list_t *vl,
1042 const char *key,
1043 const char *value)
1044 UC_WRAP(meta_data_add_string)
1045 int uc_meta_data_add_signed_int (const value_list_t *vl,
1046 const char *key,
1047 int64_t value)
1048 UC_WRAP(meta_data_add_signed_int)
1049 int uc_meta_data_add_unsigned_int (const value_list_t *vl,
1050 const char *key,
1051 uint64_t value)
1052 UC_WRAP(meta_data_add_unsigned_int)
1053 int uc_meta_data_add_double (const value_list_t *vl,
1054 const char *key,
1055 double value)
1056 UC_WRAP(meta_data_add_double)
1057 int uc_meta_data_add_boolean (const value_list_t *vl,
1058 const char *key,
1059 _Bool value)
1060 UC_WRAP(meta_data_add_boolean)
1062 int uc_meta_data_get_string (const value_list_t *vl,
1063 const char *key,
1064 char **value)
1065 UC_WRAP(meta_data_get_string)
1066 int uc_meta_data_get_signed_int (const value_list_t *vl,
1067 const char *key,
1068 int64_t *value)
1069 UC_WRAP(meta_data_get_signed_int)
1070 int uc_meta_data_get_unsigned_int (const value_list_t *vl,
1071 const char *key,
1072 uint64_t *value)
1073 UC_WRAP(meta_data_get_unsigned_int)
1074 int uc_meta_data_get_double (const value_list_t *vl,
1075 const char *key,
1076 double *value)
1077 UC_WRAP(meta_data_get_double)
1078 int uc_meta_data_get_boolean (const value_list_t *vl,
1079 const char *key,
1080 _Bool *value)
1081 UC_WRAP(meta_data_get_boolean)
1082 #undef UC_WRAP
1084 /* vim: set sw=2 ts=8 sts=2 tw=78 : */