package cfg import ( "flag" "io/ioutil" "gopkg.in/yaml.v2" ) // 配置文件数据,全局变量 var ( Debug bool Prd bool SrvAddr string SmartCanteenPay string RedisAddr string RedisPassword string DB *DBCfg IMDB *DBCfg MQ *MQCfg ES *ESCfg Log *LogCfg ) // 初始化配置文件,将cfg.yml读入到内存 func InitCfg() { //用指定的名称、默认值、使用信息注册一个string类型flag。 path := flag.String("c", "etc/cfg.yml", "config file") //解析命令行参数写入注册的flag里。 //解析之后,flag的值可以直接使用。 flag.Parse() var ( c []byte err error conf *Config ) if c, err = ioutil.ReadFile(*path); err != nil { panic(err) } //yaml.Unmarshal反序列化映射到Config if err = yaml.Unmarshal(c, &conf); err != nil { panic(err) } //数据读入内存 Prd = conf.Prd Debug = conf.Debug DB = &conf.DB IMDB = &conf.ImDB Log = &conf.Log RedisAddr = conf.RedisAddr RedisPassword = conf.RedisPassword SrvAddr = conf.SrvAddr MQ = &conf.MQ ES = &conf.ES }