golang配置文件熱更新

配置文件熱更新是服務器程序的一個基本功能,通過熱更新可以不停機調整程序的配置,特別是在生產環境可以提供極大的便利,比如發現log打得太多了可以動態調高日志等級,業務邏輯參數變化,甚至某個功能模塊的開關等都可以動態調整。

每種語言都有自己的熱更新實現方式,在golang里面我看到了有人采用了一種錯誤的實現方式,如下:

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)
        }
    }()

}

這種方式,的問題就在于config可能被多個routine同時訪問,那么loadcofig的時候直接在config上做解析是有問題的。在上面例子中,還是用的最簡單的json配置,如果是其他更復雜的配置形式,如xml,需要自己寫解析函數(而不是一個json.Unmarshal)的時候更容易放大問題。

可以采用更新的時候對config加鎖來解決這個問題:

package main

import (
    "encoding/json"
    "fmt"
    "io/ioutil"
    "log"
    "os"
    "os/signal"
    "sync"
    "syscall"
)

//用json配置測試
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
    }

    //不同的配置規則,解析復雜度不同
    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)
    }

    //熱更新配置可能有多種觸發方式,這里使用系統信號量sigusr1實現
    s := make(chan os.Signal, 1)
    signal.Notify(s, syscall.SIGUSR1)
    go func() {
        for {
            <-s
            log.Println("Reloaded config:", loadConfig())
        }
    }()
}

func main() {
    select {}
}

因為熱加載的時候,可能有很多未知的routine在使用config,通過切換配置的時候加鎖可以保證多個routine對config的正確操作,不過需要注意的是,獲取配置文件的時候也需要加鎖。

在sync/atomic包里,還真看到了一個利用原子操作動態更新配置文件的方案: https://golang.org/pkg/sync/atomic/
測試使用后感覺不錯,實現原理如下:

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
                }
        }()
}
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 230,321評論 6 543
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 99,559評論 3 429
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 178,442評論 0 383
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,835評論 1 317
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 72,581評論 6 412
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,922評論 1 328
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,931評論 3 447
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 43,096評論 0 290
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 49,639評論 1 336
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 41,374評論 3 358
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,591評論 1 374
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 39,104評論 5 364
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,789評論 3 349
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 35,196評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,524評論 1 295
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 52,322評論 3 400
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,554評論 2 379

推薦閱讀更多精彩內容