rcpp_framework/database/sqlite/sqlite3_query_result.cpp

#include "sqlite3_query_result.h"

#include <cstdio>

bool Sqlite3QueryResult::next_row() {
    return ++current_row < rows.size();
}

const char* Sqlite3QueryResult::get_cell(const int index) {
    return rows[current_row]->cells[index].c_str();
}

void Sqlite3QueryResult::query(const std::string &query, sqlite3 *conn) {
    if (sqlite3_exec(conn, query.c_str(), Sqlite3QueryResult::run_query_finished, this, &err_msg) != SQLITE_OK) {
        printf("SQLite3Database::query error: \nQuery: %s \n Error:\n %s\n", query.c_str(), err_msg);
        sqlite3_free(err_msg);
    }
}

int Sqlite3QueryResult::run_query_finished(void *data, int argc, char **argv, char **col_names) {
    Sqlite3QueryResult *res = reinterpret_cast<Sqlite3QueryResult*>(data);

    //res->col_names = col_names;

    Sqlite3QueryResultRow *r = new Sqlite3QueryResultRow();

    for (int i = 0; i < argc; ++i) {
        r->cells.push_back(argv[i]);
    }
    
    res->rows.push_back(r);

    return 0;
}

Sqlite3QueryResult::Sqlite3QueryResult() : QueryResult() {
    err_msg = nullptr;
    current_row = -1;
}

Sqlite3QueryResult::~Sqlite3QueryResult() {
    for (int i = 0; i < rows.size(); ++i) {
        delete rows[i];
    }
}