SpringBoot 自定义任务（ThreadPoolTaskScheduler）工具类 kotlin工具类

> springboot 定时任务工具类，可自定义添加、删除

```
import org.springframework.beans.factory.annotation.Autowired
import org.springframework.context.annotation.Bean
import org.springframework.core.env.Environment
import org.springframework.core.env.get
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler
import org.springframework.scheduling.support.CronTrigger
import org.springframework.stereotype.Component
import java.util.concurrent.ConcurrentHashMap
import java.util.concurrent.ScheduledFuture

/**
 * Created by vanki on 2019-05-09 10:17.
 */
@Component
class MyTaskScheduler<K>(
        env: Environment
) {
    @Autowired
    private lateinit var threadPoolTaskScheduler: ThreadPoolTaskScheduler
    private val taskMap = ConcurrentHashMap<K, ScheduledFuture<*>>()

// 线程池大小（同时执行任务数）
    private val poolSize = env["task.thread-pool.size"]?.toIntOrNull() ?: 4

@Bean
    fun threadPoolTaskScheduler(): ThreadPoolTaskScheduler {
        val threadPool = ThreadPoolTaskScheduler()
        threadPool.poolSize = this.poolSize
        return threadPool
    }

fun exists(key: K): Boolean {
        return this.taskMap.containsKey(key)
    }

fun addTask(key: K, task: Runnable, cron: CronTrigger): Boolean {
        val schedule = this.threadPoolTaskScheduler.schedule(task, cron) ?: return false
        return if (this.removeTask(key, false)) {
            taskMap[key] = schedule
            true
        } else {
            false
        }
    }

fun removeTask(key: K, mayInterruptIfRunning: Boolean = false): Boolean {
        val future = this.taskMap[key] ?: return true
        future.cancel(mayInterruptIfRunning)
        val isCancelled = future.isCancelled
        if (isCancelled) {
            this.taskMap.remove(key)
        }
        return isCancelled
    }

fun getTask(key: K): ScheduledFuture<*>? {
        return this.taskMap[key]
    }

fun count(): Int {
        return this.taskMap.size
    }

fun aliveCount(): Int {
        var count = 0
        taskMap.forEach {
            if (!it.value.isCancelled) {
                count++
            }
        }
        return count
    }
}
```

vanki

人类文明从仰望星空那一刻起，就已经距离揭示宇宙奥秘仅有一步之遥了。

创建： 2019-05-09 15:26:24

更新： 2019-05-09 15:30:27

SpringBoot 自定义任务（ThreadPoolTaskScheduler）工具类 kotlin工具类

TA的最新笔记

TA的最热笔记

欢 迎

登录

新朋友 注册

注册

老朋友 登录

SpringBoot 自定义任务（ThreadPoolTaskScheduler）工具类 kotlin工具类

TA的最新笔记

TA的最热笔记

欢迎

新朋友注册

老朋友登录