即使没有竞争条件也得不到任何输出

Not getting any output even if there is no race conditions

我正在尝试使用缓冲通道在 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 上试试这个。