From 5f9516b33f1d030ce160da709997bdb746b1d57f Mon Sep 17 00:00:00 2001 From: liuxiaolong <736321739@qq.com> Date: 星期五, 01 十一月 2019 15:58:35 +0800 Subject: [PATCH] set config --- extend/config/config.go | 37 +++++++++++++++++++++++++++++++++++++ 1 files changed, 37 insertions(+), 0 deletions(-) diff --git a/extend/config/config.go b/extend/config/config.go index d912156..2081b5b 100644 --- a/extend/config/config.go +++ b/extend/config/config.go @@ -1 +1,38 @@ package config + +import ( + "log" + "strconv" + + "github.com/spf13/viper" +) + +type server []ServerInfo + +type ServerInfo struct { + ServerId string `mapstructure: "serverId"` + ServerIp string `mapstructure: "serverIp"` + VideoPort int `mapstructure: "videoPort"` + NginxIp string `mapstructure: "ngxIp"` + NginxPort int `mapstructure: "ngxPort"` +} + +var serverList = &server{} + +var ServerMap = make(map[string]string,0) +var NgxMap = make(map[string]string,0) +func Init(env string) { + var err error + viper.SetConfigType("yaml") + viper.SetConfigName(env) + viper.AddConfigPath("./config/") + err = viper.ReadInConfig() + if err != nil { + log.Fatal("error on parsing configuration file") + } + viper.UnmarshalKey("server", serverList) + for _,s :=range *serverList { + ServerMap[s.ServerId] = "http://"+s.ServerIp+":"+strconv.Itoa(s.VideoPort)+"/getRecordVideoPath" + NgxMap[s.ServerId] = "http://"+s.NginxIp+":"+strconv.Itoa(s.NginxPort)+"/videosource" + } +} \ No newline at end of file -- Gitblit v1.8.0