首页 >后端开发 >Golang >如何在 Go 中检索和处理 Shell 命令的实时输出?

如何在 Go 中检索和处理 Shell 命令的实时输出?

Linda Hamilton
Linda Hamilton原创
2024-12-11 04:40:14888浏览

How to Retrieve and Process Real-Time Output from Shell Commands in Go?

Go 中 Shell 命令的实时输出检索

在 Golang 中,os/exec 包提供了执行 shell 命令的方法。但是,当这些命令返回实时输出或进度更新时,有必要动态检索和处理该信息。以下是如何实现这一点:

要捕获和处理 Go 中 shell 命令的实时输出,请考虑以下方法:

package main<p>导入(</p><pre class="brush:php;toolbar:false">"bufio"
"fmt"
"io"
"os"
"os/exec"
"strings"

)

func main() {

cmdName := "ffmpeg -i t.webm  -acodec aac -vcodec libx264  cmd1.mp4"
cmdArgs := strings.Fields(cmdName)

cmd := exec.Command(cmdArgs[0], cmdArgs[1:len(cmdArgs)]...)
stderr, _ := cmd.StderrPipe()
cmd.Start()
go printProgress(stderr)
cmd.Wait()

}

//stderr 换行时打印处理后的信息
func printProgress(stderr io.ReadCloser) {

r := bufio.NewReader(stderr)
for {
    line, _, err := r.ReadLine()
    if err != nil {
        fmt.Println("error in reading from stderr:", err)
        break
    }

    // Process the line (e.g., extract the progress percentage)
    percentage, err := extractPercentage(string(line))
    if err != nil {
        fmt.Println("error in extracting percentage:", err)
        continue
    }

    // Print the updated progress ratio
    fmt.Printf("%s%%", percentage)
}

}

// extractPercentage解析ffmpeg输出的行并返回进度百分比。
// 假定问题中的输出格式(例如,“frame=... size=... time=... bitrate=...”)
func extractPercentage (行字符串) (字符串, 错误) {

parts := strings.Split(line, " ")
if len(parts) < 4 {
    return "", fmt.Errorf("invalid line format: %s", line)
}

size := parts[2]
sizeUnits := []string{"B", "kB", "MB", "GB", "TB", "PB"}
for i, unit := range sizeUnits {
    if strings.HasSuffix(size, unit) {
        factor := float64(1 << (10 * i))
        sizeBytes, err := strconv.ParseFloat(strings.TrimSuffix(size, unit), 64)
        if err != nil {
            return "", err
        }

        percentage := (sizeBytes / factor) / 100
        return strconv.FormatFloat(percentage, 'f', 2, 64), nil
    }
}

return "", fmt.Errorf("unknown size unit: %s", size)

}

此方法使用管道来检索命令的 stderr 流的实时输出。它连续读取和处理输出,提取进度信息并动态打印。请注意,本示例中的 extractPercentage 函数从 ffmpeg 输出中提取百分比,假设问题中提到的特定格式。这可能需要根据你的 shell 命令的具体输出格式进行调整。

以上是如何在 Go 中检索和处理 Shell 命令的实时输出?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn