2017-04-16 16:00:44 +08:00
|
|
|
#include "sqlite_wrapper.h"
|
|
|
|
#include <cstring>
|
|
|
|
#include <cstdio>
|
|
|
|
|
2017-04-16 17:32:39 +08:00
|
|
|
// private
|
|
|
|
void SQLStatement::_set(sqlite3_stmt* p)
|
2017-04-16 16:00:44 +08:00
|
|
|
{
|
2017-04-16 17:32:39 +08:00
|
|
|
_st.reset(p,sqlite3_finalize);
|
2017-04-16 16:00:44 +08:00
|
|
|
}
|
|
|
|
|
2017-04-16 17:32:39 +08:00
|
|
|
// private
|
|
|
|
sqlite3_stmt* SQLStatement::_get() const
|
2017-04-16 16:00:44 +08:00
|
|
|
{
|
2017-04-16 17:32:39 +08:00
|
|
|
return _st.get();
|
2017-04-16 16:00:44 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void SQLStatement::_setStmt(sqlite3_stmt* p)
|
|
|
|
{
|
2017-04-16 17:32:39 +08:00
|
|
|
_set(p);
|
2017-04-16 16:00:44 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
sqlite3_stmt* SQLStatement::_getStmt() const
|
|
|
|
{
|
2017-04-16 17:32:39 +08:00
|
|
|
return _get();
|
2017-04-16 16:00:44 +08:00
|
|
|
}
|
|
|
|
|
2017-04-16 17:32:39 +08:00
|
|
|
bool SQLStatement::isReady() const
|
|
|
|
{
|
|
|
|
return _get()!=nullptr;
|
|
|
|
}
|
2017-04-16 16:00:44 +08:00
|
|
|
|
2017-04-16 17:32:39 +08:00
|
|
|
|
|
|
|
void SQLDB::_set(sqlite3* p)
|
2017-04-16 16:00:44 +08:00
|
|
|
{
|
2017-04-16 17:32:39 +08:00
|
|
|
_db.reset(p,sqlite3_close);
|
2017-04-16 16:00:44 +08:00
|
|
|
}
|
|
|
|
|
2017-04-16 17:32:39 +08:00
|
|
|
sqlite3* SQLDB::_get()
|
2017-04-16 16:00:44 +08:00
|
|
|
{
|
2017-04-16 17:32:39 +08:00
|
|
|
return _db.get();
|
2017-04-16 16:00:44 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
int SQLDB::open(const std::string& filename)
|
|
|
|
{
|
2017-04-16 17:32:39 +08:00
|
|
|
sqlite3* _temp=nullptr;
|
|
|
|
int ret=sqlite3_open(filename.c_str(),&_temp);
|
|
|
|
_set(_temp);
|
|
|
|
return ret;
|
2017-04-16 16:00:44 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
SQLStatement SQLDB::prepare(const std::string& SQLCommand)
|
|
|
|
{
|
|
|
|
SQLStatement stmt;
|
|
|
|
sqlite3_stmt* pstmt=nullptr;
|
2017-04-16 17:32:39 +08:00
|
|
|
int ret=sqlite3_prepare(_get(),SQLCommand.c_str(),SQLCommand.size(),&pstmt,NULL);
|
2017-04-16 16:00:44 +08:00
|
|
|
if(ret<0) return stmt;
|
|
|
|
stmt._setStmt(pstmt);
|
|
|
|
return stmt;
|
|
|
|
}
|
|
|
|
|
|
|
|
int SQLDB::step(const SQLStatement& Statement)
|
|
|
|
{
|
|
|
|
return sqlite3_step(Statement._getStmt());
|
|
|
|
}
|
|
|
|
|
|
|
|
int SQLDB::exec(const std::string& SQLCommand,SQLCallback callback,void* param)
|
|
|
|
{
|
|
|
|
char* errmsg=nullptr;
|
2017-04-16 17:32:39 +08:00
|
|
|
int ret=sqlite3_exec(_get(),SQLCommand.c_str(),callback,param,&errmsg);
|
2017-04-16 16:00:44 +08:00
|
|
|
printf("ErrMsg: %s\n",errmsg);
|
|
|
|
return ret;
|
|
|
|
}
|