问题描述
ComplexItemList:
- Name: First Complex item
Date: Whenever
StructItemList:
- Name: blah
Data1: 0
Data2: Silly
- Name: Yaddah
Data1: 12
Data2: Blah-blah-blah
- Name: Second Complex item
Date: Yesterday
StructItemList:
- Name: blah
Data1: 0
Data2: Silly
- Name: Yaddah
Data1: 12
Data2: Blah-blah-blah
这是我的代码的简单版本:
package main
import (
"fmt"
"github.com/spf13/viper"
)
type StructItem struct {
Name string `yaml:"Name"`
Data1 int `yaml:"Data1"`
Data2 string `yaml:"Data2"`
}
type ComplexItem struct {
Name string `yaml:"Name"`
Date string `yaml:"Date"`
StructItemList []StructItem `yaml:"StructItemList,mapstructure"`
}
type TestConfig struct {
Version string `yaml:"Version"`
ComplexItemList []ComplexItem `yaml:"ComplexItemList,mapstructure"`
}
func main() {
viper.SetConfigName("simple1") // name of config file (without extension)
viper.SetConfigType("yaml") // required if the config file does not have the extension in the name
viper.AddConfigPath(".")
err := viper.ReadInConfig() // Find and read the config file
if err != nil { // Handle errors reading the config file
panic(fmt.Errorf("Fatal error config file: %s \n",err))
}
test1 := viper.Get("ComplexItemList")
fmt.Printf("test1:\n%#v\n",test1)
// while I can easily unmarshal the information into a complex struct,// updating the file contents via viper does not seem straight forward.
// Is there a way to access a specific item in an array using viper dot notation?
config := TestConfig{}
viper.Unmarshal(&config)
// is the best way to update Data1 for "First Complex item",just to update the
// unmarshalled struct then write the whole thing back out?
}
我想使用Viper读取配置文件,但是不确定使用Get方法时如何区分“第一复杂项目”和“第二复杂项目”。
解决方法
我太复杂了。最简单的方法是解组值,更新它们,然后从对象设置数组。
config := TestConfig{}
viper.Unmarshal(&config)
config.ComplexItemList[0].StructItemList[0].Data2 = "updated item"
viper.Set("ComplexItemList",config.ComplexItemList)
或者这不是通过毒蛇处理yaml文件中的数组更新的正确方法吗?