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

Added alter method to TransactionalRef

Browse source
This commit is contained in:
Peter Vlugter 2010-04-09 05:17:36 +08:00 • committed by Jonas Bonér
parent e780ba0f82
commit ebab76b348
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
akka-core/src/main/scala/stm/TransactionalState.scala
View file

@ -102,6 +102,13 @@ class TransactionalRef[T] extends Transactional {
ref.set(elem)
}
def alter(f: T => T): T = {
ensureIsInTransaction
ensureNotNull
ref.set(f(ref.get))
ref.get
}
def get: Option[T] = {
ensureIsInTransaction
if (ref.isNull) None
@ -171,6 +178,9 @@ class TransactionalRef[T] extends Transactional {
private def ensureIsInTransaction =
if (getThreadLocalTransaction eq null) throw new NoTransactionInScopeException
private def ensureNotNull =
if (ref.isNull) throw new RuntimeException("Cannot alter Ref's value when it is null")
}
object TransactionalMap {