From 47b5f415a971e89038366048af57859c05d67bad Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期二, 24 九月 2019 20:19:45 +0800
Subject: [PATCH] 阈值参数名字

---
 main.go |   19 ++++++++++++++-----
 1 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/main.go b/main.go
index 411d714..9378715 100644
--- a/main.go
+++ b/main.go
@@ -7,7 +7,6 @@
 	"net/http"
 	_ "net/http/pprof"
 	"ruleprocess/insertdata"
-	"ruleprocess/labelFilter"
 	"ruleprocess/util"
 	"time"
 
@@ -24,6 +23,7 @@
 	"ruleprocess/cache"
 	"ruleprocess/ruleserver"
 	"sync"
+	"github.com/spf13/viper"
 )
 
 var dbIp = flag.String("dbIp", "127.0.0.1", "dbserver ip")
@@ -31,12 +31,21 @@
 var surveyPort = flag.Int("surveyPort", 40007, "survey port") //蹇冭烦
 var pubPort = flag.Int("pubPort", 50007, "pubsub port")       //鏁版嵁鏇存柊
 var initchan = make(chan bool)
+var env =  flag.String("env","pro","env set")
 
 func init() {
+	flag.Parse()
 	// 鏃ュ織鍒濆鍖�
-	logger.Config("./info.log", logger.DebugLevel)
-	logger.Info("鏃ュ織鍒濆鍖栨垚鍔燂紒")
 
+	insertdata.Init(*env)
+	var logFile = "./logger/"
+	if viper.GetString("LogBasePath") != "" {
+		logFile = viper.GetString("LogBasePath")
+	}
+	logFile = logFile + "ruleprocess.log"
+	fmt.Println("鏃ュ織鍦板潃锛�",logFile)
+	logger.Config(logFile, logger.DebugLevel)
+	logger.Info("鏃ュ織鍒濆鍖栨垚鍔燂紒")
 }
 func main() {
 	//fmt.Println("缂撳瓨鍒濆鍖栧畬鎴�",<- initchan)//dbserver鍒濆鍖栧畬姣�
@@ -95,7 +104,7 @@
 					insertdata.InsertToEs(resultMsg)
 					logger.Debug("鎻掑叆瀹孍s鎵�鐢ㄦ椂闂达細", time.Since(start))
 					//浜嬩欢鎺ㄩ��
-					go labelFilter.Judge(resultMsg)
+					//go labelFilter.Judge(resultMsg)
 				//}(msg)
 			}
 		}
@@ -118,7 +127,7 @@
 func paramFormat(msg []byte, args *ruleserver.SdkDatas) protomsg.SdkMessage {
 	defer func() {
 		if err := recover(); err != nil {
-			logger.Info("瑙e寘杩囩▼鐨勯敊璇�", err.(string))
+			logger.Info("瑙e寘杩囩▼鐨勫紓甯告崟鑾�", err.(string))
 		}
 
 	}()

--
Gitblit v1.8.0