概要
基于 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)
}
補充
上面函數的核心有以下兩點:
syscall.Exec
對于這個函數, 一般可能用的比較少, 這里稍微介紹下. 它有 3 個參數:
當 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
}
到此這篇關于golang API開發過程的中的自動重啟方式(基于gin框架)的文章就介紹到這了,更多相關golang API 自動重啟內容請搜索腳本之家以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持腳本之家！
