From 2dd1fa03c93b400ff1e8e4f44e27387965e4daa7 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期四, 19 九月 2019 17:23:59 +0800
Subject: [PATCH] get ip,port config from db,update realtime

---
 main.go |   17 ++++++++---------
 1 files changed, 8 insertions(+), 9 deletions(-)

diff --git a/main.go b/main.go
index 5f48e32..0dbb97e 100644
--- a/main.go
+++ b/main.go
@@ -4,9 +4,9 @@
 	"basic.com/dbapi.git"
 	"basic.com/valib/logger.git"
 	"flag"
-	"github.com/golang/glog"
 	"github.com/spf13/viper"
 	"strconv"
+	"webserver/cache"
 	"webserver/extend/config"
 	"webserver/router"
 )
@@ -14,6 +14,9 @@
 var envirment = flag.String("e", "dev", "")
 var dbIp = flag.String("dbIp", "127.0.0.1", "default dbIp=127.0.0.1")
 var dbPort = flag.String("dbPort", "8001", "default dbPort=8001")
+var surveyPort = flag.Int("surveyPort", 40007, "survey port") //蹇冭烦
+var pubPort = flag.Int("pubPort", 50007, "pubsub port")       //鏁版嵁鏇存柊
+var initchan = make(chan bool)
 
 func init() {
 	flag.Parse()
@@ -42,16 +45,12 @@
 	}
 	logger.SetSaveDays(logSaveDays)
 	logger.Info("loginit success !")
-
-	flag.Usage = func() {
-		glog.Exit("flag parse usage !")
-	}
-
-	flag.Lookup("alsologtostderr").Value.Set("true")
-	flag.Lookup("log_dir").Value.Set("./log")
-
 }
 func main() {
+	flag.Parse()
+	go cache.Init(initchan, *dbIp, *surveyPort,*pubPort)
+	logger.Debug("heartBeat with db done!",<-initchan)
+
 	r := router.NewRouter()
 	r.Run("0.0.0.0:8000")
 }

--
Gitblit v1.8.0