/*:*
 *: File: ./src/thrd/pool.cpp
 *: 
 *: yChat; Homepage: www.yChat.org; Version 0.7.9.5-RELEASE
 *: 
 *: Copyright (C) 2003 Paul C. Buetow, Volker Richter
 *: Copyright (C) 2004 Paul C. Buetow
 *: Copyright (C) 2005 EXA Digital Solutions GbR
 *: 
 *: This program is free software; you can redistribute it and/or
 *: modify it under the terms of the GNU General Public License
 *: as published by the Free Software Foundation; either version 2
 *: of the License, or (at your option) any later version.
 *: 
 *: This program is distributed in the hope that it will be useful,
 *: but WITHOUT ANY WARRANTY; without even the implied warranty of
 *: MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *: GNU General Public License for more details.
 *: 
 *: You should have received a copy of the GNU General Public License
 *: along with this program; if not, write to the Free Software
 *: Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 *:*/

#ifndef POOL_CPP
#define POOL_CPP

#include "pool.h"

using namespace std;

pool::pool()
{
  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_avail_threads = tool::string2int( wrap::CONF->get_elem( "httpd.thread.initpoolsize" ) );
  increase_pool(i_num_avail_threads);
}

pool::~pool()
{
  pthread_mutex_lock(&mut_queue_tasks);
  while (!queue_tasks.empty())
  {
    delete queue_tasks.front();
    queue_tasks.pop();
  }
  pthread_mutex_unlock(&mut_queue_tasks);

  pthread_mutex_destroy(&mut_threads);
  pthread_mutex_destroy(&mut_queue_tasks);
  pthread_mutex_destroy(&mut_num_avail_threads);
  pthread_cond_destroy(&cond_new_task);
}

int
pool::increase_pool(int i_num)
{
  wrap::system_message(POOLFLL + tool::int2string(i_num) +","+tool::int2string(i_num_total_threads)+")");
  int i_max_pool_size = tool::string2int( wrap::CONF->get_elem( "httpd.thread.maxpoolsize" ) );

  for ( int i = 0; i < i_num; ++i )
  {
    if ( i_max_pool_size != 0 && i_num_total_threads >= i_max_pool_size )
    {
      wrap::system_message(POOLER2+tool::int2string(i_max_pool_size)+")");
      wrap::system_message(POOLER1+tool::int2string(i)+")");
      return i;
    }

    ++i_num_total_threads;
    pthread_t p_pthread;
    pthread_create(&p_pthread, 0, wait_for_task, (void*) this );
  }

  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 (;;)
  {

#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 )
    {
      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;
    }
    pthread_mutex_unlock(&p_pool->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_unlock(&p_pool->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);
  }

  return 0;
}

void
pool::run(void* p_void)
{
  add_task(run_func, p_void);
}

void
pool::run_func(void *p_void)
{
  socketcontainer* p_sock = static_cast<socketcontainer*>(p_void);
  wrap::SOCK->read_write(p_sock);
}

bool
pool::allow_user_login()
{
  pthread_mutex_lock(&mut_num_avail_threads);
  if ( i_num_avail_threads < 2 )
  {
    int i_max_pool_size = tool::string2int( wrap::CONF->get_elem( "httpd.thread.maxpoolsize" ) );
    if ( i_max_pool_size != 0 && i_max_pool_size == i_num_total_threads )
    {
      pthread_mutex_unlock(&mut_num_avail_threads);
      return false;
    }
  }
  pthread_mutex_unlock(&mut_num_avail_threads);

  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