fix(utils): now it can be successfully restarted when deployed as systemd service

- Add `getSystemdServiceName` function to get systemd service name.
- When detected as systemd sevice, use `systemctl restart` to restart the program then exit.
This commit is contained in:
2026-06-04 12:05:09 +08:00
parent 682aa1abb9
commit 5e9bdf02f9
4 changed files with 85 additions and 2 deletions
+23 -1
View File
@@ -170,6 +170,8 @@ func IsFileExist(filePath string) bool {
return !os.IsNotExist(err)
}
var pendingSystemdRestart string
func RestartHandler(w http.ResponseWriter, r *http.Request) {
userID, err := Auth(w, r, http.MethodGet, "superuser", "admin")
if err != nil {
@@ -185,14 +187,34 @@ func RestartHandler(w http.ResponseWriter, r *http.Request) {
}
SendSuccessResponse(w, "Restarting server...", nil)
postLog.Info(fmt.Sprintf("[RestartHandler] [%d] Restarting server", userID))
if pendingSystemdRestart != "" {
serviceName := pendingSystemdRestart
go func() {
time.Sleep(500 * time.Millisecond)
cmd := exec.Command("systemctl", "restart", serviceName)
if err := cmd.Run(); err != nil {
postLog.Error(fmt.Sprintf("[RestartHandler] [%d] failed to restart systemd service %s: %v", userID, serviceName, err))
}
}()
}
}
func RestartProcess() error {
if getInitSystem() == "systemd" {
serviceName, err := getSystemdServiceName()
if err != nil {
return fmt.Errorf("failed to get systemd service name: %w", err)
}
pendingSystemdRestart = serviceName
return nil
}
executable, err := os.Executable()
if err != nil {
return fmt.Errorf("failed to get executable path: %w", err)
}
cmd := exec.Command(executable)
cmd := exec.Command(executable, os.Args[1:]...)
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
if err := cmd.Start(); err != nil {