root/software/sqlite_poc/DBManager.h @ 88cf406f
| 88cf406f | David Sorber | ||
#include <cassert>
|
|||
#include <string>
|
|||
#include <tuple>
|
|||
#include <type_traits>
|
|||
#include <vector>
|
|||
#include <sqlite3.h>
|
|||
template <size_t I>
|
|||
struct visit_impl
|
|||
{
|
|||
template <typename T, typename V>
|
|||
static void visit(T& tup, size_t idx, V value)
|
|||
{
|
|||
if (idx == I - 1)
|
|||
{
|
|||
if (std::is_same<std::get<I - 1>(tup), value>)
|
|||
{
|
|||
std::get<I - 1>(tup) = value;
|
|||
}
|
|||
else
|
|||
{
|
|||
assert(false);
|
|||
}
|
|||
}
|
|||
else
|
|||
{
|
|||
visit_impl<I - 1>::visit(tup, idx, value);
|
|||
}
|
|||
}
|
|||
};
|
|||
template <>
|
|||
struct visit_impl<0>
|
|||
{
|
|||
template <typename T, typename V>
|
|||
static void visit(T& tup, size_t idx, V fun) { assert(false); }
|
|||
};
|
|||
//~ template <typename V, typename... Ts>
|
|||
//~ void visit_at(std::tuple<Ts...> const& tup, size_t idx, V value)
|
|||
//~ {
|
|||
//~ visit_impl<sizeof...(Ts)>::visit(tup, idx, value);
|
|||
//~ }
|
|||
template <typename V, typename... Ts>
|
|||
void visit_at(std::tuple<Ts...>& tup, size_t idx, V value)
|
|||
{
|
|||
visit_impl<sizeof...(Ts)>::visit(tup, idx, value);
|
|||
}
|
|||
class DBManager
|
|||
{
|
|||
public:
|
|||
// Constructor
|
|||
DBManager(const std::string& db_path);
|
|||
// Destructor
|
|||
~DBManager();
|
|||
// execute query
|
|||
template<typename T>
|
|||
std::vector<T*>* execute_query(
|
|||
const std::string& sql_statement,
|
|||
int& return_code)
|
|||
{
|
|||
sqlite3_stmt* sqlite_statement;
|
|||
// Prepare the sqlite statement
|
|||
return_code = sqlite3_prepare_v2(db, sql_statement.c_str(), -1,
|
|||
&sqlite_statement, nullptr);
|
|||
if (return_code != SQLITE_OK)
|
|||
{
|
|||
return nullptr;
|
|||
}
|
|||
auto output_vector = new std::vector<T*>();
|
|||
int step_return = 0;
|
|||
// Iterate until we've collected all the output
|
|||
do
|
|||
{
|
|||
// Gran the
|
|||
step_return = sqlite3_step(sqlite_statement);
|
|||
// Create a new result row to be populated below
|
|||
auto result_row = new T();
|
|||
// Iterate over the columns in the row
|
|||
int num_cols = sqlite3_data_count(sqlite_statement);
|
|||
for (int col_idx = 0; col_idx < num_cols; col_idx++)
|
|||
{
|
|||
switch(sqlite3_column_type(sqlite_statement, col_idx))
|
|||
{
|
|||
case SQLITE_INTEGER:
|
|||
visit_at(*result_row, col_idx,
|
|||
sqlite3_column_int64(sqlite_statement, col_idx));
|
|||
//~ std::get<col_idx>(result_row) =
|
|||
//~ sqlite3_column_int64(sqlite_statement, col_idx);
|
|||
break;
|
|||
case SQLITE_FLOAT:
|
|||
visit_at(*result_row, col_idx, sqlite3_column_double(sqlite_statement, col_idx));
|
|||
//~ std::get<col_idx>(result_row) =
|
|||
//~ sqlite3_column_double(sqlite_statement, col_idx);
|
|||
break;
|
|||
case SQLITE_TEXT:
|
|||
visit_at(*result_row, col_idx,
|
|||
std::string(reinterpret_cast<const char*>(sqlite3_column_text(sqlite_statement, col_idx))));
|
|||
//~ std::get<col_idx>(result_row) =
|
|||
//~ std::string(sqlite3_column_text(sqlite_statement, col_idx));
|
|||
break;
|
|||
case SQLITE_BLOB:
|
|||
break;
|
|||
case SQLITE_NULL:
|
|||
break;
|
|||
}
|
|||
}
|
|||
}
|
|||
while (step_return == SQLITE_ROW);
|
|||
return output_vector;
|
|||
}
|
|||
void test();
|
|||
// templatized member function!
|
|||
template<typename T>
|
|||
std::vector<T> func(T& foo)
|
|||
{
|
|||
std::cout << "Does this work??? " << std::get<2>(foo) << std::endl;
|
|||
std::vector<T> bar;
|
|||
bar.push_back(foo);
|
|||
return bar;
|
|||
}
|
|||
private:
|
|||
sqlite3 *db;
|
|||
uint32_t numba;
|
|||
};
|