Feat: Add some parameters (naiba#294)

cmd/dashboard/main.go

@@ -2,20 +2,38 @@ package main
 
 import (
 	"context"
+	"fmt"
 	"log"
 
 	"github.com/naiba/nezha/cmd/dashboard/controller"
 	"github.com/naiba/nezha/cmd/dashboard/rpc"
 	"github.com/naiba/nezha/model"
 	"github.com/naiba/nezha/service/singleton"
 	"github.com/ory/graceful"
+	flag "github.com/spf13/pflag"
+)
+
+type DashboardCliParam struct {
+	Version          bool   // 当前版本号
+	ConfigFile       string // 配置文件路径
+	DatebaseLocation string // Sqlite3 数据库文件路径
+}
+
+var (
+	dashboardCliParam DashboardCliParam
 )
 
 func init() {
+	flag.CommandLine.ParseErrorsWhitelist.UnknownFlags = true
+	flag.BoolVarP(&dashboardCliParam.Version, "version", "v", false, "查看当前版本号")
+	flag.StringVarP(&dashboardCliParam.ConfigFile, "config", "c", "data/config.yaml", "配置文件路径")
+	flag.StringVar(&dashboardCliParam.DatebaseLocation, "db", "data/sqlite.db", "Sqlite3数据库文件路径")
+	flag.Parse()
+
 	// 初始化 dao 包
-	singleton.InitConfigFromPath("data/config.yaml")
+	singleton.InitConfigFromPath(dashboardCliParam.ConfigFile)
 	singleton.InitTimezoneAndCache()
-	singleton.InitDBFromPath("data/sqlite.db")
+	singleton.InitDBFromPath(dashboardCliParam.DatebaseLocation)
 	singleton.InitLocalizer()
 	initSystem()
 }
@@ -36,6 +54,11 @@ func initSystem() {
 }
 
 func main() {
+	if dashboardCliParam.Version {
+		fmt.Println(singleton.Version)
+		return
+	}
+
 	singleton.CleanMonitorHistory()
 	go rpc.ServeRPC(singleton.Conf.GRPCPort)
 	serviceSentinelDispatchBus := make(chan model.Monitor) // 用于传递服务监控任务信息的channel
@@ -44,13 +67,15 @@ func main() {
 	go singleton.AlertSentinelStart()
 	singleton.NewServiceSentinel(serviceSentinelDispatchBus)
 	srv := controller.ServeWeb(singleton.Conf.HTTPPort)
-	graceful.Graceful(func() error {
+	if err := graceful.Graceful(func() error {
 		return srv.ListenAndServe()
 	}, func(c context.Context) error {
 		log.Println("NEZHA>> Graceful::START")
 		singleton.RecordTransferHourlyUsage()
 		log.Println("NEZHA>> Graceful::END")
 		srv.Shutdown(c)
 		return nil
-	})
+	}); err != nil {
+		log.Printf("NEZHA>> ERROR: %v", err)
+	}
 }