package main import ( "fmt" "os" "strings" "reflect" ) //定义结构体 type MysqlConfig struct { Address string `ini:"address"` Port int `ini:"port"` Username string `ini:"username"` Password string `ini:"password"` } type RedisConfig struct { host string `ini:"host"` Port int `ini:"port"` Password string `ini:"password"` Database string `ini:"database"` } type Config struct { RedisConfig `ini:"mysql"` MysqlConfig `ini:"redis"` } func Loadini(fileName string, data interface{}) (err error) { //0.参数的校验,对函数中修改值,必须是结构体指针 t := reflect.TypeOf(data) //v := reflect.ValueOf(data) //校验指针类型 if t.Kind() != reflect.Ptr { err = fmt.Errorf("data should be pointer") return } //校验结构体类型,由于是指针,要用elem方法 if t.Elem().Kind() != reflect.Struct { err = fmt.Errorf("data should be struct") return } //1.读取文件 b, err := os.ReadFile(fileName) //读取文件拿到字节类型的数据 if err != nil { return } lineSlice := strings.Split(string(b), "\r\n") //字节类型的数据按照换行符切分 fmt.Printf("%#v\n", lineSlice) //var structName string //2.一行一行的读取数据 for idx, line := range lineSlice { //2.1如果是注释,就跳过 line = strings.TrimSpace(line) if strings.HasPrefix(line, ";") || strings.HasPrefix(line, "#") { continue } //2.2如果是[开头的,就表示节(section) if strings.HasPrefix(line, "[") { //校验1 首尾的[]都需要存在 if line[0] != '[' || line[len(line)-1] != ']' { err = fmt.Errorf("line:%d syntax error", idx+1) return } //结之间能取到的内容,比如redis或者mysql这个字符串 sectionName := strings.TrimSpace(line[1 : len(line)-1]) //校验2 []之间应该有内容 if len(sectionName) == 0 { //切片:左包含右不包含 err = fmt.Errorf("line:%d syntax error", idx+1) return } //根据字符串sectionName到data中根据反射找到结构体 for i := 0; i < t.Elem().NumField(); i++ { field := t.Elem().Field(i) if field.Tag.Get("ini") == sectionName { //找到对应的嵌套结构体 //structName = field.Name break } } } else { //2.3如果不是[,就是=分割的键值对 } } return // } func main() { var cfg Config err := Loadini("./ini_unmarshal/config.ini", &cfg) if err != nil { fmt.Println("load ini error:", err) return } fmt.Println(cfg) }