96 lines
2.0 KiB
C++
96 lines
2.0 KiB
C++
#include "sqlite3_database.h"
|
|
|
|
#include "core/database/database_manager.h"
|
|
|
|
#include "sqlite3_query_builder.h"
|
|
#include "sqlite3_query_result.h"
|
|
#include "sqlite3_table_builder.h"
|
|
|
|
Database *SQLite3Database::_creation_func() {
|
|
return new SQLite3Database();
|
|
}
|
|
|
|
void SQLite3Database::_register() {
|
|
DatabaseManager::_register_db_creation_func("sqlite", SQLite3Database::_creation_func);
|
|
}
|
|
|
|
void SQLite3Database::_unregister() {
|
|
DatabaseManager::_unregister_db_creation_func("sqlite");
|
|
}
|
|
|
|
QueryBuilder *SQLite3Database::get_query_builder() {
|
|
SQLite3QueryBuilder *b = new SQLite3QueryBuilder();
|
|
b->_db = this;
|
|
|
|
return b;
|
|
}
|
|
|
|
TableBuilder *SQLite3Database::get_table_builder() {
|
|
SQLite3TableBuilder *b = new SQLite3TableBuilder();
|
|
b->_db = this;
|
|
|
|
return b;
|
|
}
|
|
|
|
void SQLite3Database::connect(const std::string &connection_str) {
|
|
int ret = sqlite3_config(SQLITE_CONFIG_MULTITHREAD);
|
|
if (ret != SQLITE_OK) {
|
|
printf("SQLITE3 multithreading is not supported!\n");
|
|
}
|
|
|
|
ret = sqlite3_open(connection_str.c_str(), &conn);
|
|
}
|
|
|
|
QueryResult *SQLite3Database::query(const std::string &query) {
|
|
Sqlite3QueryResult *res = new Sqlite3QueryResult();
|
|
|
|
res->query(query, conn);
|
|
|
|
return res;
|
|
}
|
|
|
|
void SQLite3Database::query_run(const std::string &query) {
|
|
char *err_msg;
|
|
|
|
if (sqlite3_exec(conn, query.c_str(), NULL, NULL, &err_msg) != SQLITE_OK) {
|
|
printf("SQLite3Database::query_run error: \nQuery: %s \n Error:\n %s\n", query.c_str(), err_msg);
|
|
sqlite3_free(err_msg);
|
|
}
|
|
}
|
|
|
|
std::string SQLite3Database::escape(const std::string str) {
|
|
char *ret;
|
|
|
|
ret = sqlite3_mprintf("%q", str.c_str());
|
|
|
|
if (ret) {
|
|
std::string res(ret);
|
|
|
|
sqlite3_free(ret);
|
|
|
|
return res;
|
|
}
|
|
|
|
return "";
|
|
}
|
|
void SQLite3Database::escape(const std::string str, std::string *to) {
|
|
char *ret;
|
|
|
|
ret = sqlite3_mprintf("%q", str.c_str());
|
|
|
|
if (ret) {
|
|
to->operator=(ret);
|
|
|
|
sqlite3_free(ret);
|
|
}
|
|
}
|
|
|
|
SQLite3Database::SQLite3Database() :
|
|
Database() {
|
|
}
|
|
|
|
SQLite3Database::~SQLite3Database() {
|
|
if (conn)
|
|
sqlite3_close(conn);
|
|
}
|