Java服務如何調用系統(tǒng)指令、Bat腳本記錄

更新時間：2024年06月26日 15:00:49 作者：Aikes902

這篇文章主要介紹了Java服務如何調用系統(tǒng)指令、Bat腳本記錄,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

前言

在項目推進過程中偶爾會涉及到調用其它組件或腳本的需求，本文重點介紹Java服務調用Bat腳本、系統(tǒng)指令。

調用Bat腳本

根據(jù)需求生成BAT腳本內容文本，通過文件流寫入到新建的BAT文件中，然后利用ProcessBuilder執(zhí)行BAT命令，最后判斷刪除生成的BAT文件。

    public void executeBatScript(String cCommand, String cBatFileName, String cBasePath) throws Exception {
        //判斷路徑是否存在
        File directory = new File(cBasePath);
        if (!directory.exists()) {
            directory.mkdir();
        }
        String tBatFilePath = cBasePath + cBatFileName + ".bat"; // 批處理文件的路徑和名稱

        // 創(chuàng)建一個Writer對象來寫入批處理命令
        PrintWriter writer = new PrintWriter(new FileWriter(tBatFilePath));
        writer.println(cCommand); // 在批處理文件中寫入命令
        writer.close(); // 關閉Writer對象

        // 執(zhí)行批處理文件
        ProcessBuilder builder = new ProcessBuilder(tBatFilePath);
        // 輸出指令日志
        builder.redirectOutput(ProcessBuilder.Redirect.INHERIT);
        builder.redirectError(ProcessBuilder.Redirect.INHERIT);
        // 啟動進程并后臺執(zhí)行批處理文件
        builder.start();

        // 刪除生成的批處理文件
        File file = new File(tBatFilePath);
        if (file.delete()) {
            log.info("Batch file deleted successfully.");
        }
    }

當前模式存在兩個問題需要手動處理：

1、設置了打印指令日志會導致最終服務運行的日志很多很雜，按理說Logger應該可以配置單獨開一個日志文件，這塊暫時還沒研究到，有配置成功的同學記得評論區(qū)分享下哈。

2、執(zhí)行的BAT腳本為非阻塞型，所以針對BAT腳本運行中斷后的處理需要在BAT內容中實現(xiàn)，后面是筆者用到的BAT腳本，供大家參考。

@echo off

:: 設置重試次數(shù)和計數(shù)器變量
set max_retry=3
set retry_count=0

set remote=$remote$
set local=$local$

:retry
:: 檢查重試次數(shù)是否超過最大值
if %retry_count% gtr %max_retry% (
   echo "Maximum retries exceeded. Quitting."
   goto end
)

:: 調用ffmpeg指令  ffmpeg -i  rtmp://105if39328.51pc.cn:1935/live/0 -c copy -f flv rtmp://127.0.0.1:50001/live/aa8
ffmpeg -i %remote% -c copy -f flv %local%

:: 檢查上一個命令的退出代碼是否為0（成功）
if %errorlevel% neq nul (
   echo "remote:%remote%. local:%local%. errorlevel:%errorlevel%. Retrying in 10 seconds...This is No.%retry_count% times Retry"
   :: 等待10秒后重試
   ping -n 15 127.0.0.1 > nul
   set /a retry_count+=1
   goto retry
)

:end
echo "Done"

調用系統(tǒng)指令

執(zhí)行系統(tǒng)指令還是利用ProcessBuilder，包括日志打印的套路都與BAT腳本執(zhí)行一致，需要注意的一點是，針對指令中的空格分割需要通過List的集合進行拆分。

例如：

指令 rtmp-to-flv.exe -proxy 50003,50004，需要將其根據(jù)空格分為三份，分別裝進List集合中，然后使用processBuilder.command方法運行即可。

    public void executeSystemCommand() {
        try {
            // 執(zhí)行批處理文件
            Resource classpathResource = resourceLoader.getResource("classpath:");
            String classpathPath = classpathResource.getFile().getAbsolutePath();

            ProcessBuilder processBuilder = new ProcessBuilder();
            //定義命令內容
            List<String> command = new ArrayList<>();
            // rtmp-to-flv.exe -proxy 50003,50004

            //
            String os= System.getProperty("os.name").toLowerCase();
            if (os.contains("windows")) {
                command.add(classpathPath + "/rtmp-to-flv.exe");
            }
            else {
                command.add(classpathPath + "/rtmp-to-flv");
            }

            command.add("-proxy");
            command.add("50003,50004");
            processBuilder.command(command);
            //將標準輸入流和錯誤輸入流合并，通過標準輸入流讀取信息
            processBuilder.redirectOutput(ProcessBuilder.Redirect.INHERIT);
            processBuilder.redirectError(ProcessBuilder.Redirect.INHERIT);
            processBuilder.start(); // 啟動進程并后臺執(zhí)行批處理文件

        } catch (Exception e) {
            e.printStackTrace();
        }
    }