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 162d059112
commit b05be42f11
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")
private[akka] final def dispatchFuture(invocation: FutureInvocation): Unit = {
futures add invocation.uuid
if (active.isOff) { active.switchOn { start } }
guard withGuard {
futures add invocation.uuid
if (active.isOff) { active.switchOn { start } }
}
invocation.future.onComplete { f =>
futures remove invocation.uuid
if (futures.isEmpty && uuids.isEmpty) {
shutdownSchedule match {
case UNSCHEDULED =>
shutdownSchedule = SCHEDULED
Scheduler.scheduleOnce(shutdownAction, timeoutMs, TimeUnit.MILLISECONDS)
case SCHEDULED =>
shutdownSchedule = RESCHEDULED
case RESCHEDULED => //Already marked for reschedule
guard withGuard {
futures remove invocation.uuid
if (futures.isEmpty && uuids.isEmpty) {
shutdownSchedule match {
case UNSCHEDULED =>
shutdownSchedule = SCHEDULED
Scheduler.scheduleOnce(shutdownAction, timeoutMs, TimeUnit.MILLISECONDS)
case SCHEDULED =>
shutdownSchedule = RESCHEDULED
case RESCHEDULED => //Already marked for reschedule
}
}
}
}