feat: support to set log level in config file

This commit is contained in:
mzz2017
2023-02-05 14:03:34 +08:00
parent e2ecb80c26
commit e097acaa8e
4 changed files with 19 additions and 21 deletions

View File

@ -25,9 +25,16 @@ var (
if cfgFile == "" {
logrus.Fatalln("Argument \"--config\" or \"-c\" is required but not provided.")
}
logrus.SetLevel(logrus.DebugLevel)
log := logger.NewLogger(2, disableTimestamp)
if err := Run(log); err != nil {
// Read config from --config cfgFile.
param, err := readConfig(cfgFile)
if err != nil {
logrus.Fatalln("readConfig: %w", err)
}
log := logger.NewLogger(param.Global.LogLevel, disableTimestamp)
logrus.SetLevel(log.Level)
if err := Run(log, param); err != nil {
logrus.Fatalln(err)
}
},
@ -39,17 +46,11 @@ func init() {
runCmd.PersistentFlags().BoolVarP(&disableTimestamp, "disable-timestamp", "", false, "disable timestamp")
}
func Run(log *logrus.Logger) (err error) {
func Run(log *logrus.Logger, param *config.Params) (err error) {
// Require "sudo" if necessary.
internal.AutoSu()
// Read config from --config cfgFile.
param, err := readConfig(cfgFile)
if err != nil {
return fmt.Errorf("readConfig: %w", err)
}
// Resolve subscriptions to nodes.
nodeList := make([]string, len(param.Node))
copy(nodeList, param.Node)