go简单写个ini配置文件读取工具

直接上代码：

package main

import (
    "fmt"
    "io/ioutil"
    "reflect"
    "strconv"
    "strings"
)

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"`
}

type Config struct {
    MysqlConfig `ini:"mysql"`
    RedisConfig `ini:"redis"`
}

func loadIni(fileName string, v interface{}) error {
    // 参数校验:v需为结构体指针类型
    t := reflect.TypeOf(v)
    if t.Kind() != reflect.Ptr || t.Elem().Kind() != reflect.Struct {
        return fmt.Errorf("v should be a struct ptr")
    }

    // 读取文件得到数据后立即关闭文件
    f, err := ioutil.ReadFile(fileName)
    if err != nil {
        return err
    }
    // 逐行获取文件内容，将值赋值给对应的结构体对象
    lineSlice := strings.Split(string(f), "\n")
    var structName string
    for index, line := range lineSlice {
        line = strings.TrimSpace(line)
        // 跳过注释和空行
        if strings.HasPrefix(line, ";") || strings.HasPrefix(line, "#") || len(line) == 0 {
            continue
        }
        // 以 [ 开头,为节点,需以 ] 结尾,且除空格外存在字符
        if strings.HasPrefix(line, "[") {
            if line[len(line) - 1] != ']' {
                return fmt.Errorf("line:%d syntax error", index+1)
            }
            sectionName := strings.TrimSpace(line[1:len(line)-1])
            if len(sectionName) == 0 {
                return fmt.Errorf("line:%d syntax error", index+1)
            }
            // 根据字符串sectionName去v里面根据反射找到对应的结构体
            for i :=0 ; i < t.Elem().NumField(); i ++ {
                field := t.Elem().Field(i)
                if sectionName == field.Tag.Get("ini") {
                    structName = field.Name
                    //fmt.Println(structName)
                    break
                }
            }
        } else {
            // 不以 [ 开头,为以=分割的键值对
            if strings.Index(line, "=") == -1 || strings.HasPrefix(line, "=") {
                return fmt.Errorf("line:%d syntax error", index+1)
            }
            // 根据struct名称找到对应结构体
            data := reflect.ValueOf(v)
            structValue := data.Elem().FieldByName(structName)  // 拿到嵌套结构体的值信息
            structType := structValue.Type()  // 拿到嵌套结构体的类型信息
            if structType.Kind() != reflect.Struct {
                return fmt.Errorf("config struct field %s must be a struct", structName)
            }
            // 拆分键值对并赋值
            paramSlice := strings.Split(line, "=")
            key, value := strings.TrimSpace(paramSlice[0]), strings.TrimSpace(paramSlice[1])
            for x := 0; x < structType.NumField(); x ++ {
                // 先拿到fieldType,fieldType中可以拿到field的值类型
                fieldType := structType.Field(x)
                if fieldType.Tag.Get("ini") == key {
                    // 根据field名称拿到field字段
                    fieldV := structValue.FieldByName(fieldType.Name)
                    // 给字段赋值
                    switch fieldType.Type.Kind() {
                    case reflect.String:
                        fieldV.SetString(value)
                    case reflect.Int,reflect.Int8,reflect.Int16,reflect.Int32,reflect.Int64:
                        valueInt, err := strconv.ParseInt(value, 10, 64)
                        if err != nil {
                            return fmt.Errorf("parseInt error, err:%v", err)
                        }
                        fieldV.SetInt(valueInt)
                    case reflect.Float32:
                        valueFloat, err := strconv.ParseFloat(value, 32)
                        if err != nil {
                            return fmt.Errorf("parseFloat error, err:%v", err)
                        }
                        fieldV.SetFloat(valueFloat)
                    case reflect.Float64:
                        valueFloat, err := strconv.ParseFloat(value, 64)
                        if err != nil {
                            return fmt.Errorf("parseFloat error, err:%v", err)
                        }
                        fieldV.SetFloat(valueFloat)
                    case reflect.Bool:
                        valueBool, err := strconv.ParseBool(value)
                        if err != nil {
                            return fmt.Errorf("parseBool error, err:%v", err)
                        }
                        fieldV.SetBool(valueBool)
                    }
                    break
                }
            }
        }
    }
    return nil
}

func main() {
    var config Config
    fileName := "/home/xxx/data/demo_config/config.ini"
    err := loadIni(fileName, &config)
    if err != nil {
        fmt.Printf("load ini failed, err:%v", err)
    }
    fmt.Println("mysql config:")
    fmt.Printf("address:%s\tport:%d\tusername:%s\tpassword:%s\n", config.MysqlConfig.Address, config.MysqlConfig.Port, config.MysqlConfig.UserName, config.MysqlConfig.PassWord)

    fmt.Println("redis config:")
    fmt.Printf("host:%s\tport:%d\tpassword:%s\tdatabase:%d\n", config.RedisConfig.Host, config.RedisConfig.Port, config.RedisConfig.PassWord, config.RedisConfig.DataBase)
}