blob: 2f3ea6a1153965cda305c42c86cf94c84b6db73d [file] [log] [blame]
package org.apache.jcs.utils.threads;
/*
* Copyright 2001-2004 The Apache Software Foundation.
*
* Licensed under the Apache License, Version 2.0 (the "License")
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import java.util.*;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
/**
* A thread pool that is trying to copy the apache process management. asmuts --
* Pulled out of tomcat, since it seems to move around and hide. Trying to
* reduce the number of dependencies
*
*/
public class ThreadPool
{
private final static Log log =
LogFactory.getLog( ThreadPool.class );
/*
* Default values ...
*/
/** Description of the Field */
public final static int MAX_THREADS = 200;
/** Description of the Field */
public final static int MAX_SPARE_THREADS = 50;
/** Description of the Field */
public final static int MIN_SPARE_THREADS = 4;
/** Description of the Field */
public final static int WORK_WAIT_TIMEOUT = 60 * 1000;
/*
* Where the threads are held.
*/
/** Description of the Field */
protected Vector pool;
/*
* A monitor thread that monitors the pool for idel threads.
*/
/** Description of the Field */
protected MonitorRunnable monitor;
/*
* Max number of threads that you can open in the pool.
*/
/** Description of the Field */
protected int maxThreads;
/*
* Min number of idel threads that you can leave in the pool.
*/
/** Description of the Field */
protected int minSpareThreads;
/*
* Max number of idel threads that you can leave in the pool.
*/
/** Description of the Field */
protected int maxSpareThreads;
/*
* Number of threads in the pool.
*/
/** Description of the Field */
protected int currentThreadCount;
/*
* Number of busy threads in the pool.
*/
/** Description of the Field */
protected int currentThreadsBusy;
/*
* Flag that the pool should terminate all the threads and stop.
*/
/** Description of the Field */
protected boolean stopThePool;
/** Constructor for the ThreadPool object */
public ThreadPool()
{
maxThreads = MAX_THREADS;
maxSpareThreads = MAX_SPARE_THREADS;
minSpareThreads = MIN_SPARE_THREADS;
currentThreadCount = 0;
currentThreadsBusy = 0;
stopThePool = false;
}
/** Description of the Method */
public synchronized void start()
{
stopThePool = false;
currentThreadCount = 0;
currentThreadsBusy = 0;
maxThreads = MAX_THREADS;
maxSpareThreads = MAX_SPARE_THREADS;
minSpareThreads = MIN_SPARE_THREADS;
adjustLimits();
openThreads( minSpareThreads );
monitor = new MonitorRunnable( this );
}
/**
* Sets the maxThreads attribute of the ThreadPool object
*
* @param maxThreads The new maxThreads value
*/
public void setMaxThreads( int maxThreads )
{
this.maxThreads = maxThreads;
}
/**
* Gets the maxThreads attribute of the ThreadPool object
*
* @return The maxThreads value
*/
public int getMaxThreads()
{
return maxThreads;
}
/**
* Sets the minSpareThreads attribute of the ThreadPool object
*
* @param minSpareThreads The new minSpareThreads value
*/
public void setMinSpareThreads( int minSpareThreads )
{
this.minSpareThreads = minSpareThreads;
}
/**
* Gets the minSpareThreads attribute of the ThreadPool object
*
* @return The minSpareThreads value
*/
public int getMinSpareThreads()
{
return minSpareThreads;
}
/**
* Sets the maxSpareThreads attribute of the ThreadPool object
*
* @param maxSpareThreads The new maxSpareThreads value
*/
public void setMaxSpareThreads( int maxSpareThreads )
{
this.maxSpareThreads = maxSpareThreads;
}
/**
* Gets the maxSpareThreads attribute of the ThreadPool object
*
* @return The maxSpareThreads value
*/
public int getMaxSpareThreads()
{
return maxSpareThreads;
}
// You may wonder what you see here ... basically I am trying
// to maintain a stack of threads. This way locality in time
// is kept and there is a better chance to find residues of the
// thread in memory next time it runs.
/**
* Executes a given Runnable on a thread in the pool, block if needed.
*/
public void runIt( IThreadPoolRunnable r )
{
if ( null == r )
{
throw new NullPointerException();
}
if ( 0 == currentThreadCount || stopThePool )
{
throw new IllegalStateException();
}
ControlRunnable c = null;
// Obtain a free thread from the pool.
synchronized ( this )
{
if ( currentThreadsBusy == currentThreadCount )
{
// All threads are busy
if ( currentThreadCount < maxThreads )
{
// Not all threads were open,
// Open new threads up to the max number of idel threads
int toOpen = currentThreadCount + minSpareThreads;
openThreads( toOpen );
}
else
{
logFull( log, currentThreadCount, maxThreads );
// Wait for a thread to become idel.
while ( currentThreadsBusy == currentThreadCount )
{
try
{
this.wait();
}
// was just catch Throwable -- but no other
// exceptions can be thrown by wait, right?
// So we catch and ignore this one, since
// it'll never actually happen, since nowhere
// do we say pool.interrupt().
catch ( InterruptedException e )
{
log.error( "Unexpected exception", e );
}
// Pool was stopped. Get away of the pool.
if ( 0 == currentThreadCount || stopThePool )
{
throw new IllegalStateException();
}
}
}
}
// If we are here it means that there is a free thred. Take it.
c = ( ControlRunnable ) pool.lastElement();
pool.removeElement( c );
currentThreadsBusy++;
}
c.runIt( r );
}
static boolean logfull = true;
/** Description of the Method */
public static void logFull( Log log, int currentThreadCount, int maxThreads )
{
if ( logfull )
{
log.debug( "All threads are busy, waiting. Please " +
"increase maxThreads or check the servlet" +
" status" + currentThreadCount + " " +
maxThreads );
logfull = false;
}
}
/** Stop the thread pool */
public synchronized void shutdown()
{
if ( !stopThePool )
{
stopThePool = true;
monitor.terminate();
monitor = null;
for ( int i = 0; i < ( currentThreadCount - currentThreadsBusy ); i++ )
{
try
{
( ( ControlRunnable ) ( pool.elementAt( i ) ) ).terminate();
}
catch ( Throwable t )
{
/*
* Do nothing... The show must go on, we are shutting
* down the pool and nothing should stop that.
*/
log.error( "Ignored exception while shutting down thread pool" );
}
}
currentThreadsBusy = currentThreadCount = 0;
pool = null;
notifyAll();
}
}
/** Called by the monitor thread to harvest idle threads. */
protected synchronized void checkSpareControllers()
{
if ( stopThePool )
{
return;
}
if ( ( currentThreadCount - currentThreadsBusy ) > maxSpareThreads )
{
int toFree = currentThreadCount -
currentThreadsBusy -
maxSpareThreads;
for ( int i = 0; i < toFree; i++ )
{
ControlRunnable c = ( ControlRunnable ) pool.firstElement();
pool.removeElement( c );
c.terminate();
currentThreadCount--;
}
}
}
/**
* Returns the thread to the pool. Called by threads as they are becoming
* idel.
*/
protected synchronized void returnController( ControlRunnable c )
{
if ( 0 == currentThreadCount || stopThePool )
{
c.terminate();
return;
}
currentThreadsBusy--;
pool.addElement( c );
notify();
}
/**
* Inform the pool that the specific thread finish. Called by the
* ControlRunnable.run() when the runnable throws an exception.
*/
protected synchronized void notifyThreadEnd( ControlRunnable c )
{
currentThreadsBusy--;
currentThreadCount--;
notify();
}
/*
* Checks for problematic configuration and fix it.
* The fix provides reasonable settings for a single CPU
* with medium load.
*/
/** Description of the Method */
protected void adjustLimits()
{
if ( maxThreads <= 0 )
{
maxThreads = MAX_THREADS;
}
if ( maxSpareThreads >= maxThreads )
{
maxSpareThreads = maxThreads;
}
if ( maxSpareThreads <= 0 )
{
if ( 1 == maxThreads )
{
maxSpareThreads = 1;
}
else
{
maxSpareThreads = maxThreads / 2;
}
}
if ( minSpareThreads > maxSpareThreads )
{
minSpareThreads = maxSpareThreads;
}
if ( minSpareThreads <= 0 )
{
if ( 1 == maxSpareThreads )
{
minSpareThreads = 1;
}
else
{
minSpareThreads = maxSpareThreads / 2;
}
}
}
/** Description of the Method */
protected void openThreads( int toOpen )
{
if ( toOpen > maxThreads )
{
toOpen = maxThreads;
}
if ( 0 == currentThreadCount )
{
pool = new Vector( toOpen );
}
for ( int i = currentThreadCount; i < toOpen; i++ )
{
pool.addElement( new ControlRunnable( this ) );
}
currentThreadCount = toOpen;
}
/** Description of the Method */
void log( String s )
{
log.debug( s );
}
/**
* Periodically execute an action - cleanup in this case
*
*/
class MonitorRunnable implements Runnable
{
ThreadPool p;
Thread t;
boolean shouldTerminate;
/**
* Constructor for the MonitorRunnable object
*
* @param p
*/
MonitorRunnable( ThreadPool p )
{
shouldTerminate = false;
this.p = p;
t = new Thread( this );
t.setName( "MonitorRunnable" );
t.start();
}
/** Main processing method for the MonitorRunnable object */
public void run()
{
while ( true )
{
try
{
// Sleep for a while.
synchronized ( this )
{
this.wait( WORK_WAIT_TIMEOUT );
}
// Check if should terminate.
// termination happens when the pool is shutting down.
if ( shouldTerminate )
{
break;
}
// Harvest idle threads.
p.checkSpareControllers();
}
catch ( Throwable t )
{
log.error( "Unexpected exception" );
}
}
}
/** Stop the monitor */
public synchronized void terminate()
{
shouldTerminate = true;
this.notify();
}
}
/**
* A Thread object that executes various actions ( IThreadPoolRunnable )
* under control of ThreadPool
*
*/
class ControlRunnable implements Runnable
{
/** ThreadPool where this thread will be returned */
ThreadPool p;
/** The thread that executes the actions */
Thread t;
/** The method that is executed in this thread */
IThreadPoolRunnable toRun;
/** Stop this thread */
boolean shouldTerminate;
/** Activate the execution of the action */
boolean shouldRun;
/**
* Per thread data - can be used only if all actions are of the same
* type. A better mechanism is possible ( that would allow association
* of thread data with action type ), but right now it's enough.
*/
boolean noThData;
Object thData[] = null;
/**
* Start a new thread, with no method in it
*
* @param p
*/
ControlRunnable( ThreadPool p )
{
toRun = null;
shouldTerminate = false;
shouldRun = false;
this.p = p;
t = new Thread( this );
t.start();
noThData = true;
thData = null;
}
/** Main processing method for the ControlRunnable object */
public void run()
{
while ( true )
{
try
{
/*
* Wait for work.
*/
synchronized ( this )
{
if ( !shouldRun && !shouldTerminate )
{
this.wait();
}
}
if ( toRun == null )
{
log.debug( "No toRun ???" );
}
if ( shouldTerminate )
{
log.debug( "Terminate" );
break;
}
/*
* Check if should execute a runnable.
*/
try
{
if ( noThData )
{
log.debug( "Getting new thread data" );
thData = toRun.getInitData();
noThData = false;
}
if ( shouldRun )
{
toRun.runIt( thData );
}
}
catch ( Throwable t )
{
log.error( "Caught exception executing " + toRun.toString() + ", terminating thread" );
/*
* The runnable throw an exception (can be even a ThreadDeath),
* signalling that the thread die.
*
* The meaning is that we should release the thread from
* the pool.
*/
shouldTerminate = true;
shouldRun = false;
p.notifyThreadEnd( this );
}
finally
{
if ( shouldRun )
{
shouldRun = false;
/*
* Notify the pool that the thread is now idle.
*/
p.returnController( this );
}
}
/*
* Check if should terminate.
* termination happens when the pool is shutting down.
*/
if ( shouldTerminate )
{
break;
}
}
catch ( InterruptedException ie )
{
/*
* for the wait operation
*/
// can never happen, since we don't call interrupt
log.error( "Unexpected exception", ie );
}
}
}
/** Description of the Method */
public synchronized void runIt( IThreadPoolRunnable toRun )
{
if ( toRun == null )
{
throw new NullPointerException( "No Runnable" );
}
this.toRun = toRun;
// Do not re-init, the whole idea is to run init only once per
// thread - the pool is supposed to run a single task, that is
// initialized once.
// noThData = true;
shouldRun = true;
this.notify();
}
/** Description of the Method */
public synchronized void terminate()
{
shouldTerminate = true;
this.notify();
}
}
}