信号 goroutine 在通道关闭时停止

Posted 2023-03-14

【中文标题】信号 goroutine 在通道关闭时停止

【英文标题】：Signal goroutines to stop with channel close

【发布时间】：2019-04-15 16:02:06

【问题描述】：

我有多个 goroutines select 来自两个通道：一个通道提供数据，一个通道用于信号（一种完成/退出通道）。

我使用信号通道来捕获信号（杀死）并优雅地关闭 goroutine。

我从package a 运行'worker' goroutines，而捕获信号的goroutine 函数从package b 运行。

我使用来自https://gist.github.com/reiki4040/be3705f307d3cd136e85 的信号包。

package a

import "sync"

WorkChan := make(chan int)
QuitChan := make(chan struct)

func Stop() 
        fmt.Println("Stop called, closing channel")
        close(QuitChan)


func Work(wg *sync.WaitGroup) 
    var item int
    for 
        select 
        case item = <- WorkChan:
            ... processing
        case <- QuitChan:
            wg.Done()
            return
        
    

捕获信号并调用a.Stop()的goroutine

package b

import (
    "os/signal"
    "os"
    "syscal"
    "a"
)

func Signal() 

    sChan := make(chan os.Signal, 1)
    signal.Notify(signalChan, syscall.SIGTERM, syscall.SIGINT)

    for 
        s := <-sChan
        switch s 
        case os.Interrupt, syscall.SIGTERM:
            a.Stop()
        
    

这是我的主要功能

package main

import (
    "a"
    "b"
    "sync"
)

func main() 

    var wg sync.WaitGroup

    go b.Signal()

    wg.Add(1) // for simplicity; actual code start multiple goroutines of Work
    go a.Work(&wg)

    // wait until work is done
    wg.Wait()
    fmt.Println("Done.")

当我终止正在运行的进程时，我看到来自Quit 的打印消息。我预计一旦通道关闭，goroutines 将在某个时候 select QuitChan 案例并返回。

但他们一直在奔跑；他们继续处理来自WorkChan 的项目。似乎它被忽略了。我在这里想念什么？ 通道不会关闭吗？怎么还开着？

【问题讨论】：

当你关闭a.QuitChan，这将终止a的goroutine（迟早），但是b包中的goroutine没有终止条件，它将永远运行。此外，应用程序不会因为某些“随机” goroutine 结束而终止，应用程序会在 main goroutine 结束时终止（我们不知道您的 main goroutine 是什么）。

您能详细说明“他们一直在运行”是什么意思吗？

b 的 goroutine 中有一个循环，里面没有 return 和 break 语句，所以它会一直运行。 goroutine 或其循环不会因为独立的 goroutine 结束而神奇地结束或返回。

仅仅因为您看到"Stop called, closing channel" 打印，这并不能保证QuitChan 已关闭（因为打印在close() 调用之前）。在close()之后添加打印语句并确认您是否看到打印。

您正在复制 WaitGroup。请改用指针。我认为go vet 应该抱怨这一点。

【参考方案1】：

首先我认为你应该做一个简单的测试，然后通过它。让其他人了解您的问题会更有帮助。

我更改了您的代码，使其像 go 代码一样阅读，而不是其他语言。 现在可以了。

在您的代码中，有一些错误，我将其标记为 ERROR 注释。有些是语法错误，例如创建WorkChan。有些是类型错误。

你应该知道的一个导入设计的事情，当你想在执行Stop()后退出时，你应该关闭你发送数据到WorkChan的WorkChan，而不是在你收到日期的地方返回。

去吧

package a

import (
    "fmt"
    "sync"
)

// ERROR: can not do make in global
var WorkChan chan int
var QuitChan chan struct

// Create chan when init
func init() 
    fmt.Println("Init a")
    WorkChan = make(chan int)
    QuitChan = make(chan struct)


func Stop() 
    fmt.Println("Stop called, closing quit channel")
    close(QuitChan)


// Close the work channel where you send date
func Start(wg *sync.WaitGroup) 
    i := 0
    for 
        select 
        case <-QuitChan:
            fmt.Println("Closing work chan")
            close(WorkChan)
            wg.Done()
            return
        default:
            WorkChan <- i
            i++
        
    


// Work will exit when workchan closed
func Work(wg *sync.WaitGroup) 
    for item := range WorkChan 
        fmt.Printf("Receive %d\n", item)
    
    wg.Done()
    fmt.Println("Work exit")

b.go

package b

import (
    "github.com/shitaibin/awesome/a"
    "os"
    "os/signal"
    "syscall"
)

func Signal() 

    sChan := make(chan os.Signal, 1)
    signal.Notify(sChan, syscall.SIGTERM, syscall.SIGINT) // ERROR

    for 
        s := <-sChan
        switch s 
        case os.Interrupt, syscall.SIGTERM:
            a.Stop()
            return // should return free resource
        
    

main.go

package main

import (
    "fmt"
    "github.com/shitaibin/awesome/a"
    "github.com/shitaibin/awesome/b"
    "sync"
)

func main() 

    var wg sync.WaitGroup

    go b.Signal()

    wg.Add(1)      // for simplicity; actual code start multiple goroutines of Work
    go a.Work(&wg) // ERROR: pointer

    wg.Add(1)
    go a.Start(&wg) // Send data and close channel when stop

    // wait until work is done
    wg.Wait()
    fmt.Println("Done.")

结果

// omit
Receive 87028
Receive 87029
Receive 87030
Receive 87031
Receive 87032
Receiv^C101    <---- send signal here
Receive 87102
Receive 87103
Receive 87104
Receive 87105
Receive 87106
Receive 87107
Receive 87108
Receive 87109
Receive 87110
Stop called, closing quit channel
Receive 87111
Receive 87112
Closing work chan
Work exit
Done.

【讨论】：

我已经在使用 &wg 所以这不是问题，但是在将通道初始化移动到一个函数之后我确实取得了一些进展。我在全局范围内声明 QuitChan，在 Start() 中初始化它，但是当我现在关闭它时，我得到 panic: close of nil channel