go - 如何将curl通过管道传输到Go程序中？

Question

我想写一个Go小程序，用来美化json数据。当我使用文件时它已经可以工作了。这是代码:

package main

import (
    "bufio"
    "fmt"
    "github.com/Jeffail/gabs"
    "log"
    "os"
)

func main() {
    info, err := os.Stdin.Stat()
    if err != nil {
        log.Fatal(err)
    }

    if info.Mode()&os.ModeCharDevice != 0 || info.Size() <= 0 {
        fmt.Println("The command is intended to work with pipes.")
        fmt.Println("cat file.json | prettyjson")
        return
    }

    reader := bufio.NewReader(os.Stdin)


    input, err := reader.ReadBytes('\n')
    if err != nil {
        log.Fatal()
    }

    jsonParsed, err := gabs.ParseJSON(input)
    if err != nil {
        log.Fatal("couldn't parse json")
    }

    fmt.Println(fmt.Println(jsonParsed.StringIndent("", "  ")))
}

如果我像这样使用curl运行此代码:

curl -s "https://min-api.cryptocompare.com/data/top/exchanges?fsym=BTC&tsym=USD" | prettyjson

我得到:(23) 写入正文失败

我在this post中看到在curl写入所有数据之前管道将被关闭，但是如何优化我的Go程序以等待curl完成？

Answer 1

关于OP源代码，我会考虑更改条件来检测管道的存在。

正如https://stackoverflow.com/a/43947435/4466350中已经提供的那样正确的条件不需要检查输入的长度。考虑一下，这完全有道理，因为您可以打开标准输入而不在其上写入数据。

除了所提出的解决方案对于它试图实现的目标来说似乎毫无用处的复杂性之外，还可以漂亮地打印 json 输入。

我发现使用标准库足以实现给定测试用例的目标。

关于问题...但是我如何优化我的Go程序以等待curl完成？，OP似乎不理解文件描述符的工作方式。事实上，这个问题甚至不正确，因为理论上该进程可以保持事件状态，但主动决定关闭 Stdin。 OP 对进程的活跃度不感兴趣，相反，他应该在读取 Stdin 时简单地查找 EOF 信号，表明感兴趣的数据已正确发送。

无论如何，一个简单的解决方案看起来像这样，用 json 解码器包装 stdin，循环直到 eof 或发生错误，对于每个解码的数据，用 stdout 包装器将其编码为 json，再次出现错误中断。

package main

import (
    "encoding/json"
    "fmt"
    "io"
    "log"
    "os"
)

func main() {
    info, err := os.Stdin.Stat()
    if err != nil {
        log.Fatal(err)
    }

    if info.Mode()&os.ModeCharDevice != 0 {
        fmt.Println("The command is intended to work with pipes.")
        fmt.Println("cat file.json | prettyjson")
        return
    }

    dec := json.NewDecoder(os.Stdin)
    enc := json.NewEncoder(os.Stdout)
    enc.SetIndent("", "  ")

    for {
        data := map[string]interface{}{}
        if err := dec.Decode(&data); err != nil {
            if err == io.EOF {
                break
            }
            log.Fatalf("decode error %v", err)
        }
        if err := enc.Encode(data); err != nil {
            log.Fatalf("encod error %v", err)
        }
    }
}