Some usages of the removed signatures

akka-cluster-sharding-typed/src/test/scala/akka/cluster/sharding/typed/scaladsl/ClusterShardingPersistenceSpec.scala

@@ -47,7 +47,7 @@ object ClusterShardingPersistenceSpec {
     PersistentBehaviors.receive[Command, String, String](
       entityId,
       emptyState = "",
-      commandHandler = (_, state, cmd) ⇒ cmd match {
+      commandHandler = (state, cmd) ⇒ cmd match {
         case Add(s) ⇒ Effect.persist(s)
         case Get(replyTo) ⇒
           replyTo ! s"$entityId:$state"

akka-cluster-typed/src/test/scala/akka/cluster/typed/ClusterSingletonPersistenceSpec.scala

@@ -39,7 +39,7 @@ object ClusterSingletonPersistenceSpec {
     PersistentBehaviors.receive[Command, String, String](
       persistenceId = "TheSingleton",
       emptyState = "",
-      commandHandler = (_, state, cmd) ⇒ cmd match {
+      commandHandler = (state, cmd) ⇒ cmd match {
         case Add(s) ⇒ Effect.persist(s)
         case Get(replyTo) ⇒
           replyTo ! state

akka-persistence-typed/src/main/scala/akka/persistence/typed/scaladsl/PersistentBehaviors.scala

@@ -67,17 +67,6 @@ object PersistentBehaviors {
 
   }
 
-  /**
-   * INTERNAL API
-   */
-  @InternalApi private[akka] final class ByStateCommandHandler[Command, Event, State](
-    choice: State ⇒ CommandHandler[Command, Event, State])
-    extends CommandHandler[Command, Event, State] {
-
-    override def apply(state: State, cmd: Command): Effect[Event, State] =
-      choice(state)(state, cmd)
-
-  }
 }
 
 trait PersistentBehavior[Command, Event, State] extends DeferredBehavior[Command] {

akka-persistence-typed/src/test/java/akka/persistence/typed/javadsl/PersistentActorJavaDslTest.java

@@ -506,5 +506,4 @@ public class PersistentActorJavaDslTest extends JUnitSuite {
   }
   //event-wrapper
 
-  // FIXME test with by state command handler
 }

akka-persistence-typed/src/test/scala/docs/akka/persistence/typed/InDepthPersistentBehaviorSpec.scala

@@ -94,10 +94,8 @@ object InDepthPersistentBehaviorSpec {
 
   //#by-state-command-handler
   private val commandHandler: (BlogState, BlogCommand) ⇒ Effect[BlogEvent, BlogState] = { (state, command) ⇒
-    state match {
+    if (state.isEmpty) initial(state, command)
-      case state if state.isEmpty  ⇒ initial(state, command)
+    else postAdded(state, command)
-      case state if !state.isEmpty ⇒ postAdded(state, command)
-    }
   }
   //#by-state-command-handler