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 @   hezhixiong  阅读(619)  评论(0)    收藏  举报
编辑推荐:
· 如何统计不同电话号码的个数?—位图法
· C#高性能开发之类型系统:从 C# 7.0 到 C# 14 的类型系统演进全景
· 从零实现富文本编辑器#3-基于Delta的线性数据结构模型
· 记一次 .NET某旅行社酒店管理系统 卡死分析
· 长文讲解 MCP 和案例实战
阅读排行:
· C#高性能开发之类型系统:从 C# 7.0 到 C# 14 的类型系统演进全景
· 管理100个小程序-很难吗
· 基于Blazor实现的运输信息管理系统
· 微信支付功能的设计实现与关键实践(UniApp+Java)全代码
· 用c#从头写一个AI agent,实现企业内部自然语言数据统计分析
点击右上角即可分享
微信分享提示