mirrors/sapphire
1
Fork 0
mirror of https://github.com/SapphireServer/Sapphire.git synced 2025-04-23 21:27:45 +00:00
Code Activity

More dbm...

Browse source
This commit is contained in:
Mordred 2018-12-08 01:11:46 +01:00
parent c58edb540d
commit de77e94e56
2 changed files with 31 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
src/dbm/DbManager.cpp
View file

@ -123,18 +123,39 @@ bool DbManager::modeInit()
const std::string insertFile = "sql/schema/inserts.sql";
bool result = false;
bool dbCreated = false;
if( selectSchema() )
{
// TODO: allow init if database is empty
// select count(*)
// from information_schema.tables
// where table_type = 'BASE TABLE'
// and table_schema = 'your_database_name_here'
m_lastError = "Database already existing, use <liquidate> mode first to remove it.";
std::string query = "SELECT COUNT(*) "
"FROM information_schema.tables "
"WHERE table_type = 'BASE TABLE' "
"AND table_schema = '" + m_database + "';";
dbCreated = true;
try
{
auto stmt = m_pConnection->createStatement();
auto resultSet = stmt->executeQuery( query );
if( !resultSet->next() )
return false;
auto count = resultSet->getUInt( 1 );
if( count )
{
m_lastError = "Database " + m_database + " still contains tables. <Liquidate> it first!";
return false;
}
}
catch( std::runtime_error& e )
{
m_lastError = e.what();
return false;
}
}
if( !dbCreated )
if( !execute( "CREATE DATABASE " + m_database ) )
return false;