[go每日一库] golang viper实现动态配置更新 Marathon

viper库下载：

go get -u github.com/spf13/viper

代码结构：

.
+--- config
|   +--- config.go
|   +--- config.yaml
+--- go.mod
+--- go.sum
+--- main.go

config.yaml

app:
  app1:
    timeout: 120
    rpc: true
    compatible: true

mysql:
  host: localhost
  port: 3306
  user: root
  password: root
  database: test
  
redis:
  host: localhost
  port: 6379
  dbname: 0

config.go

package config

import (
	"fmt"
	"github.com/fsnotify/fsnotify"
	"github.com/spf13/viper"
)

type Config struct {
	App   App
	MySQL MySQL
	Redis Redis
}

type App struct {
	App1 App1
}

type App1 struct {
	Timeout    int
	Rpc        bool
	Compatible bool
}

type MySQL struct {
	Host     string
	User     string
	Password string
	Port     int
	Database string
}

type Redis struct {
	Host   string
	Port   int
	Dbname int
}

var AppConfig Config

func init()  {
	viperCfg := viper.New()

	viperCfg.SetConfigName("config")
	viperCfg.SetConfigType("yaml")
	viperCfg.AddConfigPath("D:\\demo1\\src\\demo\\demo06\\go-viper-http-watch\\config\\")

	viperCfg.ReadInConfig()

	err := viperCfg.Unmarshal(&AppConfig)
	if err != nil {
		fmt.Println(err)
	}

	viperCfg.WatchConfig()
	viperCfg.OnConfigChange(func(e fsnotify.Event) {
		fmt.Println("Config file changed:", e.Name)
		if err = viperCfg.Unmarshal(&AppConfig); err != nil {
			fmt.Println(err)
		}
	})
}

func GetConfig() Config {
	return AppConfig
}

main

package main

import (
	"github.com/gin-gonic/gin"
	. "go-viper-http-watch/config"
	"net/http"
)

func main()  {
	router := gin.Default()

	router.GET("/info", func(ctx *gin.Context) {
		ctx.JSON(http.StatusOK, gin.H{
			"code": 0,
			"msg": AppConfig.MySQL.Host,
		})
	})

	router.Run(":8080")
}

go.mod

module go-viper-http-watch

go 1.16

require (
	github.com/fsnotify/fsnotify v1.5.4
	github.com/gin-gonic/gin v1.8.1
	github.com/spf13/viper v1.12.0
)

replace go-viper-http-watch => ../go-viper-http-watch