脚本之家,脚本语言编程技术及教程分享平台!
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|

服务器之家 - 脚本之家 - Golang - 详解go基于viper实现配置文件热更新及其源码分析

详解go基于viper实现配置文件热更新及其源码分析

2020-07-11 11:19_雨落山岚 Golang

这篇文章主要介绍了详解go基于viper实现配置文件热更新及其源码分析,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧

go第三方库 github.com/spf13/viper  实现了对配置文件的读取并注入到结构中,好用方便。

其中以

?
1
2
3
4
5
6
viperInstance := viper.New()    // viper实例
viperInstance.WatchConfig()
viperInstance.OnConfigChange(func(e fsnotify.Event) {
    log.Print("Config file updated.")
    viperLoadConf(viperInstance)  // 加载配置的方法
})

可实现配置的热更新,不用重启项目新配置即可生效(实现热加载的方法也不止这一种,比如以文件的上次修改时间来判断等)。

为什么这么写?这样写为什么就能立即生效?基于这两个问题一起来看看viper是怎样实现热更新的。

上面代码的核心一共两处:WatchConfig()方法、OnConfigChange()方法。WatchConfig()方法用来开启事件监听,确定用户操作文件后该文件是否可正常读取,并将内容注入到viper实例的config字段,先来看看WatchConfig()方法:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
func (v *Viper) WatchConfig() {
    go func() {
      // 建立新的监视处理程序,开启一个协程开始等待事件
      // 从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对象有两个属性,Name为E:\etc\bizsvc\config.yml,Op为write(对文件的操作)
                case event := <-watcher.Events:
        // 清除内部的..和他前面的元素,清除当前路径.,用来判断操作的文件是否是configFile
                    if filepath.Clean(event.Name) == configFile {
        // 如果对该文件进行了创建操作或写操作
                        if event.Op&fsnotify.Write == fsnotify.Write || event.Op&fsnotify.Create == fsnotify.Create {
                            err := v.ReadInConfig()
                            if err != nil {
                                log.Println("error:", err)
                            }
                            v.onConfigChange(event)
                        }
                    }
                case err := <-watcher.Errors:
         // 有错误将打印
                    log.Println("error:", err)
                }
            }
        }()
 
        watcher.Add(configDir)
        <-done
    }()
}

其中,fsnotify是用来监控目录及文件的第三方库;  watcher, err := fsnotify.NewWatcher() 用来建立新的监视处理程序,它会开启一个协程开始等待读取事件,完成 从I / O完成端口读取任务,将事件注入到Event对象中,即Watcher.Events;

详解go基于viper实现配置文件热更新及其源码分析

执行v.ReadInConfig()后配置文件的内容将重新读取到viper实例中,如下图:

详解go基于viper实现配置文件热更新及其源码分析

执行完v.ReadInConfig()后,config字段的内容已经是用户修改的最新内容了;

其中这行v.onConfigChange(event)的onConfigChange是核心结构体Viper的一个属性,类型是func:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
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)
}

它用来传入本次event来执行你写的函数。为什么修改会立即生效?相信第二个疑问已经得到解决了。

接下来看看OnConfigChange(func(e fsnotify.Event) {...... })的运行情况:

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

方法参数为一个函数,类型为func(in fsnotify.Event)) {},这就意味着开发者需要把你自己的执行逻辑放到这个func里面,在监听到event时就会执行你写的函数,所以就可以这样写:

?
1
2
3
4
viperInstance.OnConfigChange(func(e fsnotify.Event) {
    log.Print("Config file updated.")
    viperLoadConf(viperInstance)  // viperLoadConf函数就是将最新配置注入到自定义结构体对象的逻辑
})

而OnConfigChange方法的参数会赋值给形参run并传到viper实例的onConfigChange属性,以WatchConfig()方法中的v.onConfigChange(event)来执行这个函数。

到此,第一个疑问也就解决了。

到此这篇关于详解go基于viper实现配置文件热更新及其源码分析的文章就介绍到这了,更多相关go viper文件热更新内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家!

原文链接:https://blog.csdn.net/HYZX_9987/article/details/103924392

延伸 · 阅读

精彩推荐
  • Golanggo语言制作端口扫描器

    go语言制作端口扫描器

    本文给大家分享的是使用go语言编写的TCP端口扫描器,可以选择IP范围,扫描的端口,以及多线程,有需要的小伙伴可以参考下。 ...

    脚本之家3642020-04-25
  • GolangGolang通脉之数据类型详情

    Golang通脉之数据类型详情

    这篇文章主要介绍了Golang通脉之数据类型,在编程语言中标识符就是定义的具有某种意义的词,比如变量名、常量名、函数名等等,Go语言中标识符允许由...

    4272021-11-24
  • Golanggolang的httpserver优雅重启方法详解

    golang的httpserver优雅重启方法详解

    这篇文章主要给大家介绍了关于golang的httpserver优雅重启的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,...

    helight2992020-05-14
  • Golanggolang 通过ssh代理连接mysql的操作

    golang 通过ssh代理连接mysql的操作

    这篇文章主要介绍了golang 通过ssh代理连接mysql的操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...

    a165861639710342021-03-08
  • Golanggolang json.Marshal 特殊html字符被转义的解决方法

    golang json.Marshal 特殊html字符被转义的解决方法

    今天小编就为大家分享一篇golang json.Marshal 特殊html字符被转义的解决方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧 ...

    李浩的life12792020-05-27
  • Golanggo日志系统logrus显示文件和行号的操作

    go日志系统logrus显示文件和行号的操作

    这篇文章主要介绍了go日志系统logrus显示文件和行号的操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...

    SmallQinYan12302021-02-02
  • GolangGolang中Bit数组的实现方式

    Golang中Bit数组的实现方式

    这篇文章主要介绍了Golang中Bit数组的实现方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...

    天易独尊11682021-06-09
  • Golanggolang如何使用struct的tag属性的详细介绍

    golang如何使用struct的tag属性的详细介绍

    这篇文章主要介绍了golang如何使用struct的tag属性的详细介绍,从例子说起,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看...

    Go语言中文网11352020-05-21