go - 你如何优雅地退出 go uber fx 应用程序

Question

如何像关闭整个程序一样停止 uber fx。好像除了ctrl+c没有别的办法

func main() {
    fx.New(
        fx.Invoke(register)
    ).Run
}


func register() {
    time.Sleep(5*time.Seconds)
    // shutdown somehow
}

Answer 1

docs不是特别清楚，但是有一个 Shutdowner 接口(interface)可用于任何具有 Shutdown 方法的 Fx 模块，该方法请求正常关闭应用程序。

这是 the example package 的修改部分它将在收到请求后简单地关闭:

func NewHandler(logger *log.Logger, shutdowner fx.Shutdowner) (http.Handler, error) {
    logger.Print("Executing NewHandler.")
    return http.HandlerFunc(func(http.ResponseWriter, *http.Request) {
        logger.Print("Got a request. Requesting shutdown now that I've gotten one request.")
        shutdowner.Shutdown()
    }), nil
}

编辑:以下是您可以如何修改您的解决方案:

func register(shutdowner fx.Shutdowner) {
    time.Sleep(5*time.Seconds)
    shutdowner.Shutdown()
}