78 lines
1.7 KiB
Go
78 lines
1.7 KiB
Go
package main
|
||
|
||
import (
|
||
"flag"
|
||
"fmt"
|
||
"jiacrontab/jiacrontabd"
|
||
"jiacrontab/pkg/pprof"
|
||
"jiacrontab/pkg/util"
|
||
"jiacrontab/pkg/version"
|
||
|
||
"os"
|
||
|
||
"github.com/iwannay/log"
|
||
)
|
||
|
||
func parseFlag(opt *jiacrontabd.Config) *flag.FlagSet {
|
||
|
||
var (
|
||
debug bool
|
||
cfgPath string
|
||
logLevel string
|
||
boardcastAddr string
|
||
)
|
||
|
||
flagSet := flag.NewFlagSet("jiacrontabd", flag.ExitOnError)
|
||
flagSet.Bool("version", false, "打印版本信息")
|
||
flagSet.Bool("help", false, "帮助信息")
|
||
flagSet.StringVar(&logLevel, "log_level", "warn", "日志级别(debug|info|warn|error)")
|
||
flagSet.BoolVar(&debug, "debug", false, "开启debug模式")
|
||
flagSet.StringVar(&boardcastAddr, "boardcast_addr", "", fmt.Sprintf("广播地址(default: %s:20001)", util.InternalIP()))
|
||
flagSet.StringVar(&cfgPath, "config", "./jiacrontabd.ini", "配置文件路径")
|
||
flagSet.Parse(os.Args[1:])
|
||
if flagSet.Lookup("version").Value.(flag.Getter).Get().(bool) {
|
||
fmt.Println(version.String("jiacrontab_admin"))
|
||
os.Exit(0)
|
||
}
|
||
if flagSet.Lookup("help").Value.(flag.Getter).Get().(bool) {
|
||
flagSet.Usage()
|
||
os.Exit(0)
|
||
}
|
||
|
||
opt.CfgPath = cfgPath
|
||
opt.Resolve()
|
||
|
||
// TODO： can be better
|
||
if util.HasFlagName(flagSet, "log_level") {
|
||
opt.LogLevel = logLevel
|
||
}
|
||
|
||
if util.HasFlagName(flagSet, "debug") {
|
||
opt.Debug = debug
|
||
}
|
||
|
||
if util.HasFlagName(flagSet, "boardcast_addr") {
|
||
opt.BoardcastAddr = boardcastAddr
|
||
}
|
||
|
||
if debug {
|
||
log.JSON("debug config:", opt)
|
||
}
|
||
|
||
return flagSet
|
||
}
|
||
|
||
func main() {
|
||
cfg := jiacrontabd.NewConfig()
|
||
parseFlag(cfg)
|
||
log.SetLevel(map[string]int{
|
||
"debug": 0,
|
||
"info": 1,
|
||
"warn": 2,
|
||
"error": 3,
|
||
}[cfg.LogLevel])
|
||
pprof.ListenPprof()
|
||
jiad := jiacrontabd.New(cfg)
|
||
jiad.Main()
|
||
}
|