java 自定义简单的线程池

java 自定义简单的线程池

/**
 * 简单的线程池实现类
 *
 * @author 作者名
 * @version 版本号
 * @since 创建时间
 */

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.atomic.AtomicBoolean;
 /**
  * 自定义简单的线程池
  */
class SimpleThreadPool {

    /**
     * 任务队列
     */
    private final BlockingQueue<Runnable> taskQueue;

    /**
     * 线程数组
     */
    private final Thread[] threads;

    /**
     * 线程池运行状态标识，true表示运行中，false表示已停止
     */
    private final AtomicBoolean running = new AtomicBoolean(true);

    /**
     * 构造方法，创建指定大小的线程池
     *
     * @param poolSize 线程池大小
     */
    public SimpleThreadPool(int poolSize) {
        taskQueue = new LinkedBlockingQueue<>();
        threads = new Thread[poolSize];
        for (int i = 0; i < poolSize; i++) {
            threads[i] = new Thread(() -> {
                while (running.get() || !taskQueue.isEmpty()) {
                    try {
                        taskQueue.take().run();
                    } catch (InterruptedException e) {
                        Thread.currentThread().interrupt();
                    }
                }
            });
            threads[i].start();
        }
    }

    /**
     * 执行任务
     *
     * @param task 要执行的任务
     * @throws IllegalStateException 如果线程池已停止，则抛出此异常
     */
    public void execute(Runnable task) {
        if (running.get()) {
            taskQueue.offer(task);
        } else {
            throw new IllegalStateException("Thread pool is not running.");
        }
    }

    /**
     * 关闭线程池
     * 将running设置为false，并向任务队列中添加空任务，以便线程退出循环
     */
    public void shutdown() {
        running.set(false);
        for (Thread thread : threads) {
            taskQueue.offer(() -> {});
        }
    }
}

/**
 * 示例任务类，实现Runnable接口
 *
 * @author 作者名
 * @version 版本号
 * @since 创建时间
 */
class ExampleTask implements Runnable {
    /**
     * 任务编号
     */
    private final int id;

    /**
     * 构造方法
     *
     * @param id 任务编号
     */
    public ExampleTask(int id) {
        this.id = id;
    }

    @Override
    public void run() {
        System.out.println("Task " + id + " is running on thread " + Thread.currentThread().getName());
    }
}

/**
 * 测试类
 *
 * @author 作者名
 * @version 版本号
 * @since 创建时间
 */
public class Test {
    /**
     * 主方法
     *
     * @param args 命令行参数
     */
    public static void main(String[] args) {
        SimpleThreadPool pool = new SimpleThreadPool(5);

        for (int i = 0; i < 10; i++) {
            pool.execute(new ExampleTask(i));
        }

        pool.shutdown();
    }
}