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

Add locking in dispatchFuture

Browse source
This commit is contained in:
Derek Williams 2011-02-28 11:48:51 -07:00
parent 4bd00bc92f
commit 92a858eee8
1 changed files with 15 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -78,18 +78,22 @@ trait MessageDispatcher extends Logging {
} else throw new IllegalActorStateException("Can't submit invocations to dispatcher since it's not started") } else throw new IllegalActorStateException("Can't submit invocations to dispatcher since it's not started")
private[akka] final def dispatchFuture(invocation: FutureInvocation): Unit = { private[akka] final def dispatchFuture(invocation: FutureInvocation): Unit = {
futures add invocation.uuid guard withGuard {
if (active.isOff) { active.switchOn { start } } futures add invocation.uuid
if (active.isOff) { active.switchOn { start } }
}
invocation.future.onComplete { f => invocation.future.onComplete { f =>
futures remove invocation.uuid guard withGuard {
if (futures.isEmpty && uuids.isEmpty) { futures remove invocation.uuid
shutdownSchedule match { if (futures.isEmpty && uuids.isEmpty) {
case UNSCHEDULED => shutdownSchedule match {
shutdownSchedule = SCHEDULED case UNSCHEDULED =>
Scheduler.scheduleOnce(shutdownAction, timeoutMs, TimeUnit.MILLISECONDS) shutdownSchedule = SCHEDULED
case SCHEDULED => Scheduler.scheduleOnce(shutdownAction, timeoutMs, TimeUnit.MILLISECONDS)
shutdownSchedule = RESCHEDULED case SCHEDULED =>
case RESCHEDULED => //Already marked for reschedule shutdownSchedule = RESCHEDULED
case RESCHEDULED => //Already marked for reschedule
}
} }
} }
} }