配置文件熱更新是服務(wù)器程序的一個(gè)基本功能浓瞪,通過(guò)熱更新可以不停機(jī)調(diào)整程序的配置英岭,特別是在生產(chǎn)環(huán)境可以提供極大的便利荧库,比如發(fā)現(xiàn)log打得太多了可以動(dòng)態(tài)調(diào)高日志等級(jí),業(yè)務(wù)邏輯參數(shù)變化,甚至某個(gè)功能模塊的開(kāi)關(guān)等都可以動(dòng)態(tài)調(diào)整科乎。
每種語(yǔ)言都有自己的熱更新實(shí)現(xiàn)方式空闲,在golang里面我看到了有人采用了一種錯(cuò)誤的實(shí)現(xiàn)方式,如下:
type Config struct {
Test1 string `json:"Test1"`
Test2 int `json:"Test2"`
}
var (
config *Config
)
func loadConfig() {
f, err := ioutil.ReadFile("config.json")
if err != nil {
fmt.Println("load config error: ", err)
}
err = json.Unmarshal(f, &config)
if err != nil {
fmt.Println("Para config failed: ", err)
}
}
func init() {
loadConfig()
fmt.Println("Load config: ", *config)
s := make(chan os.Signal, 1)
signal.Notify(s, syscall.SIGUSR2)
go func() {
for {
<-s
loadConfig()
fmt.Println("ReLoad config: ", *config)
}
}()
}
這種方式片迅,的問(wèn)題就在于config可能被多個(gè)routine同時(shí)訪問(wèn)空免,那么loadcofig的時(shí)候直接在config上做解析是有問(wèn)題的蹋砚。在上面例子中析恢,還是用的最簡(jiǎn)單的json配置,如果是其他更復(fù)雜的配置形式,如xml亏拉,需要自己寫解析函數(shù)(而不是一個(gè)json.Unmarshal)的時(shí)候更容易放大問(wèn)題专筷。
可以采用更新的時(shí)候?qū)onfig加鎖來(lái)解決這個(gè)問(wèn)題:
package main
import (
"encoding/json"
"fmt"
"io/ioutil"
"log"
"os"
"os/signal"
"sync"
"syscall"
)
//用json配置測(cè)試
type Config struct {
Test1 string `json:"Test1:`
Test2 int `json:"Test1:`
}
var (
config *Config
configLock = new(sync.RWMutex)
)
func loadConfig() bool {
f, err := ioutil.ReadFile("config.json")
if err != nil {
fmt.Println("load config error: ", err)
return false
}
//不同的配置規(guī)則溪烤,解析復(fù)雜度不同
temp := new(Config)
err = json.Unmarshal(f, &config)
if err != nil {
fmt.Println("Para config failed: ", err)
return false
}
configLock.Lock()
config = temp
configLock.Unlock()
return true
}
func GetConfig() *Config {
configLock.RLock()
defer configLock.RUnlock()
return config
}
func init() {
if !loadConfig() {
os.Exit(1)
}
//熱更新配置可能有多種觸發(fā)方式檬嘀,這里使用系統(tǒng)信號(hào)量sigusr1實(shí)現(xiàn)
s := make(chan os.Signal, 1)
signal.Notify(s, syscall.SIGUSR1)
go func() {
for {
<-s
log.Println("Reloaded config:", loadConfig())
}
}()
}
func main() {
select {}
}
因?yàn)闊峒虞d的時(shí)候鸳兽,可能有很多未知的routine在使用config揍异,通過(guò)切換配置的時(shí)候加鎖可以保證多個(gè)routine對(duì)config的正確操作,不過(guò)需要注意的是戚嗅,獲取配置文件的時(shí)候也需要加鎖替久。
在sync/atomic包里稼锅,還真看到了一個(gè)利用原子操作動(dòng)態(tài)更新配置文件的方案: https://golang.org/pkg/sync/atomic/
測(cè)試使用后感覺(jué)不錯(cuò)锥债,實(shí)現(xiàn)原理如下:
var config Value // holds current server configuration
// Create initial config value and store into config.
config.Store(loadConfig())
go func() {
// Reload config every 10 seconds
// and update config value with the new version.
for {
time.Sleep(10 * time.Second)
config.Store(loadConfig())
}
}()
// Create worker goroutines that handle incoming requests
// using the latest config value.
for i := 0; i < 10; i++ {
go func() {
for r := range requests() {
c := config.Load()
// Handle request r using config c.
_, _ = r, c
}
}()
}