feat: support exit timeout (#2342)

Signed-off-by: yisaer <[email protected]>
lf-edge · Oct 20, 2023 · 7f13cb8 · 7f13cb8
1 parent 4aa306c
commit 7f13cb8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/internal/server/server.go b/internal/server/server.go
@@ -214,7 +214,9 @@ func StartUp(Version string) {
 	<-sigint
 	exit <- struct{}{}
 
-	if err = srvRest.Shutdown(context.TODO()); err != nil {
+	ctx, cancel := context.WithTimeout(context.TODO(), 3*time.Second)
+	defer cancel()
+	if err = srvRest.Shutdown(ctx); err != nil {
 		logger.Errorf("rest server shutdown error: %v", err)
 	}
 	logger.Info("rest server successfully shutdown.")