Golang热加载配置实践

前言

生产环境上所输出的日志级别较高不利于定位，会碰到这样一种场景，我们改配置修改日志级别，再重启，以期望定位问题，然而问题重启后不浮现了，所以热加载配置有必要添进去。

实践

直接上代码：

package main

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

type Config struct {
Mode string `json:"Mode"`
}

var (
config     *Config
)

func loadConfig() {
f, err := ioutil.ReadFile("config.json")
if err != nil {
fmt.Println("[TEST_SIGUSR] Load config: ", err)
}
err = json.Unmarshal(f, &config)
if err != nil {
fmt.Println("[TEST_SIGUSR] Para config failed: ", err)
}

}
func init() {
loadConfig()
fmt.Println("[TEST_SIGUSR] Load config;Mode: ", config.Mode)
s := make(chan os.Signal, 1)
signal.Notify(s, syscall.SIGUSR2)
go func() {
for {
<-s
loadConfig()
fmt.Println("[TEST_SIGUSR] ReLoad config;Mode: ", config.Mode)
}
}()

}
func main() {
q := make(chan os.Signal, 1)
signal.Notify(q, syscall.SIGINT)
for {
select {
case <-q:
fmt.Println("[TEST_SIGUSR] ---ibye----")
os.Exit(1)
default:
}
fmt.Println("[TEST_SIGUSR] ---waiting to reload----")
time.Sleep(time.Second * 2)

}

}

文件目录如下：

wenhao-MA:test_sigusr wenhao$ ll
total 4944
-rw-r--r--  1 wenhao  staff       18 11 22 13:39 config.json
-rwxr-xr-x  1 wenhao  staff  2520352 11 22 13:41 sigusr
-rw-r--r--  1 wenhao  staff      973 11 22 13:47 sigusr.go

wenhao-MA:test_sigusr wenhao$ cat config.json
{"Mode":"Debug"}

编译启动后，修改

config.json

中

Mode

值，

kill -USR2 $(pidof sigusr)

结果：

wenhao-MA:test_sigusr wenhao$./sigusr
[TEST_SIGUSR] Load config;Mode:  Debug
[TEST_SIGUSR] ---waiting to reload----
[TEST_SIGUSR] ---waiting to reload----
[TEST_SIGUSR] ReLoad config;Mode:  ERROR