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

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

import com.blr19c.falowp.bot.system.Log
import com.blr19c.falowp.bot.system.api.BotApi
import com.blr19c.falowp.bot.system.plugin.TaskPluginRegister
import com.blr19c.falowp.bot.system.scheduling.api.SchedulingBotApi
import com.blr19c.falowp.bot.system.scheduling.api.SchedulingBotApiSupport
import com.blr19c.falowp.bot.system.utils.ScanUtils
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.SupervisorJob
import java.util.concurrent.CopyOnWriteArrayList
import kotlin.reflect.KClass

/**
 * 定时任务
 */
object Scheduling : Log {
    private val botList = arrayListOf()
    private val taskPlugins = CopyOnWriteArrayList()
    private val executor = CoroutineScope(Dispatchers.Default + SupervisorJob())
    private val executorTaskList = CopyOnWriteArrayList()

    suspend fun selectBot(receive: String, originalClass: KClass<*>): BotApi? {
        return botList.firstOrNull { it.supportReceive(receive) }?.bot(receive, originalClass)
    }

    suspend fun allBot(originalClass: KClass<*>): List {
        return botList.map { it.bot("", originalClass) }.toList()
    }

    fun registerTask(pluginRegister: TaskPluginRegister) {
        taskPlugins.add(pluginRegister)
    }

    fun configure() {
        log().info("初始化(周期/cron)任务")
        ScanUtils.scanPackage("com.blr19c.falowp.bot.system.adapter")
            .filter { SchedulingBotApiSupport::class.java.isAssignableFrom(it) }
            .forEach { botList.add(it.kotlin.objectInstance as SchedulingBotApiSupport) }
        botList.sortBy { it.order() }
        val tasks = taskPlugins.map(this::schedulingRunnable)
        log().info("已加载的(周期/cron)任务数量:{}", tasks.size)
        executorTaskList.addAll(tasks)
        executorTaskList.forEach(SchedulingRunnable::schedule)
        log().info("初始化(周期/cron)任务完成")
    }

    private fun schedulingRunnable(plugin: TaskPluginRegister): SchedulingRunnable {
        return SchedulingRunnable(
            { plugin.block.invoke(SchedulingBotApi(plugin.originalClass)) },
            executor,
            plugin.trigger
        )
    }
}