mirror of
https://gitcode.com/GitHub_Trending/az/azerothcore-wotlk.git
synced 2026-07-14 20:27:57 +00:00
Kargatum
2d2857ce81
* feat(Core/DBUpdater): implement db auto update * 1 * 2 * 3 * Some minor improvements * add find bin for mysql 8.0 * lic Co-authored-by: Kitzunu <24550914+Kitzunu@users.noreply.github.com>
65 lines
1.7 KiB
C++
65 lines
1.7 KiB
C++
/*
|
|
* Copyright (C) 2016+ AzerothCore <www.azerothcore.org>, released under GNU AGPL v3 license: https://github.com/azerothcore/azerothcore-wotlk/blob/master/LICENSE-AGPL3
|
|
* Copyright (C) 2021+ WarheadCore <https://github.com/WarheadCore>
|
|
*/
|
|
|
|
#ifndef DatabaseLoader_h__
|
|
#define DatabaseLoader_h__
|
|
|
|
#include "Define.h"
|
|
#include <functional>
|
|
#include <queue>
|
|
#include <stack>
|
|
#include <string>
|
|
|
|
template <class T>
|
|
class DatabaseWorkerPool;
|
|
|
|
// A helper class to initiate all database worker pools,
|
|
// handles updating, delays preparing of statements and cleans up on failure.
|
|
class AC_DATABASE_API DatabaseLoader
|
|
{
|
|
public:
|
|
DatabaseLoader(std::string const& logger, uint32 const defaultUpdateMask = 0);
|
|
|
|
// 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_MASK_ALL = DATABASE_LOGIN | DATABASE_CHARACTER | DATABASE_WORLD
|
|
};
|
|
|
|
private:
|
|
bool OpenDatabases();
|
|
bool PopulateDatabases();
|
|
bool UpdateDatabases();
|
|
bool PrepareStatements();
|
|
|
|
using Predicate = std::function<bool()>;
|
|
using Closer = std::function<void()>;
|
|
|
|
// Invokes all functions in the given queue and closes the databases on errors.
|
|
// Returns false when there was an error.
|
|
bool Process(std::queue<Predicate>& queue);
|
|
|
|
std::string const _logger;
|
|
bool const _autoSetup;
|
|
uint32 const _updateFlags;
|
|
|
|
std::queue<Predicate> _open, _populate, _update, _prepare;
|
|
std::stack<Closer> _close;
|
|
};
|
|
|
|
#endif // DatabaseLoader_h__
|