From da99edba7d05f757368b760043526e919d7f19ca Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期一, 16 九月 2019 16:14:13 +0800
Subject: [PATCH] config.Init
---
main.go | 44 ++++++++++++++++++++++----------------------
1 files changed, 22 insertions(+), 22 deletions(-)
diff --git a/main.go b/main.go
index 2d8d6da..5f48e32 100644
--- a/main.go
+++ b/main.go
@@ -2,14 +2,13 @@
import (
"basic.com/dbapi.git"
+ "basic.com/valib/logger.git"
"flag"
"github.com/golang/glog"
- "github.com/robfig/cron"
+ "github.com/spf13/viper"
"strconv"
"webserver/extend/config"
- "webserver/extend/logger"
"webserver/router"
- "webserver/service"
)
var envirment = flag.String("e", "dev", "")
@@ -17,23 +16,32 @@
var dbPort = flag.String("dbPort", "8001", "default dbPort=8001")
func init() {
- var logFile = "./logger/webserver.log"
- var logSaveDays = 15
-
- // 鏃ュ織鍒濆鍖�
- logger.Config(logFile, logger.DebugLevel)
- logger.SetSaveDays(logSaveDays)
- logger.Info("loginit success !")
flag.Parse()
- //service.InitService()
p, err := strconv.Atoi(*dbPort)
if err != nil {
p = 8001
}
- //*dbIp = "192.168.1.123"
- //*dbIp = "192.168.1.182"
dbapi.Init(*dbIp, p)
+ config.Init(*envirment)
+
+ var logFile = "./logger/"
+ if viper.GetString("LogBasePath") != "" {
+ logFile = viper.GetString("LogBasePath")
+ }
+ logFile = logFile + "/webserver.log"
+ var logSaveDays = 15
+
+ // 鏃ュ織鍒濆鍖�
+ if viper.IsSet("LogLevel") &&
+ viper.GetInt("LogLevel") >= logger.PanicLevel &&
+ viper.GetInt("LogLevel") <= logger.DebugLevel {
+ logger.Config(logFile, viper.GetInt("LogLevel"))
+ }else{
+ logger.Config(logFile, logger.DebugLevel)
+ }
+ logger.SetSaveDays(logSaveDays)
+ logger.Info("loginit success !")
flag.Usage = func() {
glog.Exit("flag parse usage !")
@@ -41,17 +49,9 @@
flag.Lookup("alsologtostderr").Value.Set("true")
flag.Lookup("log_dir").Value.Set("./log")
- config.Init(*envirment)
+
}
func main() {
- c := cron.New()
- c.AddFunc("@ daily", func() {
- res := service.InitEnableStatus()
- logger.Debug(res)
- })
- c.Start()
r := router.NewRouter()
r.Run("0.0.0.0:8000")
-
- //defer new(gorun.SimpleFaceDetect).FaceDetactClose()
}
--
Gitblit v1.8.0