在Uber FX中，实现后台进程正常关闭的正确方法是什么？这是许多人在使用Uber FX时经常遇到的问题。作为一款强大的后台任务处理框架，Uber FX提供了一种简单而有效的方法来管理和处理后台任务。php小编子墨将在本文中为您介绍如何正确关闭后台进程，确保程序的稳定性和正常运行。

问题内容

假设我的 Uber FX 应用程序中有一个服务，它应该执行一些后台活动，例如轮询外部 API。 我可以通过触发 goroutine 来运行后台任务，但是停止它们的正确方法是什么？

作为一种可能的实现，让我们考虑以下示例：

package main

import (
    "context"
    "log"
    "sync"
    "time"

    "go.uber.org/fx"
)

type AwesomeService struct {
    // context to use for background processes
    bg context.Context
    // to trigger background processes stopping
    cancel context.CancelFunc
    // to wait for background processes to gracefully finish
    wg *sync.WaitGroup
}

func New(lc fx.Lifecycle) *AwesomeService {
    bg, cancel := context.WithCancel(context.Background())
    service := &AwesomeService{
        bg:     bg,
        cancel: cancel,
        wg:     new(sync.WaitGroup),
    }

    lc.Append(fx.Hook{
        OnStart: service.start,
        OnStop:  service.stop,
    })
    return service
}

func (s *AwesomeService) start(_ context.Context) error {
    s.runBackgroundProcess()
    log.Println("Start done")
    return nil
}

func (s *AwesomeService) stop(_ context.Context) error {
    s.cancel()
    s.wg.Wait()
    log.Println("Stop done")
    return nil
}

// runBackgroundProcess does some work till context is done.
func (s *AwesomeService) runBackgroundProcess() {
    s.wg.Add(1)
    go func() {
        defer s.wg.Done()
        for {
            select {
            case <-s.bg.Done():
                return
            case <-time.After(1 * time.Second):
                log.Println("Working...")
            }
        }
    }()
}

func main() {
    fx.New(
        fx.Provide(New),
        fx.Invoke(func(*AwesomeService) {}),
    ).Run()
}

一些注意事项：

• 该服务通过使用 fx.Lifecycle 挂钩连接到应用程序生命周期。
• 我无法依赖和使用 OnStart/OnStop 方法中的上下文，因为它们是不同的上下文并且对应于启动/停止活动，而不是应用生命周期上下文。

疑虑和问题：

• 给出的示例在跟踪背景任务方面相当繁重。此外，将上下文存储在结构中是一种反模式。有什么办法可以简化吗？
• 如果没有资源可以释放，我是否应该等待完成 goroutine？

解决方法

在我看来，使用上下文就很好，但是您也可以通过通道向您想要的任何 Go 例程传达关闭信号。请参阅下面的示例代码。

是的，您还应该等待等待组计数返回零，然后再完全关闭应用程序。所以你首先要关闭通道，然后在等待组上等待。

package main

import (
    "context"
    "log"
    "sync"
    "time"

    "go.uber.org/fx"
)

type AwesomeService struct {
    // channel to shutdown background processes
    shutdown chan struct{}
    // to wait for background processes to gracefully finish
    wg *sync.WaitGroup
}

func New(lc fx.Lifecycle) *AwesomeService {
    service := &AwesomeService{
        shutdown: make(chan struct{}),
        wg:     new(sync.WaitGroup),
    }

    lc.Append(fx.Hook{
        OnStart: service.start,
        OnStop:  service.stop,
    })
    return service
}

func (s *AwesomeService) start(_ context.Context) error {
    s.runBackgroundProcess()
    log.Println("Start done")
    return nil
}

func (s *AwesomeService) stop(_ context.Context) error {
    close(s.shutdown)
    s.wg.Wait()
    log.Println("Stop done")
    return nil
}

// runBackgroundProcess does some work till context is done.
func (s *AwesomeService) runBackgroundProcess() {
    s.wg.Add(1)
    go func() {
        defer s.wg.Done()
        for {
            select {
            case <-s.shutdown:
                return
            case <-time.After(1 * time.Second):
                log.Println("Working...")
            }
        }
    }()
}

func main() {
    fx.New(
        fx.Provide(New),
        fx.Invoke(func(*AwesomeService) {}),
    ).Run()
}