詳解go基于viper實(shí)現(xiàn)配置文件熱更新及其源碼分析

viperInstance := viper.New()	// viper實(shí)例
viperInstance.WatchConfig()
viperInstance.OnConfigChange(func(e fsnotify.Event) {
	log.Print("Config file updated.")
	viperLoadConf(viperInstance)  // 加載配置的方法
})

func (v *Viper) WatchConfig() {
	go func() {
      // 建立新的監(jiān)視處理程序，開啟一個(gè)協(xié)程開始等待事件
      // 從I/O完成端口讀取，將事件注入到Event對象中：Watcher.Events
		watcher, err := fsnotify.NewWatcher()  
		if err != nil {
			log.Fatal(err)
		}
		defer watcher.Close()
 
		// we have to watch the entire directory to pick up renames/atomic saves in a cross-platform way
		filename, err := v.getConfigFile()  
		if err != nil {
			log.Println("error:", err)
			return
		}
 
		configFile := filepath.Clean(filename)    //配置文件E:\etc\bizsvc\config.yml
		configDir, _ := filepath.Split(configFile)  // E:\etc\bizsvc\

 
		done := make(chan bool)
		go func() {
			for {
				select {
        // 讀取的event對象有兩個(gè)屬性，Name為E:\etc\bizsvc\config.yml，Op為write（對文件的操作）
				case event := -watcher.Events:
		// 清除內(nèi)部的..和他前面的元素，清除當(dāng)前路徑.，用來判斷操作的文件是否是configFile
					if filepath.Clean(event.Name) == configFile {
        // 如果對該文件進(jìn)行了創(chuàng)建操作或?qū)懖僮?
						if event.Opfsnotify.Write == fsnotify.Write || event.Opfsnotify.Create == fsnotify.Create {
							err := v.ReadInConfig()
							if err != nil {
								log.Println("error:", err)
							}
							v.onConfigChange(event)
						}
					}
				case err := -watcher.Errors:
         // 有錯(cuò)誤將打印
					log.Println("error:", err)
				}
			}
		}()
 
		watcher.Add(configDir)
		-done
	}()
}

type Viper struct {
	// Delimiter that separates a list of keys
	// used to access a nested value in one go
	keyDelim string
 
	// A set of paths to look for the config file in
	configPaths []string
 
	// The filesystem to read config from.
	fs afero.Fs
 
	// A set of remote providers to search for the configuration
	remoteProviders []*defaultRemoteProvider
 
	// Name of file to look for inside the path
	configName string
	configFile string
	configType string
	envPrefix string
 
	automaticEnvApplied bool
	envKeyReplacer   *strings.Replacer
 
	config     map[string]interface{}
	override    map[string]interface{}
	defaults    map[string]interface{}
	kvstore    map[string]interface{}
	pflags     map[string]FlagValue
	env      map[string]string
	aliases    map[string]string
	typeByDefValue bool
 
	// Store read properties on the object so that we can write back in order with comments.
	// This will only be used if the configuration read is a properties file.
	properties *properties.Properties
 
	onConfigChange func(fsnotify.Event)
}

func (v *Viper) OnConfigChange(run func(in fsnotify.Event)) {
	v.onConfigChange = run
}

	viperInstance.OnConfigChange(func(e fsnotify.Event) {
		log.Print("Config file updated.")
		viperLoadConf(viperInstance)  // viperLoadConf函數(shù)就是將最新配置注入到自定義結(jié)構(gòu)體對象的邏輯
	})