mirror of
https://github.com/TrinityCore/TrinityCore.git
synced 2026-01-15 23:20:36 +01:00
73 lines
2.1 KiB
C++
73 lines
2.1 KiB
C++
/*
|
|
* Copyright (C) 2008-2015 TrinityCore <http://www.trinitycore.org/>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License as published by the
|
|
* Free Software Foundation; either version 2 of the License, or (at your
|
|
* option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef DatabaseLoader_h__
|
|
#define DatabaseLoader_h__
|
|
|
|
#include "DatabaseWorkerPool.h"
|
|
#include "DatabaseEnv.h"
|
|
|
|
#include <stack>
|
|
#include <functional>
|
|
|
|
// A helper class to initiate all database worker pools,
|
|
// handles updating, delays preparing of statements and cleans up on failure.
|
|
class DatabaseLoader
|
|
{
|
|
public:
|
|
DatabaseLoader(std::string const& logger, uint32 const defaultUpdateMask);
|
|
|
|
// Register a database to the loader (lazy implemented)
|
|
template <class T>
|
|
DatabaseLoader& AddDatabase(DatabaseWorkerPool<T>& pool, std::string const& name);
|
|
|
|
// Load all databases
|
|
bool Load();
|
|
|
|
enum DatabaseTypeFlags
|
|
{
|
|
DATABASE_NONE = 0,
|
|
|
|
DATABASE_LOGIN = 1,
|
|
DATABASE_CHARACTER = 2,
|
|
DATABASE_WORLD = 4,
|
|
DATABASE_HOTFIX = 8,
|
|
|
|
DATABASE_MASK_ALL = DATABASE_LOGIN | DATABASE_CHARACTER | DATABASE_WORLD | DATABASE_HOTFIX
|
|
};
|
|
|
|
private:
|
|
bool OpenDatabases();
|
|
bool PopulateDatabases();
|
|
bool UpdateDatabases();
|
|
bool PrepareStatements();
|
|
|
|
using Predicate = std::function<bool()>;
|
|
|
|
static bool Process(std::stack<Predicate>& stack);
|
|
|
|
std::string const _logger;
|
|
bool const _autoSetup;
|
|
uint32 const _updateFlags;
|
|
|
|
std::stack<std::pair<Predicate, std::function<void()>>> _open;
|
|
std::stack<std::function<void()>> _close;
|
|
std::stack<Predicate> _populate, _update, _prepare;
|
|
};
|
|
|
|
#endif // DatabaseLoader_h__
|