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];
}
};
|