background/config/config.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
	}
}