快捷導(dǎo)航

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

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

在測(cè)試一個(gè)文件轉(zhuǎn)換工具類的時(shí)候，發(fā)生一個(gè)有趣的現(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);
            }));

猜測(cè)可能原因

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

問題排查

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

打印結(jié)果如下：

很明顯，主進(jìn)程結(jié)束退出的時(shí)候，子進(jìn)程還沒有執(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運(yùn)行的入口，我們可以發(fā)現(xiàn)，不管Junit測(cè)試是否成功，都會(huì)調(diào)用System.exit()，而這個(gè)方法會(huì)用來結(jié)束當(dāng)前正在運(yùn)行的java虛擬機(jī)。當(dāng)status=0時(shí)表示正常結(jié)束，status=1表示異常退出(強(qiáng)制退出，程序未執(zhí)行完也會(huì)退出)。JVM都關(guān)閉了，子線程還靠什么運(yùn)行呢？所以這就是問題所在。