Golang服务器关机和热重载配置-duidaima 堆代码

Golang服务器关机和热重载配置

发布于 2个月前
 695 热度

 0 评论

蹦跶少年
0 粉丝 32 篇博客

我们因不同的目的去关闭服务。有时，关闭服务器的主要目的是用来更新配置。在Golang中，有很多关于优雅关机的帖子。然而，我发现他们并没有提到热重载配置。一般来说，shutdown graceful和hot reload是由信号控制的。这就是我把它们放在一个帖子里的原因。

起初，一个服务器由于各种原因需要关闭，常见的是操作系统的中断，我们希望能优雅地关闭服务器。最后也是最基本的是如果一个应用程序需要更新，只需要更新配置。这种情况总是发生。那个时候，如果你不需要重启服务器，那将是非常惊人的。你只需要更新你的配置，然后发送一个信号。之后，服务器将热重新加载，你的新配置就会工作。上述场景可以在Golang中使用信号包来完成。让我们动手吧。

代码实现
首先，我定义了一个Config结构并声明了一个conf变量。其代码如下：

type Config struct {
 Message string
}
var conf = &Config{Message: "Before hot reload"}

上面的代码只是一个简单的配置样本，你可以根据自己的需要定义一个复杂的结构。其次，定义一个路由器函数，用来绑定和监听8080端口。在热重载配置完成后，它也被用来显示结果。代码如下:

func router() {
 log.Println("starting up....")
 http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
  _, _ = w.Write([]byte(conf.Message))
 })

 go func() {
  log.Fatal(http.ListenAndServe(":8080", nil))
 }()
}

下一步是服务器关机和热重载配置。当一个服务器关闭时，它应该停止接收新的请求，同时完成正在进行的请求，返回其响应，然后关闭。像其他文章一样，我将在这里使用信号包。

sigCh := make(chan os.Signal, 1)

之后，我也会使用signal.Notify()。但是我一起发送更多的信号。

signal.Notify(sigCh, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM)

上述代码中，当程序被中断时，signal.Notify将向sigCh通道发送一个信号。

syscall.SIGHUP、syscall.SIGINT和syscall.SIGTERM是什么意思？

syscall.SIGINT是用来在Ctrl+C时优雅地关闭的，它也相当于os.Interrupt。
syscall.SIGTERM是常用的终止信号，也是docker容器的默认信号，Kubernetes也使用它。
syscall.SIGHUP用于热重载配置。

这一部分的全部代码如下:

func main() {
   router()
   sigCh := make(chan os.Signal, 1)
   signal.Notify(sigCh, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM)

   for {
      multiSignalHandler(<-sigCh)
   }
}

由于这个帖子也包括优雅地关闭，multiSignalHandler(<-sigCh)被用来接收chan值，然后，它将决定运行代码的哪一部分。multiSignalHandler的代码如下:

func multiSignalHandler(signal os.Signal) {
   switch signal {
   case syscall.SIGHUP:
      log.Println("Signal:", signal.String())
      log.Println("After hot reload")
      conf.Message = "Hot reload has been finished."
   case syscall.SIGINT:
      log.Println("Signal:", signal.String())
      log.Println("Interrupt by Ctrl+C")
      os.Exit(0)
   case syscall.SIGTERM:
      log.Println("Signal:", signal.String())
      log.Println("Process is killed.")
      os.Exit(0)
   default:
      log.Println("Unhandled/unknown signal")
   }
}

以上所有的代码将使优雅地关机和热重载配置工作。我把整个代码粘贴在下面，你可以很容易地把它粘贴到你的环境中。

package main
// 堆代码 www.duidaima.com
import (
    "log"
    "net/http"
    "os"
    "os/signal"
    "syscall"
)

type Config struct {
    Message string
}

var conf = & Config {
    Message: "Before hot reload"
}

func router() {
    log.Println("starting up....")
    http.HandleFunc("/", func(w http.ResponseWriter, r * http.Request) {
        _, _ = w.Write([] byte(conf.Message))
    })

    go func() {
        log.Fatal(http.ListenAndServe(":8080", nil))
    }()
}

func main() {
    router()
    sigCh: = make(chan os.Signal, 1)
    signal.Notify(sigCh, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM)

    for {
        multiSignalHandler( < -sigCh)
    }
}

func multiSignalHandler(signal os.Signal) {
    switch signal {
        case syscall.SIGHUP:
            log.Println("Signal:", signal.String())
            log.Println("After hot reload")
            conf.Message = "Hot reload has been finished."
        case syscall.SIGINT:
            log.Println("Signal:", signal.String())
            log.Println("Interrupt by Ctrl+C")
            os.Exit(0)
        case syscall.SIGTERM:
            log.Println("Signal:", signal.String())
            log.Println("Process is killed.")
            os.Exit(0)
        default:
            log.Println("Unhandled/unknown signal")
    }
}

测试优雅关闭和热重载配置
这与之前使用Golang的自动测试不同，此时我手动进行测试。这个测试需要打开两个终端窗口。

测试优雅关闭
首先，在终端一中运行服务器。如果你按上述步骤操作，你会看到与运行服务器后的截图类似的结果：