X-Git-Url: https://git.tokkee.org/?p=sysdb.git;a=blobdiff_plain;f=src%2Ftools%2Fsysdb%2Fcommand.c;h=61960be83505ca167e4045f6a225854b9aeb0946;hp=c7ca310a086230268272b46dcb7c268733b106b8;hb=0a3dd5b9b97e25156412a95bcecf25f8d75c72fc;hpb=9d80882865895b9eaebc56e878ad5c3c27f8dddb diff --git a/src/tools/sysdb/command.c b/src/tools/sysdb/command.c index c7ca310..61960be 100644 --- a/src/tools/sysdb/command.c +++ b/src/tools/sysdb/command.c @@ -33,6 +33,7 @@ #include "tools/sysdb/command.h" #include "tools/sysdb/input.h" +#include "tools/sysdb/json.h" #include "frontend/proto.h" #include "utils/error.h" @@ -47,7 +48,7 @@ #include static void -ok_printer(sdb_strbuf_t *buf) +ok_printer(sdb_input_t __attribute__((unused)) *input, sdb_strbuf_t *buf) { const char *msg = sdb_strbuf_string(buf); if (msg && *msg) @@ -57,7 +58,7 @@ ok_printer(sdb_strbuf_t *buf) } /* ok_printer */ static void -log_printer(sdb_strbuf_t *buf) +log_printer(sdb_input_t __attribute__((unused)) *input, sdb_strbuf_t *buf) { uint32_t prio = 0; @@ -72,9 +73,10 @@ log_printer(sdb_strbuf_t *buf) } /* log_printer */ static void -data_printer(sdb_strbuf_t *buf) +data_printer(sdb_input_t __attribute__((unused)) *input, sdb_strbuf_t *buf) { size_t len = sdb_strbuf_len(buf); + uint32_t type = 0; if ((! len) || (len == sizeof(uint32_t))) { /* empty command or empty reply */ @@ -86,15 +88,16 @@ data_printer(sdb_strbuf_t *buf) return; } - /* At the moment, we don't care about the result type. We simply print the - * result without further parsing it. */ + sdb_proto_unmarshal_int32(SDB_STRBUF_STR(buf), &type); sdb_strbuf_skip(buf, 0, sizeof(uint32_t)); - printf("%s\n", sdb_strbuf_string(buf)); + if (sdb_json_print(input, (int)type, buf)) + sdb_log(SDB_LOG_ERR, "Failed to print result"); + printf("\n"); } /* data_printer */ static struct { int status; - void (*printer)(sdb_strbuf_t *); + void (*printer)(sdb_input_t *, sdb_strbuf_t *); } response_printers[] = { { SDB_CONNECTION_OK, ok_printer }, { SDB_CONNECTION_LOG, log_printer }, @@ -115,7 +118,7 @@ clear_query(sdb_input_t *input) */ int -sdb_command_print_reply(sdb_client_t *client) +sdb_command_print_reply(sdb_input_t *input) { sdb_strbuf_t *recv_buf; const char *result; @@ -128,10 +131,10 @@ sdb_command_print_reply(sdb_client_t *client) if (! recv_buf) return -1; - if (sdb_client_recv(client, &rcode, recv_buf) < 0) + if (sdb_client_recv(input->client, &rcode, recv_buf) < 0) rcode = UINT32_MAX; - if (sdb_client_eof(client)) { + if (sdb_client_eof(input->client)) { sdb_strbuf_destroy(recv_buf); return -1; } @@ -141,7 +144,7 @@ sdb_command_print_reply(sdb_client_t *client) for (i = 0; i < SDB_STATIC_ARRAY_LEN(response_printers); ++i) { if (status == response_printers[i].status) { - response_printers[i].printer(recv_buf); + response_printers[i].printer(input, recv_buf); sdb_strbuf_destroy(recv_buf); return status; } @@ -200,7 +203,7 @@ sdb_command_exec(sdb_input_t *input) /* The server may send back log messages but will eventually reply to the * query, which is either DATA or ERROR. */ while (42) { - int status = sdb_command_print_reply(input->client); + int status = sdb_command_print_reply(input); if (status < 0) { sdb_log(SDB_LOG_ERR, "Failed to read reply from server"); break;