生产者速度慢，消费者速度快的情况下如何处理通道关闭同步？

Question

我是新手，找不到这个问题的答案。我正在做的是在生产者中读取 CSV 文件，做一些可能需要时间的事情，然后通过通道将输出发送给消费者。有一个生产者-消费者链，任何生产者最终都可能比它的消费者慢。producer(1 goroutine) -> chan0 -> consumer-producer-1(>1 goroutines) -> chan1 -> consumer-producer-2(>1 goroutines) -> chan2 -> consumer(>1 goroutines)这里最多可以有 15 个消费者。现在我面临的问题是，如果生产者完成了，如何在消费者方面做出决定，我们可以停止处理。我需要实现的是：一旦生产者完成，所有消费者最终应该做一些清理并在完成剩余的后退出如果消费者在特定的超时期限内没有获得任何数据，它可以退出（最好有一个信号）而不会进一步阻塞。它发生在序列中的所有生产者-消费者对上。我使用了以下方法。将信号通道与每个数据通道保持在一起，并为其下一个消费者的每个 goroutine 发布“完成”。读取后，每个消费者应该只读取通道中剩余的缓冲数据，然后将 5“完成”放在下一个信号通道上。确保每个 goroutine 只有 5 个，而不是 5 个（使用https://golang.org/pkg/sync/#Once.Do）。以下是我到这里为止能想到的。processRemaining = falsefor processRemaining == false{        select {        case stuff, ok := <-input_messages:                do_stuff(stuff)                if ok == false { // if channel has been closed                    processRemaining = true                }                if result != nil {                        //send to channel output_messages                }        case sig := <-input_signals: // if signaled to stopped.                fmt.Println("received signal", sig)                processRemaining = true        default:                fmt.Println("no activity")        }}if processRemaining {        for stuff := range input_messages {                do_stuff(stuff)                if result != nil {                        //send to channel output_messages                }        }        // send "output_routine" number of "done" to a channel "output_signals".}但即使在这种方法中，我也无法想出任何方式来表现与关闭的“input_messages”通道相同的方式，如果没有可用的时间，比如 10 秒。这种方法有没有我忽略的问题。解决这个问题的可能方法（或并发模式）是什么？确保：一旦第一个“chan0”关闭，所有后续通道都会关闭。所有的生产者在关闭他们的输出通道之前都会被更新，只有当他们都完成了他们的写入时，通道才会关闭。如果消费者在指定的超时时间内没有从通道中获取数据，它应该将其视为已关闭，并自行解除阻塞。

Answer 1

使用 async.WaitGroup来跟踪正在运行的 goroutines 的数量。每个 goroutine 在不再从通道获取数据后退出。一旦WaitGroup完成，清理就可以完成了。

是这样的：

import (

        "sync"

        "time"

)

type Data interface{} // just an example

type Consumer interface {

        Consume(Data) Data

        CleanUp()

        Count() int

        Timeout() time.Duration

}

func StartConsumers(consumer Consumer, inCh <-chan Data, outCh chan<- Data) {

        wg := sync.WaitGroup{}

        for i := 0; i < consumer.Count(); i++ {

                wg.Add(1)

                go func() {

                consumeLoop:

                        for {

                                select {

                                case v, ok := <-inCh: // 'ok' says if the channel is still open

                                        if !ok {

                                                break consumeLoop

                                        }

                                        outCh <- consumer.Consume(v)

                                case <-time.After(consumer.Timeout()):

                                        break consumeLoop

                                }

                        }

                        wg.Done()

                }()

        }

        wg.Wait()

        consumer.CleanUp()

        close(outCh)

}

在管道的每个阶段，您都可以使用与上述类似的过程来启动消费者。