提问人:HappyFace 提问时间:9/21/2020 更新时间:9/21/2020 访问量:1349
戈朗:'哎呀。Stdin.Read' 不处理由唯一的 EOF 组成的输入?
golang: `os.Stdin.Read` doesn't handle input consisting of a sole EOF?
问:
我正在编写一只猫,在接收第一个字节时超时。我让它工作,但它无法处理:echo -n
❯ echo -n | time possiblycat 1000 # 1000 is the timeout in milliseconds
possiblycat 1000 0.00s user 0.00s system 0% cpu 1.008 total; max RSS 1864
cat
本身对此没有问题;它注意到 EOF 并立即退出:
❯ echo -n | time cat
cat 0.00s user 0.00s system 71% cpu 0.003 total; max RSS 664
这是以下全部来源:possiblycat
package main
import (
"io"
"io/ioutil"
"os"
"strconv"
"time"
)
func main() {
wait := 10
if len(os.Args) >= 2 {
waitDummy, err := strconv.Atoi(os.Args[1])
if err != nil {
panic(err)
}
wait = waitDummy
}
b := make(chan byte, 1)
go scan(b)
select {
case res := <-b:
inBytes, err := ioutil.ReadAll(os.Stdin)
if err != nil {
panic(err)
}
stdin := append([]byte{res}, inBytes...)
_, err2 := os.Stdout.Write(stdin)
if err2 != nil {
panic(err2)
}
case <-time.After(time.Duration(wait) * time.Millisecond):
os.Exit(1)
}
}
func scan(out chan byte) {
var b []byte = make([]byte, 1)
_, err := os.Stdin.Read(b)
if err == io.EOF {
return
} else if err != nil {
panic(err)
}
out <- b[0]
}
相关:
答:
4赞
Marc
9/21/2020
#1
当返回时,你退出在它自己的 goroutine 中运行的函数。os.Stdin.Read
EOF
scan
但是,没有采取任何措施来告诉主 goroutine 所有输入都已处理。它正在等待通道上的数据或超时。由于没有数据传入,因此达到超时。b
b
为了正确处理这个问题,案例应该向主要的 goroutine 发出信号,表明没有更多的工作要做。一种常见的模式(但肯定不是唯一的模式)是有一个通道,指示所有工作都已完成。err == io.EOF
done
done := make(chan bool, 1)
go scan(b, done)
select {
case res := <-b:
...
case <-done:
os.Exit(1)
case <-time.After(time.Duration(wait) * time.Millisecond):
os.Exit(1)
}
}
func scan(out chan byte, done chan bool) {
var b []byte = make([]byte, 1)
_, err := os.Stdin.Read(b)
if err == io.EOF {
fmt.Println("got EOF, exiting")
done <- true
return
} else if err != nil {
...
}
另一个(甚至更简单)的替代方法是在完成后简单地关闭数据通道:
func scan(out chan byte) {
var b []byte = make([]byte, 1)
_, err := os.Stdin.Read(b)
if err == io.EOF {
fmt.Println("got EOF, exiting")
close(out)
return
} else if err != nil {
panic(err)
}
out <- b[0]
}
评论