package main

import (
	"flag"
	"strconv"
	"webserver/cache"
	"webserver/extend/config"
	"webserver/extend/sys"
	"webserver/router"

	"basic.com/dbapi.git"
	"basic.com/valib/logger.git"
	"github.com/spf13/viper"
)

var envirment = flag.String("e", "pro", "")
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()

	p, err := strconv.Atoi(*dbPort)
	if err != nil {
		p = 8001
	}
	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 !")
}

// @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")
}