You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
603 B
Go
32 lines
603 B
Go
3 years ago
|
//go:build linux || darwin
|
||
|
|
// +build linux darwin
|
||
|
|
|
||
|
|
package migrate
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"os/signal"
|
||
|
|
"syscall"
|
||
|
|
|
||
|
|
"github.com/zeromicro/go-zero/core/syncx"
|
||
|
|
"github.com/zeromicro/go-zero/tools/goctl/util/console"
|
||
|
|
)
|
||
|
|
|
||
|
|
func cancelOnSignals() {
|
||
|
|
doneChan := syncx.NewDoneChan()
|
||
|
|
defer doneChan.Close()
|
||
|
|
|
||
|
|
go func(dc *syncx.DoneChan) {
|
||
|
|
c := make(chan os.Signal)
|
||
|
|
signal.Notify(c, syscall.SIGTERM, syscall.SIGKILL, syscall.SIGINT, syscall.SIGTSTP, syscall.SIGQUIT)
|
||
|
|
select {
|
||
|
|
case <-c:
|
||
|
|
console.Error(`
|
||
|
|
migrate failed, reason: "User Canceled"`)
|
||
|
|
os.Exit(0)
|
||
|
|
case <-dc.Done():
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}(doneChan)
|
||
|
|
}
|