从stdin接收二进制数据,发送到Go中的通道

时间:2013-11-24 16:17:27

标签: binary go stdin channel

所以我有以下测试Go代码,它被设计为通过stdin从二进制文件读取,并将读取的数据发送到通道,然后进一步处理。在我给出的版本中,它只从stdin中读取前两个值,尽管就显示问题而言这很好。

package main

import (
    "fmt"
    "io"
    "os"
)

func input(dc chan []byte) {
    data := make([]byte, 2)
    var err error
    var n int
    for err != io.EOF {
        n, err = os.Stdin.Read(data)
        if n > 0 {
            dc <- data[0:n]
        }
    }
}

func main() {
    dc := make(chan []byte, 1)
    go input(dc)
    fmt.Println(<-dc)
}

为了测试它,我首先使用go build构建它,然后使用command-

将数据发送给它
./inputtest < data.bin

我目前使用的数据只是使用openssl命令创建的随机二进制数据。

我遇到的问题是它错过了Stdin的第一个值,只给出了第二个和更大的值。我认为这与通道有关,因为移除通道的相同脚本会生成正确的数据。有没有人遇到过这个?例如,运行此命令时,我得到以下输出 -

./inputtest < data.bin
[36 181]

我应该得到 -

./inputtest < data.bin
[72 218]

(两个实例中的二进制数据相同。)

1 个答案:

答案 0 :(得分:5)

你在每次读取时都会覆盖你的缓冲区,并且你有一个通道缓冲区,所以每次通道空间都会丢失数据。

尝试这样的事情(未经测试,写在平板电脑上等等):

func input(dc chan []byte) {
    defer close(dc)
    for {
        data := make([]byte, 2)
        n, err := os.Stdin.Read(data)
        if n > 0 {
            dc <- data[0:n]
        }
    }
}