我正在将数据写入网络。
写goroutine是这样的。
forend:
for{
select{
case buf,ok:=<-sendqueue:
if !ok{
break forend
}
writeBuffer(conn,buf)
}
}
Run Code Online (Sandbox Code Playgroud)
变量 conn 是一个 net.Conn。
那我想用bufio来代替net.Conn。
iowriter:=bufio.NewWriter(conn)
Run Code Online (Sandbox Code Playgroud)
iowriter 将缓存数据。为了减少延迟,我必须在发送队列中没有更多数据时立即刷新 iowriter。
所以我在编写 goroutine 中添加了一个 default case
forend:
for{
select{
case buf,ok:=<-sendqueue:
if !ok{
break forend
}
writeBuffer(iowriter,buf)
default:
iowriter.Flush()
time.Sleep(time.Millisecond)
}
}
Run Code Online (Sandbox Code Playgroud)
time.Sleep 是必须的,否则 goroutine 会运行 busy loop。
但在这种情况下,真正的需求是阻塞不睡觉。
最后,我找到了一个解决方案,有两个选择。
forend:
for{
select{
case buf,ok:=<-sendqueue:
if !ok{
break forend
}
writeBuffer(iowriter,buf)
}
nextloop:
for{
select{
case buf,ok:=<-sendqueue:
if !ok{
break forend
}
writeBuffer(iowriter,buf)
default:
iowriter.Flush()
break nextloop
}
}
}
Run Code Online (Sandbox Code Playgroud)
但是这个解决方案很复杂。第二个选择是第一个的重复,默认情况下。有没有更好的解决方案?
~~~~~~~~~~~~~~~~~~~~~~~~ 更多解释
这里的行为是这样的。如果sendqueue不为空,我继续pop data并发送。如果sendqueue 为空,我想立即刷新缓存在 iowriter 中的数据,然后再次等待 sendqueue。
这种行为可以像这样抽象。我想在通道为空时做一些事情,然后再次等待通道,并阻塞。我找到了两个重复的选择解决方案。但是当等待多个通道时,该解决方案可能会变得更加复杂.所以我正在寻找更好的解决方案。
两步解决方案让您走在正确的轨道上。
由于您想要做两件事:首先接收一个数据包(如果一个数据包尚不可用则阻塞),然后在有数据包时继续接收并在全部消耗完时刷新,您不能用一条select语句来做到这一点。
但是我们可以做一些简化,第一条select语句完全没有必要,因为它只有一个case分支:
forend:
for {
buf, ok := <-sendqueue
if !ok {
break forend
}
writeBuffer(iowriter, buf)
nextloop:
for {
select {
case buf, ok := <-sendqueue:
if !ok {
break forend
}
writeBuffer(iowriter, buf)
default:
iowriter.Flush()
break nextloop
}
}
}
Run Code Online (Sandbox Code Playgroud)
由于在两种情况下处理接收到的数据是相同的,您可以将该逻辑放入函数中以避免重复:
var buf datatype
var ok = true
handle := func() {
if ok {
writeBuffer(iowriter, buf)
}
}
for ok {
buf, ok = <-sendqueue
handle()
nextloop:
for ok {
select {
case buf, ok = <-sendqueue:
handle()
default:
iowriter.Flush()
break nextloop
}
}
}
Run Code Online (Sandbox Code Playgroud)