pekko/akka-docs/scala/code/akka/docs/actor/SchedulerDocSpec.scala

/**
 * Copyright (C) 2009-2011 Typesafe Inc. <http://www.typesafe.com>
 */
package akka.docs.actor

//#imports1
import akka.actor.Actor
import akka.actor.Props
import akka.util.duration._

//#imports1

import org.scalatest.{ BeforeAndAfterAll, WordSpec }
import org.scalatest.matchers.MustMatchers
import akka.testkit._

class SchedulerDocSpec extends AkkaSpec(Map("akka.loglevel" -> "INFO")) {
  "schedule a one-off task" in {
    //#schedule-one-off-message
    //Schedules to send the "foo"-message to the testActor after 50ms
    system.scheduler.scheduleOnce(50 milliseconds, testActor, "foo")
    //#schedule-one-off-message

    expectMsg(1 second, "foo")

    //#schedule-one-off-thunk
    //Schedules a function to be executed (send the current time) to the testActor after 50ms
    system.scheduler.scheduleOnce(50 milliseconds) {
      testActor ! System.currentTimeMillis
    }
    //#schedule-one-off-thunk

  }

  "schedule a recurring task" in {
    //#schedule-recurring
    val Tick = "tick"
    val tickActor = system.actorOf(Props(new Actor {
      def receive = {
        case Tick ⇒ //Do something
      }
    }))
    //This will schedule to send the Tick-message
    //to the tickActor after 0ms repeating every 50ms
    val cancellable =
      system.scheduler.schedule(0 milliseconds,
        50 milliseconds,
        tickActor,
        Tick)

    //This cancels further Ticks to be sent
    cancellable.cancel()
    //#schedule-recurring
    system.stop(tickActor)
  }
}