103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
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)
|
||
}
|