summaryrefslogtreecommitdiffabout
path: root/test/sqlite.h
blob: 883a2c0dee4c94fa4a2b260b85d89cc68f3b1589 (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
#include <sqlite3.h>

class sqlite3_t {
    public:
	sqlite3 *_D;

	sqlite3_t(const char *f)
	    : _D(0) {
		int r = sqlite3_open(f,&_D);
		if(r!=SQLITE_OK) {
		    string msg = sqlite3_errmsg(_D); sqlite3_close(_D);
		    throw opkele::exception(OPKELE_CP_ "Failed to open SQLite database: "+msg);
		}
	    }
	~sqlite3_t() {
	    if(_D) sqlite3_close(_D);
	}

	operator const sqlite3*(void) const { return _D; }
	operator sqlite3*(void) { return _D; }

	void exec(const char *sql) {
	    assert(_D);
	    char *errm;
	    if(sqlite3_exec(_D,sql,NULL,NULL,&errm)!=SQLITE_OK)
		throw opkele::exception(OPKELE_CP_ string("Failed to sqlite3_exec():")+errm);
	}
	void get_table(const char *sql,char ***resp,int *nr,int *nc) {
	    assert(_D);
	    char *errm;
	    if(sqlite3_get_table(_D,sql,resp,nr,nc,&errm)!=SQLITE_OK)
		throw opkele::exception(OPKELE_CP_ string("Failed to sqlite3_get_table():")+errm);
	}
};

template<typename T>
class sqlite3_mem_t {
    public:
	T _M;

	sqlite3_mem_t(T M) :_M(M) { }
	~sqlite3_mem_t() { if(_M) sqlite3_free(_M); }

	operator const T&(void) const { return _M; }
	operator T&(void) { return _M; }

	sqlite3_mem_t operator=(T M) {
	    if(_M) sqlite3_free(_M);
	    _M = M;
	}
};

class sqlite3_table_t {
    public:
	char **_T;

	sqlite3_table_t() : _T(0) { }
	sqlite3_table_t(char **T) : _T(T) { }
	~sqlite3_table_t() { if(_T) sqlite3_free_table(_T); }

	operator char**&(void) { return _T; }

	operator char ***(void) {
	    if(_T) sqlite3_free_table(_T);
	    return &_T; }

	const char *get(int r,int c,int nc) {
	    assert(_T);
	    return _T[r*nc+c];
	}
};