diff --git a/src/core/store_exec.c b/src/core/store_exec.c
index 5daaa143105558c133f6649d8d44e7c24e0f9307..c377129b825e8c6e42a2a690e6953f3a21749337 100644 (file)
--- a/src/core/store_exec.c
+++ b/src/core/store_exec.c
#include <errno.h>
#include <arpa/inet.h>
-#include <ctype.h>
#include <stdlib.h>
#include <string.h>
* private helper functions
*/
+typedef struct {
+ sdb_store_json_formatter_t *f;
+ sdb_store_obj_t *current_host;
+} iter_t;
+
+static int
+maybe_emit_host(iter_t *iter, sdb_store_obj_t *obj)
+{
+ if ((obj->type == SDB_HOST) || (obj->type == SDB_ATTRIBUTE))
+ return 0;
+ if (iter->current_host == obj->parent)
+ return 0;
+ iter->current_host = obj->parent;
+ return sdb_store_json_emit(iter->f, obj->parent);
+} /* maybe_emit_host */
+
static int
list_tojson(sdb_store_obj_t *obj,
sdb_store_matcher_t __attribute__((unused)) *filter,
void *user_data)
{
- sdb_store_json_formatter_t *f = user_data;
- return sdb_store_json_emit(f, obj);
+ iter_t *iter = user_data;
+ maybe_emit_host(iter, obj);
+ return sdb_store_json_emit(iter->f, obj);
} /* list_tojson */
static int
lookup_tojson(sdb_store_obj_t *obj, sdb_store_matcher_t *filter,
void *user_data)
{
- sdb_store_json_formatter_t *f = user_data;
- return sdb_store_json_emit_full(f, obj, filter);
+ iter_t *iter = user_data;
+ maybe_emit_host(iter, obj);
+ return sdb_store_json_emit_full(iter->f, obj, filter);
} /* lookup_tojson */
-static size_t
-sstrlen(const char *s)
-{
- return s ? strlen(s) : 0;
-} /* sstrlen */
-
/*
* query implementations
*/
sdb_store_obj_t *obj;
sdb_store_json_formatter_t *f;
+ int status = 0;
if ((! name) || ((type == SDB_HOST) && hostname)
|| ((type != SDB_HOST) && (! hostname))) {
}
sdb_strbuf_memcpy(buf, &res_type, sizeof(uint32_t));
- if (sdb_store_json_emit_full(f, obj, filter)) {
+ if (type != SDB_HOST)
+ status = sdb_store_json_emit(f, obj->parent);
+ if (status || sdb_store_json_emit_full(f, obj, filter)) {
sdb_log(SDB_LOG_ERR, "frontend: Failed to serialize "
"%s %s.%s to JSON", SDB_STORE_TYPE_TO_NAME(type),
hostname, name);
int type, sdb_store_matcher_t *filter)
{
uint32_t res_type = htonl(SDB_CONNECTION_LIST);
- sdb_store_json_formatter_t *f;
+ iter_t iter = { NULL, NULL };
- f = sdb_store_json_formatter(buf, type, SDB_WANT_ARRAY);
- if (! f) {
+ iter.f = sdb_store_json_formatter(buf, type, SDB_WANT_ARRAY);
+ if (! iter.f) {
char err[1024];
sdb_log(SDB_LOG_ERR, "frontend: Failed to create "
"JSON formatter to handle LIST command: %s",
}
sdb_strbuf_memcpy(buf, &res_type, sizeof(uint32_t));
- if (sdb_store_scan(store, type, /* m = */ NULL, filter, list_tojson, f)) {
+ if (sdb_store_scan(store, type, /* m = */ NULL, filter, list_tojson, &iter)) {
sdb_log(SDB_LOG_ERR, "frontend: Failed to serialize "
"store to JSON");
sdb_strbuf_sprintf(errbuf, "Out of memory");
- sdb_object_deref(SDB_OBJ(f));
+ sdb_object_deref(SDB_OBJ(iter.f));
return -1;
}
- sdb_store_json_finish(f);
- sdb_object_deref(SDB_OBJ(f));
+ sdb_store_json_finish(iter.f);
+ sdb_object_deref(SDB_OBJ(iter.f));
return SDB_CONNECTION_DATA;
} /* exec_list */
int type, sdb_store_matcher_t *m, sdb_store_matcher_t *filter)
{
uint32_t res_type = htonl(SDB_CONNECTION_LOOKUP);
- sdb_store_json_formatter_t *f;
+ iter_t iter = { NULL, NULL };
- f = sdb_store_json_formatter(buf, type, SDB_WANT_ARRAY);
- if (! f) {
+ iter.f = sdb_store_json_formatter(buf, type, SDB_WANT_ARRAY);
+ if (! iter.f) {
char err[1024];
sdb_log(SDB_LOG_ERR, "frontend: Failed to create "
"JSON formatter to handle LOOKUP command: %s",
sdb_strbuf_memcpy(buf, &res_type, sizeof(uint32_t));
- if (sdb_store_scan(store, type, m, filter, lookup_tojson, f)) {
+ if (sdb_store_scan(store, type, m, filter, lookup_tojson, &iter)) {
sdb_log(SDB_LOG_ERR, "frontend: Failed to lookup %ss",
SDB_STORE_TYPE_TO_NAME(type));
sdb_strbuf_sprintf(errbuf, "Failed to lookup %ss",
SDB_STORE_TYPE_TO_NAME(type));
- sdb_object_deref(SDB_OBJ(f));
+ sdb_object_deref(SDB_OBJ(iter.f));
return -1;
}
- sdb_store_json_finish(f);
- sdb_object_deref(SDB_OBJ(f));
+ sdb_store_json_finish(iter.f);
+ sdb_object_deref(SDB_OBJ(iter.f));
return SDB_CONNECTION_DATA;
} /* exec_lookup */
-static int
-exec_store(sdb_strbuf_t *buf, sdb_strbuf_t *errbuf, sdb_ast_store_t *st)
-{
- char name[sstrlen(st->hostname) + sstrlen(st->parent) + sstrlen(st->name) + 3];
- sdb_metric_store_t metric_store;
- int type = st->obj_type, status = -1;
-
- switch (st->obj_type) {
- case SDB_HOST:
- strncpy(name, st->name, sizeof(name));
- status = sdb_plugin_store_host(st->name, st->last_update);
- break;
-
- case SDB_SERVICE:
- snprintf(name, sizeof(name), "%s.%s", st->hostname, st->name);
- status = sdb_plugin_store_service(st->hostname, st->name, st->last_update);
- break;
-
- case SDB_METRIC:
- snprintf(name, sizeof(name), "%s.%s", st->hostname, st->name);
- metric_store.type = st->store_type;
- metric_store.id = st->store_id;
- status = sdb_plugin_store_metric(st->hostname, st->name,
- &metric_store, st->last_update);
- break;
-
- case SDB_ATTRIBUTE:
- type |= st->parent_type;
-
- if (st->parent)
- snprintf(name, sizeof(name), "%s.%s.%s",
- st->hostname, st->parent, st->name);
- else
- snprintf(name, sizeof(name), "%s.%s", st->hostname, st->name);
-
- switch (st->parent_type) {
- case 0:
- type |= SDB_HOST;
- status = sdb_plugin_store_attribute(st->hostname,
- st->name, &st->value, st->last_update);
- break;
-
- case SDB_SERVICE:
- status = sdb_plugin_store_service_attribute(st->hostname, st->parent,
- st->name, &st->value, st->last_update);
- break;
-
- case SDB_METRIC:
- status = sdb_plugin_store_metric_attribute(st->hostname, st->parent,
- st->name, &st->value, st->last_update);
- break;
-
- default:
- sdb_log(SDB_LOG_ERR, "store: Invalid parent type in STORE: %s",
- SDB_STORE_TYPE_TO_NAME(st->parent_type));
- return -1;
- }
- break;
-
- default:
- sdb_log(SDB_LOG_ERR, "store: Invalid object type in STORE: %s",
- SDB_STORE_TYPE_TO_NAME(st->obj_type));
- return -1;
- }
-
- if (status < 0) {
- sdb_strbuf_sprintf(errbuf, "STORE: Failed to store %s object",
- SDB_STORE_TYPE_TO_NAME(type));
- return -1;
- }
-
- if (! status) {
- sdb_strbuf_sprintf(buf, "Successfully stored %s %s",
- SDB_STORE_TYPE_TO_NAME(type), name);
- }
- else {
- char type_str[32];
- strncpy(type_str, SDB_STORE_TYPE_TO_NAME(type), sizeof(type_str));
- type_str[0] = (char)toupper((int)type_str[0]);
- sdb_strbuf_sprintf(buf, "%s %s already up to date", type_str, name);
- }
-
- return SDB_CONNECTION_OK;
-} /* exec_store */
-
static int
exec_timeseries(sdb_store_t *store, sdb_strbuf_t *buf, sdb_strbuf_t *errbuf,
const char *hostname, const char *metric,
return exec_lookup(store, buf, errbuf, SDB_AST_LOOKUP(ast)->obj_type,
q->matcher, q->filter);
- case SDB_AST_TYPE_STORE:
- if (ast->type != SDB_AST_TYPE_STORE) {
- sdb_log(SDB_LOG_ERR, "store: Invalid AST node for STORE command: %s",
- SDB_AST_TYPE_TO_STRING(ast));
- return -1;
- }
- return exec_store(buf, errbuf, SDB_AST_STORE(ast));
-
case SDB_AST_TYPE_TIMESERIES:
ts_opts.start = SDB_AST_TIMESERIES(ast)->start;
ts_opts.end = SDB_AST_TIMESERIES(ast)->end;