如何在 Go 守护进程中重启自身？

如何解决如何在 Go 守护进程中重启自身？

我使用 go-daemon 库来 fork 进程并在后台运行它。从 http 处理程序中执行更新后，我需要重新启动守护进程。

处理程序代码是

func httpUpdate(w http.ResponseWriter,req *http.Request) {
    if !isPost(req.Method) {
        http.Error(w,http.StatusText(http.StatusMethodNotAllowed),http.StatusMethodNotAllowed)
        return
    }

    if checkAuth(req) != 200 {
        http.Error(w,http.StatusText(http.StatusUnauthorized),http.StatusUnauthorized)
        return
    }

    log.Println("INFO: Update request,checking for update...")
    var err = doUpdate(UPDATE_URL,nil,false)
    if !isError(err) {
        log.Println("INFO: Update successful,exit")
        var system = RealSystem{}
        system.Run(fmt.Sprintf("(sleep 0.3s && %s start &> /test/extra.log)&disown",appFilename()))
        system.Exit(0)

        return
    }

    w.WriteHeader(http.StatusNoContent)
}

如果成功替换了可执行文件，

doUpdate() 将返回 nil。 RealSystem 只是 exec.Command 和 os.Exit() 的包装器。 appFilename() 是可执行文件名。启动应用程序的命令是 /path/to/app start。

我看到新进程启动，但执行 Context::Reborn() 失败并出现 EOF 错误。看起来一些用作实现细节的内部管道以 EOF 失败（可能是...）。

是什么原因？或者可能有更好的方法来做到这一点？

现在，一切都发生在 e2e 测试的“上下文”中的 docker 容器内，如果重要的话。我花了几个小时试图让它发挥作用，但没有成功。

解决方法

我假设您的意思是重新启动当前运行的 Go 二进制文件。您可以对基于 Unix 的系统使用系统调用，对 Windows 使用 exec.Command。

func RestartSelf() error {
    self,err := osext.Executable()
    if err != nil {
        return err
    }
    args := os.Args
    env := os.Environ()
    // Windows does not support exec syscall.
    if runtime.GOOS == "windows" {
        cmd := exec.Command(self,args[1:]...)
        cmd.Stdout = os.Stdout
        cmd.Stderr = os.Stderr
        cmd.Stdin = os.Stdin
        cmd.Env = env
        err := cmd.Run()
        if err == nil {
            os.Exit(0)
        }
        return err
    }
    return syscall.Exec(self,args,env)
}

,

该问题特定于图书馆。从子进程中生成新的 self 实例对系统来说不是问题，而是对那个库而言。

要实现这一点，必须执行类似的操作。 请注意 _GO_DAEMON=0 变量设置为零。这使得库遵循父控制流。

var cmd = exec.Command("bash","-c",fmt.Sprintf("sleep 0.5s; _GO_DAEMON=0 %s start",appFilename()))
var err = cmd.Start()

还有必要对原始库进行小的更改。这是fork。