golang+linux+pipline

linux文章/教程白墨 • 2023年3月11日 • 0 Comment

　　golang里可以通过[os/exec]包调用linux里的命令，然而对于linux里的管道（“|”）却不能直接使用。所以写了一个管道的接口如下。

func Pipline(cmds ...*exec.Cmd) ([]byte, []byte, error) {
    // At least one command
    if len(cmds) < 1 {
        return nil, nil, nil
    }

    var output bytes.Buffer
    var stderr bytes.Buffer
    var err error
    maxindex := len(cmds) - 1
    cmds[maxindex].Stdout = &output
    cmds[maxindex].Stderr = &stderr

    for i, cmd := range cmds[:maxindex] {
        if i == maxindex {
            break
        }

        cmds[i+1].Stdin, err = cmd.StdoutPipe()
        if err != nil {
            return nil, nil, err
        }
    }

    // Start each command
    for _, cmd := range cmds {
        err := cmd.Start()
        if err != nil {
            return output.Bytes(), stderr.Bytes(), err
        }
    }

    // Wait for each command to complete
    for _, cmd := range cmds {
        err := cmd.Wait()
        if err != nil {
            return output.Bytes(), stderr.Bytes(), err
        }
    }

    return output.Bytes(), stderr.Bytes(), nil
}