Rename thread pool executor and wait on tokens

aliihsancengiz · Oct 27, 2023 · 4758c9f · 4758c9f
1 parent eef3b2d
commit 4758c9f
Show file tree

Hide file tree

Showing 5 changed files with 21 additions and 15 deletions.
diff --git a/example/ExampleTasks.hpp b/example/ExampleTasks.hpp
@@ -5,7 +5,7 @@
 #include <memory>
 #include <mutex>
 #include "TaskListener.hpp"
-#include "executor/ThreadPool.hpp"
+#include "executor/ThreadPoolExecutor.hpp"
 
 std::mutex m;
 

diff --git a/example/main.cpp b/example/main.cpp
@@ -50,17 +50,23 @@ struct DeviceEventListener : TaskListenerBase
 int main()
 {
 	DefaultSchedulerPolicy policy;
-	std::shared_ptr<ThreadPool> wg = std::make_shared<ThreadPool>(policy);
-	TaskChannel ch(wg);
-	DeviceFwUploadTaskListener listener(ch);
+	auto executor = std::make_shared<ThreadPoolExecutor>(policy);
+	TaskChannel ch(executor);
+	DeviceFwUploadTaskListener deviceListener(ch);
 	DeviceEventListener evListener(ch);
+	std::vector<TokenPtr> listOfToken;
 
-	listener.get_channel().push(std::make_shared<UploadTask>("my_server_url:/upload/image/", "firmware.bin"));
-	listener.get_channel().push(std::make_shared<DownloadTask>("my_server_url:/download/image/", "firmware.bin"));
-	evListener.get_channel().push(std::make_shared<OpenedEvent>());
-	evListener.get_channel().push(std::make_shared<ClosedEvent>());
-	evListener.get_channel().push(std::make_shared<DataReceivedEvent>());
-	evListener.get_channel().push(std::make_shared<DataTransmittedEvent>());
+	listOfToken.push_back(deviceListener.get_channel().push(std::make_shared<UploadTask>("my_server_url:/upload/image/", "firmware.bin")));
+	listOfToken.push_back(deviceListener.get_channel().push(std::make_shared<DownloadTask>("my_server_url:/download/image/", "firmware.bin")));
+	listOfToken.push_back(evListener.get_channel().push(std::make_shared<OpenedEvent>()));
+	listOfToken.push_back(evListener.get_channel().push(std::make_shared<ClosedEvent>()));
+	listOfToken.push_back(evListener.get_channel().push(std::make_shared<DataReceivedEvent>()));
+	listOfToken.push_back(evListener.get_channel().push(std::make_shared<DataTransmittedEvent>()));
+
+	for (size_t i = 0; i < listOfToken.size(); i++)
+	{
+		while (!listOfToken[i]->is_set()) { }
+	}
 
 	return 0;
 }
diff --git a/include/TaskDispatcher.hpp b/include/TaskDispatcher.hpp
@@ -2,7 +2,7 @@
 
 #include <map>
 #include <typeindex>
-#include "executor/ThreadPool.hpp"
+#include "executor/ThreadPoolExecutor.hpp"
 #include "detail/TaskMapping.hpp"
 
 struct TaskDispatcher

diff --git a/include/executor/ThreadPool.hpp → include/executor/ThreadPoolExecutor.hpp b/include/executor/ThreadPool.hpp → include/executor/ThreadPoolExecutor.hpp
@@ -86,9 +86,9 @@ struct Worker
 
 } // namespace detail
 
-struct ThreadPool final : public executor::detail::ExecutorInterface
+struct ThreadPoolExecutor final : public executor::detail::ExecutorInterface
 {
-	ThreadPool(SchedulePolicy& policy, std::size_t worker_size = 4) : number_of_workers(worker_size), _policy(policy)
+	ThreadPoolExecutor(SchedulePolicy& policy, std::size_t worker_size = 4) : number_of_workers(worker_size), _policy(policy)
 	{
 		for (size_t i = 0; i < number_of_workers; i++)
 		{
@@ -103,7 +103,7 @@ struct ThreadPool final : public executor::detail::ExecutorInterface
 		}
 		_policy.set_worker_id_list(wids);
 	}
-	virtual ~ThreadPool() override
+	virtual ~ThreadPoolExecutor() override
 	{
 	}
 

diff --git a/tests/task_tests.cpp b/tests/task_tests.cpp
@@ -46,7 +46,7 @@ struct DummyTaskListener : TaskListenerBase
 TEST(TaskTestFixture, canDispatchTasks)
 {
 	DefaultSchedulerPolicy policy;
-	std::shared_ptr<ThreadPool> wg = std::make_shared<ThreadPool>(policy);
+	std::shared_ptr<ThreadPoolExecutor> wg = std::make_shared<ThreadPoolExecutor>(policy);
 	TaskChannel ch(wg);
 	DummyTaskListener listener(ch);