diff options
| author | Dan Engelbrecht <[email protected]> | 2022-03-22 13:50:08 +0100 |
|---|---|---|
| committer | GitHub <[email protected]> | 2022-03-22 13:50:08 +0100 |
| commit | 8667509907b8829b5148ee856dce675001051b01 (patch) | |
| tree | 0252caa7cd01c533809b1b17dfa41721a0cccb82 /zenhttp | |
| parent | Remove unused TotalAttachmentsSize (Mac warning) (diff) | |
| download | zen-8667509907b8829b5148ee856dce675001051b01.tar.xz zen-8667509907b8829b5148ee856dce675001051b01.zip | |
move workthreadpool to zencore (#63)
Diffstat (limited to 'zenhttp')
| -rw-r--r-- | zenhttp/httpsys.h | 2 | ||||
| -rw-r--r-- | zenhttp/workthreadpool.cpp | 77 | ||||
| -rw-r--r-- | zenhttp/workthreadpool.h | 46 |
3 files changed, 1 insertions, 124 deletions
diff --git a/zenhttp/httpsys.h b/zenhttp/httpsys.h index 0453b8740..d6bd34890 100644 --- a/zenhttp/httpsys.h +++ b/zenhttp/httpsys.h @@ -15,8 +15,8 @@ #if ZEN_WITH_HTTPSYS # define _WINSOCKAPI_ # include <zencore/windows.h> +# include <zencore/workthreadpool.h> # include "iothreadpool.h" -# include "workthreadpool.h" # include <http.h> 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 <zencore/logging.h> - -namespace zen { - -namespace detail { - struct LambdaWork : IWork - { - LambdaWork(auto Work) : WorkFunction(Work) {} - virtual void Execute() override { WorkFunction(); } - - std::function<void()> 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<IWork> Work) -{ - m_WorkQueue.Enqueue(std::move(Work)); -} - -void -WorkerThreadPool::ScheduleWork(std::function<void()>&& Work) -{ - m_WorkQueue.Enqueue(new detail::LambdaWork(Work)); -} - -void -WorkerThreadPool::WorkerThreadFunction() -{ - do - { - Ref<IWork> 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 diff --git a/zenhttp/workthreadpool.h b/zenhttp/workthreadpool.h deleted file mode 100644 index 834339d50..000000000 --- a/zenhttp/workthreadpool.h +++ /dev/null @@ -1,46 +0,0 @@ -// Copyright Epic Games, Inc. All Rights Reserved. - -#pragma once - -#include <zencore/zencore.h> - -#include <zencore/blockingqueue.h> -#include <zencore/refcount.h> - -#include <exception> -#include <functional> -#include <system_error> -#include <thread> -#include <vector> - -namespace zen { - -struct IWork : public RefCounted -{ - virtual void Execute() = 0; - - inline std::exception_ptr GetException() { return m_Exception; } - -private: - std::exception_ptr m_Exception; - - friend class WorkerThreadPool; -}; - -class WorkerThreadPool -{ -public: - WorkerThreadPool(int InThreadCount); - ~WorkerThreadPool(); - - void ScheduleWork(Ref<IWork> Work); - void ScheduleWork(std::function<void()>&& Work); - - void WorkerThreadFunction(); - -private: - std::vector<std::thread> m_WorkerThreads; - BlockingQueue<Ref<IWork>> m_WorkQueue; -}; - -} // namespace zen |