主页 / 问题 / 1447065
Accepted
Paul Shipilov
Asked:2022-09-09 18:29:04 +0000 UTC2022-09-09 18:29:04 +0000 UTC

构建具有读取元素通道和完成通道的异步架构

  • 772

遇到了一种情况，我不确定从通道顺序读取时它在 golang 中的行为方式，这应该以来自“输出通道”的完成信号结束。这是一个描述我的代码的示例：

func f() {
    doneCh := make(chan struct{})
    trafficCh := make(chan interface{})

    go write(doneCh, trafficCh)
    read(doneCh, trafficCh)
}


func write(doneCh chan<- struct{}, sendCh chan<- interface{}) {
    defer func() {
        doneCh <- struct{}{}
    }()

    for {
        // Send something in sendCh
    }
}


func read(doneCh <-chan struct{}, recvCh <-chan interface{}) {
    for {
        select {
            case <-doneCh:
                return
            case item := <-recvCh:
                // Something do with item
        }
    }
}

根据标准，如果 select 在不同的通道中收到了多条消息，那么它会随机选择一条。是否有可能当 select 选择一个频道时，它会在doneChand中看到消息recvCh，即 将发生未定义的行为，我们可以在不读取所有元素的情况下结束读取。

golang

0 个回答

  1. Best Answer

    在您的情况下，您不需要创建第二个频道。当您将所有内容写入通道时，您只需要关闭它。

    func f() {
    trafficCh := make(chan string)

    go write(trafficCh)
    read(trafficCh)
}

func write(sendCh chan<- string) {
    // Записать всё что нужно
    for i := 0; i < 100; i++ {
        sendCh <- fmt.Sprintf("HELLO %d", i)
    }
    close(sendCh)
}

func read(recvCh <-chan string) {
    for str := range recvCh {
        fmt.Println(str)
    }
}

    对于未来... 通常需要优先考虑应该首先阅读哪个频道。要做到这一点，这样做：（乍一看它看起来很奇怪）

    for {
        select {
        case item := <-recvCh:
            // Something do with item
        default:
        }
        select {
        case <-doneCh:
            return
        case item := <-recvCh:
             // Something do with item
        }
    }
    • 1

相关问题