转载

Java 的线程工厂 ThreadFactory原理及源码详解

在JDK的源码使用工厂模式,ThreadFactory就是其中一种。

在我们一般的使用中,创建一个线程,通常有两种方式:
继承Thread类,覆盖run方法,实现我们需要的业务
继承Runnable接口,实现run方法,实现我们需要的业务,并且调用new Thread(Runnable)方法,将其包装为一个线程执行

设想这样一种场景,我们需要一个线程池,并且对于线程池中的线程对象,赋予统一的线程优先级、统一的名称、甚至进行统一的业务处理或和业务方面的初始化工作,这时工厂方法就是最好用的方法了

ThreadFactory接口
ThreadFactory接口很简单,源码如下:
public interface ThreadFactory {
    Thread newThread(Runnable r);
}

我们可以看到ThreadFactory中,只有一个newThread方法,它负责接收一个Runnable对象,并将其封装到Thread对象中,进行执行。

最简单的ThreadFactory实现
我们可以实现上述接口,做一个最简单的线程工厂出来,源码如下:

public class SimpleThreadFactory implements ThreadFactory{
    @Override
    public Thread newThread(Runnable r) {
        return new Thread(r);
    }
}

上述线程工厂,只是创建了一个新线程,其他什么都没干。实际使用时,一般不会创建这么简单的线程工厂。

public class FixCountThreadFactory implements ThreadFactory{
    private final int MAX_THREAD;
    
    private final AtomicInteger count = new AtomicInteger(0);
    
    public FixCountThreadFactory(int maxThread) {
        MAX_THREAD = maxThread;
    }
    
    @Override
    public Thread newThread(Runnable r) {
        int incrementAndGet = count.incrementAndGet();
        if(incrementAndGet > MAX_THREAD)
        {
            count.decrementAndGet();
            return null;
        }
        
        return new Thread(r);
    }
}

上述线程工厂,可以控制创建线程的总数。
JDK中默认的线程工厂
在Executors工具类中,JDK提供了一个非常简单的线程工程,源码如下:

 static class DefaultThreadFactory implements ThreadFactory {
    static final AtomicInteger poolNumber = new AtomicInteger(1);
    final ThreadGroup group;
    final AtomicInteger threadNumber = new AtomicInteger(1);
    final String namePrefix;
 
    DefaultThreadFactory() {
        SecurityManager s = System.getSecurityManager();
        group = (s != null)? s.getThreadGroup() :
                             Thread.currentThread().getThreadGroup();
        namePrefix = "pool-" +
                      poolNumber.getAndIncrement() +
                     "-thread-";
    }
 
    public Thread newThread(Runnable r) {
        Thread t = new Thread(group, r,
                              namePrefix + threadNumber.getAndIncrement(),
                              0);
        if (t.isDaemon())
            t.setDaemon(false);
        if (t.getPriority() != Thread.NORM_PRIORITY)
            t.setPriority(Thread.NORM_PRIORITY);
        return t;
    }
}
正文到此结束
本文目录