编写ThreadService实现暴力结束线程Daemon,join,interrupt的使用

编写ThreadService实现暴力结束线程

上一篇文章最后留下了一个问题，就是如果线程后面被阻塞了，该怎么办？这结合之前学过的ThreadApi写一个ThreadService来暴力结束线程。大致思路是：任务线程设置为调用线程的守护线程，通过main来启动调用线程，从而执行任务线程。当需要停止线程的时候只需要中断调用线程即可，调用线程生命周期结束时，为守护线程的任务线程会自动停止。需要注意的是任务线程可能执行完调用线程就结束了，这里任务线程就需要join一下了。虽然解释起来有点绕但是多看几遍，多敲几遍就一览众山小了。

代码如下：

/**
 * @program: ThreadDemo
 * @description: 暴力关闭线程
 * @author: hs96.cn@Gmail.com
 * @create: 2020-09-05
 */
public class ThreadService {
    // 调用线程
    private Thread executeThread;

    private boolean finished = false;

    /**
     * 执行线程
     *
     * @param task 任务
     */
    public void execute(Runnable task) {
        this.executeThread = new Thread(() -> {
            // 任务执行线程
            Thread runner = new Thread(task);
            runner.setDaemon(true);
            runner.start();
            try {
                runner.join();
                finished = true;
            } catch (InterruptedException e) {
            }
        });
        this.executeThread.start();
    }

    /**
     * 关闭线程
     *
     * @param mills 线程最长执行时间（单位：毫秒）
     */
    public void shutdown(long mills) {
        long currentTime = System.currentTimeMillis();
        while (!finished) {
            if (System.currentTimeMillis() - currentTime >= mills) {
                System.out.println("任务超时，需要结束线程！");
                executeThread.interrupt();
                break;
            }
            try {
                Thread.sleep(1);
            } catch (InterruptedException e) {
                System.out.println("执行线程被打断！");
                break;
            }
        }
        finished = false;
    }

    public static void main(String[] args) {
        ThreadService service = new ThreadService();
        long startTime = System.currentTimeMillis();
        service.execute(() -> {
            // 1. load a very heavy resource.
            /*while (true) {
            }*/
            // 2. 线程执行5s后结束
            try {
                Thread.sleep(5_000);
            } catch (InterruptedException e) {
            }
        });
        service.shutdown(10_000);
        long endTime = System.currentTimeMillis();
        System.out.println("执行时间为：" + (endTime - startTime));
    }
}

其实这里面有很多细节的地方，比如runner是守护线程，他的生命周期依赖于executeThread，所以runner在start后要join()一下。比如finished作为标识来配合Thread.sleep(1)来实现时间的巧妙校验，里面还有很多细节，慢慢细品。最近就不贴运行代码了，因为家里的笔记本太卡了。