golang API开发过程的中的自动重启方式(基于gin框架)
作者:返回主页千里之行,始于足下 时间:2024-02-03 02:56:48
概要
基于 golang Gin 框架开发 web 服务时, 需要时不时的 go build , 然后重启服务查看运行结果.
go build 的过程集成在编辑器中(emacs), 可以通过快捷键迅速完成, 但是每次重启服务都切换到命令行中操作.
因此, 希望能够编译通过之后自动重启服务.
这里并不是部署阶段的服务重启, 所以不用过多考虑是否正常退出其中的协程.
实现方式
在开源的 illuminant 项目中, 已经将相应的代码集成到 gin 的 debug mode 中.
代码文件: https://gitee.com/wangyubin/illuminant/blob/dev/server_cmd.go
func setupWatcher() (chan struct{}, error) {
file, err := osext.Executable()
if err != nil {
return nil, err
}
log.Printf("watching %q\n", file)
w, err := fsnotify.NewWatcher()
if err != nil {
return nil, err
}
done := make(chan struct{})
go func() {
select {
case e := <-w.Events:
log.Printf("watcher received: %+v", e)
err := syscall.Exec(file, os.Args, os.Environ())
if err != nil {
log.Fatal(err)
}
case err := <-w.Errors:
log.Printf("watcher error: %+v", err)
case <-done:
log.Print("watcher shutting down")
return
}
}()
err = w.Add(file)
if err != nil {
return nil, err
}
return done, nil
}
在 gin debug mode 下, 使用此方法自动重启服务
if c.Bool("prod") {
gin.SetMode(gin.ReleaseMode)
// start route
return routes.Routes(cnf.Server.Port)
} else {
gin.SetMode(gin.DebugMode)
watcher, err := setupWatcher()
if err != nil {
// do something sensible
log.Fatal(err)
}
defer close(watcher)
return routes.Routes(cnf.Server.Port)
}
补充
上面函数的核心有以下两点:
w, err := fsnotify.NewWatcher(): 创建监控文件变化的 watcher, err = w.Add(file) 并将当前二进制文件加入到监控文件列表中
err := syscall.Exec(file, os.Args, os.Environ()) 接受到文件变化的事件时, 重新调用一次自己, 使用上次一样的参数和环境变量
syscall.Exec
对于这个函数, 一般可能用的比较少, 这里稍微介绍下. 它有 3 个参数:
args[0]: 可执行文件的路径(相对路径, 绝对路径或者 PATH 中的路径都可以)
args[1]: 命令的参数
args[2]: 命令的执行的环境变量, os.Environ() 表示继承 caller 的环境变量
当 syscall.Exec 执行时, 在它之前的所有未执行完的程序都会被中止(包括在 go routine 中执行的程序),
然后执行 syscall.Exec 调用的命令, 该命令还保持在之前程序的 PID 下执行.
syscall.Exec 是最后一条执行的代码, 重启时在它之后可以有代码, 但是都不会被执行到, 包括 defer 中的代码.
下面是个小例子(通过这个例子可以验证上面的结论):
package main
import (
"fmt"
"log"
"os"
"syscall"
"time"
"github.com/fsnotify/fsnotify"
"github.com/kardianos/osext"
)
func syscallExec() {
watcher, err := setupWatcher()
if err != nil {
log.Fatal(err)
}
defer finally(watcher)
fmt.Printf("current pid: %d\n", os.Getpid())
var count = 0
go func(count int) {
for {
fmt.Printf(">>> count in GO ROUTINE: %d\n", count)
count++
time.Sleep(1 * time.Second)
}
}(count)
for {
fmt.Printf(">>> count in MAIN: %d\n", count)
count++
time.Sleep(1 * time.Second)
}
}
func finally(watcher chan struct{}) {
// 重启时没有执行此函数
fmt.Println("exit original exec")
close(watcher)
}
func setupWatcher() (chan struct{}, error) {
file, err := osext.Executable()
if err != nil {
return nil, err
}
log.Printf("watching %q\n", file)
w, err := fsnotify.NewWatcher()
if err != nil {
return nil, err
}
done := make(chan struct{})
go func() {
select {
case e := <-w.Events:
log.Printf("watcher received: %v", e)
err := syscall.Exec(file, os.Args, os.Environ())
if err != nil {
log.Fatal(err)
}
case err := <-w.Errors:
log.Printf("watcher error: %+v", err)
case <-done:
log.Print("watcher shutting down")
return
}
}()
err = w.Add(file)
if err != nil {
return nil, err
}
return done, nil
}
来源:https://www.cnblogs.com/wang_yb/archive/2020/12/14/14132498.html
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
基于Python绘制子图及子图刻度的变换等的问题
![](https://img.aspxhome.com/file/2023/3/90443_0s.png)
js读取图片的宽和高
Django中使用group_by的方法
Python之两种模式的生产者消费者模型详解
将Python中的数据存储到系统本地的简单方法
在JavaScript中,为什么要尽可能使用局部变量?
![](https://img.aspxhome.com/file/UploadPic/20093/1/variable_write_performance2-17s.png)
OracleEXP和IMP用法和介绍
python中list常用操作实例详解
Python 静态方法和类方法实例分析
![](https://img.aspxhome.com/file/2023/3/131573_0s.png)
python实现基于信息增益的决策树归纳
微信小程序实现给嵌套template模板传递数据的方式总结
Python安装图文教程 Pycharm安装教程
![](https://img.aspxhome.com/file/2023/4/90434_0s.jpg)
python使用paramiko模块通过ssh2协议对交换机进行配置的方法
MySQL数据库必备之条件查询语句
vue使用svg文件补充-svg放大缩小操作(使用d3.js)
![](https://img.aspxhome.com/file/2023/3/130203_0s.jpg)
sqlserver分页的两种写法分别介绍
解决Python中pandas读取*.csv文件出现编码问题
![](https://img.aspxhome.com/file/2023/0/94180_0s.png)