java中自定义线程池的实现

Question

出于学习的目的，我正在尝试用java实现自己的线程池。下面是我已经实现的。我对这个实现有几个问题：

>

虽然我像内置java一样使用BlockingQueue执行器希望我们提供Runnable对象（通过执行方法）。但在我的情况下，我觉得我可以创建任何对象而不是Runnable。那么为什么Java执行器期望Runnable，我尝试查看源代码，但还不能弄清楚。

这个原始实现还有什么问题吗？

请找到密码。

公共类自定义线程池{

private final BlockingQueue<Runnable> blockingQueue;
private final Worker[] workers;

public CustomThreadPool(final int numOfThreads) {
    blockingQueue = new LinkedBlockingQueue<>();
    workers = new Worker[numOfThreads];

    for (int i = 0; i < numOfThreads; i++) {
        workers[i] = new Worker();
        workers[i].start();
    }
}

public void execute(final Runnable task) {
    blockingQueue.add(task);
}

public void shutdownImmediately() {
    for (int i = 0; i < workers.length; i++) {
        workers[i].shutdownSignal = true;
        workers[i] = null;
    }
}

 private class Worker extends Thread {
    private Runnable taskToPerform = null;
    boolean shutdownSignal = false;

    @Override
    public void run() {
        while(true && !shutdownSignal) {
            taskToPerform = blockingQueue.poll();
            if (taskToPerform != null) {
                taskToPerform.run();
            }
            if(shutdownSignal) {
                break;
            }
        }
    }
}

public static void main(String[] args) throws Exception {
    final CustomThreadPool threadPool = new CustomThreadPool(5);
    for (int i = 0; i < 20; i++) {
        threadPool.execute(() -> System.out.println(Thread.currentThread().getName()));
    }
    Thread.sleep(1*1000);
    threadPool.shutdownImmediately();
}

}

Answer 1

>

Executor期望Runnable或Callable，因为它在运行您提交的任务时将调用这些接口的run或call方法。

在您的实现中，您不使用BlockingQueue的阻塞方面。线程池线程将在上持续旋转（占用cpu时间），而

您应该使用其中一种阻塞方法，而不是poll（）。

您可以使用轮询（long timeout， TimeUnit）方法，该方法需要超时参数。在这种情况下，如果您在任何线程池线程等待此调用时调用您的Shutdown Im权方法。他们将等待超时持续时间，轮询将返回null给他们他们将看到关闭信号正在设置并退出循环。如果您不想让他们等待超时，您也可以调用中断方法。

 // in run method
 try {
     taskToPerform = blockingQueue.poll(5, TimeUnit.SECONDS);
 } catch (InterruptedException e) {
     break; // this thread is interrupted. Time to leave this loop
 }

// in shutdownImmediately method 
workers[i].shutdownSignal = true;
// If You don't call interrupt, Threads will wait at max 5 sec for this case.
workers[i].interrupt(); 
workers[i] = null;

或者您可以使用take（）方法，当队列中没有任何内容时，该方法会进行阻塞。但是在这种情况下，当您调用立即关闭方法时，必须中断可能正在等待take方法调用的线程。否则，他们将无法执行take（）调用。

// in run method
try {
   taskToPerform = blockingQueue.take();
} catch (InterruptedException e) {
   break; // this thread is interrupted. Time to leave this loop
}

// in shutdownImmediately method 
workers[i].shutdownSignal = true;
workers[i].interrupt(); // this is crucial for this case
workers[i] = null;