使用 close(ch) 无法停止 goroutine 的情况

golang学习网今天将给大家带来《使用 close(ch) 无法停止 goroutine 的情况》，感兴趣的朋友请继续看下去吧！以下内容将会涉及到等等知识点，如果你是正在学习Golang或者已经是大佬级别了，都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家！

一旦另一个 goroutine 关闭通道，我就会尝试结束多个 goroutine。但是，在收到关闭信号后，我最终陷入无限循环。我不明白为什么。

我知道可以使用 context.context 但我尝试通过关闭通道来尝试。

围棋演示：https://play.golang.org/p/c6pcygglng9

package main

import (
    "fmt"
    "time"
    "sync"
)


func runner(id int, ch <-chan struct{}, wg *sync.WaitGroup) {
    for {
        select {
            case <-time.Tick(time.Second):
                fmt.Println("worker ", id)
            case <- ch:
                fmt.Println("closing worker ", id)
                break
        }
    }

    wg.Done()

}

func main() {
    fmt.Println("Hello, playground")
    ch := make(chan struct{})
    var wg sync.WaitGroup

    wg.Add(1)
    go runner(1, ch, &wg)

    wg.Add(1)
    go runner(2, ch, &wg)

    time.Sleep(5*time.Second)

    close(ch)

    wg.Wait()
}

解决方案

问题在于您的 break 的范围：

func runner(id int, ch <-chan struct{}, wg *sync.waitgroup) {
    for {
        select {
            case <-time.tick(time.second):
                fmt.println("worker ", id)
            case <- ch:
                fmt.println("closing worker ", id)
                break
        }
    }

    wg.done()
}

您想跳出 for 循环，但实际上只是跳出 select。要解决这个问题，您有两种选择：

1. 向 for 循环添加标签，并显式中断：

   func runner(id int, ch <-chan struct{}, wg *sync.waitgroup) {
   loop:  // <---------- add a label
       for {
           select {
           case <-time.tick(time.second):
               fmt.println("worker ", id)
           case <-ch:
               fmt.println("closing worker ", id)
               break loop  // <---------- and break from it explicitly
           }
       }
   
       wg.done()
   }

2. 可能是一个更惯用和更强大的解决方案，只需在完成后返回即可。这意味着必须推迟 wg.done() 调用。

   func runner(id int, ch <-chan struct{}, wg *sync.WaitGroup) {
       defer wg.Done() // <--- Defer the wg.Done() call, so it happens on return
       for {
           select {
           case <-time.Tick(time.Second):
               fmt.Println("worker ", id)
           case <-ch:
               fmt.Println("closing worker ", id)
               return   // <--- replace `break` with `return`
           }
       }
   }

到这里，我们也就讲完了《使用 close(ch) 无法停止 goroutine 的情况》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！