background/config/config.go

152 lines
3.1 KiB
Go

package config
import (
"log"
"os"
"runtime"
wsConf "tcptemplate/config"
"gopkg.in/yaml.v2"
)
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"` // 日志
Redis1 EntityRedis `yaml:"redis1"`
Mysql MysqlConfig `yaml:"mysql"` // 认证配置
ElasticSerach ElasticSearch `yaml:"elasticsearch"`
Mysql1 MysqlConfig `yaml:"mysql1"` // 认证配置
MongoConf MongoConfig `yaml:"mongo"`
WsServerConf wsConf.Config `yaml:"ws_server_conf"`
CaCert string `yaml:"ca_cert"`
ClientCert string `yaml:"client_cert"`
ClientKey string `yaml:"client_key"'`
init bool
}
type ElasticSearch struct{
Address string `yaml:"address"`
User string `yaml:"user"`
Password string `yaml:"password"`
}
type EntityRedis struct {
Addr string `yaml:"addr"`
Pwd string `yaml:"password"`
DB int `yaml:"db"`
PoolSize int `yaml:"poolsize"`
}
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"`
}
// mongodb://myuser:mypass@localhost:40001
type MongoConfig struct {
Addr string `yaml:"addr"`
Port int64 `yaml:"port"`
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 ApiConfig() ConfAPI{
return gConf
}
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())
log.Print(filec)
file.Read(filec)
e = yaml.Unmarshal(filec, &gConf)
if nil != e {
log.Println(e.Error())
}
gConf.init = true
log.Print(gConf)
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 GetMysqlBlogConfig() *MysqlConfig {
if gConf.init {
return &gConf.Mysql1
} else {
return nil
}
}
func GETCaCert() string{
return gConf.CaCert
}
func GetMongoDBConfig() *MongoConfig{
if gConf.init{
return &gConf.MongoConf
} else {
return nil
}
}
func GetRedis1() *EntityRedis {
if gConf.init {
return &gConf.Redis1
} else {
return nil
}
}
func GetLogConfig() *LogConfig {
if gConf.init {
return &gConf.Logs
} else {
return nil
}
}
func GetElkConfig() *ElasticSearch{
if gConf.init{
return &gConf.ElasticSerach
}else {
return nil
}
}