mirror of
https://github.com/EvernodeXRPL/sashimono.git
synced 2026-04-29 15:38:00 +00:00
Initial project structure. (#1)
This commit is contained in:
Savinda Senevirathne
committed by
GitHub
GitHub
parent
5da5d4e303
commit
220d65bd0b
257
src/sqlite.cpp
Normal file
257
src/sqlite.cpp
Normal file
@@ -0,0 +1,257 @@
|
||||
#include "sqlite.hpp"
|
||||
#include "salog.hpp"
|
||||
|
||||
namespace sqlite
|
||||
{
|
||||
constexpr const char *COLUMN_DATA_TYPES[]{"INT", "TEXT", "BLOB"};
|
||||
constexpr const char *CREATE_TABLE = "CREATE TABLE IF NOT EXISTS ";
|
||||
constexpr const char *CREATE_INDEX = "CREATE INDEX ";
|
||||
constexpr const char *CREATE_UNIQUE_INDEX = "CREATE UNIQUE INDEX ";
|
||||
constexpr const char *BEGIN_TRANSACTION = "BEGIN TRANSACTION;";
|
||||
constexpr const char *COMMIT_TRANSACTION = "COMMIT;";
|
||||
constexpr const char *ROLLBACK_TRANSACTION = "ROLLBACK;";
|
||||
constexpr const char *INSERT_INTO = "INSERT INTO ";
|
||||
constexpr const char *PRIMARY_KEY = "PRIMARY KEY";
|
||||
constexpr const char *NOT_NULL = "NOT NULL";
|
||||
constexpr const char *VALUES = "VALUES";
|
||||
constexpr const char *SELECT_ALL = "SELECT * FROM ";
|
||||
constexpr const char *SQLITE_MASTER = "sqlite_master";
|
||||
constexpr const char *WHERE = " WHERE ";
|
||||
constexpr const char *AND = " AND ";
|
||||
|
||||
/**
|
||||
* Opens a connection to a given databse and give the db pointer.
|
||||
* @param db_name Database name to be connected.
|
||||
* @param db Pointer to the db pointer which is to be connected and pointed.
|
||||
* @param writable Whether the database must be opened in a writable mode or not.
|
||||
* @returns returns 0 on success, or -1 on error.
|
||||
*/
|
||||
int open_db(std::string_view db_name, sqlite3 **db, const bool writable)
|
||||
{
|
||||
int ret;
|
||||
const int flags = writable ? (SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE) : SQLITE_OPEN_READONLY;
|
||||
if ((ret = sqlite3_open_v2(db_name.data(), db, flags, 0)) != SQLITE_OK)
|
||||
{
|
||||
LOG_ERROR << ret << ": Sqlite error when opening database " << db_name;
|
||||
*db = NULL;
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* Executes given sql query.
|
||||
* @param db Pointer to the db.
|
||||
* @param sql Sql query to be executed.
|
||||
* @param callback Callback funcion which is called for each result row.
|
||||
* @param callback_first_arg First data argumat to be parced to the callback (void pointer).
|
||||
* @returns returns 0 on success, or -1 on error.
|
||||
*/
|
||||
int exec_sql(sqlite3 *db, std::string_view sql, int (*callback)(void *, int, char **, char **), void *callback_first_arg)
|
||||
{
|
||||
char *err_msg;
|
||||
if (sqlite3_exec(db, sql.data(), callback, (callback != NULL ? (void *)callback_first_arg : NULL), &err_msg) != SQLITE_OK)
|
||||
{
|
||||
LOG_ERROR << "SQL error occured: " << err_msg;
|
||||
sqlite3_free(err_msg);
|
||||
return -1;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
int begin_transaction(sqlite3 *db)
|
||||
{
|
||||
return sqlite::exec_sql(db, BEGIN_TRANSACTION);
|
||||
}
|
||||
|
||||
int commit_transaction(sqlite3 *db)
|
||||
{
|
||||
return sqlite::exec_sql(db, COMMIT_TRANSACTION);
|
||||
}
|
||||
|
||||
int rollback_transaction(sqlite3 *db)
|
||||
{
|
||||
return sqlite::exec_sql(db, ROLLBACK_TRANSACTION);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a table with given table info.
|
||||
* @param db Pointer to the db.
|
||||
* @param table_name Table name to be created.
|
||||
* @param column_info Column info of the table.
|
||||
* @returns returns 0 on success, or -1 on error.
|
||||
*/
|
||||
int create_table(sqlite3 *db, std::string_view table_name, const std::vector<table_column_info> &column_info)
|
||||
{
|
||||
std::string sql;
|
||||
sql.append(CREATE_TABLE).append(table_name).append(" (");
|
||||
|
||||
for (auto itr = column_info.begin(); itr != column_info.end(); ++itr)
|
||||
{
|
||||
sql.append(itr->name);
|
||||
sql.append(" ");
|
||||
sql.append(COLUMN_DATA_TYPES[itr->column_type]);
|
||||
|
||||
if (itr->is_key)
|
||||
{
|
||||
sql.append(" ");
|
||||
sql.append(PRIMARY_KEY);
|
||||
}
|
||||
|
||||
if (!itr->is_null)
|
||||
{
|
||||
sql.append(" ");
|
||||
sql.append(NOT_NULL);
|
||||
}
|
||||
|
||||
if (itr != column_info.end() - 1)
|
||||
sql.append(",");
|
||||
}
|
||||
sql.append(")");
|
||||
|
||||
const int ret = exec_sql(db, sql);
|
||||
if (ret == -1)
|
||||
LOG_ERROR << "Error when creating sqlite table " << table_name;
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int create_index(sqlite3 *db, std::string_view table_name, std::string_view column_names, const bool is_unique)
|
||||
{
|
||||
std::string index_name = std::string("idx_").append(table_name).append("_").append(column_names);
|
||||
std::replace(index_name.begin(), index_name.end(), ',', '_');
|
||||
|
||||
std::string sql;
|
||||
sql.append(is_unique ? CREATE_UNIQUE_INDEX : CREATE_INDEX)
|
||||
.append(index_name)
|
||||
.append(" ON ")
|
||||
.append(table_name)
|
||||
.append("(")
|
||||
.append(column_names)
|
||||
.append(")");
|
||||
|
||||
const int ret = exec_sql(db, sql);
|
||||
if (ret == -1)
|
||||
LOG_ERROR << "Error when creating sqlite index '" << index_name << "' in table " << table_name;
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
/**
|
||||
* Inserts mulitple rows to a table.
|
||||
* @param db Pointer to the db.
|
||||
* @param table_name Table name to be populated.
|
||||
* @param column_names_string Comma seperated string of colums (eg: "col_1,col_2,...").
|
||||
* @param value_strings Vector of comma seperated values (wrap in single quotes for TEXT type) (eg: ["r1val1,'r1val2',...", "r2val1,'r2val2',..."]).
|
||||
* @returns returns 0 on success, or -1 on error.
|
||||
*/
|
||||
int insert_rows(sqlite3 *db, std::string_view table_name, std::string_view column_names_string, const std::vector<std::string> &value_strings)
|
||||
{
|
||||
std::string sql;
|
||||
|
||||
sql.append(INSERT_INTO);
|
||||
sql.append(table_name);
|
||||
sql.append("(");
|
||||
sql.append(column_names_string);
|
||||
sql.append(") ");
|
||||
sql.append(VALUES);
|
||||
|
||||
for (auto itr = value_strings.begin(); itr != value_strings.end(); ++itr)
|
||||
{
|
||||
sql.append("(");
|
||||
sql.append(*itr);
|
||||
sql.append(")");
|
||||
|
||||
if (itr != value_strings.end() - 1)
|
||||
sql.append(",");
|
||||
}
|
||||
|
||||
/* Execute SQL statement */
|
||||
return exec_sql(db, sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* Inserts a row to a table.
|
||||
* @param db Pointer to the db.
|
||||
* @param table_name Table name to be populated.
|
||||
* @param column_names_string Comma seperated string of colums (eg: "col_1,col_2,...").
|
||||
* @param value_string comma seperated values as per column order (wrap in single quotes for TEXT type) (eg: "r1val1,'r1val2',...").
|
||||
* @returns returns 0 on success, or -1 on error.
|
||||
*/
|
||||
int insert_row(sqlite3 *db, std::string_view table_name, std::string_view column_names_string, std::string_view value_string)
|
||||
{
|
||||
std::string sql;
|
||||
// Reserving the space for the query before construction.
|
||||
sql.reserve(sizeof(INSERT_INTO) + table_name.size() + column_names_string.size() + sizeof(VALUES) + value_string.size() + 5);
|
||||
|
||||
sql.append(INSERT_INTO);
|
||||
sql.append(table_name);
|
||||
sql.append("(");
|
||||
sql.append(column_names_string);
|
||||
sql.append(") ");
|
||||
sql.append(VALUES);
|
||||
sql.append("(");
|
||||
sql.append(value_string);
|
||||
sql.append(")");
|
||||
|
||||
/* Execute SQL statement */
|
||||
return exec_sql(db, sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* Checks whether table exist in the database.
|
||||
* @param db Pointer to the db.
|
||||
* @param table_name Table name to be checked.
|
||||
* @returns returns true is exist, otherwise false.
|
||||
*/
|
||||
bool is_table_exists(sqlite3 *db, std::string_view table_name)
|
||||
{
|
||||
std::string sql;
|
||||
// Reserving the space for the query before construction.
|
||||
sql.reserve(sizeof(SELECT_ALL) + sizeof(SQLITE_MASTER) + sizeof(WHERE) + sizeof(AND) + table_name.size() + 19);
|
||||
|
||||
sql.append(SELECT_ALL);
|
||||
sql.append(SQLITE_MASTER);
|
||||
sql.append(WHERE);
|
||||
sql.append("type='table'");
|
||||
sql.append(AND);
|
||||
sql.append("name='");
|
||||
sql.append(table_name);
|
||||
sql.append("'");
|
||||
|
||||
sqlite3_stmt *stmt;
|
||||
|
||||
if (sqlite3_prepare_v2(db, sql.data(), -1, &stmt, 0) == SQLITE_OK &&
|
||||
stmt != NULL && sqlite3_step(stmt) == SQLITE_ROW)
|
||||
{
|
||||
// Finalize and distroys the statement.
|
||||
sqlite3_finalize(stmt);
|
||||
return true;
|
||||
}
|
||||
|
||||
// Finalize and distroys the statement.
|
||||
sqlite3_finalize(stmt);
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* Closes a connection to a given databse.
|
||||
* @param db Pointer to the db.
|
||||
* @returns returns 0 on success, or -1 on error.
|
||||
*/
|
||||
int close_db(sqlite3 **db)
|
||||
{
|
||||
if (*db == NULL)
|
||||
return 0;
|
||||
|
||||
if (sqlite3_close(*db) != SQLITE_OK)
|
||||
{
|
||||
LOG_ERROR << "Can't close database: " << sqlite3_errmsg(*db);
|
||||
return -1;
|
||||
}
|
||||
|
||||
*db = NULL;
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user