即使没有竞争，也没有产出

在使用 Go 语言实现生产者-消费者消息队列系统时，即使不存在竞争条件，也无法获得输出。这是因为生产者在完成发送值后没有关闭消息队列通道，也没有在 done 通道上发送值，导致主程序无法立即终止。修改后的代码通过关闭消息队列通道并仅在消费完成后发送 done 值，解决了这个问题。

我正在尝试使用缓冲通道在 golang 中创建生产者-消费者消息队列系统。这是我的实现。

package main

import "fmt"

type MessageQueue struct {
    storage chan int
    count   int
}

var done = make(chan bool)

func NewMessageQueue(count int) *MessageQueue {
    ret := &MessageQueue{
        count:   count,
        storage: make(chan int, count),
    }
    return ret
}

func (m *MessageQueue) Produce() {
    for i := 0; i < m.count; i++ {
        m.storage <- i + 1
    }
    done <- true
}

func (m *MessageQueue) Consume(f func(int) int) {
    for each := range m.storage {
        fmt.Printf("%d ", f(each))
    }
}

func main() {
    op1 := func(a int) int {
        return a * a
    }
    msq := NewMessageQueue(10)
    go msq.Produce()
    go msq.Consume(op1)
    <-done
}

但不幸的是，当我运行 go run main.go 时，我无法获得输出，但是要检查是否存在任何竞争条件，当我尝试 go run -race main.go 时，我确实得到了输出。我无法理解为什么会发生这种情况。有人可以帮我吗？

正确答案

当您的生产者可以发送值时，它会在 done 通道上发送一个值，以便您的应用可以立即终止。

相反，当生产者完成后，它应该关闭 m.storage 通道，表示不会发送更多值，并且不要在 done 上发送值，因为你还没有完成！

当值被消耗时，您就完成了，因此请在 consume() 中的 done 上发送一个值：

func (m *messagequeue) produce() {
    for i := 0; i < m.count; i++ {
        m.storage <- i + 1
    }
    close(m.storage)
}

func (m *messagequeue) consume(f func(int) int) {
    for each := range m.storage {
        fmt.printf("%d ", f(each))
    }
    done <- true
}

这将输出（在 Go Playground 上尝试）：

1 4 9 16 25 36 49 64 81 100

done 通道是必需的，因为 main goroutine 中不会发生消费，并且 main goroutine 必须等待它结束。

如果您在 main goroutine 上进行消费，则可以删除 done 通道：

msq := NewMessageQueue(10)
go msq.Produce()
msq.Consume(op1)

在 Go Playground 上试试这个。

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~