快捷導航

Junit測試多線程無法得到結(jié)果的問題解決

更新時間：2021年05月25日 09:08:00 作者：Mr_Right_

在測試一個文件轉(zhuǎn)換工具類的時候，發(fā)生一個有趣的現(xiàn)象，同樣的輸入，使用Main函數(shù)可以正確解析，得到結(jié)果，使用Junit卻無法得到結(jié)果，神奇的是，即使捕獲Throwable，也無法捕獲到仍和異常。

class Main {
    public static void main(String[] args) {
        String trxFileDir = args[0];
        String targetDir = args[1];
        boolean isDecode = Boolean.parseBoolean(args[2]);
        ParseMojo parseMojo = new ParseMojo(trxFileDir,targetDir,isDecode);
        parseMojo.execute();
    }
}

    @Test
    public void executeEncode() {
        String trxFileDir = "E:\\MAE\\code\\MyMojoParse\\src\\main\\resources\\inputFile\\";
        String targetDir = "E:\\MAE\\code\\MyMojoParse\\src\\main\\resources\\outputFile\\";
        boolean isEncode = true;
        ParseMojo parser = new ParseMojo(trxFileDir, targetDir, isEncode);
        parser.execute();
    }

解析工具支持多線程，核心代碼如下

		 Arrays.stream(trxFiles).forEach(trx -> PARSE_POOL.execute(() -> {
                    convertToBinaryFile(trx);
            }));

猜測可能原因

首先當前問題和業(yè)務(wù)邏輯無關(guān)，初步懷疑是多線程和UT的問題，可能Junit進行測試的時候，主線程結(jié)束會導致子線程也終止。

問題排查

添加打印，查看文件的中斷處

打印結(jié)果如下：

很明顯，主進程結(jié)束退出的時候，子進程還沒有執(zhí)行完成，且每次執(zhí)行到的位置不一致

問題原因

分析Junit源碼

    public static void main(String args[]) {
        TestRunner aTestRunner = new TestRunner();
        try {
            TestResult r = aTestRunner.start(args);
            if (!r.wasSuccessful()) {
                System.exit(FAILURE_EXIT);
            }
            System.exit(SUCCESS_EXIT);
        } catch (Exception e) {
            System.err.println(e.getMessage());
            System.exit(EXCEPTION_EXIT);
        }
    }

這是Junit運行的入口，我們可以發(fā)現(xiàn)，不管Junit測試是否成功，都會調(diào)用System.exit()，而這個方法會用來結(jié)束當前正在運行的java虛擬機。當status=0時表示正常結(jié)束，status=1表示異常退出(強制退出，程序未執(zhí)行完也會退出)。JVM都關(guān)閉了，子線程還靠什么運行呢？所以這就是問題所在。