向当前 Golang 进程发送自定义信号

Question

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

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

例如

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

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

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

谢谢

Answer 1

如评论中所述，信号可能不是实现此目的的最佳方式。我假设您仍然需要信号。

您可以使用 standard 用户信号：SIGUSR1 启用维护，SIGUSR2 禁用它。

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

// Brief example code. Real code might be structured differently
// (perhaps pack up maint and http.Server in one type MyServer).

var maint uint32 // atomic: 1 if in maintenance mode

func handleMaintSignals() {
    ch := make(chan os.Signal, 1)
    go func() { // FIXME: use Server.RegisterOnShutdown to terminate this
        for sig := range ch {
            switch sig { // FIXME: add logging
            case syscall.SIGUSR1:
                atomic.StoreUint32(&maint, 1)
            case syscall.SIGUSR2:
                atomic.StoreUint32(&maint, 0)
            }
        }
    }()
    signal.Notify(ch, syscall.SIGUSR1, syscall.SIGUSR2)
}

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

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

你可以在每个路由的基础上应用这个中间件，或者直接应用到服务器的 root handler:

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

你不需要像 changeMyServerStatus 这样的第二个可执行文件。使用操作系统的工具发送信号，例如 pkill:

$ nohup myserver &

$ curl http://localhost:17990/
404 page not found

$ pkill -USR1 myserver

$ curl http://localhost:17990/
Down for maintenance

$ pkill -USR2 myserver

$ curl http://localhost:17990/
404 page not found

但是手动处理 nohup 和 pkill 是乏味且容易出错的。相反，使用服务管理器，例如 systemd to manage your process. Systemd lets you send arbitrary signals with systemctl kill:

systemctl kill -s SIGUSR1 myserver

向当前 Golang 进程发送自定义信号

Send a custom signal to curreng Golang process

signals

http

reload

go

server