select - 如何在 golang 的 select default case 中阻塞？

Question

我正在向网络写入数据。
编写goroutine是这样的。

forend:
for{
    select{
    case buf,ok:=<-sendqueue:
        if !ok{
            break forend
        }
        writeBuffer(conn,buf)
    }
}

变量conn是一个net.Conn。
那我想用bufio来代替net.Conn。

iowriter:=bufio.NewWriter(conn)

iowriter会缓存数据。为了减少延迟，我必须在sendqueue中没有更多数据时立即刷新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)
    }
}

time.sleep 是必须的，否则goroutine会忙循环。
但是在这种情况下，真正的需求是block not to sleep。
最后，我找到了一个解决方案，有两个选择。

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
        }
    }
}

但是这个解决方案很复杂。第二个 select 是第一个 default case 的副本。有没有更好的解决方案？

~~~~~~~~~~~~~~~~~~~~~~~ 更多解释
这里的行为是这样的，如果sendqueue不为空，我继续pop数据发送。如果sendqueue为空，我想立即刷新iowriter中缓存的数据，然后再次等待sendqueue。
这种行为可以像这样抽象。我想在 channel 为空时做一些事情，然后再次等待 channel ，阻塞。我找到了两个重复的选择解决方案。但是当等待多个 channel 时，该解决方案可能会变得更加复杂.所以我正在寻找更好的解决方案。

Answer 1

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

因为你想做两件事:首先接收一个数据包(如果一个数据包还不可用则阻塞)，然后在有数据包的情况下继续接收，如果全部消耗完则刷新，你不能用一个 选择语句。

但是我们可以做一些简化，第一个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
            }
        }
    }

由于在这两种情况下处理接收到的数据是相同的，您可以将该逻辑放入一个函数中以避免重复:

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
        }
    }
}