src/daemon/utils_threshold.c

   1 /**
   2  * collectd - src/utils_threshold.c
   3  * Copyright (C) 2014       Pierre-Yves Ritschard
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a
   6  * copy of this software and associated documentation files (the "Software"),
   7  * to deal in the Software without restriction, including without limitation
   8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   9  * and/or sell copies of the Software, and to permit persons to whom the
  10  * Software is furnished to do so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be included in
  13  * all copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  21  * DEALINGS IN THE SOFTWARE.
  22  *
  23  * Authors:
  24  *   Pierre-Yves Ritschard <pyr at spootnik.org>
  25  **/
  26
  27 #include "collectd.h"
  28
  29 #include "common.h"
  30 #include "utils_avltree.h"
  31 #include "utils_threshold.h"
  32
  33 #include <pthread.h>
  34
  35 /*
  36  * Exported symbols
  37  * {{{ */
  38 c_avl_tree_t   *threshold_tree = NULL;
  39 pthread_mutex_t threshold_lock = PTHREAD_MUTEX_INITIALIZER;
  40 /* }}} */
  41
  42 /*
  43  * threshold_t *threshold_get
  44  *
  45  * Retrieve one specific threshold configuration. For looking up a threshold
  46  * matching a value_list_t, see "threshold_search" below. Returns NULL if the
  47  * specified threshold doesn't exist.
  48  */
  49 threshold_t *threshold_get (const char *hostname,
  50     const char *plugin, const char *plugin_instance,
  51     const char *type, const char *type_instance)
  52 { /* {{{ */
  53   char name[6 * DATA_MAX_NAME_LEN];
  54   threshold_t *th = NULL;
  55
  56   format_name (name, sizeof (name),
  57       (hostname == NULL) ? "" : hostname,
  58       (plugin == NULL) ? "" : plugin, plugin_instance,
  59       (type == NULL) ? "" : type, type_instance);
  60   name[sizeof (name) - 1] = '\0';
  61
  62   if (c_avl_get (threshold_tree, name, (void *) &th) == 0)
  63     return (th);
  64   else
  65     return (NULL);
  66 } /* }}} threshold_t *threshold_get */
  67
  68 /*
  69  * threshold_t *threshold_search
  70  *
  71  * Searches for a threshold configuration using all the possible variations of
  72  * "Host", "Plugin" and "Type" blocks. Returns NULL if no threshold could be
  73  * found.
  74  * XXX: This is likely the least efficient function in collectd.
  75  */
  76 threshold_t *threshold_search (const value_list_t *vl)
  77 { /* {{{ */
  78   threshold_t *th;
  79
  80   if ((th = threshold_get (vl->host, vl->plugin, vl->plugin_instance,
  81           vl->type, vl->type_instance)) != NULL)
  82     return (th);
  83   else if ((th = threshold_get (vl->host, vl->plugin, vl->plugin_instance,
  84           vl->type, NULL)) != NULL)
  85     return (th);
  86   else if ((th = threshold_get (vl->host, vl->plugin, NULL,
  87           vl->type, vl->type_instance)) != NULL)
  88     return (th);
  89   else if ((th = threshold_get (vl->host, vl->plugin, NULL,
  90           vl->type, NULL)) != NULL)
  91     return (th);
  92   else if ((th = threshold_get (vl->host, "", NULL,
  93           vl->type, vl->type_instance)) != NULL)
  94     return (th);
  95   else if ((th = threshold_get (vl->host, "", NULL,
  96           vl->type, NULL)) != NULL)
  97     return (th);
  98   else if ((th = threshold_get ("", vl->plugin, vl->plugin_instance,
  99           vl->type, vl->type_instance)) != NULL)
 100     return (th);
 101   else if ((th = threshold_get ("", vl->plugin, vl->plugin_instance,
 102           vl->type, NULL)) != NULL)
 103     return (th);
 104   else if ((th = threshold_get ("", vl->plugin, NULL,
 105           vl->type, vl->type_instance)) != NULL)
 106     return (th);
 107   else if ((th = threshold_get ("", vl->plugin, NULL,
 108           vl->type, NULL)) != NULL)
 109     return (th);
 110   else if ((th = threshold_get ("", "", NULL,
 111           vl->type, vl->type_instance)) != NULL)
 112     return (th);
 113   else if ((th = threshold_get ("", "", NULL,
 114           vl->type, NULL)) != NULL)
 115     return (th);
 116
 117   return (NULL);
 118 } /* }}} threshold_t *threshold_search */
 119
 120 int ut_search_threshold (const value_list_t *vl, /* {{{ */
 121     threshold_t *ret_threshold)
 122 {
 123   threshold_t *t;
 124
 125   if (vl == NULL)
 126     return (EINVAL);
 127
 128   /* Is this lock really necessary? */
 129   pthread_mutex_lock (&threshold_lock);
 130   t = threshold_search (vl);
 131   if (t == NULL) {
 132     pthread_mutex_unlock (&threshold_lock);
 133     return (ENOENT);
 134   }
 135
 136   memcpy (ret_threshold, t, sizeof (*ret_threshold));
 137   pthread_mutex_unlock (&threshold_lock);
 138
 139   ret_threshold->next = NULL;
 140
 141   return (0);
 142 } /* }}} int ut_search_threshold */
 143
 144