go解析 ini 配置文件
conf.ini
[mysql]
address=10.20.30.40
port=3306
username=root
password=123456
#redis config
[redis]
xxxx=
host=127.0.0.10
port=22
password=3333
databases=0
test=false
package main
import (
"errors"
"fmt"
"io/ioutil"
"reflect"
"strconv"
"strings"
)
//init 文件配置
type MysqlConfig struct {
Address string `ini:"address"`
Port int `ini:"port"`
Username string `ini:"username"`
Password string `ini:"password"`
}
type RediConfig struct {
Host string `ini:"host"`
Port int `ini:"port"`
Password string `ini:"password"`
Database int `ini:"databases"`
Test bool `ini:"test"`
}
// 整个配置文件的结构体
type Config struct {
//int `ini:"mysql"`
MysqlConfig `ini:"mysql"`
RediConfig `ini:"redis"`
}
func loadIni(filname string, data interface{}) (err error) {
//0; 参数的校验
//0.1 传进来的参数必须是指针类型,(因为需要在函数中对其赋值)
// 任意值通过reflect.TypeOf()获得反射对象信息后,如果它的类型是结构体,
//可以通过反射值对象(reflect.Type)的NumField()和Field()方法获得结构体成员的详细信息。
t := reflect.TypeOf(data)
fmt.Println(t, t.Kind())
fmt.Println("---------------------")
if t.Kind() != reflect.Ptr {
err = errors.New("data param should be a pointer.")
return
}
//0.2 传进来的data 参数必须是结构体类型, (因为配置文件中有各种键值对需要赋值给结构体的字段)
if t.Elem().Kind() != reflect.Struct {
err = errors.New("data param should be a struct .")
return
}
//1.读取文件得到字节类型数据
b, err := ioutil.ReadFile(filname) // 把文件读取 出来然后关闭了
if err != nil {
err = errors.New("read file error .")
return
}
// 把文件内容 转化为字符串
lineSlice := strings.Split(string(b), "\r\n")
fmt.Println("%#v\n", lineSlice)
//2.一行一行的读取数据
var structName string
for index, line := range lineSlice {
// 去掉字符串首尾的空格
line = strings.TrimSpace(line)
// r如果空行直接 跳过
if len(line) == 0 {
continue
}
//2.1如果是注释 就跳过
if strings.HasPrefix(line, ";") || strings.HasPrefix(line, "#") {
continue
}
//2.2 如果是[]开头的就表示个 段落section
if strings.HasPrefix(line, "[") {
if line[0] != '[' || line[len(line)-1] != ']' {
err = fmt.Errorf("line : %d has syntax error ", index+1)
return
}
// 把这一行首尾的[] 去掉, 取到中间的内容,把首尾的空格去掉,拿到内容
sectionName := strings.TrimSpace(line[1 : len(line)-1])
if len(sectionName) == 0 {
err = fmt.Errorf("line : %d has syntax error ", index+1)
return
}
// 根据字符串sectionName 去data 里面根据反射找到对应的结构体
for i := 0; i < t.Elem().NumField(); i++ {
field := t.Elem().Field(i)
if sectionName == field.Tag.Get("ini") { // 找到 配置文件中的mysql 跟结构体对应
// 找到了对应的结构体, 切片名字记下来
structName = field.Name
fmt.Println("找到 %s 对应的嵌套结构体 %s ", sectionName, structName)
break
}
}
} else { //2.3 读取每一行 如果不是[]开头的就是 = 分割的键值对
// 1. 以等号分割这一行, 等号左边是key, 等号右边是value
if strings.Index(line, "=") == -1 || strings.HasPrefix(line, "=") {
err = fmt.Errorf("line: %d syntax error", index+1)
return
}
index := strings.Index(line, "=")
key := strings.TrimSpace(line[:index])
value := strings.TrimSpace(line[index+1:])
// 2. 根据structName 去data 里面把对应的嵌套结构体给取出来
v := reflect.ValueOf(data)
sValue := v.Elem().FieldByName(structName) // 拿到嵌套结构体的值信息
sType := sValue.Type() // 拿到嵌套结构体的类型信息
structObj := v.Elem().FieldByName(structName)
if structObj.Kind() != reflect.Struct {
fmt.Println(" data 中的%s 字段应该是个结构体", structName)
}
var fieldName string
var fileType reflect.StructField
// 3. 遍历 嵌套结构体的每一个字段, 判断 tag 是不是等于key
for i := 0; i < sValue.NumField(); i++ {
filed := sType.Field(i) // tag 信息存储在类型信息中
fileType = filed
if filed.Tag.Get("ini") == key {
// 找到对应的字段
fieldName = filed.Name
break
}
}
// 4. 如果key = tag ,给这个字段赋值。
//4.1 根据 fieldName 去取出这个字段
if len(fieldName) == 0 { //如果是 xxx= 就跳过
continue
}
fileObj := sValue.FieldByName(fieldName)
// 4.2 对其赋值
fmt.Println(fieldName, fileType.Type.Name())
switch fileType.Type.Kind() {
case reflect.String:
fileObj.SetString(value)
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
var valueInt int64
valueInt, err := strconv.ParseInt(value, 10, 64)
if err != nil {
err = fmt.Errorf("line: %d value type erro", index+1)
return err
}
fileObj.SetInt(valueInt)
case reflect.Bool: // 如果是 bool 类型的 ,这里我们没用到哦,可以作为扩展
var valueBool bool
valueBool, err = strconv.ParseBool(value)
if err != nil {
err = fmt.Errorf("line: %d value type erro", index+1)
return err
}
fileObj.SetBool(valueBool)
}
}
}
return
}
func testini() {
var cfg Config
iniFilePath := "C:\\Users\\chaoren\\GolandProjects\\mylogger\\demo\\inidemo\\conf.ini"
//1. 加载配置文件
err := loadIni(iniFilePath, &cfg)
if err != nil {
fmt.Printf("load ini failed , err: %v\n", err)
return
}
fmt.Printf("%#v \n", cfg)
//2. 把配置文件映射成 MysqlConfig
}
func main() {
testini()
}