diff options
Diffstat (limited to 'src/base')
-rw-r--r-- | src/base/database.cpp | 20 | ||||
-rw-r--r-- | src/base/database.hpp | 8 |
2 files changed, 28 insertions, 0 deletions
diff --git a/src/base/database.cpp b/src/base/database.cpp index 96f40745..ec2e8493 100644 --- a/src/base/database.cpp +++ b/src/base/database.cpp @@ -82,6 +82,26 @@ namespace usdx sqlite3_finalize(sqliteStatement); } + const bool StatDatabase::sqlite_table_exists(const std::string table) + { + std::string sql = "select [name] from [sqlite_master] where [type] = 'table' and [tbl_name] = ?1;"; + sqlite3_stmt *sqliteStatement = sqlite_prepare(sql); + + // bind table name to parameter 1 and execute statement + sqlite3_bind_text(sqliteStatement, 1, table.c_str(), table.length(), SQLITE_TRANSIENT); + int rc = sqlite3_step(sqliteStatement); + + // if rc is SQLITE_ROW, than result has at lease one row and so + // the table exists + bool result = false; + if (rc == SQLITE_ROW) { + result = true; + } + + sqlite3_finalize(sqliteStatement); + return result; + } + void StatDatabase::init(const std::string filename) { LOG4CXX_DEBUG(log, "Initializing Database: " << filename); diff --git a/src/base/database.hpp b/src/base/database.hpp index 3bf699ad..d5313609 100644 --- a/src/base/database.hpp +++ b/src/base/database.hpp @@ -72,6 +72,14 @@ namespace usdx */ void sqlite_exec(const std::string sqlStatement); + /** + * Check if the given table exists in the database. + * + * @param table Name to check if exists + * @return true, if table exists, false if not + */ + const bool sqlite_table_exists(const std::string table); + // Singleton StatDatabase(std::string filename); |