net.nemerosa.ontrack.job.orchestrator.JobOrchestrator.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ontrack-job Show documentation

Abstract management of identified jobs.

There is a newer version: 4.4.5

package net.nemerosa.ontrack.job.orchestrator

import net.nemerosa.ontrack.job.*
import org.slf4j.Logger
import org.slf4j.LoggerFactory
import java.util.*
import kotlin.streams.toList
import org.springframework.transaction.PlatformTransactionManager
import org.springframework.transaction.support.TransactionTemplate

class JobOrchestrator(
    private val jobScheduler: JobScheduler,
    private val name: String,
    private val jobOrchestratorSuppliers: Collection,
    transactionManager: PlatformTransactionManager,
) : Job {

    private val cache = HashSet()
    private val transactionTemplate = TransactionTemplate(transactionManager)
    private val logger: Logger = LoggerFactory.getLogger(JobOrchestrator::class.java)

    override fun getKey(): JobKey {
        return JobCategory.CORE.getType("orchestrator").withName("Orchestrator").getKey(name)
    }

    override fun getTask(): JobRun {
        return JobRun { this.orchestrate(it) }
    }

    @Synchronized
    fun orchestrate(runListener: JobRunListener) {
        // Complete list of registrations
        val registrations = transactionTemplate.execute {
            jobOrchestratorSuppliers.flatMap {
                try {
                    it.collectJobRegistrations().toList()
                } catch (any: Exception) {
                    // Does not collect the jobs for this collector for now, but marks the exception
                    logger.error("Cannot collect jobs for ${it::class.java.name}", any)
                    emptyList()
                }
            }
        } ?: return
        // List of registration keys
        val keys = registrations.map { registration -> registration.job.key }
        // Jobs to unschedule
        val toRemove = HashSet(cache)
        toRemove.removeAll(keys)
        toRemove.forEach { jobScheduler.unschedule(it) }
        // Jobs to add / update
        val toRegister = HashSet(keys)
        toRegister.removeAll(toRemove)
        registrations
            .filter { jobRegistration -> toRegister.contains(jobRegistration.job.key) }
            .forEach { jobRegistration -> schedule(jobRegistration, runListener) }
        // Resets the cache
        cache.clear()
        cache.addAll(keys)
    }

    private fun schedule(jobRegistration: JobRegistration, runListener: JobRunListener) {
        runListener.message("Scheduling: %s", jobRegistration.job.key)
        jobScheduler.schedule(jobRegistration.job, jobRegistration.schedule)
    }

    override fun getDescription(): String {
        return name
    }

    override fun isDisabled(): Boolean {
        return false
    }

}