TyleoDvDelaware3217/GoDoxy
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2025-07-15 01:54:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: utils.WaitExit

Browse source
This commit is contained in:
yusing 2025-03-28 05:59:04 +08:00
parent 40aa937f54
commit fd223c7542
2 changed files with 18 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cmd/main.go
View file

@ -138,17 +138,7 @@ func main() {
config.WatchChanges() config.WatchChanges()
sig := make(chan os.Signal, 1) task.WaitExit(cfg.Value().TimeoutShutdown)
signal.Notify(sig, syscall.SIGINT)
signal.Notify(sig, syscall.SIGTERM)
signal.Notify(sig, syscall.SIGHUP)
// wait for signal
<-sig
// gracefully shutdown
logging.Info().Msg("shutting down")
_ = task.GracefulShutdown(time.Second * time.Duration(cfg.Value().TimeoutShutdown))
} }
func prepareDirectory(dir string) { func prepareDirectory(dir string) {

17
internal/task/utils.go
View file

@ -4,6 +4,9 @@ import (
"context" "context"
"encoding/json" "encoding/json"
"errors" "errors"
"os"
"os/signal"
"syscall"
"time" "time"
"github.com/yusing/go-proxy/internal/logging" "github.com/yusing/go-proxy/internal/logging"
@ -73,3 +76,17 @@ func GracefulShutdown(timeout time.Duration) (err error) {
} }
} }
} }
func WaitExit(shutdownTimeout int) {
sig := make(chan os.Signal, 1)
signal.Notify(sig, syscall.SIGINT)
signal.Notify(sig, syscall.SIGTERM)
signal.Notify(sig, syscall.SIGHUP)
// wait for signal
<-sig
// gracefully shutdown
logging.Info().Msg("shutting down")
_ = GracefulShutdown(time.Second * time.Duration(shutdownTimeout))
}