package tech.powerjob.worker.core.executor;

import com.google.common.util.concurrent.ThreadFactoryBuilder;
import lombok.Getter;
import tech.powerjob.worker.common.PowerJobWorkerConfig;

import java.util.concurrent.*;

/**
 * @author Echo009
 * @since 2022/9/23
 */
@Getter
public class ExecutorManager {
    /**
     * 执行 Worker 底层核心任务
     */
    private final ScheduledExecutorService coreExecutor;
    /**
     * 执行轻量级任务状态上报
     */
    private final ScheduledExecutorService lightweightTaskStatusCheckExecutor;
    /**
     * 执行轻量级任务
     */
    private final ExecutorService lightweightTaskExecutorService;


    public ExecutorManager(PowerJobWorkerConfig workerConfig){


        final int availableProcessors = Runtime.getRuntime().availableProcessors();
        // 初始化定时线程池
        ThreadFactory coreThreadFactory = new ThreadFactoryBuilder().setNameFormat("powerjob-worker-core-%d").build();
        coreExecutor =  new ScheduledThreadPoolExecutor(3, coreThreadFactory);

        ThreadFactory lightTaskReportFactory = new ThreadFactoryBuilder().setNameFormat("powerjob-worker-light-task-status-check-%d").build();
        // 都是 io 密集型任务
        lightweightTaskStatusCheckExecutor =  new ScheduledThreadPoolExecutor(availableProcessors * 10, lightTaskReportFactory);

        ThreadFactory lightTaskExecuteFactory = new ThreadFactoryBuilder().setNameFormat("powerjob-worker-light-task-execute-%d").build();
        // 大部分任务都是 io 密集型
        lightweightTaskExecutorService = new ThreadPoolExecutor(availableProcessors * 10,availableProcessors * 10, 120L, TimeUnit.SECONDS,
                new ArrayBlockingQueue<>((workerConfig.getMaxLightweightTaskNum() * 2),true), lightTaskExecuteFactory, new ThreadPoolExecutor.AbortPolicy());

    }



    public void shutdown(){
        coreExecutor.shutdownNow();
    }

}