golang入门笔记—ini配置文件解析
package main
//ini配置文件解析器
import (
"errors"
"fmt"
"io/ioutil"
"reflect"
"strconv"
"strings"
)
//Mysql Config MySQL配置结构体
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 int `ini:"database"`
Test bool `ini:"test"`
}
type Config struct {
MysqlConfig `ini:"mysql"`
RedisConfig `ini:"redis"`
}
func loadIni(fileName string, data interface{}) (err error) {
//0.参数的校验
//0.1.传进来的data参数必须是指针类型(因为需要在函数中对其赋值)
t := reflect.TypeOf(data)
fmt.Println(t, t.Kind())
if t.Kind() != reflect.Ptr {
err = fmt.Errorf("data should be a pointer.") //格式化输出之后返回一个error类型
return
}
//0.2.传进来的data参数必须是结构体类型指针(因为配置文件中各种键值对需要赋值给结构体的字段)
if t.Elem().Kind() != reflect.Struct {
err = errors.New("data param should be a pointer")
return
}
//1.读文件得到字节类型的数据
b, err := ioutil.ReadFile(fileName)
if err != nil {
return
}
lineSlice := strings.Split(string(b), "\r\n")
fmt.Printf("%#v", lineSlice)
//2.一行一行地读数据
var structName string
for idx, line := range lineSlice {
//2.1.去掉字符串首尾的空格
line = strings.TrimSpace(line)
if len(line) == 0 {
continue
}
//2.2.如果是注释就忽略跳过
if strings.HasPrefix(line, ";") || strings.HasPrefix(line, "#") {
continue
}
//2.3.如果是[开头的就表示是字节(section)
if strings.HasPrefix(line, "[") {
if !strings.HasSuffix(line, "]") {
err = fmt.Errorf("line:%d syntax error", idx+1)
return
}
//把这一行首尾的[]去掉，取到中间的内容去掉空格拿到的内容
sectionName := line[1 : len(line)-1]
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 sectionName == field.Tag.Get("ini") {
//说明找到了对应的嵌套结构体，把字段名记下来
structName = field.Name
fmt.Printf("找到%s对应的嵌套结构体%s\n", sectionName, structName)
}
}
} else {
//2.4.如果不是[开头就是=分割的键值对
//1.以等号分割这一行，等号左边是key，等号右边是value
if strings.Index(line, "=") == -1 || strings.HasPrefix(line, "=") {
err = fmt.Errorf("line:%d syntax error", idx+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 := sValue.Type()
if structObj.Kind() != reflect.Struct {
fmt.Printf("data中的%s字段应该是一个结构体\n", structName)
return
}
var fieldName string
var fileType reflect.StructField
//3.遍历嵌套结构体的每一个字段，判断tag是否等于key
for i := 0; i < structObj.NumField(); i++ {
filed := sType.Field(i) //Tag信息是存储在类型信息中
fileType = filed
if filed.Tag.Get("ini") == key {
//找到对应的字段
fieldName = filed.Name
break
}
}
//4.如果key=tag，给这个字段赋值
//4.1.根据fileName取出这个字段
if len(fieldName) == 0 {
//在结构体中找不到对应的字段
continue
}
fieldObj := sValue.FieldByName(fieldName)
//4.2.对其赋值
fmt.Println(fieldName, fileType.Type.Kind())
switch fileType.Type.Kind() {
case reflect.String:
fieldObj.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 error", idx+1)
return
}
fieldObj.SetInt(valueInt)
case reflect.Bool:
var valueBool bool
valueBool, err = strconv.ParseBool(value)
if err != nil {
err = fmt.Errorf("line:%d value type error", idx+1)
return
}
fieldObj.SetBool(valueBool)
case reflect.Float32, reflect.Float64:
var valueFloat float64
valueFloat, err = strconv.ParseFloat(value, 64)
if err != nil {
err = fmt.Errorf("line:%d value type error", idx+1)
return
}
fieldObj.SetFloat(valueFloat)
}
}
}
return
}
func main() {
var mc Config
err := loadIni("./config.ini", &mc)
if err != nil {
fmt.Printf("load ini failed,err:%v\n", err)
return
}
fmt.Println(mc)
}