From 8667509907b8829b5148ee856dce675001051b01 Mon Sep 17 00:00:00 2001 From: Dan Engelbrecht Date: Tue, 22 Mar 2022 13:50:08 +0100 Subject: move workthreadpool to zencore (#63) --- zenhttp/workthreadpool.cpp | 77 ---------------------------------------------- 1 file changed, 77 deletions(-) delete mode 100644 zenhttp/workthreadpool.cpp (limited to 'zenhttp/workthreadpool.cpp') diff --git a/zenhttp/workthreadpool.cpp b/zenhttp/workthreadpool.cpp deleted file mode 100644 index 41eaaae94..000000000 --- a/zenhttp/workthreadpool.cpp +++ /dev/null @@ -1,77 +0,0 @@ -// Copyright Epic Games, Inc. All Rights Reserved. - -#include "workthreadpool.h" - -#include - -namespace zen { - -namespace detail { - struct LambdaWork : IWork - { - LambdaWork(auto Work) : WorkFunction(Work) {} - virtual void Execute() override { WorkFunction(); } - - std::function WorkFunction; - }; -} // namespace detail - -WorkerThreadPool::WorkerThreadPool(int InThreadCount) -{ - for (int i = 0; i < InThreadCount; ++i) - { - m_WorkerThreads.emplace_back(&WorkerThreadPool::WorkerThreadFunction, this); - } -} - -WorkerThreadPool::~WorkerThreadPool() -{ - m_WorkQueue.CompleteAdding(); - - for (std::thread& Thread : m_WorkerThreads) - { - Thread.join(); - } - - m_WorkerThreads.clear(); -} - -void -WorkerThreadPool::ScheduleWork(Ref Work) -{ - m_WorkQueue.Enqueue(std::move(Work)); -} - -void -WorkerThreadPool::ScheduleWork(std::function&& Work) -{ - m_WorkQueue.Enqueue(new detail::LambdaWork(Work)); -} - -void -WorkerThreadPool::WorkerThreadFunction() -{ - do - { - Ref Work; - if (m_WorkQueue.WaitAndDequeue(Work)) - { - try - { - Work->Execute(); - } - catch (std::exception& e) - { - Work->m_Exception = std::current_exception(); - - ZEN_WARN("Caught exception in worker thread: {}", e.what()); - } - } - else - { - return; - } - } while (true); -} - -} // namespace zen \ No newline at end of file -- cgit v1.2.3