Merge pull request #39 from milobella/fix/config

fix: parsing of log level + configname
milobella · Oct 5, 2022 · 3dd5d6d · 3dd5d6d
2 parents c169f7d + bf8a9eb
commit 3dd5d6d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/pkg/config/config.go b/pkg/config/config.go
@@ -21,8 +21,8 @@ func (c Config) String() string {
 }
 
 type Server struct {
-	Port     int          `mapstructure:"port"`
-	LogLevel logrus.Level `mapstructure:"log_level"`
+	Port     int    `mapstructure:"port"`
+	LogLevel string `mapstructure:"log_level"`
 }
 
 type Tool struct {

diff --git a/pkg/config/viper.go b/pkg/config/viper.go
@@ -9,7 +9,7 @@ import (
 	"github.com/spf13/viper"
 )
 
-const configName = "server"
+const configName = "ability"
 
 func Read(extensions ...interface{}) *Config {
 	e := enviper.New(viper.New())
@@ -39,7 +39,15 @@ func Read(extensions ...interface{}) *Config {
 		}
 	}
 
-	logrus.SetLevel(config.Server.LogLevel)
+	var level logrus.Level
+	level, err = logrus.ParseLevel(config.Server.LogLevel)
+	if err != nil {
+		logrus.SetLevel(logrus.InfoLevel)
+		logrus.WithError(err).Errorf("Error occured while parsing log level. Default to INFO")
+	} else {
+		logrus.SetLevel(level)
+	}
+
 	return &config
 }