go 解析配置文件

package main

import (
   "bufio"
   "errors"
   "io"
   "io/ioutil"
   "reflect"
   "strconv"
   "strings"

   //"io/ioutil"
   "os"
   //"strings"
   "fmt"
)
type config struct{
   Redis redisconfig `ini:"redis"`
   Mysql mysqlconfig `ini:"mysql"`
}

type redisconfig struct{
   Host string `ini:"host"`
   Port int `ini:"port"`
   Password string `ini:"password"`
}

type mysqlconfig struct{
   Host string `ini:"host"`
   Port int `ini:"port"`
}
func loadinit1(fileName string,conf interface{})error{
   v := reflect.ValueOf(conf)
   // 指针类型数据 需要使用 Elem() 方法
   if v.Elem().Kind() != reflect.Struct {
      return errors.New("请传入一个指针的类型结构体变量")
   }
   t := v.Type()
   if t.Kind() != reflect.Ptr  {
      err := errors.New("参数错误1")
      return err
   }
   
   b,err :=ioutil.ReadFile(fileName)
   if err != nil{
      return err
   }

   data :=strings.Split(string(b),"\n")
   var structName string
   for _,line := range data{
      if len(line) == 0{
         continue
      }
      line = strings.TrimSpace(line)
      if strings.HasPrefix(line,";") ||  strings.HasPrefix(line,"#"){
         continue
      }
      if(line[0] == '['){
         if  line[len(line)-1] !=']' || len(line) <=1{
            err := errors.New("段错误2")
            return err
         }
         sectionName := line[1:len(line)-1]
         for i := 0; i < t.Elem().NumField(); i++ {
            field := t.Elem().Field(i)
            if sectionName == field.Tag.Get("ini"){
               structName = field.Name
            }
         }
      } else{
         if strings.Index(line,"=") == -1 || strings.HasPrefix(line,"="){
            err := errors.New("数据格式错误3")
            return err
         }
         structObj := v.Elem().FieldByName(structName)
         if structObj.Kind() != reflect.Struct{
            err := errors.New("不是结构体4")
            return err
         }
         para := strings.Split(line,"=")
         var filedName string
         stype := structObj.Type()
         for i:=0;i<stype.NumField();i++{
            field := stype.Field(i)
            if field.Tag.Get("ini") == para[0]{
               filedName = field.Name
               fieldObj := structObj.FieldByName(filedName)
               switch field.Type.Kind(){
               case reflect.String:
                  fieldObj.SetString(para[1])
               case reflect.Int:
                  valint,err := strconv.ParseInt(para[1],10,64)
                  if err != nil{
                     return err
                  }
                  fieldObj.SetInt(valint)
               }
               break
            }
         }
      }
   }

   return nil
}

func main(){
   con :=config{}
   err :=loadinit1("/Users/admin/code/goproject/src/go_code/basic/string/reflect.ini",&con)
   if err != nil{
      fmt.Println(err)
   }
   fmt.Println(con)
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值