![[tokkee]](http://tokkee.org/images/avatar.png){kind=avatar}
index db24903f5513785852a22748f05de88b1a4b0f91..8ab7f69e20b5f728a1d439922c1a22d849e7eaa5 100644 (file)
/*
* SysDB - src/plugins/timeseries/rrdtool.c
- * Copyright (C) 2014 Sebastian 'tokkee' Harl <sh@tokkee.org>
+ * Copyright (C) 2014-2016 Sebastian 'tokkee' Harl <sh@tokkee.org>
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
#include <errno.h>
#include <stdlib.h>
+#include <string.h>
+#include <strings.h>
#include <rrd.h>
#ifdef HAVE_RRD_CLIENT_H
# include <rrd_client.h>
/* Current versions of RRDtool do not support multiple RRDCacheD client
* connections. Use this to guard against multiple configured RRDCacheD
* instances. */
-static _Bool rrdcached_in_use = 0;
+static bool rrdcached_in_use = 0;
+
+static bool
+rrdcached_connect(char *addr)
+{
+#ifdef HAVE_RRD_CLIENT_H
+ rrd_clear_error();
+ if (! rrdc_is_connected(addr)) {
+ if (rrdc_connect(addr)) {
+ sdb_log(SDB_LOG_ERR, "Failed to connectd to RRDCacheD at %s: %s",
+ addr, rrd_get_error());
+ return 0;
+ }
+ }
+#else
+ sdb_log(SDB_LOG_ERR, "Callback called with RRDCacheD address "
+ "but your build of SysDB does not support that");
+ return 0;
+#endif
+ return 1;
+} /* rrdcached_connect */
/*
* plugin API
*/
+static sdb_timeseries_info_t *
+sdb_rrd_describe(const char *id, sdb_object_t *user_data)
+{
+ rrd_info_t *info, *iter;
+ char filename[strlen(id) + 1];
+ sdb_timeseries_info_t *ts_info;
+
+ strncpy(filename, id, sizeof(filename));
+
+ if (user_data) {
+ /* -> use RRDCacheD */
+ char *addr = SDB_OBJ_WRAPPER(user_data)->data;
+
+ if (! rrdcached_connect(addr))
+ return NULL;
+
+#ifdef HAVE_RRD_CLIENT_H
+ /* TODO: detect and use rrdc_info if possible */
+ sdb_log(SDB_LOG_ERR, "DESCRIBE not yet supported via RRDCacheD");
+ return NULL;
+#endif
+ }
+ else {
+ rrd_clear_error();
+ info = rrd_info_r(filename);
+ }
+ if (! info) {
+ sdb_log(SDB_LOG_ERR, "Failed to extract header information from '%s': %s",
+ filename, rrd_get_error());
+ return NULL;
+ }
+
+ ts_info = calloc(1, sizeof(*ts_info));
+ if (! ts_info) {
+ sdb_log(SDB_LOG_ERR, "Failed to allocate memory");
+ rrd_info_free(info);
+ return NULL;
+ }
+
+ for (iter = info; iter != NULL; iter = iter->next) {
+ size_t len, n, m;
+ char *ds_name;
+ char **tmp;
+
+ /* Parse the DS name. The raw value is not exposed via the rrd_info
+ * interface. */
+ n = strlen("ds[");
+ if (strncmp(iter->key, "ds[", n))
+ continue;
+
+ len = strlen(iter->key);
+ m = strlen("].index");
+ if ((len < m) || strcmp(iter->key + len - m, "].index"))
+ continue;
+
+ ds_name = iter->key + n;
+ len -= n;
+ ds_name[len - m] = '\0';
+
+ /* Append the new datum. */
+ tmp = realloc(ts_info->data_names,
+ (ts_info->data_names_len + 1) * sizeof(*ts_info->data_names));
+ if (! tmp) {
+ sdb_log(SDB_LOG_ERR, "Failed to allocate memory");
+ sdb_timeseries_info_destroy(ts_info);
+ rrd_info_free(info);
+ return NULL;
+ }
+
+ ts_info->data_names = tmp;
+ ts_info->data_names[ts_info->data_names_len] = strdup(ds_name);
+ if (! ts_info->data_names[ts_info->data_names_len]) {
+ sdb_log(SDB_LOG_ERR, "Failed to allocate memory");
+ sdb_timeseries_info_destroy(ts_info);
+ rrd_info_free(info);
+ return NULL;
+ }
+ ts_info->data_names_len++;
+ }
+ rrd_info_free(info);
+
+ return ts_info;
+} /* sdb_rrd_describe */
+
static sdb_timeseries_t *
sdb_rrd_fetch(const char *id, sdb_timeseries_opts_t *opts,
sdb_object_t *user_data)
rrd_value_t *data = NULL, *data_ptr;
if (user_data) {
-#ifdef HAVE_RRD_CLIENT_H
/* -> use RRDCacheD */
char *addr = SDB_OBJ_WRAPPER(user_data)->data;
- rrd_clear_error();
- if (! rrdc_is_connected(addr)) {
- if (rrdc_connect(addr)) {
- sdb_log(SDB_LOG_ERR, "timeseries::rrdtool: Failed to "
- "connectd to RRDCacheD at %s: %s",
- addr, rrd_get_error());
- return NULL;
- }
- }
+ if (! rrdcached_connect(addr))
+ return NULL;
+#ifdef HAVE_RRD_CLIENT_H
if (rrdc_flush(id)) {
- sdb_log(SDB_LOG_ERR, "timeseries::rrdtool: Failed to flush "
- "'%s' through RRDCacheD: %s", id, rrd_get_error());
+ sdb_log(SDB_LOG_ERR, "Failed to flush '%s' through RRDCacheD: %s",
+ id, rrd_get_error());
return NULL;
}
-#else
- sdb_log(SDB_LOG_ERR, "timeseries::rrdtool: Callback called with "
- "RRDCacheD address but your build of SysDB does not support "
- "that");
- return NULL;
#endif
}
rrd_freemem(data); \
} while (0)
+ /* limit to about 1000 data-points for now
+ * TODO: make this configurable */
+ step = (end - start) / 1000;
+
if (rrd_fetch_r(id, "AVERAGE", &start, &end, &step,
&ds_cnt, &ds_namv, &data)) {
char errbuf[1024];
sdb_strerror(errno, errbuf, sizeof(errbuf));
- sdb_log(SDB_LOG_ERR, "rrdtool plugin: Failed to fetch data "
- "from %s: %s", id, errbuf);
+ sdb_log(SDB_LOG_ERR, "Failed to fetch data from %s: %s", id, errbuf);
return NULL;
}
if (! ts) {
char errbuf[1024];
sdb_strerror(errno, errbuf, sizeof(errbuf));
- sdb_log(SDB_LOG_ERR, "rrdtool plugin: Failed to allocate "
- "time-series object: %s", errbuf);
+ sdb_log(SDB_LOG_ERR, "Failed to allocate time-series object: %s", errbuf);
FREE_RRD_DATA();
return NULL;
}
return ts;
} /* sdb_rrd_fetch */
+static sdb_timeseries_fetcher_t fetcher_impl = {
+ sdb_rrd_describe, sdb_rrd_fetch,
+};
+
static int
sdb_rrdcached_shutdown(sdb_object_t __attribute__((unused)) *user_data)
{
char *addr = NULL;
if (rrdcached_in_use) {
- sdb_log(SDB_LOG_ERR, "timeseries::rrdtool: RRDCacheD does "
- "not support multiple connections");
+ sdb_log(SDB_LOG_ERR, "RRDCacheD does not support multiple connections");
return -1;
}
#ifndef HAVE_RRD_CLIENT_H
- sdb_log(SDB_LOG_ERR, "timeseries::rrdtool: RRDCacheD client "
- "support not available in your SysDB build");
+ sdb_log(SDB_LOG_ERR, "RRDCacheD client support not available in your SysDB build");
return -1;
#else
if (oconfig_get_string(ci, &addr)) {
- sdb_log(SDB_LOG_ERR, "timeseries::unixsock: RRDCacheD requires "
- "a single string argument\n\tUsage <RRDCacheD ADDR>");
+ sdb_log(SDB_LOG_ERR, "RRDCacheD requires a single string argument\n"
+ "\tUsage <RRDCacheD ADDR>");
return -1;
}
if ((*addr != '/') && strncmp(addr, "unix:", strlen("unix:"))) {
/* XXX: add (optional) support for rrdc_fetch if available */
- sdb_log(SDB_LOG_ERR, "timeseries::unixsock: RRDCacheD only "
- "supports local (UNIX socket) addresses");
+ sdb_log(SDB_LOG_ERR, "RRDCacheD only supports local (UNIX socket) addresses");
return -1;
}
addr = strdup(addr);
if (! addr) {
char errbuf[1024];
- sdb_log(SDB_LOG_ERR, "timeseries::unixsock: Failed to duplicate "
- "string: %s", sdb_strerror(errno, errbuf, sizeof(errbuf)));
+ sdb_log(SDB_LOG_ERR, "Failed to duplicate string: %s",
+ sdb_strerror(errno, errbuf, sizeof(errbuf)));
return -1;
}
if (ci->children_num)
- sdb_log(SDB_LOG_WARNING, "timeseries::unixsock: RRDCacheD does "
- "not support any child config options");
+ sdb_log(SDB_LOG_WARNING, "RRDCacheD does not support any child config options");
ud = sdb_object_create_wrapper("rrdcached-addr", addr, free);
if (! ud) {
char errbuf[1024];
- sdb_log(SDB_LOG_ERR, "timeseries::unixsock: Failed to create "
- "user-data object: %s",
+ sdb_log(SDB_LOG_ERR, "Failed to create user-data object: %s",
sdb_strerror(errno, errbuf, sizeof(errbuf)));
free(addr);
return -1;
}
- sdb_plugin_register_ts_fetcher("rrdcached", sdb_rrd_fetch, ud);
+ sdb_plugin_register_timeseries_fetcher("rrdcached", &fetcher_impl, ud);
sdb_plugin_register_shutdown("rrdcached", sdb_rrdcached_shutdown, NULL);
sdb_object_deref(ud);
rrdcached_in_use = 1;
if (! strcasecmp(child->key, "RRDCacheD"))
sdb_rrd_config_rrdcached(child);
else
- sdb_log(SDB_LOG_WARNING, "timeseries::rrdtool: Ignoring "
- "unknown config option '%s'.", child->key);
+ sdb_log(SDB_LOG_WARNING, "Ignoring unknown config option '%s'.", child->key);
}
return 0;
} /* sdb_rrd_config */
sdb_plugin_set_info(info, SDB_PLUGIN_INFO_VERSION, SDB_VERSION);
sdb_plugin_set_info(info, SDB_PLUGIN_INFO_PLUGIN_VERSION, SDB_VERSION);
- sdb_plugin_register_ts_fetcher("rrdtool", sdb_rrd_fetch, NULL);
+ sdb_plugin_register_timeseries_fetcher("rrdtool", &fetcher_impl, NULL);
sdb_plugin_register_config(sdb_rrd_config);
return 0;
} /* sdb_module_init */