X-Git-Url: https://git.tokkee.org/?p=sysdb.git;a=blobdiff_plain;f=src%2Ffrontend%2Fquery.c;h=56cd4ec86638ef537d8bdc09fabc15e4b90ed0b0;hp=97698fd300be6af87607c13d9b7c2336cea54b48;hb=df77814daebd2e4ff336b0f1184458fa4b01d226;hpb=4404b790e34bebc820c02d5749c84c0c975ba417 diff --git a/src/frontend/query.c b/src/frontend/query.c index 97698fd..56cd4ec 100644 --- a/src/frontend/query.c +++ b/src/frontend/query.c @@ -31,7 +31,7 @@ #include "sysdb.h" -#include "core/store.h" +#include "core/plugin.h" #include "frontend/connection-private.h" #include "parser/ast.h" #include "parser/parser.h" @@ -40,6 +40,7 @@ #include "utils/strbuf.h" #include +#include #include /* @@ -52,10 +53,100 @@ sstrdup(const char *s) return s ? strdup(s) : NULL; } /* sstrdup */ +static size_t +sstrlen(const char *s) +{ + return s ? strlen(s) : 0; +} /* sstrlen */ + static int -query_exec(sdb_conn_t *conn, sdb_ast_node_t *ast) +exec_store(sdb_ast_store_t *st, sdb_strbuf_t *buf, sdb_strbuf_t *errbuf) +{ + 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_query(sdb_conn_t *conn, sdb_ast_node_t *ast) { - sdb_store_query_t *q; sdb_strbuf_t *buf; int status; @@ -64,21 +155,15 @@ query_exec(sdb_conn_t *conn, sdb_ast_node_t *ast) return -1; } - q = sdb_store_query_prepare(ast); - if (! q) { - /* this shouldn't happen */ - sdb_strbuf_sprintf(conn->errbuf, "failed to compile AST"); - sdb_log(SDB_LOG_ERR, "frontend: failed to compile AST"); - return -1; - } - buf = sdb_strbuf_create(1024); if (! buf) { sdb_strbuf_sprintf(conn->errbuf, "Out of memory"); - sdb_object_deref(SDB_OBJ(q)); return -1; } - status = sdb_store_query_execute(q, buf, conn->errbuf); + if (ast->type == SDB_AST_TYPE_STORE) + status = exec_store(SDB_AST_STORE(ast), buf, conn->errbuf); + else + status = sdb_plugin_query(ast, buf, conn->errbuf); if (status < 0) { char query[conn->cmd_len + 1]; strncpy(query, sdb_strbuf_string(conn->buf), conn->cmd_len); @@ -90,16 +175,15 @@ query_exec(sdb_conn_t *conn, sdb_ast_node_t *ast) (uint32_t)sdb_strbuf_len(buf), sdb_strbuf_string(buf)); sdb_strbuf_destroy(buf); - sdb_object_deref(SDB_OBJ(q)); return status < 0 ? status : 0; -} /* query_exec */ +} /* exec_query */ /* * public API */ int -sdb_fe_query(sdb_conn_t *conn) +sdb_conn_query(sdb_conn_t *conn) { sdb_llist_t *parsetree; sdb_ast_node_t *ast = NULL; @@ -143,15 +227,15 @@ sdb_fe_query(sdb_conn_t *conn) } if (ast) { - status = query_exec(conn, ast); + status = exec_query(conn, ast); sdb_object_deref(SDB_OBJ(ast)); } sdb_llist_destroy(parsetree); return status; -} /* sdb_fe_query */ +} /* sdb_conn_query */ int -sdb_fe_fetch(sdb_conn_t *conn) +sdb_conn_fetch(sdb_conn_t *conn) { sdb_ast_node_t *ast; char hostname[conn->cmd_len + 1]; @@ -182,13 +266,13 @@ sdb_fe_fetch(sdb_conn_t *conn) hostname[0] ? strdup(hostname) : NULL, name[0] ? strdup(name) : NULL, /* filter = */ NULL); - status = query_exec(conn, ast); + status = exec_query(conn, ast); sdb_object_deref(SDB_OBJ(ast)); return status; -} /* sdb_fe_fetch */ +} /* sdb_conn_fetch */ int -sdb_fe_list(sdb_conn_t *conn) +sdb_conn_list(sdb_conn_t *conn) { sdb_ast_node_t *ast; uint32_t type = SDB_HOST; @@ -208,13 +292,13 @@ sdb_fe_list(sdb_conn_t *conn) } ast = sdb_ast_list_create((int)type, /* filter = */ NULL); - status = query_exec(conn, ast); + status = exec_query(conn, ast); sdb_object_deref(SDB_OBJ(ast)); return status; -} /* sdb_fe_list */ +} /* sdb_conn_list */ int -sdb_fe_lookup(sdb_conn_t *conn) +sdb_conn_lookup(sdb_conn_t *conn) { sdb_ast_node_t *ast, *m; const char *matcher; @@ -237,38 +321,30 @@ sdb_fe_lookup(sdb_conn_t *conn) matcher = sdb_strbuf_string(conn->buf) + sizeof(uint32_t); matcher_len = conn->cmd_len - sizeof(uint32_t); - m = sdb_parser_parse_conditional(matcher, (int)matcher_len, conn->errbuf); + m = sdb_parser_parse_conditional((int)type, + matcher, (int)matcher_len, conn->errbuf); if (! m) { - char expr[matcher_len + 1]; - strncpy(expr, matcher, sizeof(expr)); - expr[sizeof(expr) - 1] = '\0'; - sdb_log(SDB_LOG_ERR, "frontend: Failed to parse lookup condition '%s': %s", - expr, sdb_strbuf_string(conn->errbuf)); - return -1; - } - - ast = sdb_ast_lookup_create((int)type, m, /* filter = */ NULL); - /* run analyzer using the full context */ - if (ast && sdb_parser_analyze(ast, conn->errbuf)) { char expr[matcher_len + 1]; char err[sdb_strbuf_len(conn->errbuf) + sizeof(expr) + 64]; strncpy(expr, matcher, sizeof(expr)); expr[sizeof(expr) - 1] = '\0'; snprintf(err, sizeof(err), "Failed to parse lookup condition '%s': %s", expr, sdb_strbuf_string(conn->errbuf)); + sdb_log(SDB_LOG_ERR, "frontend: %s", err); sdb_strbuf_sprintf(conn->errbuf, "%s", err); - status = -1; + return -1; } - else - status = query_exec(conn, ast); + + ast = sdb_ast_lookup_create((int)type, m, /* filter = */ NULL); + status = exec_query(conn, ast); if (! ast) sdb_object_deref(SDB_OBJ(m)); sdb_object_deref(SDB_OBJ(ast)); return status; -} /* sdb_fe_lookup */ +} /* sdb_conn_lookup */ int -sdb_fe_store(sdb_conn_t *conn) +sdb_conn_store(sdb_conn_t *conn) { sdb_ast_node_t *ast; const char *buf = sdb_strbuf_string(conn->buf); @@ -367,10 +443,10 @@ sdb_fe_store(sdb_conn_t *conn) status = sdb_parser_analyze(ast, conn->errbuf); if (! status) - status = query_exec(conn, ast); + status = exec_query(conn, ast); sdb_object_deref(SDB_OBJ(ast)); return status; -} /* sdb_fe_store */ +} /* sdb_conn_store */ /* vim: set tw=78 sw=4 ts=4 noexpandtab : */