You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
459 B
26 lines
459 B
2 months ago
|
package graceful_shutdown
|
||
|
|
|
||
|
|
import (
|
||
|
|
"git.rinsvent.ru/rinsvent/gol"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
"os"
|
||
|
|
"os/signal"
|
||
|
|
"syscall"
|
||
|
|
)
|
||
|
|
|
||
|
|
type FuncShutdowner struct{}
|
||
|
|
|
||
|
|
func (sh *FuncShutdowner) Wait(f func()) {
|
||
|
|
sigChan := make(chan os.Signal, 2)
|
||
|
|
signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
|
||
|
|
sig := <-sigChan
|
||
|
|
|
||
|
|
gol.With(zap.String("sig", sig.String())).Info("Gracefully shutting down")
|
||
|
|
|
||
|
|
f()
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewFuncShutdowner() FuncShutdowner {
|
||
|
|
return FuncShutdowner{}
|
||
|
|
}
|