对于下面的代码段,收到^ 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
}
}
答案 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...