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