diff options
Diffstat (limited to 'src/thrd/pool.cpp')
| -rw-r--r-- | src/thrd/pool.cpp | 113 |
1 files changed, 53 insertions, 60 deletions
diff --git a/src/thrd/pool.cpp b/src/thrd/pool.cpp index a27beac..4574620 100644 --- a/src/thrd/pool.cpp +++ b/src/thrd/pool.cpp @@ -1,7 +1,7 @@ /*:* *: File: ./src/thrd/pool.cpp *: - *: yChat; Homepage: www.yChat.org; Version 0.7.9.5-RELEASE + *: yChat; Homepage: www.yChat.org; Version 0.8.3-CURRENT *: *: Copyright (C) 2003 Paul C. Buetow, Volker Richter *: Copyright (C) 2004 Paul C. Buetow @@ -29,26 +29,44 @@ using namespace std; -pool::pool() +int pool::i_num_avail_threads; +int pool::i_num_total_threads; + +int pool::i_max_queue_size; +int pool::i_cur_queue_index; +int pool::i_free_queue_index; + +task** pool::queue_tasks; + +pthread_mutex_t pool::mut_threads; +pthread_mutex_t pool::mut_queue_tasks; +pthread_mutex_t pool::mut_num_avail_threads; +pthread_cond_t pool::cond_new_task; + +void +pool::init() { pthread_mutex_init(&mut_threads, 0); pthread_mutex_init(&mut_queue_tasks, 0); pthread_mutex_init(&mut_num_avail_threads, 0); pthread_cond_init(&cond_new_task, 0); - i_num_total_threads = 0; + i_num_total_threads = i_cur_queue_index = i_free_queue_index = 0; i_num_avail_threads = tool::string2int( wrap::CONF->get_elem( "httpd.thread.initpoolsize" ) ); + i_max_queue_size = tool::string2int( wrap::CONF->get_elem( "httpd.thread.maxqueuesize" ) ); + + queue_tasks = new task*[i_max_queue_size]; + for (int i = 0; i < i_max_queue_size; ++i) + queue_tasks[i] = NULL; + increase_pool(i_num_avail_threads); } -pool::~pool() +void +pool::destroy() { pthread_mutex_lock(&mut_queue_tasks); - while (!queue_tasks.empty()) - { - delete queue_tasks.front(); - queue_tasks.pop(); - } + delete [] queue_tasks; pthread_mutex_unlock(&mut_queue_tasks); pthread_mutex_destroy(&mut_threads); @@ -74,60 +92,43 @@ pool::increase_pool(int i_num) ++i_num_total_threads; pthread_t p_pthread; - pthread_create(&p_pthread, 0, wait_for_task, (void*) this ); + pthread_create(&p_pthread, 0, wait_for_task, (void*) p_pthread ); } return i_num; } -void -pool::add_task( void(*p_func)(void*), void* p_void ) -{ - pthread_mutex_lock(&mut_queue_tasks); - queue_tasks.push(new task(p_func, p_void)); - pthread_mutex_unlock(&mut_queue_tasks); - - pthread_cond_signal(&cond_new_task); - -} - void* pool::wait_for_task( void* p_void ) { - pool* p_pool = static_cast<pool*>(p_void); - for (;;) { + pthread_mutex_lock(&mut_threads); + pthread_cond_wait(&cond_new_task, &mut_threads); -#ifdef NCURSES - p_pool->print_pool_size(); -#endif - - pthread_mutex_lock(&p_pool->mut_threads); - pthread_cond_wait(&p_pool->cond_new_task, &p_pool->mut_threads); - - pthread_mutex_lock(&p_pool->mut_num_avail_threads); - if ( --p_pool->i_num_avail_threads < 5 ) + pthread_mutex_lock(&mut_num_avail_threads); + if ( --i_num_avail_threads < 5 ) { - int i_size = 9 - p_pool->i_num_avail_threads; - i_size = p_pool->increase_pool(i_size); - p_pool->i_num_avail_threads += i_size; + int i_size = 9 - i_num_avail_threads; + i_size = increase_pool(i_size); + i_num_avail_threads += i_size; } - pthread_mutex_unlock(&p_pool->mut_num_avail_threads); + pthread_mutex_unlock(&mut_num_avail_threads); - pthread_mutex_lock(&p_pool->mut_queue_tasks); - task* p_task = p_pool->queue_tasks.front(); - p_pool->queue_tasks.pop(); - pthread_mutex_unlock(&p_pool->mut_queue_tasks); + pthread_mutex_lock(&mut_queue_tasks); + task* p_task = queue_tasks[i_cur_queue_index]; + queue_tasks[i_cur_queue_index++] = NULL; + i_cur_queue_index %= i_max_queue_size; + pthread_mutex_unlock(&mut_queue_tasks); - pthread_mutex_unlock(&p_pool->mut_threads); + pthread_mutex_unlock(&mut_threads); (*(p_task->p_func))(p_task->p_void); delete p_task; - pthread_mutex_lock(&p_pool->mut_num_avail_threads); - p_pool->i_num_avail_threads++; - pthread_mutex_unlock(&p_pool->mut_num_avail_threads); + pthread_mutex_lock(&mut_num_avail_threads); + i_num_avail_threads++; + pthread_mutex_unlock(&mut_num_avail_threads); } return 0; @@ -136,13 +137,19 @@ pool::wait_for_task( void* p_void ) void pool::run(void* p_void) { - add_task(run_func, p_void); + pthread_mutex_lock(&mut_queue_tasks); + queue_tasks[i_free_queue_index++] = new task(run_func, p_void); + i_free_queue_index %= i_max_queue_size; + + pthread_mutex_unlock(&mut_queue_tasks); + + pthread_cond_signal(&cond_new_task); } void pool::run_func(void *p_void) { - socketcontainer* p_sock = static_cast<socketcontainer*>(p_void); + _socket* p_sock = static_cast<_socket*>(p_void); wrap::SOCK->read_write(p_sock); } @@ -164,18 +171,4 @@ pool::allow_user_login() return true; } -#ifdef NCURSES -void -pool::print_pool_size() -{ - if ( wrap::NCUR->is_ready() ) - { - pthread_mutex_lock(&mut_num_avail_threads); - mvprintw( NCUR_POOL_WAIT_X,NCUR_POOL_WAIT_Y, "Wait/Tot: %d/%d ", i_num_avail_threads, i_num_total_threads); - mvprintw( NCUR_POOL_RUNNING_X,NCUR_POOL_RUNNING_Y, "Running: %d ", i_num_total_threads-i_num_avail_threads); - pthread_mutex_unlock(&mut_num_avail_threads); - refresh(); - } -} -#endif #endif |
