PingPongActorMain

PingPongActorMain.scala

import akka.NotUsed
import akka.actor.typed.scaladsl.Behaviors
import akka.actor.typed.{ActorRef, ActorSystem, Behavior}

import scala.concurrent.duration.DurationInt


object PingPongActorMain extends App {

  case class Ping(counter: Int, replyTo: ActorRef[Pong])
  case class Pong(counter: Int, replyTo: ActorRef[Ping])

  val pingBehaviour = Behaviors.receive[Ping] {
    (context, message) => {
      println(s"Received Ping ${message.counter}")
      context.scheduleOnce(2 second, message.replyTo, Pong(message.counter, context.self))
      Behaviors.same[Ping]
    }
  }

  val pongBehaviour = Behaviors.receive[Pong] {
    (context, message) => {
      println(s"Received Pong ${message.counter}")
      context.scheduleOnce(2 second, message.replyTo, Ping(message.counter + 1, context.self))
      Behaviors.same[Pong]
    }
  }

  def setupPingPong(): Behavior[NotUsed] = {
    Behaviors.setup[NotUsed] {
      e => {
        val pingActor = e.spawn(pingBehaviour, "PingActor")
        val pongActor = e.spawn(pongBehaviour, "PongActor")

        pingActor.tell(Ping(0, pongActor))

        Behaviors.same
      }
    }
  }

  val system = ActorSystem(setupPingPong(), "hello")

}