From: Florian Forster Date: Sun, 6 Sep 2009 08:54:01 +0000 (+0200) Subject: unixsock plugin: Implement the “GETTHRESHOLD” command. X-Git-Tag: collectd-4.8.0~11 X-Git-Url: https://git.octo.it/?p=collectd.git;a=commitdiff_plain;h=f4dbf81c3905ad55805a4c8bf7362e2594c7ec98 unixsock plugin: Implement the “GETTHRESHOLD” command. Example session: -> GETTHRESHOLD "leeloo.lan.home.verplant.org/cpu-3/cpu-idle" <- 6 Threshold found <- Type: cpu <- Type Instance: idle <- Warning Min: 5 <- Warning Max: 100 <- Failure Min: 1 <- Failure Max: 120 --- diff --git a/src/Makefile.am b/src/Makefile.am index 25711d6c..e91df833 100644 --- a/src/Makefile.am +++ b/src/Makefile.am @@ -972,6 +972,7 @@ pkglib_LTLIBRARIES += unixsock.la unixsock_la_SOURCES = unixsock.c \ utils_cmd_flush.h utils_cmd_flush.c \ utils_cmd_getval.h utils_cmd_getval.c \ + utils_cmd_getthreshold.h utils_cmd_getthreshold.c \ utils_cmd_listval.h utils_cmd_listval.c \ utils_cmd_putval.h utils_cmd_putval.c \ utils_cmd_putnotif.h utils_cmd_putnotif.c diff --git a/src/unixsock.c b/src/unixsock.c index 57f34501..be36abc8 100644 --- a/src/unixsock.c +++ b/src/unixsock.c @@ -26,6 +26,7 @@ #include "utils_cmd_flush.h" #include "utils_cmd_getval.h" +#include "utils_cmd_getthreshold.h" #include "utils_cmd_listval.h" #include "utils_cmd_putval.h" #include "utils_cmd_putnotif.h" @@ -241,6 +242,10 @@ static void *us_handle_client (void *arg) { handle_getval (fhout, buffer); } + else if (strcasecmp (fields[0], "getthreshold") == 0) + { + handle_getthreshold (fhout, buffer); + } else if (strcasecmp (fields[0], "putval") == 0) { handle_putval (fhout, buffer); diff --git a/src/utils_cmd_getthreshold.c b/src/utils_cmd_getthreshold.c new file mode 100644 index 00000000..e8c29fa3 --- /dev/null +++ b/src/utils_cmd_getthreshold.c @@ -0,0 +1,178 @@ +/** + * collectd - src/utils_cms_getthreshold.c + * Copyright (C) 2008,2009 Florian octo Forster + * + * This program is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License as published by the + * Free Software Foundation; only version 2 of the License is applicable. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program; if not, write to the Free Software Foundation, Inc., + * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + * + * Author: + * Florian octo Forster + **/ + +#include "collectd.h" +#include "common.h" +#include "plugin.h" + +#include "utils_threshold.h" +#include "utils_parse_option.h" /* for `parse_string' */ +#include "utils_cmd_getthreshold.h" + +#define print_to_socket(fh, ...) \ + if (fprintf (fh, __VA_ARGS__) < 0) { \ + char errbuf[1024]; \ + WARNING ("handle_getthreshold: failed to write to socket #%i: %s", \ + fileno (fh), sstrerror (errno, errbuf, sizeof (errbuf))); \ + return -1; \ + } + +int handle_getthreshold (FILE *fh, char *buffer) +{ + char *command; + char *identifier; + char *identifier_copy; + + char *host; + char *plugin; + char *plugin_instance; + char *type; + char *type_instance; + + value_list_t vl; + threshold_t threshold; + + int status; + size_t i; + + if ((fh == NULL) || (buffer == NULL)) + return (-1); + + DEBUG ("utils_cmd_getthreshold: handle_getthreshold (fh = %p, buffer = %s);", + (void *) fh, buffer); + + command = NULL; + status = parse_string (&buffer, &command); + if (status != 0) + { + print_to_socket (fh, "-1 Cannot parse command.\n"); + return (-1); + } + assert (command != NULL); + + if (strcasecmp ("GETTHRESHOLD", command) != 0) + { + print_to_socket (fh, "-1 Unexpected command: `%s'.\n", command); + return (-1); + } + + identifier = NULL; + status = parse_string (&buffer, &identifier); + if (status != 0) + { + print_to_socket (fh, "-1 Cannot parse identifier.\n"); + return (-1); + } + assert (identifier != NULL); + + if (*buffer != 0) + { + print_to_socket (fh, "-1 Garbage after end of command: %s\n", buffer); + return (-1); + } + + /* parse_identifier() modifies its first argument, + * returning pointers into it */ + identifier_copy = sstrdup (identifier); + + status = parse_identifier (identifier_copy, &host, + &plugin, &plugin_instance, + &type, &type_instance); + if (status != 0) + { + DEBUG ("handle_getthreshold: Cannot parse identifier `%s'.", identifier); + print_to_socket (fh, "-1 Cannot parse identifier `%s'.\n", identifier); + sfree (identifier_copy); + return (-1); + } + + memset (&vl, 0, sizeof (vl)); + sstrncpy (vl.host, host, sizeof (vl.host)); + sstrncpy (vl.plugin, plugin, sizeof (vl.plugin)); + if (plugin_instance != NULL) + sstrncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance)); + sstrncpy (vl.type, type, sizeof (vl.type)); + if (type_instance != NULL) + sstrncpy (vl.type_instance, type_instance, sizeof (vl.type_instance)); + sfree (identifier_copy); + + memset (&threshold, 0, sizeof (threshold)); + status = ut_search_threshold (&vl, &threshold); + if (status == ENOENT) + { + print_to_socket (fh, "-1 No threshold found for identifier %s\n", + identifier); + return (0); + } + else if (status != 0) + { + print_to_socket (fh, "-1 Error while looking up threshold: %i\n", + status); + return (-1); + } + + /* Lets count the number of lines we'll return. */ + i = 0; + if (threshold.host[0] != 0) i++; + if (threshold.plugin[0] != 0) i++; + if (threshold.plugin_instance[0] != 0) i++; + if (threshold.type[0] != 0) i++; + if (threshold.type_instance[0] != 0) i++; + if (threshold.data_source[0] != 0) i++; + if (!isnan (threshold.warning_min)) i++; + if (!isnan (threshold.warning_max)) i++; + if (!isnan (threshold.failure_min)) i++; + if (!isnan (threshold.failure_max)) i++; + if (threshold.hysteresis > 0.0) i++; + if (threshold.hits > 1) i++; + + /* Print the response */ + print_to_socket (fh, "%zu Threshold found\n", i); + + if (threshold.host[0] != 0) + print_to_socket (fh, "Host: %s\n", threshold.host) + if (threshold.plugin[0] != 0) + print_to_socket (fh, "Plugin: %s\n", threshold.plugin) + if (threshold.plugin_instance[0] != 0) + print_to_socket (fh, "Plugin Instance: %s\n", threshold.plugin_instance) + if (threshold.type[0] != 0) + print_to_socket (fh, "Type: %s\n", threshold.type) + if (threshold.type_instance[0] != 0) + print_to_socket (fh, "Type Instance: %s\n", threshold.type_instance) + if (threshold.data_source[0] != 0) + print_to_socket (fh, "Data Source: %s\n", threshold.data_source) + if (!isnan (threshold.warning_min)) + print_to_socket (fh, "Warning Min: %g\n", threshold.warning_min) + if (!isnan (threshold.warning_max)) + print_to_socket (fh, "Warning Max: %g\n", threshold.warning_max) + if (!isnan (threshold.failure_min)) + print_to_socket (fh, "Failure Min: %g\n", threshold.failure_min) + if (!isnan (threshold.failure_max)) + print_to_socket (fh, "Failure Max: %g\n", threshold.failure_max) + if (threshold.hysteresis > 0.0) + print_to_socket (fh, "Hysteresis: %g\n", threshold.hysteresis) + if (threshold.hits > 1) + print_to_socket (fh, "Hits: %i\n", threshold.hits) + + return (0); +} /* int handle_getthreshold */ + +/* vim: set sw=2 sts=2 ts=8 et : */ diff --git a/src/utils_cmd_getthreshold.h b/src/utils_cmd_getthreshold.h new file mode 100644 index 00000000..5481cfd9 --- /dev/null +++ b/src/utils_cmd_getthreshold.h @@ -0,0 +1,31 @@ +/** + * collectd - src/utils_cmd_getthreshold.h + * Copyright (C) 2009 Florian octo Forster + * + * This program is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License as published by the + * Free Software Foundation; only version 2 of the License is applicable. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program; if not, write to the Free Software Foundation, Inc., + * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + * + * Author: + * Florian octo Forster + **/ + +#ifndef UTILS_CMD_GETTHRESHOLD_H +#define UTILS_CMD_GETTHRESHOLD_H 1 + +#include + +int handle_getthreshold (FILE *fh, char *buffer); + +#endif /* UTILS_CMD_GETTHRESHOLD_H */ + +/* vim: set sw=2 sts=2 ts=8 : */