go - 如何仅在满足某些条件的情况下才执行 `case`语句中的 `select`

Question

我有一个 channel :

aChan := make(chan struct{})

和超时持续时间var t time.Duration。我希望程序在 channel 关闭或达到 t 超时时退出，如果 t 是正持续时间。

我知道我可以使用外部 if else 循环，但这看起来非常多余:

    if t >= time.Duration(0) {
        select {
        case <-time.After(t):
            fmt.Fprintln(os.Stdout, "timeout!"))
            close(timeoutChan)
        case <-aChan:
            fmt.Fprintln(os.Stdout, "aChan is closed"))
            return
        }
    } else {
        select {
        case <-aChan:
            fmt.Fprintln(os.Stdout, "aChan is closed"))
            return
        }
    }

有更优雅的写法吗？

Answer 1

当持续时间小于零时，使用nil channel 进行超时。 nil channel 的超时情况不会执行，因为 nil channel 上的接收永远不会准备好。

var after <-chan time.Time
if t >= 0 {
    after = time.After(t)
}
select {
case <-after:
    fmt.Println("timeout!")
    close(timeoutChan)
case <-aChan:
    fmt.Println("aChan is closed")
    return
}