关闭未知长度的通道

时间:2015-12-15 07:24:41

标签: go concurrency channel

我不知道关于它的信息时关闭频道 长度

package main

import (
    "fmt"
    "time"
)

func gen(ch chan int) {
    var i int
    for {
        time.Sleep(time.Millisecond * 10)
        ch <- i
        i++
        // when no more data (e.g. from db, or event stream)
        if i > 100 {
            break
        }
    }

    // hot to close it properly?
    close(ch)
}

func receiver(ch chan int) {
    for i := range ch {
        fmt.Println("received:", i)
    }
}

func main() {
    ch := make(chan int)

    for i := 0; i < 10; i++ {
        go gen(ch)
    }

    receiver(ch)
}

它给了我错误

panic: send on closed channel

goroutine 8 [running]:
main.gen(0xc82001a0c0)
    /home/exu/src/github.com/exu/go-workshops/100-concurrency-channels/16-close-problem.go:12 +0x57
created by main.main
    /home/exu/src/github.com/exu/go-workshops/100-concurrency-channels/16-close-problem.go:35 +0xbd

goroutine 1 [panicwait]:
runtime.gopark(0x0, 0x0, 0x50b8e0, 0x9, 0x10, 0x1)
    /usr/lib/go/src/runtime/proc.go:185 +0x163
runtime.main()
    /usr/lib/go/src/runtime/proc.go:121 +0x2f4
runtime.goexit()
    /usr/lib/go/src/runtime/asm_amd64.s:1696 +0x1

goroutine 6 [sleep]:
time.Sleep(0x989680)
    /usr/lib/go/src/runtime/time.go:59 +0xf9
main.gen(0xc82001a0c0)
    /home/exu/src/github.com/exu/go-workshops/100-concurrency-channels/16-close-problem.go:11 +0x29
created by main.main
    /home/exu/src/github.com/exu/go-workshops/100-concurrency-channels/16-close-problem.go:33 +0x79

goroutine 7 [sleep]:
time.Sleep(0x989680)
    /usr/lib/go/src/runtime/time.go:59 +0xf9
main.gen(0xc82001a0c0)
    /home/exu/src/github.com/exu/go-workshops/100-concurrency-channels/16-close-problem.go:11 +0x29
created by main.main
    /home/exu/src/github.com/exu/go-workshops/100-concurrency-channels/16-close-problem.go:34 +0x9b
exit status 2

它是合乎逻辑的 - 当第二个试图发送给它时,第一个goroutine关闭通道。在这种情况下关闭渠道的最佳方法是什么?

3 个答案:

答案 0 :(得分:12)

频道关闭后,您无法在其上发送更多值,否则会发生恐慌。这就是你的体验。

这是因为您启动了多个使用相同通道的goroutine,并且它们会在其上发送值。然后你关闭每个频道。而且由于它们不同步,一旦第一个goroutine到达关闭它的点,其他人可能(并且他们将继续)继续发送它的值:恐慌!

您只能关闭频道一次(尝试关闭已经关闭的频道也会发生恐慌)。当你在其上发送所有值的goroutine完成时,你应该这样做。为此,您需要检测何时完成所有发送方goroutine。检测此问题的惯用方法是使用sync.WaitGroup

对于每个已启动的发件人goroutine,我们使用WaitGroup.Add()WaitGroup添加1。发送值的每个goroutine都可以通过调用WaitGroup.Done()来发出信号。最好将此作为延迟声明,所以如果你的goroutine会突然终止(例如恐慌),WaitGroup.Done()仍会被调用,并且不会让其他goroutines挂起(等待赦免 - &#34;缺失& #34; WaitGroup.Done()打电话永远不会......)。

并且WaitGroup.Wait()将等待所有发送方goroutine完成,并且仅在此之后且仅一次将关闭该通道。我们想要发现这个&#34;全球&#34;完成事件并关闭频道,同时处理在其上发送的值正在进行中,因此我们必须在自己的goroutine中执行此操作。

接收器goroutine将一直运行,直到通道关闭,因为我们在通道上使用了for ... range构造。并且由于它在主goroutine中运行,因此程序将不会退出,直到从通道正确接收和处理所有值。 for ... range构造循环,直到收到在关闭通道之前发送的所有值。

请注意,下面的解决方案也可以使用缓冲和非缓冲通道而无需修改(尝试使用带ch := make(chan int, 100)的缓冲通道。)

正确的解决方案(在Go Playground上试试):

func gen(ch chan int, wg *sync.WaitGroup) {
    defer wg.Done()
    var i int
    for {
        time.Sleep(time.Millisecond * 10)
        ch <- i
        i++
        // when no more data (e.g. from db, or event stream)
        if i > 100 {
            break
        }
    }
}

func receiver(ch chan int) {
    for i := range ch {
        fmt.Println("received:", i)
    }
}

func main() {
    ch := make(chan int)
    wg := &sync.WaitGroup{}

    for i := 0; i < 10; i++ {
        wg.Add(1)
        go gen(ch, wg)
    }

    go func() {
        wg.Wait()
        close(ch)
    }()

    receiver(ch)
}

注意:

请注意,receiver(ch)在主goroutine中运行,以及等待WaitGroup并在其自己(非主要)goroutine中关闭通道的代码非常重要;而不是相反。如果要切换这两个,可能会导致提前退出&#34;,并非所有值都可能从通道接收和处理。这是因为当主goroutine完成时,Go程序退出(spec:Program execution)。它不会等待其他(非主要)goroutines完成。因此,如果等待和关闭通道将在主goroutine中,在关闭通道之后,程序可以随时退出,而不是等待另一个goroutine,在这种情况下将循环以从通道接收值。

答案 1 :(得分:0)

使用select而不是range。

for{
    i,ok:=<-ch
    //process with i
    if !ok {
        break
    }
}

答案 2 :(得分:0)

使用Go频道的一个一般原则是,不要从接收方关闭频道,如果该频道有多个并发发件人,则不要关闭频道。

每个通道在标记为要清除后最终都会被GC,因此可以将通道保持关闭状态,这唯一的区别是,经过几个周期后,该通道将可用于gc如果没有明确关闭。

但是,如果您可以关闭通道,那总是好的。请通过以下链接进行详细说明。

thisthis条展示了在1:N,N:1或M:N(发送方:接收方)的情况下关闭频道的各种方法