Some Random Crypto Guy
45 lines
1.3 KiB
C++
45 lines
1.3 KiB
C++
#include <iostream>
|
|
#include "threadpool_boost.h"
|
|
|
|
ThreadPool::ThreadPool(size_t numThreads)
|
|
: workGuard(boost::asio::make_work_guard(ioService)) {
|
|
for (size_t i = 0; i < numThreads; ++i) {
|
|
workers.emplace_back([this]() {
|
|
std::cerr << "Thread started" << std::endl;
|
|
ioService.run();
|
|
std::cerr << "Thread finished" << std::endl;
|
|
});
|
|
}
|
|
}
|
|
|
|
void ThreadPool::enqueue(std::function<void()> task) {
|
|
ioService.post([task]() {
|
|
try {
|
|
task(); // Run the task
|
|
} catch (const std::exception& e) {
|
|
std::cerr << "Exception in thread pool task: " << e.what() << std::endl;
|
|
} catch (...) {
|
|
std::cerr << "Unknown exception in thread pool task!" << std::endl;
|
|
}
|
|
});
|
|
}
|
|
|
|
bool ThreadPool::isStopping() const {
|
|
return ioService.stopped(); // Check if io_context has stopped
|
|
}
|
|
|
|
void ThreadPool::waitForCompletion() {
|
|
std::cout << "Waiting for completion...\n";
|
|
workGuard.reset(); // Allow ioService to stop when no more tasks
|
|
ioService.run(); // Ensure no threads are left hanging
|
|
|
|
for (auto &worker : workers) {
|
|
if (worker.joinable()) worker.join();
|
|
}
|
|
std::cout << "All threads joined.\n";
|
|
}
|
|
|
|
ThreadPool::~ThreadPool() {
|
|
waitForCompletion();
|
|
}
|