随笔- 100 文章- 0 评论- 85 阅读- 119万

一起聊聊3个线程依次打印1、2、3...的故事

3个线程依次打印1、2、3…这个问题，常常被作为面试题，题目如下：

三个线程，一个线程负责打印1，4，7，……；第二个负责打印2，5，8，……，第三个负责打印3，6，9，……，要求在控制台中按顺序输出1，2，3，4，5，6……。

这个题目肯定是要启动3个线程的，那怎么让这3个线程“协作”按顺序打印1、2、3呢？从大的方面来讲，这种“协作”可分为以下两种：

竞争型：每个线程都抢着去打印，如果发现不该自己打印，则准备下一轮抢。由于大家都是竞争的，因此需要用锁机制来保护。
协同型：当前线程线程打印之后通知下一个线程去打印，这种需要确认好第一个线程打印时机。由于是协同型的因此可以不用锁机制来保护，但是需要一个通知机制。

竞争型打印

多个线程竞争型打印，优势是代码简单易懂，劣势是线程争抢是CPU调度进行的，可能该某个线程打印时结果该线程迟迟未被CPU调度，结果其他线程被CPU调度到但是由于不能执行打印操作而继续争抢，造成CPU性能浪费。示例代码如下：

public class DemoTask implements Runnable {

    // 这里将lock对象换成 Lock(ReentrantLock) 进行lock/unlock也是可以的
    private static final Object lock = new Object();
    private static final int MAX = 1024;
    private static int current = 1;

    private int index;
    public DemoTask(int index) {
        this.index = index;
    }

    @Override
    public void run() {
        while (current <= MAX) {
            synchronized (lock) {
                if ((current <= MAX) && (current % 3 == index)) {
                    System.out.println(current++);
                }
            }
        }
    }

    public static void main(String[] args) throws Exception {
        List<Thread> threadList = Arrays.asList(
                new Thread(new DemoTask(0)),
                new Thread(new DemoTask(1)),
                new Thread(new DemoTask(2))
        );

        threadList.forEach(Thread::start);
        for (Thread thread : threadList) {
            thread.join();
        }
    }
}

协同型打印

多个线程协同型打印，优势是各个线程使用“通知”机制进行协同分工，理论上执行效率较高，不过要使用对应的“通知”机制。关于如何“通知”，第一种是可使用Java对象的 wait/notify 或者Conditon对象的 await/signal ，第二种是以事件或者提交任务的方式（比如通过提交“待打印数字”这个任务给下一个线程）。

下面以第二种方式进行代码分析，比如当前线程通过submit给下一个线程一个“待打印数字”的任务，这样很容易想到使用只包含1个线程的线程池来实现，示例代码如下：

public class DemoTask implements Runnable {

    // 主线程要等待线程打印完毕，使用CountDownLatch通知机制
    private static CountDownLatch countDownLatch = new CountDownLatch(1);
    private static List<ExecutorService> threadList = new ArrayList<>();
    private static final int MAX = 1024;
    private static int current = 1;

    @Override
    public void run() {
        if (current <= MAX) {
            System.out.println(current++);
            threadList.get(current % threadList.size()).submit(new DemoTask());
        } else {
            countDownLatch.countDown();
        }
    }

    public static void main(String[] args) throws InterruptedException {
        for (int i = 0; i < 3; i++) {
            threadList.add(Executors.newFixedThreadPool(1));
        }

        threadList.get(0).submit(new DemoTask());
        countDownLatch.await();
        threadList.forEach(ExecutorService::shutdown);
    }
}