X-Git-Url: https://git.octo.it/?p=collection4.git;a=blobdiff_plain;f=src%2Fdp_rrdtool.c;h=f90b79e913a979194117c81078492ab8b54c7cce;hp=79c8e4187a7937fe5cc3c0083e82c89df434fceb;hb=2ef734476e3511b6f1c2611875e340b6b8ca2d4c;hpb=d586e8f07c42e4ee8dc6144fe2a0be8d69dc8000 diff --git a/src/dp_rrdtool.c b/src/dp_rrdtool.c index 79c8e41..f90b79e 100644 --- a/src/dp_rrdtool.c +++ b/src/dp_rrdtool.c @@ -23,16 +23,19 @@ #include "graph_types.h" #include "graph_ident.h" -#include "graph_list.h" #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 +133,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 +199,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 +270,81 @@ 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; + + dp_data_point_t *dp = NULL; + size_t dp_num = 0; + + 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); \ + free (dp); \ + 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; + + dp_num = (size_t) data_length; + dp = calloc (dp_num, sizeof (*dp)); + if (dp == NULL) + BAIL_OUT (ENOMEM); + + for (data_index = 0; data_index < data_length; data_index++) + { + unsigned long index = (ds_count * data_index) + ds_index; + + dp[data_index].time.tv_sec = rrd_start + (data_index * step); + dp[data_index].time.tv_nsec = 0; + dp[data_index].value = (double) data[index]; + } - config = NULL; + status = (*cb) (ident, ds_name, dp, dp_num, 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 +357,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; @@ -208,7 +377,7 @@ int dp_rrdtool_config (oconfig_item_t *ci) dp.private_data = conf; - gl_register_data_provider ("rrdtool", &dp); + data_provider_register ("rrdtool", &dp); return (0); } /* }}} int dp_rrdtool_config */