如何在golang中阻止选择默认情况?

海牙客*_*客移库 3 select channel go

我正在将数据写入网络。
写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。
这种行为可以像这样抽象。我想在通道为空时做一些事情,然后再次等待通道,并阻塞。我找到了两个重复的选择解决方案。但是当等待多个通道时,该解决方案可能会变得更加复杂.所以我正在寻找更好的解决方案。

icz*_*cza 5

两步解决方案让您走在正确的轨道上。

由于您想要做两件事:首先接收一个数据包(如果一个数据包尚不可用则阻塞),然后在有数据包时继续接收并在全部消耗完时刷新,您不能用一条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)