Horrible hack to access latestSeqNr in typed persistence akka#25113

johanandren · Mar 5, 2019 · 9df2087 · 9df2087
1 parent 6071d89
commit 9df2087
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 3 deletions.
diff --git a/akka-actor-typed/src/main/scala/akka/actor/typed/internal/adapter/ActorAdapter.scala b/akka-actor-typed/src/main/scala/akka/actor/typed/internal/adapter/ActorAdapter.scala
@@ -40,6 +40,7 @@ import akka.util.OptionVal
   import Behavior._
 
   protected var behavior: Behavior[T] = _initialBehavior
+  def getBehavior: Behavior[T] = behavior
 
   private var _ctx: ActorContextAdapter[T] = _
   def ctx: ActorContextAdapter[T] =
@@ -211,7 +212,7 @@ import akka.util.OptionVal
   }
 
   protected def initializeContext(): Unit = {
-    _ctx = new ActorContextAdapter[T](context)
+    _ctx = new ActorContextAdapter[T](context, this)
   }
 }
 

diff --git a/akka-actor-typed/src/main/scala/akka/actor/typed/internal/adapter/ActorContextAdapter.scala b/akka-actor-typed/src/main/scala/akka/actor/typed/internal/adapter/ActorContextAdapter.scala
@@ -17,10 +17,12 @@ import scala.concurrent.duration._
 /**
  * INTERNAL API. Wrapping an [[akka.actor.ActorContext]] as an [[TypedActorContext]].
  */
-@InternalApi private[akka] final class ActorContextAdapter[T](val untypedContext: untyped.ActorContext) extends ActorContextImpl[T] {
+@InternalApi private[akka] final class ActorContextAdapter[T](val untypedContext: untyped.ActorContext, adapter: ActorAdapter[T]) extends ActorContextImpl[T] {
 
   import ActorRefAdapter.toUntyped
 
+  private[akka] def currentBehavior: Behavior[T] = adapter.getBehavior
+
   // lazily initialized
   private var actorLogger: OptionVal[Logger] = OptionVal.None
 

diff --git a/akka-actor-typed/src/main/scala/akka/actor/typed/scaladsl/ActorContext.scala b/akka-actor-typed/src/main/scala/akka/actor/typed/scaladsl/ActorContext.scala
@@ -36,7 +36,7 @@ import akka.annotation.InternalApi
  */
 @DoNotInherit
 @ApiMayChange
-trait ActorContext[T] extends TypedActorContext[T] { this: akka.actor.typed.javadsl.ActorContext[T] ⇒
+trait ActorContext[T] extends TypedActorContext[T] {
 
   /**
    * Get the `javadsl` of this `ActorContext`.

diff --git a/...istence-typed/src/main/scala/akka/persistence/typed/scaladsl/PersistentActorContext.scala b/...istence-typed/src/main/scala/akka/persistence/typed/scaladsl/PersistentActorContext.scala
@@ -0,0 +1,39 @@
+/*
+ * Copyright (C) 2009-2019 Lightbend Inc. <https://www.lightbend.com>
+ */
+
+package akka.persistence.typed.scaladsl
+
+import akka.actor.typed.internal.adapter.ActorContextAdapter
+import akka.actor.typed.scaladsl.ActorContext
+import akka.persistence.typed.internal.Running
+
+object PersistentActorContext {
+
+  /**
+   * Access the to details about the persistent actor from the actor context, can only be used in response
+   * to a command, will throw `IllegalState` for other cases
+   */
+  def apply(context: ActorContext[_]): PersistentActorContext = {
+    context match {
+      case impl: ActorContextAdapter[_] ⇒
+        impl.currentBehavior match {
+          case w: Running.WithStateAccessible ⇒
+            new PersistentActorContext {
+              def lastSequenceNr: Long = w.state.seqNr
+            }
+
+          case s ⇒ throw new IllegalStateException(s"Cannot extract the PersistentActorContext in state ${s.getClass.getName}")
+
+        }
+
+      case c ⇒ throw new IllegalStateException(s"Cannot extract the PersistentActorContext from context ${c.getClass.getName}")
+    }
+  }
+
+}
+
+trait PersistentActorContext {
+  def lastSequenceNr: Long
+}
+