summaryrefslogtreecommitdiff
path: root/libsqlitepp/selectcommand.cpp
blob: a3d5ac75277e40b904b721f209adccbee5db9b8e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
#include "selectcommand.h"
#include "connection.h"
#include "error.h"
#include <string.h>

class Column : public DB::Column {
	public:
		Column(const Glib::ustring & n, unsigned int i, sqlite3_stmt * s) :
			DB::Column(n, i),
			stmt(s)
		{
		}

		bool isNull() const {
			return (SQLITE_NULL == sqlite3_column_type(stmt, colNo));
		}

		void apply(DB::HandleField & h) const {
			switch (sqlite3_column_type(stmt, colNo)) {
				case SQLITE_INTEGER:
					h.integer(sqlite3_column_int64(stmt, colNo));
					return;
				case SQLITE_FLOAT:
					h.floatingpoint(sqlite3_column_double(stmt, colNo));
					return;
				case SQLITE_TEXT:
					{
						auto t = sqlite3_column_text(stmt, colNo);
						auto l = sqlite3_column_bytes(stmt, colNo);
						h.string(reinterpret_cast<const char *>(t), l);
						return;
					}
				case SQLITE_NULL:
					h.null();
					return;
				case SQLITE_BLOB:
					throw std::runtime_error("Blobs not supported");
			}
			
		}

		void rebind(DB::Command*, unsigned int) const {
			throw std::runtime_error("Not implemented");
		}

	private:
		sqlite3_stmt * const stmt;
};

SQLite::SelectCommand::SelectCommand(const Connection * conn, const std::string & sql) :
	DB::Command(sql),
	DB::SelectCommand(sql),
	SQLite::Command(conn, sql)
{
}

void
SQLite::SelectCommand::execute()
{
	// No explicit execute required
}

bool
SQLite::SelectCommand::fetch()
{
	switch (sqlite3_step(stmt)) {
		case SQLITE_ROW:
			if (columns.empty()) {
				for (int c = sqlite3_data_count(stmt) - 1; c >= 0; c -= 1) {
					columns.insert(DB::ColumnPtr(new Column(sqlite3_column_name(stmt, c), c, stmt)));
				}
			}
			return true;
		case SQLITE_DONE:
			return false;
		default:
			throw Error(sqlite3_errmsg(c->db));
	}
}