src/graph_instance.[ch]: Implement "inst_data_to_json".
[collection4.git] / src / dp_rrdtool.c
index 79c8e41..f90b79e 100644 (file)
 
 #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 <stdlib.h>
 #include <stdio.h>
 #include <string.h>
 #include <limits.h>
 #include <errno.h>
+#include <assert.h>
+
+#include <rrd.h>
 
 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 */