How to make boost::thread_group smaller having boost::asio::io_service::run in its threads?
Generally as I've seen it is common way to create thread pools via "io_service + thread_group". Its really great for const size thread pools. Or pools that can only get bigger. But I wonder how to make such pool smaller with out stopping all io_service?
So we have as shown
// class variables
asio::io_service io_service;
boost::thread_group threads;
asio::io_service::work *work;
// some pool init function
work = new asio::io_service::work(io_service);
int cores_number = boost::thread::hardware_concurrency();
for (std::size_t i = 0; i < cores_number; ++i)
threads.create_thread(boost::bind(&asio::io_service::run, &io_service));
// and now we can simply post tasks
io_service.post(boost::bind(&class_name::an_expensive_calculation, this, 42));
io_service.post(boost::bind(&class_name::a_long_running_task, this, 123));
// and it is really eazy to make pool biger - just call (mutexes may be required)
threads.create_thread(boost::bind(&asio::io_service::run, &io_service));
But what if we want to remove threads from our threads pool? we can not just simply call threads.remove_thread(thread* thrd);
because it will not stop running in it &asio::io_service::run
(IMHO) so I wonder - is it possible and how to really remove threads from such pool? (not just interrupting tham but waiting until current thread task is out of scope)?
Update:
Here is some simple compilable code: thread pool, with threads desired life time.
#include <stdio.h>
#include <iostream>
#include <fstream>
//Boost
#include <boost/asio.hpp>
#include <boost/thread.hpp>
#include <boost/thread/locks.hpp>
boost::asio::io_service io_service;
boost::asio::io_service::work *work;
boost::thread_group threads;
boost::mutex threads_creation;
int time_limit;
int calculate_the_answer_to_life_the_universe_and_everything(int i)
{
boost::this_thread::sleep(boost::posix_time::milliseconds(i));
std::cout << i << std::endl;
return i;
}
void run(boost::shared_ptr<boost::thread> thread_ptr)
{
try
{
io_service.run();
}
catch(std::exception &e)
{
std::cout << "exeption: " << e.what() << std::endl;
boost::mutex::scoped_lock lock(threads_creation);
threads.remove_thread(thread_ptr.get());
lock.unlock();
std::cout << "thread removed from group" << std::endl;
return;
}
}
void pool_item( int i)
{
boost::packaged_task<int> pt(boost::bind(calculate_the_answer_to_life_the_universe_and_everything, i));
boost::unique_future<int> fi=pt.get_future();
boost::thread *task = new boost::thread(std::move(pt)); // launch task on a thread
if(fi.timed_wait(boost::posix_time::milliseconds(time_limit)))
{
std::cout << "sucsess function returned: " << fi.get() << std::endl;
}
else
{
std::cout << "request took way 2 long!" << std::endl;
std::cout << "current group size:" << threads.size() << std::endl;
boost::shared_ptr<boost::thread> thread;
boost::packaged_task<void> pt(boost::bind(run, thread));
thread = boost::shared_ptr<boost::thread>( new boost::thread(std::move(pt)));
boost::mutex::scoped_lock lock(threads_creation);
threads.add_thread(thread.get());
lock.unlock();
task->join();
throw std::runtime_error("killed joined thread");
}
}
int main()
{
time_limit = 500;
work = new boost::asio::io_service::work(io_service);
int cores_number = boost::thread::hardware_concurrency();
for (std::size_t i = 0; i < cores_number; ++i)
{
boost::shared_ptr<boost::thread> thread;
boost::packaged_task<void> pt(boost::bind(run, thread));
thread = boost::shared_ptr<boost::thread>( new boost::thread(std::move(pt)));
threads.add_thread(thread.get());
}
int i = 800;
io_service.post(boost::bind(pool_item, i));
boost::this_thread::sleep(boost::posix_time::milliseconds(i*2));
std::cout << "thread should be removed by now." << std::endl
<< "group size:" << threads.size() << std::endl;
std::cin.get();
return 0;
}
As you can see threads are not removed from threads pool even after .remove_thread(ptr);
call.=( Why?
Update #2:
Well any way I ended up with costume thread group...
#include <stdio.h>
#include <iostream>
#include <fstream>
#include <set>
//Boost
#include <boost/asio.hpp>
#include <boost/thread.hpp>
#include <boost/thread/locks.hpp>
//cf service interface
//#include <service.hpp>
//cf-server
//#include <server.h>
#include <boost/foreach.hpp>
class thread_group
{
public:
void add( boost::shared_ptr<boost::thread> to_add)
{
boost::mutex::scoped_lock lock(m);
ds_.insert(to_add);
}
void remove( boost::shared_ptr<boost::thread> to_remove)
{
boost::mutex::scoped_lock lock(m);
ds_.erase(to_remove);
}
int size()
{
boost::mutex::scoped_lock lock(m);
return ds_.size();
}
void join_all(boost::posix_time::milliseconds interuption_time=boost::posix_time::milliseconds(1000))
{
boost::mutex::scoped_lock lock(m);
BOOST_FOREACH(boost::shared_ptr<boost::thread> t, ds_)
{
boost::thread interrupter(boost::bind(&thread_group::interupt_thread, this, t, interuption_time));
}
}
private:
std::set< boost::shared_ptr<boost::thread> > ds_;
boost::mutex m;
void interupt_thread(boost::shared_ptr<boost::thread> t, boost::posix_time::milliseconds interuption_time)
{
try
{
if(!t->timed_join(interuption_time))
t->interrupt();
}
catch(std::exception &e)
{
}
}
};
boost::asio::io_service io_service;
boost::asio::io_service::work *work;
thread_group threads;
int time_limit;
int calculate_the_answer_to_life_the_universe_and_everything(int i)
{
boost::this_thread::sleep(boost::posix_time::milliseconds(i));
std::cout << i << std::endl;
return i;
}
void run(boost::shared_ptr<boost::thread> thread_ptr)
{
try
{
io_service.run();
}
catch(std::exception &e)
{
std::cout << "exeption: " << e.what() << std::endl;
threads.remove(thread_ptr);
std::cout << "thread removed from group" << std::endl;
return;
}
}
void pool_item( int i)
{
boost::packaged_task<int> pt(boost::bind(calculate_the_answer_to_life_the_universe_and_everything, i));
boost::unique_future<int> fi=pt.get_future();
boost::thread *task = new boost::thread(std::move(pt)); // launch task on a thread
if(fi.timed_wait(boost::posix_time::milliseconds(time_limit)))
{
std::cout << "sucsess function returned: " << fi.get() << std::endl;
}
else
{
std::cout << "request took way 2 long!" << std::endl;
std::cout << "current group size:" << threads.size() << std::endl;
std::cout << "we want to add thread!" << std::endl;
boost::shared_ptr<boost::thread> thread;
boost::packaged_task<void> pt(boost::bind(run, thread));
threads.add(thread);
std::cout << "thread added" << std::endl
<< "current group size:" << threads.size() << std::endl;
task->join();
throw std::runtime_error("killed joined thread");
}
}
int main()
{
time_limit = 500;
work = new boost::asio::io_service::work(io_se开发者_开发技巧rvice);
int cores_number = boost::thread::hardware_concurrency();
for (std::size_t i = 0; i < cores_number; ++i)
{
boost::shared_ptr<boost::thread> thread;
boost::packaged_task<void> pt(boost::bind(run, thread));
thread = boost::shared_ptr<boost::thread>( new boost::thread(std::move(pt)));
threads.add(thread);
}
int i = 800;
io_service.post(boost::bind(pool_item, i));
boost::this_thread::sleep(boost::posix_time::milliseconds(i*2));
std::cout << "thread should be removed by now." << std::endl
<< "group size:" << threads.size() << std::endl;
std::cin.get();
return 0;
}
I have been able to achieve this in the past by leveraging the fact that run()
will exit if a callback throws an exception. Instead of launching run()
directly in the thread, I call a utility function that exits the thread if the appropriate exception is thrown:
void RunIOService()
{
try
{
io_service.run();
}
catch(std::exception ex)
{
}
}
Then all you have to do is schedule a callback that will throw an exception:
static void KillThreadCallback()
{
// throw some exception that you catch above
}
io_service.post(&KillThreadCallback);
This will cause the thread that executes this callback to exit, essentially lowering the thread pool count size by 1. Using this, you can expand and contract the io_service
thread pool pretty easily.
One pattern that could be used for cleanly shutting down the I/O service is (using C++0x lambdas):
void ThreadLoop()
{
while(m_keepRunning) {
try {
io_service.run_one();
} catch(const std::exception& e) {
// error handling
}
}
}
void Stop()
{
// Using C++0x lambdas
io_service.post([=]{ m_keepRunning = false; });
// or
io_service.post(boost::bind(&ThisClass::StopCallback, this));
}
void StopCallback()
{
m_keepRunning = false;
}
Where m_keepRunning is a member variable. Should only be touched in the I/O service thread.
精彩评论