mirror of
https://github.com/TrinityCore/TrinityCore.git
synced 2026-01-15 23:20:36 +01:00
115 lines
2.5 KiB
C++
115 lines
2.5 KiB
C++
/*
|
|
* This file is part of the TrinityCore Project. See AUTHORS file for Copyright information
|
|
*
|
|
* 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/>.
|
|
*/
|
|
|
|
#include "MapUpdater.h"
|
|
#include "DatabaseEnv.h"
|
|
#include "Map.h"
|
|
#include "Metric.h"
|
|
|
|
class MapUpdateRequest
|
|
{
|
|
private:
|
|
|
|
Map& m_map;
|
|
MapUpdater& m_updater;
|
|
uint32 m_diff;
|
|
|
|
public:
|
|
|
|
MapUpdateRequest(Map& m, MapUpdater& u, uint32 d)
|
|
: m_map(m), m_updater(u), m_diff(d)
|
|
{
|
|
}
|
|
|
|
void call()
|
|
{
|
|
TC_METRIC_TIMER("map_update_time_diff", TC_METRIC_TAG("map_id", std::to_string(m_map.GetId())));
|
|
m_map.Update (m_diff);
|
|
m_updater.update_finished();
|
|
}
|
|
};
|
|
|
|
void MapUpdater::activate(size_t num_threads)
|
|
{
|
|
for (size_t i = 0; i < num_threads; ++i)
|
|
_workerThreads.emplace_back(&MapUpdater::WorkerThread, this);
|
|
}
|
|
|
|
void MapUpdater::deactivate()
|
|
{
|
|
_cancelationToken = true;
|
|
|
|
wait();
|
|
|
|
_queue.Cancel();
|
|
|
|
for (auto& thread : _workerThreads)
|
|
thread.join();
|
|
}
|
|
|
|
void MapUpdater::wait()
|
|
{
|
|
std::unique_lock lock(_lock);
|
|
|
|
_condition.wait(lock, [&] { return pending_requests == 0; });
|
|
}
|
|
|
|
void MapUpdater::schedule_update(Map& map, uint32 diff)
|
|
{
|
|
std::scoped_lock lock(_lock);
|
|
|
|
++pending_requests;
|
|
|
|
_queue.Push(new MapUpdateRequest(map, *this, diff));
|
|
}
|
|
|
|
bool MapUpdater::activated() const
|
|
{
|
|
return !_workerThreads.empty();
|
|
}
|
|
|
|
void MapUpdater::update_finished()
|
|
{
|
|
std::scoped_lock lock(_lock);
|
|
|
|
--pending_requests;
|
|
|
|
_condition.notify_all();
|
|
}
|
|
|
|
void MapUpdater::WorkerThread()
|
|
{
|
|
LoginDatabase.WarnAboutSyncQueries(true);
|
|
CharacterDatabase.WarnAboutSyncQueries(true);
|
|
WorldDatabase.WarnAboutSyncQueries(true);
|
|
HotfixDatabase.WarnAboutSyncQueries(true);
|
|
|
|
while (true)
|
|
{
|
|
MapUpdateRequest* request = nullptr;
|
|
|
|
_queue.WaitAndPop(request);
|
|
|
|
if (_cancelationToken)
|
|
return;
|
|
|
|
request->call();
|
|
|
|
delete request;
|
|
}
|
|
}
|