禁用选择语句的默认部分

Question

我想要一种在选择语句的默认部分进行处理的方法，然后稍后禁用默认值。我在下面的代码中使用了 if 语句，但是，在完成后，我们只有一个繁忙的循环（当通道为空时）将继续旋转测试完成的标志，这将始终为真。这似乎不是正确的方法。for {    select {    case p := <-subscriber.Next:        ...    case err := <subsriber.OnError:        ...    default:        if !finished {            subscriber.Subscribe(name, params)            ...            if ... { finished = true }        }            }}我知道你不能禁用默认部分，但必须有更好的方法来做到这一点。

Answer 1

您不能禁用default子句，但可以case通过分配nil给 chan 来禁用子句。也许像下面的代码一样创建一个非常快速的代码。下面的代码将 chanC从股票代码分配给defaultCh然后通过分配禁用它nil。

    ticker := time.NewTicker(time.Microsecond)

    defaultCh := ticker.C

    for {

        select {

            case p := <-subscriber.Next:

                ...

            case err := <subscriber.OnError:

                ...

            case <-defaultCh:

                subscriber.Subscribe(name, params)

                ...

                if finished {

                    defaultCh = nil

                    ticker.Stop()

                }

            }

        }

    }

Answer 2

我不知道有任何内置功能可以执行此操作，但您可以根据您的状态输入两个不同的选择语句。

for {

    // Without default case, wait only for cases

    if finished {

        select {

        case p := <-subscriber.Next:

            //...

        case err := <-subsriber.OnError:

            //...

        }

    } else {

        // Allow default case

        select {

        case p := <-subscriber.Next:

            //...

        case err := <-subsriber.OnError:

            //...

        default:

            subscriber.Subscribe(name, params)

        }

    }

}

这当然会导致代码重复，但至少它不会进入循环只是发现什么都不需要做。