mirror of
https://github.com/tstack/lnav.git
synced 2024-09-11 13:05:51 +03:00
79 lines
1.7 KiB
C++
79 lines
1.7 KiB
C++
#include <iostream>
|
|
|
|
#include <assert.h>
|
|
#include <sqlite3.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "auto_mem.hh"
|
|
#include "base/injector.hh"
|
|
#include "regexp_vtab.hh"
|
|
#include "sqlite-extension-func.hh"
|
|
#include "xpath_vtab.hh"
|
|
|
|
struct callback_state {
|
|
int cs_row;
|
|
};
|
|
|
|
static int
|
|
sql_callback(void* ptr, int ncols, char** colvalues, char** colnames)
|
|
{
|
|
struct callback_state* cs = (struct callback_state*) ptr;
|
|
|
|
printf("Row %d:\n", cs->cs_row);
|
|
for (int lpc = 0; lpc < ncols; lpc++) {
|
|
printf(" Column %10s: %s\n", colnames[lpc], colvalues[lpc]);
|
|
}
|
|
|
|
cs->cs_row += 1;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
main(int argc, char* argv[])
|
|
{
|
|
int retval = EXIT_SUCCESS;
|
|
auto_mem<sqlite3> db(sqlite3_close);
|
|
std::string stmt;
|
|
|
|
log_argv(argc, argv);
|
|
|
|
if (argc == 2) {
|
|
stmt = argv[1];
|
|
} else {
|
|
std::getline(std::cin, stmt, '\0');
|
|
}
|
|
|
|
if (sqlite3_open(":memory:", db.out()) != SQLITE_OK) {
|
|
fprintf(stderr, "error: unable to make sqlite memory database\n");
|
|
retval = EXIT_FAILURE;
|
|
} else {
|
|
auto_mem<char> errmsg(sqlite3_free);
|
|
struct callback_state state;
|
|
|
|
memset(&state, 0, sizeof(state));
|
|
|
|
{
|
|
int register_collation_functions(sqlite3 * db);
|
|
|
|
register_sqlite_funcs(db.in(), sqlite_registration_funcs);
|
|
register_collation_functions(db.in());
|
|
}
|
|
|
|
register_regexp_vtab(db.in());
|
|
register_xpath_vtab(db.in());
|
|
|
|
if (sqlite3_exec(
|
|
db.in(), stmt.c_str(), sql_callback, &state, errmsg.out())
|
|
!= SQLITE_OK)
|
|
{
|
|
fprintf(stderr, "error: sqlite3_exec failed -- %s\n", errmsg.in());
|
|
retval = EXIT_FAILURE;
|
|
}
|
|
}
|
|
|
|
return retval;
|
|
}
|