如何优雅的关闭GoWeb服务器
以下内容均为Let’s Go Further内容节选以及作者本人理解。
这里创建了一个后台进程用于捕获关闭信号，在后台进程中，主要内容为：
- 创建一个缓冲通道 quit
- 使用signal.Notify函数监听并捕获关机信号SIGINT,SIGTERM，在捕获关机信号后，缓冲通道取值操作才会执行，取消阻塞。
- 随后使用shutdown函数关闭服务器。工作原理是首先关闭所有打开的侦听器，然后关闭所有空闲连接，然后无限期地等待连接返回空闲状态，然后关闭。
- 如果关闭连接时发生错误，则将错误存储进入通道，主程序取出错误，取消阻塞。返回错误。
- 如果正常关闭连接，app.wg.Wait()等待全部后台进程全部运行结束后放行，随后向错误通道存储空值，放行主程序。此时连接和后台进程全部关闭，打印日志，结束。
package mainimport ("context""errors""fmt""net/http""os""os/signal""syscall""time"
)func (app *application) serve() error {srv := http.Server{Addr: fmt.Sprintf(":%d", app.config.port),Handler: app.routes(),IdleTimeout: time.Minute,ReadTimeout: 10 * time.Second,WriteTimeout: 30 * time.Second,//服务器可以编写自己的日志消息,实现Write方法变成io.Write接口可以将自定义日志传递进去//ErrorLog: log.New(logger, "", 0),}shutdownErr := make(chan error)//开启一个后台进程 捕获关闭信号，关闭所有连接和后台进程，实现优雅的关机go func() {//创建一个通道保存os.Signal值/*在这里使用缓冲通道，如果使用非缓冲通道，quit 通道 在信号发送的时刻没有准备去接收，会错过信号*/quit := make(chan os.Signal, 1)//使用signal.Notify监听SIGINT and SIGTERM信号并将他们传入quit通道。其他信号不会被捕获并保留他们默认的行为。signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)//从通道中读取值，这个操作将被阻塞直到通道接收一个值s := <-quitapp.logger.PrintInfo("shutting down server", map[string]string{"signal": s.String(),})//创建一个5秒的上下文ctx, cancelFunc := context.WithTimeout(context.Background(), 5*time.Second)defer cancelFunc()//服务调用shutdown，如果成功关闭将返回nil，或者返回错误（当关闭一个监听器时遇到问题，或者在到达截止日期时没有完成）/*工作原理是首先关闭所有打开的侦听器，然后关闭所有空闲连接，然后无限期地等待连接返回空闲状态，然后关闭。*//*1. 关闭连接时没有返回错误，说明连接全部关闭，但是还有后台进程，告诉正在执行关闭后台进程，需要等待wg归零。2. 关闭连接出现错误。将err存入通道，放行主程序*///shutdown返回的是一个error类型的变量err := srv.Shutdown(ctx)//通道中存入错误，主程序取消阻塞，继续执行if err != nil {shutdownErr <- err}//正在完成后台进程app.logger.PrintInfo("completing background tasks", map[string]string{"addr": srv.Addr,})app.wg.Wait()//后台进程全部关闭后，放行主程序shutdownErr <- nil}()app.logger.PrintInfo("starting server", map[string]string{"addr": srv.Addr,"env": app.config.env,})//调用shutdown会造成ListenAndServer立刻返回一个http.ErrServerClosed错误。如果是这个错误，表明我们的程序优雅的关闭了err := srv.ListenAndServe()if !errors.Is(err, http.ErrServerClosed) {return err}//阻塞 关闭后才能继续执行err = <-shutdownErrif err != nil {return err}app.logger.PrintInfo("stopped server", map[string]string{"addr": srv.Addr,})return nil
}关于使用后台进程运行其他任务时，使用函数background来启动后台进程
func (app *application) background(fn func()) {//开启一个后台进程app.wg.Add(1)go func() {defer app.wg.Done()//使用延迟函数捕获可能出现的panicdefer func() {if err := recover(); err != nil {app.logger.PrintError(fmt.Errorf("%s", err), nil)}}()fn()}()
}