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

Removed boilerplate, added final optmization

Browse source
This commit is contained in:
Viktor Klang 2010-10-25 00:38:48 +02:00
parent a630caecf3
commit b075b80db9
1 changed files with 14 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

27
akka-actor/src/main/scala/dispatch/MessageHandling.scala
View file

@ -62,9 +62,10 @@ object MessageDispatcher {
* @author <a href="http://jonasboner.com">Jonas Bon&#233;r</a>
*/
trait MessageDispatcher extends MailboxFactory with Logging {
import MessageDispatcher._
protected val uuids = new ConcurrentSkipListSet[Uuid]
protected val guard = new ReentrantGuard
private val shutdownSchedule = new AtomicInteger(MessageDispatcher.UNSCHEDULED)
private var shutdownSchedule = UNSCHEDULED //This can be non-volatile since it is protected by guard withGuard
protected val active = new Switch(false)
/**
@ -99,13 +100,13 @@ trait MessageDispatcher extends MailboxFactory with Logging {
if (uuids remove actorRef.uuid) {
actorRef.mailbox = null
if (uuids.isEmpty){
shutdownSchedule.get() match {
case MessageDispatcher.UNSCHEDULED =>
shutdownSchedule.set(MessageDispatcher.SCHEDULED)
shutdownSchedule match {
case UNSCHEDULED =>
shutdownSchedule = SCHEDULED
Scheduler.scheduleOnce(shutdownAction, timeoutMs, TimeUnit.MILLISECONDS)
case MessageDispatcher.SCHEDULED =>
shutdownSchedule.set(MessageDispatcher.RESCHEDULED)
case MessageDispatcher.RESCHEDULED => //Already marked for reschedule
case SCHEDULED =>
shutdownSchedule = RESCHEDULED
case RESCHEDULED => //Already marked for reschedule
}
}
}
@ -128,18 +129,18 @@ trait MessageDispatcher extends MailboxFactory with Logging {
private val shutdownAction = new Runnable {
def run = guard withGuard {
shutdownSchedule.get() match {
case MessageDispatcher.RESCHEDULED =>
shutdownSchedule.set(MessageDispatcher.SCHEDULED)
shutdownSchedule match {
case RESCHEDULED =>
shutdownSchedule = SCHEDULED
Scheduler.scheduleOnce(this, timeoutMs, TimeUnit.MILLISECONDS)
case MessageDispatcher.SCHEDULED =>
case SCHEDULED =>
if (uuids.isEmpty()) {
active switchOff {
shutdown // shut down in the dispatcher's references is zero
}
}
shutdownSchedule.set(MessageDispatcher.UNSCHEDULED)
case MessageDispatcher.UNSCHEDULED => //Do nothing
shutdownSchedule = UNSCHEDULED
case UNSCHEDULED => //Do nothing
}
}
}