java-多线程使用

发布于:2024-07-27 ⋅ 阅读:(42) ⋅ 点赞:(0)
@Slf4j
public class RunnableMaster {
    /**
     * 默认线程池
     */
    private static final ExecutorService DEFAULT_EXECUTOR_SERVICE = Executors.newFixedThreadPool(16);

    /**
     * 实例线程池
     */
    private final ExecutorService executorService;

    /**
     * 任务数
     */
    private final AtomicInteger taskCounter = new AtomicInteger(0);

    public RunnableMaster() {
        this(DEFAULT_EXECUTOR_SERVICE);
    }

    public RunnableMaster(ExecutorService executorService) {
        this.executorService = executorService;
    }

    /**
     * 提交任务
     *
     * @param task
     */
    public void submit(Runnable task) {
        taskCounter.incrementAndGet();
        executorService.submit(() -> {
            try {
                task.run();
            } catch (Exception e) {
                log.error("Task Execute Error ! taskOrder: {}, error:{}", e);
            }
            taskCounter.decrementAndGet();
        });
    }

    /**
     * 等待所有任务执行完毕
     * @return
     */
    public boolean await() {
        while (true) {
            if (taskCounter.get() == 0) {
                return true;
            }
        }
    }
}


使用
private ExecutorService executorService = Executors.newFixedThreadPool(8);
RunnableMaster runnableMaster = new RunnableMaster(executorService);
runnableMaster.submit(() -> monthOutput(monthIndicatorResp));
 runnableMaster.await();