Merge pull request #1 from ostafen/fix/init

Use default value instead of panicking when level is invalid

logger.go

@@ -26,7 +26,7 @@ const (
 )
 
 func init() {
-	InitLog(os.Getenv("LOG_LEVEL"))
+	InitLog(getLogLevel())
 }
 
 // Logger logger interface
@@ -133,3 +133,22 @@ func FatalfIf(cond bool, fmt string, args ...interface{}) {
 func FatalIfError(err error) {
 	FatalfIf(err != nil, "fatal error: %v", err)
 }
+
+const logLevelDefault = "INFO"
+
+func isValidLevel(level string) bool {
+	switch strings.ToUpper(level) {
+	case "DEBUG", "INFO", "WARN", "ERROR":
+		return true
+	}
+	return false
+}
+
+func getLogLevel() string {
+	level := os.Getenv("LOG_LEVEL")
+	if !isValidLevel(level) {
+		fmt.Fprintf(os.Stderr, "invalid log level: %s, switching to default: %s\n", level, logLevelDefault)
+		level = logLevelDefault
+	}
+	return level
+}