feat(Core/MapUpdate): switch from ACE_Method_Request to PCQ (#3459)

This commit is contained in:
Kargatum
2020-11-14 03:56:26 +07:00
committed by GitHub
parent fc63d935b9
commit 543753b641
6 changed files with 90 additions and 281 deletions
-121
View File
@@ -1,121 +0,0 @@
#include <ace/Thread_Mutex.h>
#include <ace/Log_Msg.h>
#include "Threading.h"
#include "DelayExecutor.h"
DelayExecutor* DelayExecutor::instance()
{
static DelayExecutor instance;
return &instance;
}
DelayExecutor::DelayExecutor()
: pre_svc_hook_(0), post_svc_hook_(0), activated_(false), mqueue_(1 * 1024 * 1024, 1 * 1024 * 1024), queue_(&mqueue_)
{
}
DelayExecutor::~DelayExecutor()
{
if (pre_svc_hook_)
delete pre_svc_hook_;
if (post_svc_hook_)
delete post_svc_hook_;
deactivate();
}
int DelayExecutor::deactivate()
{
if (!activated())
return -1;
activated(false);
queue_.queue()->deactivate();
wait();
return 0;
}
int DelayExecutor::svc()
{
if (pre_svc_hook_)
pre_svc_hook_->call();
for (;;)
{
ACE_Method_Request* rq = queue_.dequeue();
if (!rq)
break;
rq->call();
delete rq;
}
if (post_svc_hook_)
post_svc_hook_->call();
return 0;
}
int DelayExecutor::start(int num_threads, ACE_Method_Request* pre_svc_hook, ACE_Method_Request* post_svc_hook)
{
if (activated())
return -1;
if (num_threads < 1)
return -1;
if (pre_svc_hook_)
delete pre_svc_hook_;
if (post_svc_hook_)
delete post_svc_hook_;
pre_svc_hook_ = pre_svc_hook;
post_svc_hook_ = post_svc_hook;
queue_.queue()->activate();
// pussywizard:
//acore::ThreadPriority tp;
//int _priority = tp.getPriority(acore::Priority_Highest);
//if (ACE_Task_Base::activate(THR_NEW_LWP | THR_JOINABLE, num_threads, 0, _priority) == -1)
// return -1;
if (ACE_Task_Base::activate(THR_NEW_LWP | THR_JOINABLE | THR_INHERIT_SCHED, num_threads) == -1)
return -1;
activated(true);
return true;
}
int DelayExecutor::execute(ACE_Method_Request* new_req)
{
if (new_req == NULL)
return -1;
// pussywizard: NULL as param for enqueue - wait until the action is possible!
// new tasks are added to the queue during map update (schedule_update in MapInstanced::Update)
// the queue can be momentarily blocked by map threads constantly waiting for tasks (for (;;) { queue_.dequeue();... } in DelayExecutor::svc())
// so just wait a moment, don't drop the task xDddd
if (queue_.enqueue(new_req, /*(ACE_Time_Value*)&ACE_Time_Value::zero*/ NULL) == -1)
{
delete new_req;
ACE_ERROR_RETURN((LM_ERROR, ACE_TEXT("(%t) %p\n"), ACE_TEXT("DelayExecutor::execute enqueue")), -1);
}
return 0;
}
bool DelayExecutor::activated()
{
return activated_;
}
void DelayExecutor::activated(bool s)
{
activated_ = s;
}
-39
View File
@@ -1,39 +0,0 @@
#ifndef _M_DELAY_EXECUTOR_H
#define _M_DELAY_EXECUTOR_H
#include <ace/Task.h>
#include <ace/Activation_Queue.h>
#include <ace/Method_Request.h>
class DelayExecutor : protected ACE_Task_Base
{
public:
DelayExecutor();
virtual ~DelayExecutor();
static DelayExecutor* instance();
int execute(ACE_Method_Request* new_req);
int start(int num_threads = 1, ACE_Method_Request* pre_svc_hook = NULL, ACE_Method_Request* post_svc_hook = NULL);
int deactivate();
bool activated();
virtual int svc();
private:
ACE_Method_Request* pre_svc_hook_;
ACE_Method_Request* post_svc_hook_;
bool activated_;
ACE_Message_Queue<ACE_SYNCH> mqueue_;
ACE_Activation_Queue queue_;
void activated(bool s);
};
#endif // _M_DELAY_EXECUTOR_H
+1 -5
View File
@@ -708,11 +708,9 @@ void Map::VisitNearbyCellsOf(WorldObject* obj, TypeContainerVisitor<acore::Objec
void Map::Update(const uint32 t_diff, const uint32 s_diff, bool /*thread*/) void Map::Update(const uint32 t_diff, const uint32 s_diff, bool /*thread*/)
{ {
uint32 mapId = GetId(); // pussywizard: for crashlogs
sLog->outDebug(LOG_FILTER_POOLSYS, "%u", mapId); // pussywizard: for crashlogs
if (t_diff) if (t_diff)
_dynamicTree.update(t_diff); _dynamicTree.update(t_diff);
/// update worldsessions for existing players /// update worldsessions for existing players
for (m_mapRefIter = m_mapRefManager.begin(); m_mapRefIter != m_mapRefManager.end(); ++m_mapRefIter) for (m_mapRefIter = m_mapRefManager.begin(); m_mapRefIter != m_mapRefManager.end(); ++m_mapRefIter)
{ {
@@ -837,8 +835,6 @@ void Map::Update(const uint32 t_diff, const uint32 s_diff, bool /*thread*/)
sScriptMgr->OnMapUpdate(this, t_diff); sScriptMgr->OnMapUpdate(this, t_diff);
BuildAndSendUpdateForObjects(); // pussywizard BuildAndSendUpdateForObjects(); // pussywizard
sLog->outDebug(LOG_FILTER_POOLSYS, "%u", mapId); // pussywizard: for crashlogs
} }
void Map::HandleDelayedVisibility() void Map::HandleDelayedVisibility()
+3 -3
View File
@@ -52,9 +52,9 @@ void MapManager::Initialize()
{ {
int num_threads(sWorld->getIntConfig(CONFIG_NUMTHREADS)); int num_threads(sWorld->getIntConfig(CONFIG_NUMTHREADS));
// Start mtmaps if needed. // Start mtmaps if needed
if (num_threads > 0 && m_updater.activate(num_threads) == -1) if (num_threads > 0)
abort(); m_updater.activate(num_threads);
} }
void MapManager::InitializeVisibilityDistanceInfo() void MapManager::InitializeVisibilityDistanceInfo()
+66 -90
View File
@@ -1,88 +1,62 @@
/*
* Copyright (C) 2016+ AzerothCore <www.azerothcore.org>, released under GNU GPL v2 license: https://github.com/azerothcore/azerothcore-wotlk/blob/master/LICENSE-GPL2
* Copyright (C) 2008-2020 TrinityCore <http://www.trinitycore.org/>
* Copyright (C) 2005-2009 MaNGOS <http://getmangos.com/>
*/
#include "MapUpdater.h" #include "MapUpdater.h"
#include "DelayExecutor.h"
#include "Map.h" #include "Map.h"
#include "DatabaseEnv.h"
#include "LFGMgr.h" #include "LFGMgr.h"
#include "AvgDiffTracker.h" #include "AvgDiffTracker.h"
#include <ace/Guard_T.h> class UpdateRequest
#include <ace/Method_Request.h>
class WDBThreadStartReq1 : public ACE_Method_Request
{ {
public: public:
UpdateRequest() = default;
virtual ~UpdateRequest() = default;
WDBThreadStartReq1() virtual void call() = 0;
{
}
virtual int call()
{
return 0;
}
}; };
class WDBThreadEndReq1 : public ACE_Method_Request class MapUpdateRequest : public UpdateRequest
{ {
public: public:
WDBThreadEndReq1()
{
}
virtual int call()
{
return 0;
}
};
class MapUpdateRequest : public ACE_Method_Request
{
private:
Map& m_map;
MapUpdater& m_updater;
uint32 m_diff;
uint32 s_diff;
public:
MapUpdateRequest(Map& m, MapUpdater& u, uint32 d, uint32 sd) MapUpdateRequest(Map& m, MapUpdater& u, uint32 d, uint32 sd)
: m_map(m), m_updater(u), m_diff(d), s_diff(sd) : m_map(m), m_updater(u), m_diff(d), s_diff(sd)
{ {
} }
virtual int call() void call() override
{ {
m_map.Update (m_diff, s_diff); m_map.Update(m_diff, s_diff);
m_updater.update_finished(); m_updater.update_finished();
return 0;
} }
};
class LFGUpdateRequest : public ACE_Method_Request
{
private: private:
Map& m_map;
MapUpdater& m_updater; MapUpdater& m_updater;
uint32 m_diff; uint32 m_diff;
uint32 s_diff;
};
class LFGUpdateRequest : public UpdateRequest
{
public: public:
LFGUpdateRequest(MapUpdater& u, uint32 d) : m_updater(u), m_diff(d) {} LFGUpdateRequest(MapUpdater& u, uint32 d) : m_updater(u), m_diff(d) {}
virtual int call() void call()
{ {
uint32 startTime = getMSTime(); uint32 startTime = getMSTime();
sLFGMgr->Update(m_diff, 1); sLFGMgr->Update(m_diff, 1);
uint32 totalTime = getMSTimeDiff(startTime, getMSTime()); uint32 totalTime = getMSTimeDiff(startTime, getMSTime());
lfgDiffTracker.Update(totalTime); lfgDiffTracker.Update(totalTime);
m_updater.update_finished(); m_updater.update_finished();
return 0;
} }
private:
MapUpdater& m_updater;
uint32 m_diff;
}; };
MapUpdater::MapUpdater(): MapUpdater::MapUpdater(): pending_requests(0)
m_executor(), m_mutex(), m_condition(m_mutex), pending_requests(0)
{ {
} }
@@ -91,80 +65,82 @@ MapUpdater::~MapUpdater()
deactivate(); deactivate();
} }
int MapUpdater::activate(size_t num_threads) void MapUpdater::activate(size_t num_threads)
{ {
return m_executor.start((int)num_threads, new WDBThreadStartReq1, new WDBThreadEndReq1); for (size_t i = 0; i < num_threads; ++i)
{
_workerThreads.push_back(std::thread(&MapUpdater::WorkerThread, this));
}
} }
int MapUpdater::deactivate() void MapUpdater::deactivate()
{ {
_cancelationToken = true;
wait(); wait();
return m_executor.deactivate(); _queue.Cancel();
for (auto& thread : _workerThreads)
{
thread.join();
}
} }
int MapUpdater::wait() void MapUpdater::wait()
{ {
ACORE_GUARD(ACE_Thread_Mutex, m_mutex); std::unique_lock<std::mutex> guard(_lock);
while (pending_requests > 0) while (pending_requests > 0)
m_condition.wait(); _condition.wait(guard);
return 0; guard.unlock();
} }
int MapUpdater::schedule_update(Map& map, uint32 diff, uint32 s_diff) void MapUpdater::schedule_update(Map& map, uint32 diff, uint32 s_diff)
{ {
ACORE_GUARD(ACE_Thread_Mutex, m_mutex); std::lock_guard<std::mutex> guard(_lock);
++pending_requests; ++pending_requests;
if (m_executor.execute(new MapUpdateRequest(map, *this, diff, s_diff)) == -1) _queue.Push(new MapUpdateRequest(map, *this, diff, s_diff));
{
ACE_DEBUG((LM_ERROR, ACE_TEXT("(%t) \n"), ACE_TEXT("Failed to schedule Map Update")));
--pending_requests;
return -1;
}
return 0;
} }
int MapUpdater::schedule_lfg_update(uint32 diff) void MapUpdater::schedule_lfg_update(uint32 diff)
{ {
ACORE_GUARD(ACE_Thread_Mutex, m_mutex); std::lock_guard<std::mutex> guard(_lock);
++pending_requests; ++pending_requests;
if (m_executor.execute(new LFGUpdateRequest(*this, diff)) == -1) _queue.Push(new LFGUpdateRequest(*this, diff));
{
ACE_DEBUG((LM_ERROR, ACE_TEXT("(%t) \n"), ACE_TEXT("Failed to schedule LFG Update")));
--pending_requests;
return -1;
}
return 0;
} }
bool MapUpdater::activated() bool MapUpdater::activated()
{ {
return m_executor.activated(); return _workerThreads.size() > 0;
} }
void MapUpdater::update_finished() void MapUpdater::update_finished()
{ {
ACORE_GUARD(ACE_Thread_Mutex, m_mutex); std::lock_guard<std::mutex> lock(_lock);
if (pending_requests == 0)
{
ACE_ERROR((LM_ERROR, ACE_TEXT("(%t)\n"), ACE_TEXT("MapUpdater::update_finished BUG, report to devs")));
sLog->outMisc("WOOT! pending_requests == 0 before decrement!");
m_condition.broadcast();
return;
}
--pending_requests; --pending_requests;
m_condition.broadcast(); _condition.notify_all();
} }
void MapUpdater::WorkerThread()
{
while (1)
{
UpdateRequest* request = nullptr;
_queue.WaitAndPop(request);
if (_cancelationToken)
return;
request->call();
delete request;
}
}
+20 -23
View File
@@ -1,43 +1,40 @@
#ifndef _MAP_UPDATER_H_INCLUDED #ifndef _MAP_UPDATER_H_INCLUDED
#define _MAP_UPDATER_H_INCLUDED #define _MAP_UPDATER_H_INCLUDED
#include <ace/Thread_Mutex.h> #include "Define.h"
#include <ace/Condition_Thread_Mutex.h> #include "PCQueue.h"
#include <condition_variable>
#include "DelayExecutor.h" #include <mutex>
#include "World.h" #include <thread>
class Map; class Map;
class UpdateRequest;
class MapUpdater class MapUpdater
{ {
public: public:
MapUpdater(); MapUpdater();
virtual ~MapUpdater(); virtual ~MapUpdater();
friend class MapUpdateRequest; void schedule_update(Map& map, uint32 diff, uint32 s_diff);
friend class LFGUpdateRequest; void schedule_lfg_update(uint32 diff);
void wait();
int schedule_update(Map& map, uint32 diff, uint32 s_diff); void activate(size_t num_threads);
int schedule_lfg_update(uint32 diff); void deactivate();
int wait();
int activate(size_t num_threads);
int deactivate();
bool activated(); bool activated();
void update_finished();
private: private:
void WorkerThread();
DelayExecutor m_executor; ProducerConsumerQueue<UpdateRequest*> _queue;
ACE_Thread_Mutex m_mutex;
ACE_Condition_Thread_Mutex m_condition; std::vector<std::thread> _workerThreads;
std::atomic<bool> _cancelationToken;
std::mutex _lock;
std::condition_variable _condition;
size_t pending_requests; size_t pending_requests;
void update_finished();
}; };
#endif //_MAP_UPDATER_H_INCLUDED #endif //_MAP_UPDATER_H_INCLUDED