从 Golang 流中读取第一个字节

珍惜时间，勤奋学习！今天给大家带来《从 Golang 流中读取第一个字节》，正文内容主要涉及到等等，如果你正在学习Golang，或者是对Golang有疑问，欢迎大家关注我！后面我会持续更新相关内容的，希望都能帮到正在学习的大家！

我在 golang 中有以下函数：

func ReadFromStdinIfAvailable(cmd *cobra.Command, args []string) ([]byte, error) {
    if len(args) == 0 {
        r := bufio.NewReader(os.Stdin)
        firstByte, err := r.Peek(1)
        if err != nil {
            return nil, err
        }

        if len(firstByte) == 0 {
            cmd.Println(userstrings.NoFilenameProvidedErrorString)
            return nil, errors.New(userstrings.NoFilenameProvidedErrorString)
        }

        var bytesResult []byte
        scanner := bufio.NewScanner(r)
        for scanner.Scan() {
            // Read from stdin - allocate a buffer to read into
            bytesResult = append(bytesResult, scanner.Bytes()...)
        }
        return bytesResult, nil
    }
    return nil, errors.New(userstrings.NoStdInProvidedErrorString)
}

这是一个非常小的数据量（<10 mb 是肯定的）。但是，数据应该在第一次读取时就存在 - 如果不是，则会出现错误。

但是，当我这样做时，它挂在 scanner.scan() 上，不出所料，它是一个流。

在开始读取之前，如何查看 os.stdin 流以查看其中是否有任何数据？

正确答案

这就是我最终所做的 - 如果这对任何人有帮助（希望有人做得更好）：

func ReadFromStdinIfAvailable(cmd *cobra.Command, args []string) ([]byte, error) {
    if len(args) == 0 {
        r := bufio.NewReader(RootCmd.InOrStdin())
        var bytesResult []byte
        scanner := bufio.NewScanner(r)

        // buffered channel of dataStream
        dataStream := make(chan []byte, 1)

        // Run scanner.Bytes() function in it's own goroutine and pass back it's
        // response into dataStream channel.
        go func() {
            for scanner.Scan() {
                dataStream <- scanner.Bytes()
            }
            close(dataStream)
        }()

        // Listen on dataStream channel AND a timeout channel - which ever happens first.
        timedOut := false
        select {
        case res := <-dataStream:
            bytesResult = append(bytesResult, res...)
        case <-time.After(time.Duration(10) * time.Millisecond): //nolint:gomnd // 10ms timeout
            timedOut = true
        }

        if timedOut {
            cmd.Println(userstrings.NoFilenameProvidedErrorString)
            return nil, errors.New(userstrings.NoFilenameProvidedErrorString)
        }
        for scanner.Scan() {
            bytesResult = append(bytesResult, scanner.Bytes()...)
        }

        return bytesResult, nil
    }
    return nil, fmt.Errorf("should not be possible, args should be empty")
}

到这里，我们也就讲完了《从 Golang 流中读取第一个字节》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！