在Go中收到SIGINT时是否调用了延迟函数?

时间:2018-05-18 08:07:02

标签: go goroutine deferred-execution resource-cleanup

对于下面的代码段,收到^ C时不会进行延期通话。清理是否可能引入竞争条件?如果是,那么在接收中断时可能有更好的清理模式?

  func fn() {
    // some code
    defer cleanup()
    go func() {
       c := make(chan os.Signal, 1)
       signal.Notify(c, os.Interrupt)

       // Block until a signal is received.
       _ = <-c
       cleanup()
     }
     for {
        // Infinite loop. Returns iff an error is encountered in the 
        // body
     }
}

1 个答案:

答案 0 :(得分:6)

请注意,如果你&#34;安装&#34;您的信号频道为signal.Notify(),默认行为将被禁用。这意味着如果您执行此操作,for函数中的fn()循环将不会中断,它将继续运行。

因此,当您在注册频道上收到一个值时,您必须使for循环终止,以便您可以执行&#34;清理&#34;清理。否则cleanup()应该释放的资源可能仍会在for中使用,最有可能导致错误或恐慌。

执行此操作后,您甚至不必手动调用cleanup(),因为从fn()返回将正确运行延迟功能。

以下是一个例子:

var shutdownCh = make(chan struct{})

func fn() {
    defer cleanup()

    go func() {
        c := make(chan os.Signal, 1)
        signal.Notify(c, os.Interrupt)
        <-c
        close(shutdownCh)
    }()

    for {
        select {
        case <-shutdownCh:
            return
            // Other cases might be listed here..
        default:
        }
        time.Sleep(time.Millisecond)
    }
}

当然,上述示例并不保证应用终止。你应该有一些代码来监听shutdownCh并终止应用程序。此代码还应该等待所有goroutine优雅地完成。为此你可以使用sync.WaitGroup:当你启动一个应该在退出时等待的goroutine时加1,并在这样的goroutine结束时调用WaitGroup.Done()

此外,由于在真正的应用程序中可能会有很多这样的应用程序,因此信号处理应该转移到一个中心位置。在每个地方放置并没有完成。

以下是如何做到这一点的完整示例:

var shutdownCh = make(chan struct{})
var wg = &sync.WaitGroup{}

func main() {
    wg.Add(1)
    go func() {
        defer wg.Done()
        fn()
    }()

    c := make(chan os.Signal, 1)
    signal.Notify(c, os.Interrupt)
    <-c
    close(shutdownCh)
    wg.Wait()
}

func fn() {
    defer cleanup()
    for {
        select {
        case <-shutdownCh:
            return
            // Other cases might be listed here..
        default:
        }
        fmt.Println("working...")
        time.Sleep(time.Second)
    }
}

func cleanup() {
    fmt.Println("cleaning up...")
}

这是以上应用的示例输出,在启动后3秒按 CTRL + C

working...
working...
working...
^Ccleaning up...