com.github.eikek.calev.akka.internal.CalevTimerSchedulerImpl.scala Maven / Gradle / Ivy

Go to download
package com.github.eikek.calev.akka.internal

import java.time.{Clock, ZonedDateTime}

import scala.concurrent.duration.FiniteDuration

import akka.actor.typed.scaladsl.TimerScheduler
import com.github.eikek.calev.CalEvent
import com.github.eikek.calev.akka.dsl.CalevTimerScheduler

private[akka] class CalevTimerSchedulerImpl[T](
    val scheduler: TimerScheduler[T],
    clock: Clock,
    minInterval: Option[FiniteDuration]
) extends CalevTimerScheduler[T] {

  private val upcomingEventProvider =
    new UpcomingEventProvider(clock, minInterval)

  def startSingleTimer(calEvent: CalEvent, msgFactory: ZonedDateTime => T): Unit =
    upcomingEventProvider(calEvent)
      .foreach { case (instant, delay) =>
        scheduler.startSingleTimer(msgFactory.apply(instant), delay)
      }

  def startSingleTimer[K](
      key: K,
      calEvent: CalEvent,
      msgFactory: (K, ZonedDateTime) => T
  ): Unit =
    upcomingEventProvider(calEvent)
      .foreach { case (instant, delay) =>
        scheduler.startSingleTimer(key, msgFactory(key, instant), delay)
      }

}