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

fixed misc bugs and completed first iteration of transactions

Browse source
This commit is contained in:
Jonas Boner 2009-03-26 20:22:49 +01:00
parent 6cb38f6ce9
commit 123fa5bd50
8 changed files with 173 additions and 98 deletions
Download patch file Download diff file Expand all files Collapse all files

54
kernel/src/main/scala/Transaction.scala
View file

@ -21,6 +21,7 @@ object TransactionStatus {
* @author <a href="http://jonasboner.com">Jonas Bon&#233;r</a>
*/
object TransactionIdFactory {
// FIXME: will not work in distributed env
private val currentId = new AtomicLong(0L)
def newId = currentId.getAndIncrement
}
@ -31,8 +32,10 @@ object TransactionIdFactory {
* @author <a href="http://jonasboner.com">Jonas Bon&#233;r</a>
*/
class Transaction extends Logging {
val stateful= classOf[se.scalablesolutions.akka.annotation.stateful]
val id = TransactionIdFactory.newId
log.debug("Creating a new transaction [%s]", id)
private[this] var parent: Option[Transaction] = None
private[this] var oldActorVersions = new HashMap[GenericServerContainer, GenericServer]
private[this] var precommitted: List[GenericServerContainer] = Nil
@ -43,45 +46,52 @@ class Transaction extends Logging {
if (status == TransactionStatus.Completed) throw new IllegalStateException("Can't begin COMPLETED transaction")
if (status == TransactionStatus.New) log.debug("Actor [%s] is starting NEW transaction", server)
else log.debug("Actor [%s] is participating in transaction", server)
val oldVersion = server.cloneServerAndReturnOldVersion
oldActorVersions.put(server, oldVersion)
if (server.getServer.getClass.isAnnotationPresent(stateful)) {
val oldVersion = server.cloneServerAndReturnOldVersion
oldActorVersions.put(server, oldVersion)
}
status = TransactionStatus.Active
}
def precommit(server: GenericServerContainer) = synchronized {
ensureIsActive
log.debug("Pre-committing transaction for actor [%s]", server)
precommitted ::= server
if (status == TransactionStatus.Active) {
log.debug("Pre-committing transaction for actor [%s]", server)
precommitted ::= server
}
}
def commit(server: GenericServerContainer) = synchronized {
ensureIsActive
log.debug("Committing transaction for actor [%s]", server)
val haveAllPreCommitted =
if (oldActorVersions.size == precommitted.size) {{
for (server <- oldActorVersions.keys) yield {
if (precommitted.exists(_.id == server.id)) true
else false
}}.exists(_ == false)
} else false
if (haveAllPreCommitted) status = TransactionStatus.Completed
else rollback(server)
if (status == TransactionStatus.Active) {
log.debug("Committing transaction for actor [%s]", server)
val haveAllPreCommitted =
if (oldActorVersions.size == precommitted.size) {{
for (server <- oldActorVersions.keys) yield {
if (precommitted.exists(_.id == server.id)) true
else false
}}.exists(_ == false)
} else false
if (haveAllPreCommitted) status = TransactionStatus.Completed
else rollback(server)
}
}
def rollback(server: GenericServerContainer) = synchronized {
ensureIsActive
ensureIsActiveOrAborted
log.debug("Actor [%s] has initiated transaction rollback, rolling back [%s]" , server, oldActorVersions.keys)
oldActorVersions.foreach(entry => {
val (server, backup) = entry
server.swapServer(backup)
val (server, backup) = entry
server.swapServer(backup)
})
status = TransactionStatus.Aborted
}
private def ensureIsActive = if (status == TransactionStatus.Active)
private def ensureIsActive = if (status != TransactionStatus.Active)
throw new IllegalStateException("Expected ACTIVE transaction - current status [" + status + "]")
private def ensureIsActiveOrAborted =
if (!(status == TransactionStatus.Active || status == TransactionStatus.Aborted))
throw new IllegalStateException("Expected ACTIVE or ABORTED transaction - current status [" + status + "]")
override def equals(that: Any): Boolean =
that != null &&
that.isInstanceOf[Transaction] &&