You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
177 lines
3.4 KiB
177 lines
3.4 KiB
package ConfigUtil
|
|
|
|
import (
|
|
"dsAutoCode/Utils/CommonUtil"
|
|
"fmt"
|
|
"gopkg.in/ini.v1"
|
|
"os"
|
|
)
|
|
|
|
var (
|
|
// mysql数据库服务器配置
|
|
MysqlIp string
|
|
MysqlPort string
|
|
MysqlUser string
|
|
MysqlPwd string
|
|
MysqlDataBase string
|
|
WorkingPath string
|
|
WorkingDataBase string
|
|
)
|
|
|
|
// 判断所给路径文件/文件夹是否存在
|
|
func Exists(path string) bool {
|
|
_, err := os.Stat(path) //os.Stat获取文件信息
|
|
if err != nil {
|
|
if os.IsExist(err) {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
func init() {
|
|
//判断是不是单元测试,如果是的话,那么配置文件的路径需要加上一个.
|
|
var configIniFile = CommonUtil.GetCurrentPath()+"/Config/Config.ini"
|
|
//判断文件不是存在
|
|
if !Exists(configIniFile) {
|
|
configIniFile = CommonUtil.GetCurrentParentPath()+ "/Config/Config.ini"
|
|
}
|
|
|
|
iniParser := IniParser{}
|
|
if err := iniParser.Load(configIniFile); err != nil {
|
|
fmt.Printf("try load config file[%s] error[%s]\n", configIniFile, err.Error())
|
|
return
|
|
}
|
|
//主机
|
|
MysqlIp = iniParser.GetString("mysql", "ip")
|
|
//端口
|
|
MysqlPort = iniParser.GetString("mysql", "port")
|
|
//数据库名称
|
|
MysqlDataBase = iniParser.GetString("mysql", "database")
|
|
//用户
|
|
MysqlUser = iniParser.GetString("mysql", "user")
|
|
//密码
|
|
MysqlPwd = iniParser.GetString("mysql", "pwd")
|
|
//数据库
|
|
WorkingDataBase =iniParser.GetString("mysql","workingDataBase")
|
|
}
|
|
|
|
type IniParser struct {
|
|
confReader *ini.File // config reader
|
|
}
|
|
|
|
type IniParserError struct {
|
|
errorInfo string
|
|
}
|
|
|
|
func (e *IniParserError) Error() string { return e.errorInfo }
|
|
|
|
func (this *IniParser) Load(configFileName string) error {
|
|
conf, err := ini.Load(configFileName)
|
|
if err != nil {
|
|
this.confReader = nil
|
|
return err
|
|
}
|
|
this.confReader = conf
|
|
return nil
|
|
}
|
|
|
|
func (this *IniParser) GetString(section string, key string) string {
|
|
if this.confReader == nil {
|
|
return ""
|
|
}
|
|
|
|
s := this.confReader.Section(section)
|
|
if s == nil {
|
|
return ""
|
|
}
|
|
|
|
return s.Key(key).String()
|
|
}
|
|
|
|
func (this *IniParser) GetInt32(section string, key string) int32 {
|
|
if this.confReader == nil {
|
|
return 0
|
|
}
|
|
|
|
s := this.confReader.Section(section)
|
|
if s == nil {
|
|
return 0
|
|
}
|
|
|
|
valueInt, _ := s.Key(key).Int()
|
|
|
|
return int32(valueInt)
|
|
}
|
|
|
|
func (this *IniParser) GetUint32(section string, key string) uint32 {
|
|
if this.confReader == nil {
|
|
return 0
|
|
}
|
|
|
|
s := this.confReader.Section(section)
|
|
if s == nil {
|
|
return 0
|
|
}
|
|
|
|
valueInt, _ := s.Key(key).Uint()
|
|
|
|
return uint32(valueInt)
|
|
}
|
|
|
|
func (this *IniParser) GetInt64(section string, key string) int64 {
|
|
if this.confReader == nil {
|
|
return 0
|
|
}
|
|
|
|
s := this.confReader.Section(section)
|
|
if s == nil {
|
|
return 0
|
|
}
|
|
|
|
valueInt, _ := s.Key(key).Int64()
|
|
return valueInt
|
|
}
|
|
|
|
func (this *IniParser) GetUint64(section string, key string) uint64 {
|
|
if this.confReader == nil {
|
|
return 0
|
|
}
|
|
|
|
s := this.confReader.Section(section)
|
|
if s == nil {
|
|
return 0
|
|
}
|
|
|
|
valueInt, _ := s.Key(key).Uint64()
|
|
return valueInt
|
|
}
|
|
|
|
func (this *IniParser) GetFloat32(section string, key string) float32 {
|
|
if this.confReader == nil {
|
|
return 0
|
|
}
|
|
|
|
s := this.confReader.Section(section)
|
|
if s == nil {
|
|
return 0
|
|
}
|
|
|
|
valueFloat, _ := s.Key(key).Float64()
|
|
return float32(valueFloat)
|
|
}
|
|
|
|
func (this *IniParser) GetFloat64(section string, key string) float64 {
|
|
if this.confReader == nil {
|
|
return 0
|
|
}
|
|
|
|
s := this.confReader.Section(section)
|
|
if s == nil {
|
|
return 0
|
|
}
|
|
|
|
valueFloat, _ := s.Key(key).Float64()
|
|
return valueFloat
|
|
}
|