X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fdp_rrdtool.c;h=b72e512c9fe9a2a570050889470d35cfb045cf15;hb=6b69fb3290de5d4208dc751f2fdbf8a37212fed8;hp=79c8e4187a7937fe5cc3c0083e82c89df434fceb;hpb=d586e8f07c42e4ee8dc6144fe2a0be8d69dc8000;p=collection4.git diff --git a/src/dp_rrdtool.c b/src/dp_rrdtool.c index 79c8e41..b72e512 100644 --- a/src/dp_rrdtool.c +++ b/src/dp_rrdtool.c @@ -27,12 +27,16 @@ #include "data_provider.h" #include "filesystem.h" #include "oconfig.h" +#include "common.h" #include #include #include #include #include +#include + +#include struct dp_rrdtool_s { @@ -130,6 +134,50 @@ static int scan_host_cb (const char *base_dir, return (fs_foreach_dir (abs_dir, scan_plugin_cb, data)); } /* }}} int scan_host_cb */ +static int ident_to_rrdfile (const graph_ident_t *ident, /* {{{ */ + dp_rrdtool_t *config, + char *buffer, size_t buffer_size) +{ + const char *plugin_instance; + const char *type_instance; + + plugin_instance = ident_get_plugin_instance (ident); + if ((plugin_instance != NULL) && (plugin_instance[0] == 0)) + plugin_instance = NULL; + + type_instance = ident_get_type_instance (ident); + if ((type_instance != NULL) && (type_instance[0] == 0)) + type_instance = NULL; + + buffer[0] = 0; + + strlcat (buffer, config->data_dir, buffer_size); + strlcat (buffer, "/", buffer_size); + + strlcat (buffer, ident_get_host (ident), buffer_size); + strlcat (buffer, "/", buffer_size); + strlcat (buffer, ident_get_plugin (ident), buffer_size); + if (plugin_instance != NULL) + { + strlcat (buffer, "-", buffer_size); + strlcat (buffer, plugin_instance, buffer_size); + } + strlcat (buffer, "/", buffer_size); + strlcat (buffer, ident_get_type (ident), buffer_size); + if (type_instance != NULL) + { + strlcat (buffer, "-", buffer_size); + strlcat (buffer, type_instance, buffer_size); + } + + strlcat (buffer, ".rrd", buffer_size); + + return (0); +} /* }}} int ident_to_rrdfile */ + +/* + * Callback functions + */ static int get_idents (void *priv, dp_get_idents_callback cb, void *ud) { /* {{{ */ @@ -152,11 +200,68 @@ static int get_idents (void *priv, static int get_ident_ds_names (void *priv, graph_ident_t *ident, dp_list_get_ident_ds_names_callback cb, void *ud) { /* {{{ */ - priv = NULL; - ident = NULL; - cb = NULL; - ud = NULL; - return (0); + dp_rrdtool_t *config = priv; + char file[PATH_MAX + 1]; + int status; + + char *rrd_argv[] = { "info", file, NULL }; + int rrd_argc = (sizeof (rrd_argv) / sizeof (rrd_argv[0])) - 1; + + rrd_info_t *info; + rrd_info_t *ptr; + + memset (file, 0, sizeof (file)); + status = ident_to_rrdfile (ident, config, file, sizeof (file)); + if (status != 0) + return (status); + + info = rrd_info (rrd_argc, rrd_argv); + if (info == NULL) + { + printf ("%s: rrd_info (%s) failed.\n", __func__, file); + return (-1); + } + + for (ptr = info; ptr != NULL; ptr = ptr->next) + { + size_t keylen; + size_t dslen; + char *ds; + + if (ptr->key[0] != 'd') + continue; + + if (strncmp ("ds[", ptr->key, strlen ("ds[")) != 0) + continue; + + keylen = strlen (ptr->key); + if (keylen < strlen ("ds[?].index")) + continue; + + dslen = keylen - strlen ("ds[].index"); + assert (dslen >= 1); + + if (strcmp ("].index", ptr->key + (strlen ("ds[") + dslen)) != 0) + continue; + + ds = malloc (dslen + 1); + if (ds == NULL) + continue; + + memcpy (ds, ptr->key + strlen ("ds["), dslen); + ds[dslen] = 0; + + status = (*cb) (ident, ds, ud); + + free (ds); + + if (status != 0) + break; + } + + rrd_info_free (info); + + return (status); } /* }}} int get_ident_ds_names */ static int get_ident_data (void *priv, @@ -166,16 +271,74 @@ static int get_ident_data (void *priv, { /* {{{ */ dp_rrdtool_t *config = priv; - ident = NULL; - ds_name = NULL; - begin.tv_sec = 0; - end.tv_sec = 0; - cb = NULL; - ud = NULL; + char filename[PATH_MAX + 1]; + const char *cf = "AVERAGE"; /* FIXME */ + time_t rrd_start; + time_t rrd_end; + unsigned long step; + unsigned long ds_count; + char **ds_namv; + rrd_value_t *data; + int status; + + unsigned long ds_index; + unsigned long data_index; + unsigned long data_length; + + status = ident_to_rrdfile (ident, config, filename, sizeof (filename)); + if (status != 0) + return (status); + + rrd_start = (time_t) begin.tv_sec; + rrd_end = (time_t) end.tv_sec; + step = 0; + ds_count = 0; + ds_namv = NULL; + data = NULL; + + status = rrd_fetch_r (filename, cf, + &rrd_start, &rrd_end, + &step, &ds_count, &ds_namv, + &data); + if (status != 0) + return (status); + +#define BAIL_OUT(ret_status) do { \ + unsigned long i; \ + for (i = 0; i < ds_count; i++) \ + free (ds_namv[i]); \ + free (ds_namv); \ + free (data); \ + return (ret_status); \ +} while (0) + + for (ds_index = 0; ds_index < ds_count; ds_index++) + if (strcmp (ds_name, ds_namv[ds_index]) == 0) + break; + + if (ds_index >= ds_count) + BAIL_OUT (ENOENT); + + /* Number of data points returned. */ + data_length = (rrd_end - rrd_start) / step; + + for (data_index = 0; data_index < data_length; data_index++) + { + dp_data_point_t dp; + unsigned long index = (ds_count * data_index) + ds_index; - config = NULL; + memset (&dp, 0, sizeof (dp)); + dp.time.tv_sec = rrd_start + (data_index * step); + dp.time.tv_nsec = 0; + dp.value = (double) data[index]; + + status = (*cb) (ident, ds_name, &dp, ud); + if (status != 0) + BAIL_OUT (status); + } - return (EINVAL); + BAIL_OUT (0); +#undef BAIL_OUT } /* }}} int get_ident_data */ static int print_graph (void *priv, @@ -188,7 +351,7 @@ static int print_graph (void *priv, return (-1); } /* }}} int print_graph */ -int dp_rrdtool_config (oconfig_item_t *ci) +int dp_rrdtool_config (const oconfig_item_t *ci) { /* {{{ */ dp_rrdtool_t *conf;