golang+linux+pipline

  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
}
View Code

  接口调用方法:

    // linux comand: df | grep -w "/" | cut -d "%", -f 1
    df := exec.Command("df")
    grep := exec.Command("grep", "-w", "/")
    cut := exec.Command("cut", "-d", "%", "-f", "1")

    result, _, err := utils.Pipline(df, grep, cut)
    if err != nil {
        log.Error(err)    
        return
    }
     // 输出最终结果
     fmt.Println(string(result)) 
View Code

 

posted @ 2015-11-21 21:43  hezhixiong  阅读(616)  评论(0编辑  收藏  举报