向 curreng Golang 进程发送自定义信号

Question

我正在使用 Go 创建一个 HTTP 服务器。每当我进行数据库维护时，我希望服务器将所有流量重定向到“当前正在进行维护”页面。

目前，这是通过秘密管理页面（例如http://myhome/secret）完成的，但我想知道是否可以使用类似于 TERM 信号的信号来完成此操作，但暂时重定向而不是实际终止进程。

例如。

/home/myhome> nohup startServer &
... 
/home/myhome> changeMyServerStatus "maintenance"

我假设会有两个可执行文件..“startServer”和“changeMyServerStatus”

所以，这类似于服务。（比如重新加载）但是，这可能吗？如果是这样，你能给我一些提示吗？

谢谢

Answer 1

正如评论中所指出的，信号可能不是实现这一目标的最佳方式。I\xe2\x80\x99m 假设你仍然需要信号。

\n\n

您可以使用标准用户信号：SIGUSR1启用维护和SIGUSR2禁用维护。

\n\n

用于os/signal获取这些信号的通知并更新程序状态：

\n\n

// Brief example code. Real code might be structured differently\n// (perhaps pack up maint and http.Server in one type MyServer).\n\nvar maint uint32 // atomic: 1 if in maintenance mode\n\nfunc handleMaintSignals() {\n    ch := make(chan os.Signal, 1)\n    go func() { // FIXME: use Server.RegisterOnShutdown to terminate this\n        for sig := range ch {\n            switch sig { // FIXME: add logging\n            case syscall.SIGUSR1:\n                atomic.StoreUint32(&maint, 1)\n            case syscall.SIGUSR2:\n                atomic.StoreUint32(&maint, 0)\n            }\n        }\n    }()\n    signal.Notify(ch, syscall.SIGUSR1, syscall.SIGUSR2)\n}\n

\n\n

让中间件查看该状态并做出相应响应：

\n\n

func withMaint(next http.Handler) http.Handler {\n    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n        if atomic.LoadUint32(&maint) == 1 {\n            http.Error(w, "Down for maintenance", http.StatusServiceUnavailable)\n            return\n        }\n        next.ServeHTTP(w, r)\n    })\n}\n

\n\n

您可以在每个路由的基础上应用此中间件，或直接应用到 server\xe2\x80\x99s根处理程序：

\n\n

func main() {\n    handleMaintSignals()\n    srv := http.Server{\n        Addr:    ":17990",\n        Handler: withMaint(http.DefaultServeMux),\n    }\n    srv.ListenAndServe()\n}\n

\n\n\n\n

您不需要\xe2\x80\x99t 需要第二个可执行文件，例如changeMyServerStatus. 使用您的操作系统\xe2\x80\x99s工具发送信号，例如pkill：

\n\n

// Brief example code. Real code might be structured differently\n// (perhaps pack up maint and http.Server in one type MyServer).\n\nvar maint uint32 // atomic: 1 if in maintenance mode\n\nfunc handleMaintSignals() {\n    ch := make(chan os.Signal, 1)\n    go func() { // FIXME: use Server.RegisterOnShutdown to terminate this\n        for sig := range ch {\n            switch sig { // FIXME: add logging\n            case syscall.SIGUSR1:\n                atomic.StoreUint32(&maint, 1)\n            case syscall.SIGUSR2:\n                atomic.StoreUint32(&maint, 0)\n            }\n        }\n    }()\n    signal.Notify(ch, syscall.SIGUSR1, syscall.SIGUSR2)\n}\n

\n\n

但手动操作既繁琐nohup又pkill容易出错。相反，请使用systemd等服务管理器来管理您的进程。Systemd 允许您通过以下方式发送任意信号systemctl kill：

\n\n

func withMaint(next http.Handler) http.Handler {\n    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {\n        if atomic.LoadUint32(&maint) == 1 {\n            http.Error(w, "Down for maintenance", http.StatusServiceUnavailable)\n            return\n        }\n        next.ServeHTTP(w, r)\n    })\n}\n

\n