114 lines
3.6 KiB
C++
114 lines
3.6 KiB
C++
#ifndef BOOST_NETWORK_UTILS_THREAD_POOL_HPP_20101020
|
|
#define BOOST_NETWORK_UTILS_THREAD_POOL_HPP_20101020
|
|
|
|
// Copyright 2010 Dean Michael Berris.
|
|
// Distributed under the Boost Software License, Version 1.0.
|
|
// (See accompanying file LICENSE_1_0.txt or copy at
|
|
// http://www.boost.org/LICENSE_1_0.txt)
|
|
|
|
#include <cstddef>
|
|
#include <boost/network/tags.hpp>
|
|
#include <boost/thread/thread.hpp>
|
|
#include <boost/shared_ptr.hpp>
|
|
#include <boost/function.hpp>
|
|
#include <boost/asio/io_service.hpp>
|
|
#include <boost/scope_exit.hpp>
|
|
|
|
namespace boost { namespace network { namespace utils {
|
|
|
|
typedef boost::shared_ptr<boost::asio::io_service> io_service_ptr;
|
|
typedef boost::shared_ptr<boost::thread_group> worker_threads_ptr;
|
|
typedef boost::shared_ptr<boost::asio::io_service::work> sentinel_ptr;
|
|
|
|
template <class Tag>
|
|
struct basic_thread_pool {
|
|
basic_thread_pool(
|
|
std::size_t threads = 1,
|
|
io_service_ptr io_service = io_service_ptr(),
|
|
worker_threads_ptr worker_threads = worker_threads_ptr()
|
|
)
|
|
: threads_(threads)
|
|
, io_service_(io_service)
|
|
, worker_threads_(worker_threads)
|
|
, sentinel_()
|
|
{
|
|
bool commit = false;
|
|
BOOST_SCOPE_EXIT_TPL((&commit)(&io_service_)(&worker_threads_)(&sentinel_)) {
|
|
if (!commit) {
|
|
sentinel_.reset();
|
|
io_service_.reset();
|
|
if (worker_threads_.get()) {
|
|
worker_threads_->interrupt_all();
|
|
worker_threads_->join_all();
|
|
}
|
|
worker_threads_.reset();
|
|
}
|
|
} BOOST_SCOPE_EXIT_END
|
|
|
|
if (!io_service_.get()) {
|
|
io_service_.reset(new boost::asio::io_service);
|
|
}
|
|
|
|
if (!worker_threads_.get()) {
|
|
worker_threads_.reset(new boost::thread_group);
|
|
}
|
|
|
|
if (!sentinel_.get()) {
|
|
sentinel_.reset(new boost::asio::io_service::work(*io_service_));
|
|
}
|
|
|
|
for (std::size_t counter = 0; counter < threads_; ++counter)
|
|
worker_threads_->create_thread(
|
|
boost::bind(
|
|
&boost::asio::io_service::run,
|
|
io_service_
|
|
)
|
|
);
|
|
|
|
commit = true;
|
|
}
|
|
|
|
std::size_t thread_count() const {
|
|
return threads_;
|
|
}
|
|
|
|
void post(boost::function<void()> f) {
|
|
io_service_->post(f);
|
|
}
|
|
|
|
~basic_thread_pool() throw () {
|
|
sentinel_.reset();
|
|
try {
|
|
worker_threads_->join_all();
|
|
} catch (...) {
|
|
BOOST_ASSERT(false && "A handler was not supposed to throw, but one did.");
|
|
}
|
|
}
|
|
|
|
void swap(basic_thread_pool & other) {
|
|
std::swap(other.threads_, threads_);
|
|
std::swap(other.io_service_, io_service_);
|
|
std::swap(other.worker_threads_, worker_threads_);
|
|
std::swap(other.sentinel_, sentinel_);
|
|
}
|
|
protected:
|
|
std::size_t threads_;
|
|
io_service_ptr io_service_;
|
|
worker_threads_ptr worker_threads_;
|
|
sentinel_ptr sentinel_;
|
|
|
|
private:
|
|
basic_thread_pool(basic_thread_pool const &); // no copies please
|
|
basic_thread_pool & operator=(basic_thread_pool); // no assignment please
|
|
};
|
|
|
|
typedef basic_thread_pool<tags::default_> thread_pool;
|
|
|
|
} /* utils */
|
|
|
|
} /* network */
|
|
|
|
} /* boost */
|
|
|
|
#endif /* BOOST_NETWORK_UTILS_THREAD_POOL_HPP_20101020 */
|