go - Goでいくつかのコマンドをパイプする方法は?

Question

Goで複数の外部コマンドを一緒にパイプするにはどうすればよいですか? このコードを試してみましたが、というエラーが表示されますexit status 1。

package main

import (
    "io"
    "log"
    "os"
    "os/exec"
)

func main() {
    c1 := exec.Command("ls")
    stdout1, err := c1.StdoutPipe()
    if err != nil {
        log.Fatal(err)
    }

    if err = c1.Start(); err != nil {
        log.Fatal(err)
    }
    if err = c1.Wait(); err != nil {
        log.Fatal(err)
    }

    c2 := exec.Command("wc", "-l")
    c2.Stdin = stdout1

    stdout2, err := c2.StdoutPipe()
    if err != nil {
        log.Fatal(err)
    }

    if err = c2.Start(); err != nil {
        log.Fatal(err)
    }
    if err = c2.Wait(); err != nil {
        log.Fatal(err)
    }

    io.Copy(os.Stdout, stdout2)
}

score 140 · Accepted Answer

単純なシナリオでは、次のアプローチを使用できます。

bash -c "echo 'your command goes here'"

たとえば、次の関数は、パイプコマンドを使用して CPU モデル名を取得します。

func getCPUmodel() string {
        cmd := "cat /proc/cpuinfo | egrep '^model name' | uniq | awk '{print substr($0, index($0,$4))}'"
        out, err := exec.Command("bash","-c",cmd).Output()
        if err != nil {
                return fmt.Sprintf("Failed to execute command: %s", cmd)
        }
        return string(out)
}

score 62 · Accepted Answer

StdoutPipe は、コマンドの開始時にコマンドの標準出力に接続されるパイプを返します。Wait がコマンドの終了を確認した後、パイプは自動的に閉じられます。

( http://golang.org/pkg/os/exec/#Cmd.StdinPipeから)

あなたがするという事実はをc1.Wait閉じますstdoutPipe。

実際の例を作成しました（単なるデモ、エラーキャッチを追加してください！）：

package main

import (
    "bytes"
    "io"
    "os"
    "os/exec"
)

func main() {
    c1 := exec.Command("ls")
    c2 := exec.Command("wc", "-l")

    r, w := io.Pipe() 
    c1.Stdout = w
    c2.Stdin = r

    var b2 bytes.Buffer
    c2.Stdout = &b2

    c1.Start()
    c2.Start()
    c1.Wait()
    w.Close()
    c2.Wait()
    io.Copy(os.Stdout, &b2)
}

score 58 · Accepted Answer

package main

import (
    "os"
    "os/exec"
)

func main() {
    c1 := exec.Command("ls")
    c2 := exec.Command("wc", "-l")
    c2.Stdin, _ = c1.StdoutPipe()
    c2.Stdout = os.Stdout
    _ = c2.Start()
    _ = c1.Run()
    _ = c2.Wait()
}

score 8 · Accepted Answer

最初の回答と同様ですが、最初のコマンドが開始され、ゴルーチンで待機しました。これにより、パイプが幸せに保たれます。

package main

import (
    "io"
    "os"
    "os/exec"
)

func main() {
    c1 := exec.Command("ls")
    c2 := exec.Command("wc", "-l")

    pr, pw := io.Pipe()
    c1.Stdout = pw
    c2.Stdin = pr
    c2.Stdout = os.Stdout

    c1.Start()
    c2.Start()

    go func() {
        defer pw.Close()

        c1.Wait()
    }()
    c2.Wait()
}

score 5 · Accepted Answer

これは完全に機能する例です。このExecute関数は任意の数のexec.Cmdインスタンスを取り (可変引数関数を使用)、それらをループ処理して、stdout の出力を次のコマンドの stdin に正しくアタッチします。これは、関数が呼び出される前に行う必要があります。

次に call 関数はループ内でコマンドを呼び出し、defer を使用して再帰的に呼び出し、パイプを適切に閉じるようにします。

package main

import (
    "bytes"
    "io"
    "log"
    "os"
    "os/exec"
)

func Execute(output_buffer *bytes.Buffer, stack ...*exec.Cmd) (err error) {
    var error_buffer bytes.Buffer
    pipe_stack := make([]*io.PipeWriter, len(stack)-1)
    i := 0
    for ; i < len(stack)-1; i++ {
        stdin_pipe, stdout_pipe := io.Pipe()
        stack[i].Stdout = stdout_pipe
        stack[i].Stderr = &error_buffer
        stack[i+1].Stdin = stdin_pipe
        pipe_stack[i] = stdout_pipe
    }
    stack[i].Stdout = output_buffer
    stack[i].Stderr = &error_buffer

    if err := call(stack, pipe_stack); err != nil {
        log.Fatalln(string(error_buffer.Bytes()), err)
    }
    return err
}

func call(stack []*exec.Cmd, pipes []*io.PipeWriter) (err error) {
    if stack[0].Process == nil {
        if err = stack[0].Start(); err != nil {
            return err
        }
    }
    if len(stack) > 1 {
        if err = stack[1].Start(); err != nil {
             return err
        }
        defer func() {
            if err == nil {
                pipes[0].Close()
                err = call(stack[1:], pipes[1:])
            }
        }()
    }
    return stack[0].Wait()
}

func main() {
    var b bytes.Buffer
    if err := Execute(&b,
        exec.Command("ls", "/Users/tyndyll/Downloads"),
        exec.Command("grep", "as"),
        exec.Command("sort", "-r"),
    ); err != nil {
        log.Fatalln(err)
    }
    io.Copy(os.Stdout, &b)
}

この要点で利用可能

https://gist.github.com/tyndyll/89fbb2c2273f83a074dc

~ のようなシェル変数は補間されないことを知っておくとよいでしょう。

go - Goでいくつかのコマンドをパイプする方法は?

8 に答える 8

Related

Reference