我想捕获从控制台发送的 Ctrl+C (
SIGINT
) 信号并打印出一些部分运行总计。
这有效:
package main
import (
"fmt"
"os"
"os/signal"
"syscall"
"time" // or "runtime"
)
func cleanup() {
fmt.Println("cleanup")
}
func main() {
c := make(chan os.Signal)
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
go func() {
<-c
cleanup()
os.Exit(1)
}()
for {
fmt.Println("sleeping...")
time.Sleep(10 * time.Second) // or runtime.Gosched() or similar per @misterbee
}
}
稍微补充一下其他答案,如果您确实想捕获 SIGTERM (kill 命令发送的默认信号),您可以使用
syscall.SIGTERM
代替 os.Interrupt。请注意,系统调用接口是特定于系统的,可能无法在任何地方工作(例如在 Windows 上)。但它可以很好地捕获两者:
c := make(chan os.Signal, 2)
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
....
上面接受的答案中有(在发布时)一两个小错别字,所以这是清理后的版本。在此示例中,我在收到 Ctrl+C 时停止 CPU 分析器。
// capture ctrl+c and stop CPU profiler
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt)
go func() {
for sig := range c {
log.Printf("captured %v, stopping profiler and exiting..", sig)
pprof.StopCPUProfile()
os.Exit(1)
}
}()
以上所有内容在拼接时似乎都可以工作,但是gobyexample的信号页面有一个非常干净且完整的信号捕获示例。值得添加到此列表中。
package main
import (
"fmt"
"os"
"os/signal"
"syscall"
)
func main() {
sigs := make(chan os.Signal, 1)
done := make(chan bool, 1)
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
go func() {
sig := <-sigs
fmt.Println()
fmt.Println(sig)
done <- true
}()
fmt.Println("awaiting signal")
<-done
fmt.Println("exiting")
}
来源:gobyexample.com/signals
当我们运行这个程序时,它会阻塞等待信号。通过输入 Ctrl+C(终端显示为 ^C)我们可以发送 SIGINT 信号,使程序打印中断然后退出。
信号。 Notify 注册给定通道以接收指定信号的通知。
package main
import (
"fmt"
"os"
"os/signal"
"syscall"
)
func main() {
sig := make(chan os.Signal, 1)
done := make(chan bool, 1)
signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
go func() {
sig := <-sig
fmt.Println()
fmt.Println(sig)
done <- true
fmt.Println("ctrl+c")
}()
fmt.Println("awaiting signal")
<-done
fmt.Println("exiting")
}
检测 HTTP 请求取消
package main
import (
"fmt"
"net/http"
"time"
)
func main() {
mux := http.NewServeMux()
mux.HandleFunc("/path", func(writer http.ResponseWriter, request *http.Request) {
time.Sleep(time.Second * 5)
select {
case <-time.After(time.Millisecond * 10):
fmt.Println("started")
return
case <-request.Context().Done():
fmt.Println("canceled")
}
})
http.ListenAndServe(":8000", mux)
}
您可以使用不同的 goroutine 来检测 syscall.SIGINT 和 syscall.SIGTERM 信号,并使用 signal.Notify 将它们中继到通道。您可以使用通道将钩子发送到该 goroutine 并将其保存在函数切片中。当在通道上检测到关闭信号时,您可以在切片中执行这些功能。这可用于清理资源、等待运行的 goroutine 完成、保留数据或打印部分运行总计。
我编写了一个小而简单的实用程序来在关闭时添加和运行挂钩。希望它能有所帮助。
https://github.com/ankit-arora/go-utils/blob/master/go-shutdown-hook/shutdown-hook.go
您可以以“推迟”的方式执行此操作。
优雅关闭服务器的示例:
srv := &http.Server{}
go_shutdown_hook.ADD(func() {
log.Println("shutting down server")
srv.Shutdown(nil)
log.Println("shutting down server-done")
})
l, err := net.Listen("tcp", ":3090")
log.Println(srv.Serve(l))
go_shutdown_hook.Wait()
这是另一个版本,可以在您有一些任务需要清理时使用。代码将在其方法中留下清理过程。
package main
import (
"fmt"
"os"
"os/signal"
"syscall"
)
func main() {
_,done1:=doSomething1()
_,done2:=doSomething2()
//do main thread
println("wait for finish")
<-done1
<-done2
fmt.Print("clean up done, can exit safely")
}
func doSomething1() (error, chan bool) {
//do something
done:=make(chan bool)
c := make(chan os.Signal, 2)
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
go func() {
<-c
//cleanup of something1
done<-true
}()
return nil,done
}
func doSomething2() (error, chan bool) {
//do something
done:=make(chan bool)
c := make(chan os.Signal, 2)
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
go func() {
<-c
//cleanup of something2
done<-true
}()
return nil,done
}
如果您需要清理主函数,您还需要使用 go func() 在主线程中捕获信号。
从 Go 1.16 开始,你可以使用
signal.NotifyContext
,例如:
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
defer stop()
Death是一个简单的库,它使用通道和等待组来等待关闭信号。收到信号后,它将对您想要清理的所有结构调用 close 方法。
如果有人需要一种在 Windows 上处理信号的方法,仅供记录。
我需要处理程序 A 通过 os/exec 调用程序 B,但程序 B 始终无法正常终止,因为 Windows 不支持通过
cmd.Process.Signal(syscall.SIGTERM)
发送信号或其他信号。
我通过创建一个临时文件作为信号来解决这个问题。例如,程序 A 创建文件
.signal.term
,程序 B 需要检查该文件是否存在于间隔基础上。如果文件存在,它将退出程序并根据需要进行清理。
我确信还有其他方法,但这做到了。