在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()
}
以上就是在 Uber FX 中实现后台进程正常关闭的正确方法是什么?的详细内容,更多请关注编程网其它相关文章!