migrate from github

internal/utils/shutdown.go

@@ -0,0 +1,36 @@
+package utils
+
+import (
+	"context"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/sirupsen/logrus"
+)
+
+type ShutdownHook struct {
+	Name string
+	Fn   func() error
+}
+
+func ShutdownContext() (context.Context, context.CancelFunc) {
+	return signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
+}
+
+func WaitForShutdownSignal() {
+	ctx, stop := ShutdownContext()
+	defer stop()
+	<-ctx.Done()
+}
+
+func RunShutdownHooks(hooks ...ShutdownHook) {
+	for _, hook := range hooks {
+		if hook.Fn == nil {
+			continue
+		}
+		if err := hook.Fn(); err != nil {
+			logrus.Errorf("Close %s failed: %v", hook.Name, err)
+		}
+	}
+}