From eb2c3a8890a0e9695c0cd2bc3b3dc310c6fbcfa9 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期六, 26 十月 2019 11:28:23 +0800
Subject: [PATCH] fix user timeout

---
 main.go |   40 ++++++++++++++++++++++++++--------------
 1 files changed, 26 insertions(+), 14 deletions(-)

diff --git a/main.go b/main.go
index 5f48e32..425e46d 100644
--- a/main.go
+++ b/main.go
@@ -1,19 +1,24 @@
 package main
 
 import (
+	"flag"
+	"strconv"
+	"webserver/cache"
+	"webserver/extend/config"
+	"webserver/extend/sys"
+	"webserver/router"
+
 	"basic.com/dbapi.git"
 	"basic.com/valib/logger.git"
-	"flag"
-	"github.com/golang/glog"
 	"github.com/spf13/viper"
-	"strconv"
-	"webserver/extend/config"
-	"webserver/router"
 )
 
 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()
@@ -37,21 +42,28 @@
 		viper.GetInt("LogLevel") >= logger.PanicLevel &&
 		viper.GetInt("LogLevel") <= logger.DebugLevel {
 		logger.Config(logFile, viper.GetInt("LogLevel"))
-	}else{
+	} else {
 		logger.Config(logFile, logger.DebugLevel)
 	}
 	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")
-
 }
+
+// @title Swagger API
+// @version 2.0
+// @description analysis webserver
+// @securityDefinitions.apikey ApiKeyAuth
+// @in header
+// @name Authorization
+// @BasePath /
 func main() {
+	flag.Parse()
+	go cache.Init(initchan, *dbIp, *surveyPort, *pubPort)
+	logger.Debug("heartBeat with db done!", <-initchan)
+
+	// 缁熻绯荤粺杩愯鐘舵��
+	go sys.GatherStat()
+
 	r := router.NewRouter()
 	r.Run("0.0.0.0:8000")
 }

--
Gitblit v1.8.0