background/config/config.go

82 lines
1.6 KiB
Go
Raw Normal View History

2019-01-25 09:11:15 +00:00
package config
import (
"log"
"gopkg.in/yaml.v2"
"os"
"runtime"
)
var ostype = runtime.GOOS
var conf ConfAPI
type ConfAPI struct {
ListenSvr int `yaml:"listen_svr"` // 服务监听端口
ListenApi int `yaml:"listen_api"` // 服务监听端口
RunMode string `yaml:"runmode"` // 服务运行模式
MaxConn int `yaml:"max_conn"`
Logs LogConfig `yaml:"logs"` // 日志
Redis RedisConfig `yaml:"redis"`
Mysql MysqlConfig `yaml:"mysql"` // 认证配置
init bool
}
type RedisConfig struct {
Addr string `yaml:"addr"`
Pwd string `yaml:"password"`
DB int64 `yaml:"db"`
}
type LogConfig struct {
Dir string `yaml:"dir"`
File string `yaml:"file"`
Level int `yaml:"level"`
SaveFile bool `yaml:"savefile"`
}
type MysqlConfig struct {
Addr string `yaml:"addr"`
UserName string `yaml:"user"`
Password string `yaml:"password"`
Db string `yaml:"db"`
MaxOpen int `yaml:"max_open"`
MaxIdle int `yaml:"max_idle"`
}
var gConf ConfAPI
func Init(path string) error {
file,e := os.Open(path)
if nil != e{
log.Println(e.Error())
return e
}
stat,_ := file.Stat()
filec := make([]byte, stat.Size())
file.Read(filec)
e = yaml.Unmarshal(filec,&gConf)
if nil != e{
log.Println(e.Error())
}
gConf.init = true
return nil
}
func GetPort() int {
if gConf.init{
return gConf.ListenApi
}else {
return 8001
}
}
func GetMysqlConfig() *MysqlConfig{
if gConf.init{
return &gConf.Mysql
}else {
return nil
}
2019-01-29 13:57:53 +00:00
}
func GetLogConfig() *LogConfig {
if gConf.init{
return &gConf.Logs
}else{
return nil
}
2019-01-25 09:11:15 +00:00
}