diff --git a/src/core/store_exec.c b/src/core/store_exec.c
index 6cf3a820bd3a013e770d3f172de2a1728649ff19..d4e308806bac8ece40c9a12a20885df768c2dffe 100644 (file)
--- a/src/core/store_exec.c
+++ b/src/core/store_exec.c
*/
static int
-exec_fetch(sdb_store_t *store, sdb_strbuf_t *buf, sdb_strbuf_t *errbuf,
+exec_fetch(sdb_store_t *store,
+ sdb_store_writer_t *w, sdb_object_t *wd, sdb_strbuf_t *errbuf,
int type, const char *hostname, const char *name,
sdb_store_matcher_t *filter)
{
- uint32_t res_type = htonl(SDB_CONNECTION_FETCH);
-
sdb_store_obj_t *host;
sdb_store_obj_t *obj;
- sdb_store_json_formatter_t *f;
int status = 0;
if ((! name) || ((type == SDB_HOST) && hostname)
}
host = NULL;
- f = sdb_store_json_formatter(buf, type, /* flags = */ 0);
- if (! f) {
- char err[1024];
- sdb_log(SDB_LOG_ERR, "frontend: Failed to create "
- "JSON formatter to handle FETCH command: %s",
- sdb_strerror(errno, err, sizeof(err)));
-
- sdb_strbuf_sprintf(errbuf, "Out of memory");
- sdb_object_deref(SDB_OBJ(obj));
- return -1;
- }
-
- sdb_strbuf_memcpy(buf, &res_type, sizeof(uint32_t));
if (type != SDB_HOST)
- status = sdb_store_emit(obj->parent,
- &sdb_store_json_writer, SDB_OBJ(f));
- if (status || sdb_store_emit_full(obj, filter,
- &sdb_store_json_writer, SDB_OBJ(f))) {
+ status = sdb_store_emit(obj->parent, w, wd);
+ if (status || sdb_store_emit_full(obj, filter, w, wd)) {
sdb_log(SDB_LOG_ERR, "frontend: Failed to serialize "
"%s %s.%s to JSON", SDB_STORE_TYPE_TO_NAME(type),
hostname, name);
sdb_strbuf_sprintf(errbuf, "Out of memory");
- sdb_object_deref(SDB_OBJ(f));
sdb_object_deref(SDB_OBJ(obj));
return -1;
}
sdb_object_deref(SDB_OBJ(obj));
- sdb_store_json_finish(f);
- sdb_object_deref(SDB_OBJ(f));
-
return SDB_CONNECTION_DATA;
} /* exec_fetch */
static int
-exec_list(sdb_store_t *store, sdb_strbuf_t *buf, sdb_strbuf_t *errbuf,
+exec_list(sdb_store_t *store,
+ sdb_store_writer_t *w, sdb_object_t *wd, sdb_strbuf_t *errbuf,
int type, sdb_store_matcher_t *filter)
{
- uint32_t res_type = htonl(SDB_CONNECTION_LIST);
- iter_t iter = { NULL, &sdb_store_json_writer, NULL };
- sdb_store_json_formatter_t *f;
+ iter_t iter = { NULL, w, wd };
- f = sdb_store_json_formatter(buf, type, SDB_WANT_ARRAY);
- if (! f) {
- char err[1024];
- sdb_log(SDB_LOG_ERR, "frontend: Failed to create "
- "JSON formatter to handle LIST command: %s",
- sdb_strerror(errno, err, sizeof(err)));
-
- sdb_strbuf_sprintf(errbuf, "Out of memory");
- return -1;
- }
-
- iter.wd = SDB_OBJ(f);
- sdb_strbuf_memcpy(buf, &res_type, sizeof(uint32_t));
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));
return -1;
}
- sdb_store_json_finish(f);
- sdb_object_deref(SDB_OBJ(f));
-
return SDB_CONNECTION_DATA;
} /* exec_list */
static int
-exec_lookup(sdb_store_t *store, sdb_strbuf_t *buf, sdb_strbuf_t *errbuf,
+exec_lookup(sdb_store_t *store,
+ sdb_store_writer_t *w, sdb_object_t *wd, sdb_strbuf_t *errbuf,
int type, sdb_store_matcher_t *m, sdb_store_matcher_t *filter)
{
- uint32_t res_type = htonl(SDB_CONNECTION_LOOKUP);
- iter_t iter = { NULL, &sdb_store_json_writer, NULL };
- sdb_store_json_formatter_t *f;
-
- f = sdb_store_json_formatter(buf, type, SDB_WANT_ARRAY);
- if (! f) {
- char err[1024];
- sdb_log(SDB_LOG_ERR, "frontend: Failed to create "
- "JSON formatter to handle LOOKUP command: %s",
- sdb_strerror(errno, err, sizeof(err)));
+ iter_t iter = { NULL, w, wd };
- sdb_strbuf_sprintf(errbuf, "Out of memory");
- return -1;
- }
-
- iter.wd = SDB_OBJ(f);
- sdb_strbuf_memcpy(buf, &res_type, sizeof(uint32_t));
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));
return -1;
}
- sdb_store_json_finish(f);
- sdb_object_deref(SDB_OBJ(f));
-
return SDB_CONNECTION_DATA;
} /* exec_lookup */
-static int
-exec_timeseries(sdb_store_t *store, sdb_strbuf_t *buf, sdb_strbuf_t *errbuf,
- const char *hostname, const char *metric,
- sdb_timeseries_opts_t *opts)
-{
- uint32_t res_type = htonl(SDB_CONNECTION_TIMESERIES);
-
- sdb_strbuf_memcpy(buf, &res_type, sizeof(uint32_t));
- if (sdb_store_fetch_timeseries(store, hostname, metric, opts, buf)) {
- sdb_log(SDB_LOG_ERR, "frontend: Failed to fetch time-series");
- sdb_strbuf_sprintf(errbuf, "Failed to fetch time-series");
- return -1;
- }
-
- return SDB_CONNECTION_DATA;
-} /* exec_timeseries */
-
/*
* public API
*/
int
sdb_store_query_execute(sdb_store_t *store, sdb_store_query_t *q,
- sdb_strbuf_t *buf, sdb_strbuf_t *errbuf)
+ sdb_store_writer_t *w, sdb_object_t *wd, sdb_strbuf_t *errbuf)
{
- sdb_timeseries_opts_t ts_opts;
sdb_ast_node_t *ast;
if (! q)
ast = q->ast;
switch (ast->type) {
case SDB_AST_TYPE_FETCH:
- return exec_fetch(store, buf, errbuf, SDB_AST_FETCH(ast)->obj_type,
+ return exec_fetch(store, w, wd, errbuf, SDB_AST_FETCH(ast)->obj_type,
SDB_AST_FETCH(ast)->hostname, SDB_AST_FETCH(ast)->name,
q->filter);
case SDB_AST_TYPE_LIST:
- return exec_list(store, buf, errbuf, SDB_AST_LIST(ast)->obj_type,
+ return exec_list(store, w, wd, errbuf, SDB_AST_LIST(ast)->obj_type,
q->filter);
case SDB_AST_TYPE_LOOKUP:
- return exec_lookup(store, buf, errbuf, SDB_AST_LOOKUP(ast)->obj_type,
+ return exec_lookup(store, w, wd, errbuf, SDB_AST_LOOKUP(ast)->obj_type,
q->matcher, q->filter);
- case SDB_AST_TYPE_TIMESERIES:
- ts_opts.start = SDB_AST_TIMESERIES(ast)->start;
- ts_opts.end = SDB_AST_TIMESERIES(ast)->end;
- return exec_timeseries(store, buf, errbuf,
- SDB_AST_TIMESERIES(ast)->hostname,
- SDB_AST_TIMESERIES(ast)->metric, &ts_opts);
-
default:
sdb_log(SDB_LOG_ERR, "store: Invalid query of type %s",
SDB_AST_TYPE_TO_STRING(ast));