From c2ced9eaa6a8f174866ecbdb77d7ce7b02bb969c Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期二, 17 十二月 2019 16:40:52 +0800
Subject: [PATCH] add sdk config

---
 proc/master.go |   26 +++++++++++++++++++++++---
 1 files changed, 23 insertions(+), 3 deletions(-)

diff --git a/proc/master.go b/proc/master.go
index 7c69c0a..eec37c2 100644
--- a/proc/master.go
+++ b/proc/master.go
@@ -4,6 +4,8 @@
 	"analysis/logo"
 	"analysis/util"
 	"context"
+
+	"github.com/spf13/viper"
 )
 
 func reaper(ctxt context.Context) {
@@ -18,17 +20,35 @@
 
 	util.InitDBAPI()
 
+	const (
+		configFilePath = "/opt/vasystem/config/"
+		configFileName = "sdkconfig"
+		configFileType = "yaml"
+	)
+	viper.SetConfigType(configFileType)
+	viper.SetConfigName(configFileName)
+	viper.AddConfigPath(configFilePath)
+	viper.AddConfigPath("./")
+
+	envString := ""
+
 	sdks := util.SDKInfo()
 	for k, v := range sdks {
+
+		if err := viper.ReadInConfig(); err == nil {
+
+			envString = viper.GetString(v.SdkType)
+
+		}
+		logo.Errorln("MASTER ANALYSIS START SLAVE TYPE: ", v.SdkType, " SDK CONFIG: ", envString)
+
 		args := []string{
 			`-role=slave`,
 			"-sdk=" + v.SdkType,
 			"-id=" + v.IpcId,
 		}
-
 		args = append(args, (*util.GetParams())...)
-
-		pid, err := runProc(ctx, "./analysis", args)
+		pid, err := runProc(ctx, "./analysis", args, &envString)
 
 		if err != nil {
 			logo.Errorf("ANALYSIS START SLAVE PROC %s IPC: %s error %+v\n", v.SdkType, v.IpcId, err)

--
Gitblit v1.8.0