82 lines
1.6 KiB
Go
82 lines
1.6 KiB
Go
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
|
|
}
|
|
}
|
|
func GetLogConfig() *LogConfig {
|
|
if gConf.init{
|
|
return &gConf.Logs
|
|
}else{
|
|
return nil
|
|
}
|
|
} |