2 /* $Id: thread_pool_observer.cpp,v 1.9 2005-11-04 11:06:52 adam Exp $
3 Copyright (c) 2005, Index Data.
16 #include <boost/thread/thread.hpp>
17 #include <boost/thread/mutex.hpp>
18 #include <boost/thread/condition.hpp>
23 #include <yaz++/socket-observer.h>
26 #include "thread_pool_observer.hpp"
29 class ThreadPoolSocketObserver::Worker {
31 Worker(ThreadPoolSocketObserver *s) : m_s(s) {};
32 ThreadPoolSocketObserver *m_s;
33 void operator() (void) {
38 class ThreadPoolSocketObserver::Rep : public boost::noncopyable {
39 friend class ThreadPoolSocketObserver;
41 Rep(yazpp_1::ISocketObservable *obs);
44 yazpp_1::ISocketObservable *m_socketObservable;
46 boost::thread_group m_thrds;
47 boost::mutex m_mutex_input_data;
48 boost::condition m_cond_input_data;
49 boost::mutex m_mutex_output_data;
50 std::deque<IThreadPoolMsg *> m_input;
51 std::deque<IThreadPoolMsg *> m_output;
58 using namespace yazpp_1;
61 ThreadPoolSocketObserver::Rep::Rep(ISocketObservable *obs)
62 : m_socketObservable(obs)
66 ThreadPoolSocketObserver::Rep::~Rep()
70 IThreadPoolMsg::~IThreadPoolMsg()
75 ThreadPoolSocketObserver::ThreadPoolSocketObserver(ISocketObservable *obs,
80 obs->addObserver(m_p->m_fd[0], this);
81 obs->maskObserver(this, SOCKET_OBSERVE_READ);
83 m_p->m_stop_flag = false;
84 m_p->m_no_threads = no_threads;
86 for (i = 0; i<no_threads; i++)
89 m_p->m_thrds.add_thread(new boost::thread(w));
93 ThreadPoolSocketObserver::~ThreadPoolSocketObserver()
96 boost::mutex::scoped_lock input_lock(m_p->m_mutex_input_data);
97 m_p->m_stop_flag = true;
98 m_p->m_cond_input_data.notify_all();
100 m_p->m_thrds.join_all();
102 m_p->m_socketObservable->deleteObserver(this);
108 void ThreadPoolSocketObserver::socketNotify(int event)
110 if (event & SOCKET_OBSERVE_READ)
113 read(m_p->m_fd[0], buf, 1);
116 boost::mutex::scoped_lock output_lock(m_p->m_mutex_output_data);
117 out = m_p->m_output.front();
118 m_p->m_output.pop_front();
125 void ThreadPoolSocketObserver::run(void *p)
129 IThreadPoolMsg *in = 0;
131 boost::mutex::scoped_lock input_lock(m_p->m_mutex_input_data);
132 while (!m_p->m_stop_flag && m_p->m_input.size() == 0)
133 m_p->m_cond_input_data.wait(input_lock);
134 if (m_p->m_stop_flag)
137 in = m_p->m_input.front();
138 m_p->m_input.pop_front();
140 IThreadPoolMsg *out = in->handle();
142 boost::mutex::scoped_lock output_lock(m_p->m_mutex_output_data);
143 m_p->m_output.push_back(out);
144 write(m_p->m_fd[1], "", 1);
149 void ThreadPoolSocketObserver::put(IThreadPoolMsg *m)
151 boost::mutex::scoped_lock input_lock(m_p->m_mutex_input_data);
152 m_p->m_input.push_back(m);
153 m_p->m_cond_input_data.notify_one();
158 * indent-tabs-mode: nil
159 * c-file-style: "stroustrup"
161 * vim: shiftwidth=4 tabstop=8 expandtab