Java多线程监听K8s Job运行状态

最新推荐文章于 2024-10-23 17:34:29 发布

原创最新推荐文章于 2024-10-23 17:34:29 发布 · 1.8k 阅读

1 ·

CC 4.0 BY-SA版权

K8s 专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种使用Java多线程技术监听Kubernetes Job运行状态的方法，通过配置线程池并创建监听线程，实现对Job状态的实时监控，一旦Job运行完成，将发送消息并关闭线程。

Java多线程监听K8s Job的状态，如果在项目中需要知道Job运行状态，可以利用线程池启动线程进行检测，完成后推送消息，然后关闭线程；

1，配置线程池

@Configuration
@EnableAsync
public class TaskPoolConfig {

    @Bean("taskExecutor")
    public Executor taskExecutro(){
        ThreadPoolTaskExecutor taskExecutor = new ThreadPoolTaskExecutor();
        taskExecutor.setCorePoolSize(10);
        taskExecutor.setMaxPoolSize(50);
        taskExecutor.setQueueCapacity(200);
        taskExecutor.setKeepAliveSeconds(60);
        taskExecutor.setThreadNamePrefix("taskExecutor--");
        taskExecutor.setWaitForTasksToCompleteOnShutdown(true);
        taskExecutor.setAwaitTerminationSeconds(60);
        return taskExecutor;
    }
}

2，线程类

/**
 * Job监听线程，当评测任务结束关闭线程
 */
@Slf4j
@Component
public class JobStateMonitorThread {

    @Autowired
    private static EvaluationService evaluationService;

    @Autowired
    private static RedisUtil redisUtil;

    @Resource
    public void setEvaluationService(EvaluationService evaluationService) {
        JobStateMonitorThread.evaluationService = evaluationService;
    }

    @Resource
    public void setRedisUtil(RedisUtil redisUtil) {
        JobStateMonitorThread.redisUtil = redisUtil;
    }

    @Async("taskExecutor")
    public void tesTask(String no,String jobName) throws IOException, InterruptedException, ApiException {
        log.info("Job：{}监控线程启动{}" ,jobName,Thread.currentThread().getName());
        Thread.sleep(10000);
        boolean jobActive = true;
        while (jobActive) {
            String state = K8sUtil.getJobState(jobName);
            log.info("获取【{}】运行状态时间{}，运行状态：{}" ,jobName,new Date(),state);
            if (state.equals("running")){
                log.info("{}运行中" ,jobName);
                Thread.sleep(60000);
            }else {
                log.info("{}运行状态为完成" ,jobName);
                if (state.equals("success")){
                    //运行成功
                    //放到队列
                    redisUtil.pub("JOB_START",no);
                }else {
                    //运行失败
                    log.info("{}运行状态为失败，推送消息" ,no);
                    evaluationService.addResult(no);
                }
                log.info("{}线程关闭" ,jobName);
                jobActive = false;
                Thread.currentThread().interrupt();
                throw new RuntimeException("Job任务运行完成，异常法强制关闭当前监听线程！");
            }
        }

    }

    public JobStateMonitorThread(){

    }


}

3，调用线程

        //启动线程监听
        log.info("启动线程监听-{}",jobName);
        JobStateMonitorThread jobStateMonitorThread = new JobStateMonitorThread();
        jobStateMonitorThread.tesTask(no,jobName);