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