码迷,mamicode.com
首页 > 系统相关 > 详细

golang+linux+pipline

时间:2015-11-21 22:25:04      阅读:264      评论:0      收藏:0      [点我收藏+]

标签:

  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

 

golang+linux+pipline

标签:

原文地址:http://www.cnblogs.com/hezhixiong/p/4984828.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!