com.blr19c.falowp.bot.system.scheduling.cron.Trigger.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of falowp-bot-system Show documentation

FalowpBot system infrastructure

There is a newer version: 2.0.0-RC2

package com.blr19c.falowp.bot.system.scheduling.cron

import com.blr19c.falowp.bot.system.cache.OnlyReadOnceReference
import com.blr19c.falowp.bot.system.systemConfigProperty
import java.time.Instant
import java.time.ZoneOffset
import kotlin.time.Duration
import kotlin.time.Duration.Companion.seconds

interface Trigger {

    /**
     * 获取下次执行时间
     */
    fun nextExecutionTime(triggerContext: TriggerContext): Instant?

    /**
     * 任务执行时间
     */
    fun scheduledDate(triggerContext: TriggerContext): Instant {
        var completion = triggerContext.lastCompletionTime() ?: Instant.now()
        val scheduled = triggerContext.lastScheduledExecutionTime()
        //防止时钟毫秒内回拨
        if (scheduled != null && completion.isBefore(scheduled)) {
            completion = scheduled
        }
        return completion
    }

}

/**
 * Cron表达式
 */
class CronTrigger(cron: String) : Trigger {
    private val cronExpression = CronExpression.parse(cron)

    override fun nextExecutionTime(triggerContext: TriggerContext): Instant? {
        val zoneOffset = ZoneOffset.of(systemConfigProperty("zoneOffset"))
        val date = scheduledDate(triggerContext).atZone(zoneOffset).toLocalDateTime()
        return cronExpression.next(date)?.toInstant(zoneOffset)
    }

}

/**
 * 周期循环
 */
class PeriodicTrigger(
    /**
     * 固定速率执行
     */
    private val fixedRate: Boolean = false,
    /**
     * 执行周期
     */
    private val period: Duration,
    /**
     * 首次执行间隔
     */
    private val initialDelay: Duration = 0.seconds
) : Trigger {

    override fun nextExecutionTime(triggerContext: TriggerContext): Instant? {
        val lastExecution = triggerContext.lastScheduledExecutionTime()
            ?: return Instant.ofEpochMilli(System.currentTimeMillis() + initialDelay.inWholeMilliseconds)
        return if (fixedRate) lastExecution.plusMillis(period.inWholeMilliseconds)
        else scheduledDate(triggerContext).plusMillis(period.inWholeMilliseconds)
    }
}


/**
 * 在程序完全启动之后执行
 */
class ApplicationInitTrigger : Trigger {

    private val onlyReadOnceReference by OnlyReadOnceReference {
        Instant.ofEpochMilli(System.currentTimeMillis())
    }

    override fun nextExecutionTime(triggerContext: TriggerContext): Instant? {
        return onlyReadOnceReference
    }
}