mirror of
https://github.com/TrinityCore/TrinityCore.git
synced 2026-01-22 18:15:31 +01:00
f2233f5d14
* The error is triggered when it fails to open an update for hashing
* See 'https://community.trinitycore.org/topic/12352-worldserver-crashes-on-database-update/'
for details
* And a minor cleanup in the UpdateFetcher
(cherry picked from commit c43b808665)
133 lines
4.0 KiB
C++
133 lines
4.0 KiB
C++
/*
|
|
* Copyright (C) 2008-2016 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 UpdateFetcher_h__
|
|
#define UpdateFetcher_h__
|
|
|
|
#include <DBUpdater.h>
|
|
|
|
#include <functional>
|
|
#include <string>
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
class UpdateFetcher
|
|
{
|
|
typedef boost::filesystem::path Path;
|
|
|
|
public:
|
|
UpdateFetcher(Path const& updateDirectory,
|
|
std::function<void(std::string const&)> const& apply,
|
|
std::function<void(Path const& path)> const& applyFile,
|
|
std::function<QueryResult(std::string const&)> const& retrieve);
|
|
|
|
UpdateResult Update(bool const redundancyChecks, bool const allowRehash,
|
|
bool const archivedRedundancy, int32 const cleanDeadReferencesMaxCount) const;
|
|
|
|
private:
|
|
enum UpdateMode
|
|
{
|
|
MODE_APPLY,
|
|
MODE_REHASH
|
|
};
|
|
|
|
enum State
|
|
{
|
|
RELEASED,
|
|
ARCHIVED
|
|
};
|
|
|
|
struct AppliedFileEntry
|
|
{
|
|
AppliedFileEntry(std::string const& name_, std::string const& hash_, State state_, uint64 timestamp_)
|
|
: name(name_), hash(hash_), state(state_), timestamp(timestamp_) { }
|
|
|
|
std::string const name;
|
|
|
|
std::string const hash;
|
|
|
|
State const state;
|
|
|
|
uint64 const timestamp;
|
|
|
|
static inline State StateConvert(std::string const& state)
|
|
{
|
|
return (state == "RELEASED") ? RELEASED : ARCHIVED;
|
|
}
|
|
|
|
static inline std::string StateConvert(State const state)
|
|
{
|
|
return (state == RELEASED) ? "RELEASED" : "ARCHIVED";
|
|
}
|
|
|
|
std::string GetStateAsString() const
|
|
{
|
|
return StateConvert(state);
|
|
}
|
|
};
|
|
|
|
struct DirectoryEntry
|
|
{
|
|
DirectoryEntry(Path const& path_, State state_) : path(path_), state(state_) { }
|
|
|
|
Path const path;
|
|
|
|
State const state;
|
|
};
|
|
|
|
typedef std::pair<Path, State> LocaleFileEntry;
|
|
|
|
struct PathCompare
|
|
{
|
|
inline bool operator() (LocaleFileEntry const& left, LocaleFileEntry const& right) const
|
|
{
|
|
return left.first.filename().string() < right.first.filename().string();
|
|
}
|
|
};
|
|
|
|
typedef std::set<LocaleFileEntry, PathCompare> LocaleFileStorage;
|
|
typedef std::unordered_map<std::string, std::string> HashToFileNameStorage;
|
|
typedef std::unordered_map<std::string, AppliedFileEntry> AppliedFileStorage;
|
|
typedef std::vector<UpdateFetcher::DirectoryEntry> DirectoryStorage;
|
|
|
|
LocaleFileStorage GetFileList() const;
|
|
void FillFileListRecursively(Path const& path, LocaleFileStorage& storage,
|
|
State const state, uint32 const depth) const;
|
|
|
|
DirectoryStorage ReceiveIncludedDirectories() const;
|
|
AppliedFileStorage ReceiveAppliedFiles() const;
|
|
|
|
std::string ReadSQLUpdate(Path const& file) const;
|
|
std::string CalculateHash(std::string const& query) const;
|
|
|
|
uint32 Apply(Path const& path) const;
|
|
|
|
void UpdateEntry(AppliedFileEntry const& entry, uint32 const speed = 0) const;
|
|
void RenameEntry(std::string const& from, std::string const& to) const;
|
|
void CleanUp(AppliedFileStorage const& storage) const;
|
|
|
|
void UpdateState(std::string const& name, State const state) const;
|
|
|
|
Path const _sourceDirectory;
|
|
|
|
std::function<void(std::string const&)> const _apply;
|
|
std::function<void(Path const& path)> const _applyFile;
|
|
std::function<QueryResult(std::string const&)> const _retrieve;
|
|
};
|
|
|
|
#endif // UpdateFetcher_h__
|