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