X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fdp_rrdtool.c;h=d7ad5ba612360bb05a3853602176ab7594b8ef55;hb=99b10b920e1e1687003edaf1f882d14b08bcd70c;hp=bbfe59e07fba9d655af79416d0d1fb662658bded;hpb=dc75a248afd3d8c35698c4ddf751565074b483f4;p=collection4.git diff --git a/src/dp_rrdtool.c b/src/dp_rrdtool.c index bbfe59e..d7ad5ba 100644 --- a/src/dp_rrdtool.c +++ b/src/dp_rrdtool.c @@ -21,13 +21,7 @@ * Florian octo Forster **/ -#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 "config.h" #include #include @@ -38,6 +32,16 @@ #include +#include "graph_types.h" +#include "graph_ident.h" +#include "data_provider.h" +#include "filesystem.h" +#include "oconfig.h" +#include "common.h" + +#include +#include + struct dp_rrdtool_s { char *data_dir; @@ -53,23 +57,27 @@ struct dp_get_idents_data_s typedef struct dp_get_idents_data_s dp_get_idents_data_t; static int scan_type_cb (__attribute__((unused)) const char *base_dir, - const char *sub_dir, void *ud) + const char *file, void *ud) { /* {{{ */ dp_get_idents_data_t *data = ud; - size_t sub_dir_len; + size_t file_len; char type_copy[1024]; + size_t type_copy_len; char *type_inst; - sub_dir_len = strlen (sub_dir); - if (sub_dir_len < 5) + file_len = strlen (file); + if (file_len < 5) return (0); /* Ignore files that don't end in ".rrd". */ - if (strcasecmp (".rrd", sub_dir + (sub_dir_len - 4)) != 0) + if (strcasecmp (".rrd", file + (file_len - 4)) != 0) return (0); - strncpy (type_copy, sub_dir, sizeof (type_copy)); - type_copy[sub_dir_len - 4] = 0; + strncpy (type_copy, file, sizeof (type_copy)); + type_copy_len = file_len - 4; + if (type_copy_len > (sizeof (type_copy) - 1)) + type_copy_len = sizeof (type_copy) - 1; + type_copy[type_copy_len] = 0; type_inst = strchr (type_copy, '-'); if (type_inst != NULL) @@ -134,7 +142,7 @@ 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, +static int ident_to_rrdfile (const graph_ident_t *ident, /* {{{ */ dp_rrdtool_t *config, char *buffer, size_t buffer_size) { @@ -218,7 +226,8 @@ static int get_ident_ds_names (void *priv, graph_ident_t *ident, info = rrd_info (rrd_argc, rrd_argv); if (info == NULL) { - printf ("%s: rrd_info (%s) failed.\n", __func__, file); + fprintf (stderr, "%s: rrd_info (%s) failed.\n", __func__, file); + fflush (stderr); return (-1); } @@ -235,13 +244,13 @@ static int get_ident_ds_names (void *priv, graph_ident_t *ident, continue; keylen = strlen (ptr->key); - if (keylen < strlen ("ds[?].index")) + if (keylen < strlen ("ds[?].type")) continue; - dslen = keylen - strlen ("ds[].index"); + dslen = keylen - strlen ("ds[].type"); assert (dslen >= 1); - if (strcmp ("].index", ptr->key + (strlen ("ds[") + dslen)) != 0) + if (strcmp ("].type", ptr->key + (strlen ("ds[") + dslen)) != 0) continue; ds = malloc (dslen + 1); @@ -271,16 +280,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; - config = NULL; + 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]; + } + + 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, @@ -293,7 +367,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; @@ -313,7 +387,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 */