main.misk.cron.CronTask.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of misk-cron Show documentation

Open source application container in Kotlin

There is a newer version: 2024.09.17.200749-4708422

package misk.cron

import com.google.common.util.concurrent.AbstractIdleService
import misk.clustering.weights.ClusterWeightProvider
import misk.tasks.RepeatedTaskQueue
import misk.tasks.Status
import wisp.lease.LeaseManager
import wisp.logging.getLogger
import java.time.Clock
import java.time.Duration
import jakarta.inject.Inject
import jakarta.inject.Singleton

@Singleton
internal class CronTask @Inject constructor() : AbstractIdleService() {
  @Inject private lateinit var clock: Clock
  @Inject private lateinit var cronManager: CronManager
  @Inject private lateinit var leaseManager: LeaseManager
  @Inject @ForMiskCron private lateinit var taskQueue: RepeatedTaskQueue
  @Inject private lateinit var clusterWeight: ClusterWeightProvider

  override fun startUp() {
    logger.info { "Starting CronTask" }
    var lastRun = clock.instant()
    taskQueue.scheduleWithBackoff(INTERVAL) {
      if (clusterWeight.get() == 0) {
        logger.info { "CronTask is running on a passive node. Skipping." }
        return@scheduleWithBackoff Status.OK
      }
      val lease = leaseManager.requestLease(CRON_CLUSTER_LEASE_NAME)

      val now = clock.instant()
      var leaseHeld = lease.checkHeld()
      if (!leaseHeld) {
        leaseHeld = lease.acquire()
      }
      if (leaseHeld) {
        cronManager.runReadyCrons(lastRun)
      }
      lastRun = now
      Status.OK
    }
  }

  override fun shutDown() {
    logger.info { "Stopping CronTask" }
    cronManager.removeAllCrons()
  }

  companion object {
    val INTERVAL: Duration = Duration.ofSeconds(60L)
    private const val CRON_CLUSTER_LEASE_NAME = "misk.cron.lease"

    private val logger = getLogger()
  }
}