From d671ea8ec8658da21990d5418c981d74f6015045 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期三, 11 十二月 2019 11:50:43 +0800
Subject: [PATCH] add conf path

---
 extend/config/config.go |   36 ++++++++++++++++++++++++++++++++++++
 1 files changed, 36 insertions(+), 0 deletions(-)

diff --git a/extend/config/config.go b/extend/config/config.go
index d912156..54b3bf9 100644
--- a/extend/config/config.go
+++ b/extend/config/config.go
@@ -1 +1,37 @@
 package config
+
+import (
+	"github.com/spf13/viper"
+	"log"
+)
+
+type server []ServerInfo
+
+type ServerInfo struct {
+	ServerId string `mapstructure: "serverId"`
+	ServerIp string `mapstructure: "serverIp"`
+	VideoPort string   `mapstructure: "videoPort"`
+	NgxIp string  `mapstructure: "ngxIp"`
+	NgxPort string   `mapstructure: "ngxPort"`
+}
+
+var serverList = &server{}
+
+var ServerMap = make(map[string]string,0)
+var NgxMap = make(map[string]string,0)
+func Init(env string, configPath string) {
+	var err error
+	viper.SetConfigType("yaml")
+	viper.SetConfigName(env)
+	viper.AddConfigPath(configPath)
+	err = viper.ReadInConfig()
+	if err != nil {
+		log.Fatal("error on parsing configuration file",err)
+	}
+	viper.UnmarshalKey("server", serverList)
+	for _,s :=range *serverList {
+		si := s
+		ServerMap[si.ServerId] = "http://"+si.ServerIp+":"+si.VideoPort+"/getRecordVideoPath"
+		NgxMap[si.ServerId] = "http://"+si.NgxIp+":"+si.NgxPort+"/videosource"
+	}
+}
\ No newline at end of file

--
Gitblit v1.8.0