testing - 如何在 Go 中查看或测试优雅重启？

Question

我通过gin 的 https://github.com/fvbock/endless提供 HTTP 服务。我想看看与基本 HTTP 服务器的区别。

我已经发送syscall.SIGUSR1了信号：

syscall.Kill(getPid(), syscall.SIGUSR1)

该应用程序没有退出，但我无法检测到重新启动。

我要做的是在 toml 配置文件更改时为应用程序初始化新配置。

我的代码如下：

package main

import (
    "os"
    "fmt"
    "syscall"

    "github.com/gin-gonic/gin"
    "github.com/fvbock/endless"
    "github.com/BurntSushi/toml"
)

type Config struct {
    Age  int
    Cats []string
}

var cfg Config

func restart(c *gin.Context) {
    syscall.Kill(os.Getpid(), syscall.SIGUSR1)
}

func init() {
    toml.DecodeFile("config.toml", &cfg)
    fmt.Println("Testing", cfg)
}

func main() {
    router := gin.New()

    router.GET("/restart", restart)

    if err := endless.ListenAndServe("localhost:7777", router); err != nil {
        panic(err)
    }
}

当我点击重启端点时，我希望打印出 toml 配置。

Answer 1

根据您对问题的更改更新答案。无限库可以让您默认处理该信号。您将需要注册一个钩子。我在下面扩展了您的示例代码：

package main

import (
    "os"
    "fmt"
    "syscall"

    "github.com/gin-gonic/gin"
    "github.com/fvbock/endless"
    "github.com/BurntSushi/toml"
)

type Config struct {
    Age  int
    Cats []string
}

var cfg Config

func restart(c *gin.Context) {
    syscall.Kill(os.Getpid(), syscall.SIGUSR1)
}

func readConfig() {
    toml.DecodeFile("config.toml", &cfg)
    fmt.Println("Testing", cfg)
}

func main() {
        readConfig()
    router := gin.New()

    router.GET("/restart", restart)

    srv  := endless.NewServer("localhost:7777", router)

        srv.SignalHooks[endless.PRE_SIGNAL][syscall.SIGUSR1] = append(
                srv.SignalHooks[endless.PRE_SIGNAL][syscall.SIGUSR1],
                readConfig)

        if err := srv.ListenAndServe(); err != nil {
                panic(err)
        }

}

现在，当您调用重启端点时，您应该会在标准输出中看到对配置文件的更改。但是，为了查看文件的更改，您需要使用fsnotify 之类的东西