Java在Linux環境下執行MySQL命令無法獲取結果的問題

小海子l發表於2020-11-28

背景

最近專案中包含一些匯出功能,一些功能需要多表查詢才可以滿足需求,也有一些資料僅是單表查詢。在此之前想到過兩種方案,第一種是查詢出資料後通過EasyExcle寫入檔案,另一種是使用MySQL自帶的匯出功能。但在嘗試第二種方案時遇到一些問題,記錄如下。

問題

在執行單表匯出的過程中,我使用了MySQL自帶的命令完成。即在程式碼中通過字串拼接命令。例如:
在這裡插入圖片描述

即便是這條語句通過Java程式呼叫還是會執行失敗。失敗的原因有兩個:

  1. 通過Java程式碼呼叫命令會出現程式卡死,導致後面的程式無法執行,需要手動處理
  2. 命令是字串,需要告訴作業系統這並非字串而是命令

解決

import java.io.*;

/**
 * @description:
 * @author: 582895699@qq.com
 * @time: 2020/11/26 下午 11:17
 */
public class Test {
    public static void main(String[] args) {
        executeCommand();
    }

    public static void executeCommand() {
        StringBuilder stringBuilder = new StringBuilder();
        stringBuilder.append("mysql -uroot -proot -D test -e 'select * from goods where id=1'");
        stringBuilder.append(" > ");
        stringBuilder.append("/tmp/20201128.csv");
        // 告訴作業系統並非字串,而是命令
        String[] command = {"/bin/bash", "-c", stringBuilder.toString()};
        try {
            Process process = Runtime.getRuntime().exec(command);
            printMessage(process.getInputStream());
            printMessage(process.getErrorStream());
            // 等待命令執行完畢
            process.waitFor();
        } catch (IOException | InterruptedException e) {
            e.printStackTrace();
        }
    }

    /**
     * 另起一個執行緒輸出錯誤流
     *
     * @param input
     */
    private static void printMessage(final InputStream input) {
        new Thread(() -> {
            Reader reader = new InputStreamReader(input);
            BufferedReader bf = new BufferedReader(reader);
            String line = null;
            try {
                while ((line = bf.readLine()) != null) {
                    System.out.println(line);
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }).start();
    }
}

如有問題,歡迎指正
原文地址:https://www.haicheng.website/passages/execute-command-on-linux/

相關文章