summaryrefslogtreecommitdiffabout
path: root/test/sqlite.h
Side-by-side diff
Diffstat (limited to 'test/sqlite.h') (more/less context) (show whitespace changes)
-rw-r--r--test/sqlite.h71
1 files changed, 71 insertions, 0 deletions
diff --git a/test/sqlite.h b/test/sqlite.h
new file mode 100644
index 0000000..883a2c0
--- a/dev/null
+++ b/test/sqlite.h
@@ -0,0 +1,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];
+ }
+};