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

2
deps/mysqlConnector/Statement.cpp vendored
View file

@ -60,7 +60,7 @@ std::shared_ptr< Mysql::ResultSet > Mysql::Statement::getResultSet()
{
if( errNo() != 0 )
throw std::runtime_error( "Error during getResultSet() : " + std::to_string( errNo() ) + ": " +
m_pConnection->getError() );
m_pConnection->getError() );
return std::make_shared< ResultSet >( mysql_store_result( m_pConnection->getRawCon() ), shared_from_this() );
}

39
src/dbm/DbManager.cpp
View file

@ -123,20 +123,41 @@ 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.";
return false;
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( !execute( "CREATE DATABASE " + m_database ) )
return false;
if( !dbCreated )
if( !execute( "CREATE DATABASE " + m_database ) )
return false;
if( !selectSchema() )
{