go viper涉及多个配置文件时使用watchconfig 配置丢失问题

最新推荐文章于 2024-08-16 23:35:32 发布

夏日风帆

最新推荐文章于 2024-08-16 23:35:32 发布

阅读量3.5k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： golang 文章标签： golang 开发语言后端

本文链接：https://blog.csdn.net/chsummer699/article/details/122652943

背景：

最近在项目使用中需要同时使用两个配置文件一个是程序启动时需要的配置如日志级别另一个是会更新的业务配置文件使用同一个viper对象管理发现只要这个文件变动触发viper更新则第一个文件的所有配置项丢失

原因：

即使读取第二个文件使用MergeInConfig 问题还会存在

查看源码发现


func (v *Viper) WatchConfig() {
	initWG := sync.WaitGroup{}
	initWG.Add(1)
	go func() {
		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.Printf("error: %v\n", err)
			initWG.Done()
			return
		}

		configFile := filepath.Clean(filename)
		configDir, _ := filepath.Split(configFile)
		realConfigFile, _ := filepath.EvalSymlinks(filename)

		eventsWG := sync.WaitGroup{}
		eventsWG.Add(1)
		go func() {
			for {
				select {
				case event, ok := <-watcher.Events:
					if !ok { // 'Events' channel is closed
						eventsWG.Done()
						return
					}