backdoor restart delay (#237)

shawnfeng · Mar 24, 2021 · b472a06 · b472a06
1 parent 50c155f
commit b472a06
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/util/service/admin.go b/util/service/admin.go
@@ -71,9 +71,13 @@ func FactoryRestart() xhttp.HandleRequest {
 
 func (m *Restart) Handle(r *xhttp.HttpRequest) xhttp.HttpResponse {
 	xlog.Infof(context.Background(), "RECEIVE RESTART COMMAND")
-	server.sbase.Stop()
-	os.Exit(1)
-	// 这里的代码执行不到了，因为之前已经退出了
+	// 延迟退出, 保证接口正常返回
+	go func() {
+		time.Sleep(1 * time.Second)
+		server.sbase.Stop()
+		os.Exit(1)
+	}()
+
 	return xhttp.NewHttpRespString(200, "{}")
 }