Golang热更新

1、什么是热更新

网络上有这么一个例子来形容热更新,我觉得很形象很贴切:

一架行驶在高速上的大卡车,行驶过程中突然遭遇爆胎,热更新则是要求在不停车的情况下将车胎修补好,且补胎过程中卡车需要保持正常行驶。

软件的热更新就是指在保持系统正常运行的情况下对系统进行更新升级。常见的情况有:系统服务升级、修复现有逻辑、服务配置更新等。

2、热更新原理

先来看下 Nginx 热更新是如何做的? Nginx 支持运行中接收信号,方便开发者控制进程。

  • 1)首先备份原有的 Nginx 二进制文件,并用新编译好的 Nginx 二进制文件替换旧的
  • 2)然后向 master 进程发送USR2信号。此时 Nginx 进程会启动一个新版本 Nginx,该新版本 Nginx 进程会发起一个新的 master 进程与 work 进程。即此时会有两个 Nginx 实例在运行,一起处理新来的请求。
  • 3)再向原 master 进程发送WINCH信号,它会逐渐关闭相关 work 进程,此时原 master 进程仍保持监听新请求但不会发送至其下 work 进程,而是交给新的 work 进程
  • 4)最后等到所有原 work 进程全部关闭,向原 master 进程发送QUIT信号,终止原 master 进程,至此,完成 Nginx 热升级。

:在 * nix 系统中,信号(Signal)是一种进程间通信机制,它给应用程序提供一种异步的软件中断,使应用程序有机会接受其他程序或终端发送的命令 (即信号)。

同样地,Golang 热更新也可以采取类似的处理。如上篇所述,都是利用用户自定义信号USR2

:Plugin 包方式的 Golang 热更新本文暂不讨论。

3、热更新实现

Golang 热更新可以细分为服务热『更新』(即热升级,类比 Nginx 的 restart 命令)与配置文件热更新(类比 Nginx 的 reload 命令)。接下来从实现细节处依次讨论。

3.1 服务热更新

大致流程如下:

  • 1)Golang 服务进程运行时监听USR2信号
  • 2)进程收到USR2信号后,fork 子进程(启动新版本服务),并将当前 socket 句柄等进程环境交给它
  • 3)新进程开始监听 socket 请求
  • 4)等待旧服务连接停止

主要代码示例如下: 监听USR2信号

func (a *app) signalHandler(wg *sync.WaitGroup) {
    ch := make(chan os.Signal, 10)
    signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM, syscall.SIGUSR2)
    for {
        sig := <-ch
        switch sig {
        case syscall.SIGINT, syscall.SIGTERM:
            // 确保接收到INT/TERM信号时可以触发Golang标准的进程终止行为
            signal.Stop(ch)
            a.term(wg)
            return
        case syscall.SIGUSR2:
            err := a.preStartProcess()
            if err != nil {
                a.errors <- err
            }
            // 发起新进程
            if _, err := a.net.StartProcess(); err != nil {
                a.errors <- err
            }
        }
    }
}

复制当前进程 socket 连接,发起新进程

execSpec := &syscall.ProcAttr{
Env: os.Environ(),
Files: []uintptr{os.Stdin.Fd(), os.Stdout.Fd(), os.Stderr.Fd()},
}
fork, err := syscall.ForkExec(os.Args[0], os.Args, execSpec)
...

详细源码可见:https://scalingo.com/articles…

以上仅为代码示例,目前已经成熟的开源实现主要有:endless 和 facebook 的 grace,原理基本类似,fork 一个子进程,子进程监听原有父进程 socket 端口,父进程优雅退出。

在实际的生产环境中推荐使用以上开源库,关于热更新开源库的使用非常方便,下面是 facebook 的 grace 库的例子: 引入github.com/facebookgo/grace/gracehttp

func main() {
    app := gin.New()// 项目中时候的是gin框架
    router.Route(app)
    var server *http.Server
    server = &http.Server{
        Addr:    ":8080",
        Handler: app,
    }
    gracehttp.Serve(server)
}

利用 go build 命令编译,生成服务的可执行文件。 然后再用 shell 封装一下服务命令,生成 restat.sh 命令文件

#!/bin/sh

ps aux | grep wingo
count=`ps -ef | grep "wingo" | grep -v "grep" | wc -l`
echo ""

if [ 0 == $count ]; then
    echo "Wingo starting..."
    sudo ./wingo &
    echo "Wingo started"
else
    echo "Wingo Restarting..."
    sudo kill -USR2 $(ps -ef | grep "wingo" | grep -v grep | awk '{print $2}')
    echo "Wingo Restarted"
fi

sleep 1

ps aux | grep wingo

:其中 wingo 为服务的二进制名称。

于是,便可通过执行./restart.sh 命令,达到对服务的热升级目的。

3.2 配置文件热更新

配置文件热更新是指在不停止服务的情况下,重新加载服务所有配置文件。 与 3.1 服务热升级原理一样,利用用户自定义信号:USR1,即可实现服务的配置文件热更新。

  • 1)服务监听USR1信号
  • 2)服务接收到USR1信号后,停止接受新的连接,等待当前连接停止,重新载入配置文件,重启服务器,从而实现相对平滑的不停服的更改。

主要代码实现:

// LoadAllConf 调用加载配置文件函数
// load为具体加载配置文件方法
func LoadAllConf(load func(bool)) {
    load(true)
    listenSIGUSR1(load)
}

// listenSIGUSR1 监听SIGUSR1信号
func listenSIGUSR1(f func(bool)) {
    s := make(chan os.Signal, 1)
    signal.Notify(s, syscall.SIGUSR1)
    go func() {
        for {
            <-s
            f(false)
            log.Println("Reloaded")
        }
    }()
}

详细源码可见:https://www.openmymind.net/Go…

利用 go build 命令编译,生成服务的可执行文件。 然后再用 shell 封装一下配置重载命令,生成 reload.sh 命令文件

#!/bin/sh

ps aux | grep wingo
echo ""

echo "Wingo Reloading..."
sudo kill -USR1 $(ps -ef | grep "wingo" | grep -v grep | awk '{print $2}')
echo "Wingo Reloaded"
echo ""

sleep 1

ps aux | grep wingo

于是,便可通过执行./reload.sh 命令,达到对服务的配置文件热升级目的。

4、总结

本文主要描述了 Golang 服务热升级与配置文件热更新原理与主要代码实现,本质上也不是什么新内容,如果之前读过《Unix 环境高级编程》,就会觉得很亲切。底层原理基本上是利用了信号这个软件中断机制,在运行中改变常驻进程的行为。

References

https://scalingo.com/articles… http://kuangchanglang.com/gol… https://blog.csdn.net/black_O… https://www.openmymind.net/Go… https://blog.csdn.net/qq_1543… https://wrfly.kfd.me/posts/%E…

原链接: https://segmentfault.com/a/1190000017228287

本文链接:参与评论 »

--EOF--

提醒:本文最后更新于 1928 天前,文中所描述的信息可能已发生改变,请谨慎使用。

专题「golang」的其它文章 »

评论列表