From 032150859e985bedfb5061320b3aed5984dda476 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期三, 18 九月 2019 17:51:02 +0800
Subject: [PATCH] mian函数中加一个异常处理,插入es时大图只插入一次

---
 main.go |   20 +++++++++++++++++---
 1 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index 34393ee..d0caff6 100644
--- a/main.go
+++ b/main.go
@@ -23,6 +23,7 @@
 	"ruleprocess/cache"
 	"ruleprocess/ruleserver"
 	"sync"
+	"github.com/spf13/viper"
 )
 
 var dbIp = flag.String("dbIp", "127.0.0.1", "dbserver ip")
@@ -30,17 +31,30 @@
 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"
+	logger.Config(logFile, logger.DebugLevel)
+	logger.Info("鏃ュ織鍒濆鍖栨垚鍔燂紒")
 }
 func main() {
 	//fmt.Println("缂撳瓨鍒濆鍖栧畬鎴�",<- initchan)//dbserver鍒濆鍖栧畬姣�
+	defer func() {
+		if err := recover(); err != nil {
+			logger.Error("鏈�澶栧眰鐨勫紓甯告崟鎹�",err.(string))
+		}
+	}()
 	go func() {
-		http.ListenAndServe("0:0:0:0:8069",nil)
+		http.ListenAndServe("0.0.0.0:8899",nil)
 	}()
 	flag.Parse()
 	wg := sync.WaitGroup{}

--
Gitblit v1.8.0