1.增加RegisterOnRestart注册

2.增加TERM/HUP信号量默认处理

server/server.go

@@ -79,6 +79,7 @@ type Server struct {
 
 	inShutdown int32
 	onShutdown []func(s *Server)
+	onRestart  []func(s *Server)
 
 	// TLSConfig for creating tls tcp connection.
 	tlsConfig *tls.Config
@@ -174,13 +175,22 @@ func (s *Server) startShutdownListener() {
 	go func(s *Server) {
 		log.Info("server pid:", os.Getpid())
 		c := make(chan os.Signal, 1)
-		signal.Notify(c, syscall.SIGTERM)
+		signal.Notify(c, syscall.SIGTERM, syscall.SIGHUP)
 		si := <-c
-		if si.String() == "terminated" {
-			if nil != s.onShutdown && len(s.onShutdown) > 0 {
-				for _, sd := range s.onShutdown {
-					sd(s)
-				}
+		var customFuncs []func(s *Server)
+		switch si.String() {
+		case "terminated":
+			customFuncs = append(s.onShutdown, func(s *Server) {
+				s.Shutdown(context.Background())
+			})
+		case "hangup":
+			customFuncs = append(s.onRestart, func(s *Server) {
+				s.Restart(context.Background())
+			})
+		}
+		if nil != customFuncs && len(customFuncs) > 0 {
+			for _, fn := range customFuncs {
+				fn(s)
 			}
 		}
 	}(s)
@@ -700,6 +710,13 @@ func (s *Server) RegisterOnShutdown(f func(s *Server)) {
 	s.mu.Unlock()
 }
 
+// RegisterOnRestart registers a function to call on Restart.
+func (s *Server) RegisterOnRestart(f func(s *Server)) {
+	s.mu.Lock()
+	s.onRestart = append(s.onRestart, f)
+	s.mu.Unlock()
+}
+
 var shutdownPollInterval = 1000 * time.Millisecond
 
 // Shutdown gracefully shuts down the server without interrupting any