你如何优雅地退出一个超级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

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

下面是示例包的修改部分，它将在收到请求时将其简单地关闭：

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()

}

Answer 2

您可以将其包装在 go 例程中，并使用上下文优雅地退出。

 import (

    "context"

    "log"

    " go.uber.org/fx"

)

func main() {

    f := fx.New(fx.Invoke(register))

    go func() {

        f.Run()

    }()

    stopCh := make(chan os.Signal)

    signal.Notify(stopCh, syscall.SIGINT, syscall.SIGTERM)

    <-stopCh

    if err := f.Stop(context.Background()); err != nil {

        log.Printf("error stopping gracefully")

    }

}

func register() {

    time.Sleep(5*time.Seconds)

    // shutdown somehow

}