首頁 > 後端開發 > Golang > 主體

如何使用 Go 即時串流命令輸出?

Mary-Kate Olsen
發布: 2024-11-22 00:04:17
原創
884 人瀏覽過

How to Stream Command Output in Real-Time Using Go?

線上串流指令輸出

考慮以下執行指令並擷取其輸出的程式碼:

cmdParams := [][]string{
    {filepath.Join(dir, path), "npm", "install"},
    {filepath.Join(pdir, n.path), "gulp"},
}
for _, cp := range cmdParams {
    log.Printf("Starting %s in folder %s...", cp[1:], cp[0])
    cmd := exec.Command(cp[1], cp[2:]...)
    cmd.Dir = cp[0]
    // Wait to finish, get output:
    out, err := cmd.Output()
    if err != nil {
        log.Printf("Error running %s: %v\n", cp[1:], err)
        return
    }
    log.Println("Finished %s, output: %s", cp[1:], out)
}
登入後複製

此程式碼捕獲命令的完整輸出,並僅在命令完成執行後列印它。但是,需要即時串流輸出。

要實現即時輸出串流,我們可以修改程式碼如下:

cmdParams := [][]string{
    {filepath.Join(dir, path), "npm", "install"},
    {filepath.Join(pdir, n.path), "gulp"},
}
for _, cp := range cmdParams {
    log.Printf("Starting %s in folder %s...", cp[1:], cp[0])
    cmd := exec.Command(cp[1], cp[2:]...)
    cmd.Dir = cp[0]

    stdout, err := cmd.StdoutPipe()
    if err != nil {
        log.Printf("%s cmd.StdoutPipe() error: %v\n", cp[1:], err)
        return
    }
    // Start command:
    if err = cmd.Start(); err != nil {
        log.Printf("%s start error: %v\n", cp[1:], err)
        return
    }

    // Stream command output:
    scanner := bufio.NewScanner(stdout)
    scanner.Split(bufio.ScanRunes)
    for scanner.Scan() {
        fmt.Print(scanner.Text())
    }
    if scanner.Err() != nil {
        log.Printf("Reading %s stdout error: %v\n", cp[1:], err)
        return
    }

    // Get execution success or failure:
    if err = cmd.Wait(); err != nil {
        log.Printf("Error running %s: %v\n", cp[1:], err)
        return
    }
    log.Printf("Finished %s", cp[1:])
}
登入後複製

以下是更改的說明:

  • cmd.StdoutPipe() 建立一個用於讀取命令標準的管道輸出。
  • cmd.Start() 在背景啟動指令。
  • bufio.NewScanner(stdout) 建立一個掃描器,用於從 stdout 管道讀取。
  • 掃描器。 Split(bufio.ScanRunes) 設定掃描器以符文讀取,提供即時串流媒體。
  • 掃描器的Scan() 迴圈不斷讀取並列印命令產生的每個符文(字元)。
  • 循環完成後(寫入標準輸出的命令已完成),scanner.Err() 檢查是否有任何錯誤。
  • cmd.Wait() 等待指令完成並檢查錯誤。
  • 最後的 log.Println() 語句記錄指令的完成。

以上是如何使用 Go 即時串流命令輸出?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板