raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 11

Added tests are fixed some bugs

Browse source
This commit is contained in:
Viktor Klang 2010-08-13 13:13:27 +02:00
parent f0ac45f374
commit b6444b12a5
4 changed files with 176 additions and 68 deletions
Download patch file Download diff file Expand all files Collapse all files

57
akka-core/src/main/scala/dispatch/ThreadPoolBuilder.scala
View file

@ -10,7 +10,7 @@ import atomic.{AtomicLong, AtomicInteger}
import ThreadPoolExecutor.CallerRunsPolicy
import se.scalablesolutions.akka.actor.IllegalActorStateException
import se.scalablesolutions.akka.util.Logging
import se.scalablesolutions.akka.util.{Logger, Logging}
trait ThreadPoolBuilder {
val name: String
@ -22,7 +22,7 @@ trait ThreadPoolBuilder {
private var threadPoolBuilder: ThreadPoolExecutor = _
private var boundedExecutorBound = -1
private var inProcessOfBuilding = false
@volatile private var inProcessOfBuilding = false
private var blockingQueue: BlockingQueue[Runnable] = _
private lazy val threadFactory = new MonitorableThreadFactory(name)
@ -34,7 +34,6 @@ trait ThreadPoolBuilder {
def buildThreadPool(): Unit = synchronized {
ensureNotActive
inProcessOfBuilding = false
threadPoolBuilder.allowCoreThreadTimeOut(true)
if (boundedExecutorBound > 0) {
val boundedExecutor = new BoundedExecutorDecorator(threadPoolBuilder, boundedExecutorBound)
boundedExecutorBound = -1
@ -103,46 +102,58 @@ trait ThreadPoolBuilder {
this
}
def configureIfPossible(f: (ThreadPoolBuilder) => Unit): Boolean = synchronized {
if(inProcessOfBuilding) {
f(this)
true
}
else {
Logger(getClass).info("Tried to configure an already started ThreadPoolBuilder")
false
}
}
/**
* Default is 16.
*/
def setCorePoolSize(size: Int): ThreadPoolBuilder = synchronized {
ensureNotActive
verifyInConstructionPhase
threadPoolBuilder.setCorePoolSize(size)
this
}
def setCorePoolSize(size: Int): ThreadPoolBuilder =
setThreadPoolExecutorProperty(_.setCorePoolSize(size))
/**
* Default is 128.
*/
def setMaxPoolSize(size: Int): ThreadPoolBuilder = synchronized {
ensureNotActive
verifyInConstructionPhase
threadPoolBuilder.setMaximumPoolSize(size)
this
}
def setMaxPoolSize(size: Int): ThreadPoolBuilder =
setThreadPoolExecutorProperty(_.setMaximumPoolSize(size))
/**
* Default is 60000 (one minute).
*/
def setKeepAliveTimeInMillis(time: Long): ThreadPoolBuilder = synchronized {
ensureNotActive
verifyInConstructionPhase
threadPoolBuilder.setKeepAliveTime(time, MILLISECONDS)
this
}
def setKeepAliveTimeInMillis(time: Long): ThreadPoolBuilder =
setThreadPoolExecutorProperty(_.setKeepAliveTime(time, MILLISECONDS))
/**
* Default ThreadPoolExecutor.CallerRunsPolicy. To allow graceful backing off when pool is overloaded.
*/
def setRejectionPolicy(policy: RejectedExecutionHandler): ThreadPoolBuilder = synchronized {
def setRejectionPolicy(policy: RejectedExecutionHandler): ThreadPoolBuilder =
setThreadPoolExecutorProperty(_.setRejectedExecutionHandler(policy))
/**
* Default false, set to true to conserve thread for potentially unused dispatchers
*/
def setAllowCoreThreadTimeout(allow: Boolean) =
setThreadPoolExecutorProperty(_.allowCoreThreadTimeOut(allow))
/**
* Default ThreadPoolExecutor.CallerRunsPolicy. To allow graceful backing off when pool is overloaded.
*/
protected def setThreadPoolExecutorProperty(f: (ThreadPoolExecutor) => Unit): ThreadPoolBuilder = synchronized {
ensureNotActive
verifyInConstructionPhase
threadPoolBuilder.setRejectedExecutionHandler(policy)
f(threadPoolBuilder)
this
}
protected def verifyNotInConstructionPhase = {
if (inProcessOfBuilding) throw new IllegalActorStateException("Is already in the process of building a thread pool")
inProcessOfBuilding = true