add level type function

Author: mobus

log/log.go

@@ -15,7 +15,6 @@
 package log
 
 import (
-	"fmt"
 	stdlog "log"
 	"os"
 )
@@ -45,13 +44,13 @@ func (l Level) String() string {
 	default:
 		return ""
 	case LevelDebug:
-		return "[DBG]"
+		return "DBG"
 	case LevelError:
-		return "[ERR]"
+		return "ERR"
 	case LevelInfo:
-		return "[INF]"
+		return "INF"
 	case LevelFatal:
-		return "[FTL]"
+		return "FTL"
 	}
 }
 
@@ -62,7 +61,7 @@ var (
 	// default log level is info
 	level = LevelInfo
 
-	prefix = "[Slog]"
+	prefix = "Slog"
 )
 
 type defaultLogLogger struct{}
@@ -76,22 +75,24 @@ func (t *defaultLogLogger) Logf(format string, v ...interface{}) {
 }
 
 func init() {
-	switch os.Getenv("SLOG_LOG_LEVEL") {
+	level = LevelType(os.Getenv("SLOG_LOG_LEVEL"))
+}
+
+func LevelType(level string) Level {
+	switch level {
+	default:
+		return LevelInfo
 	case "debug":
-		level = LevelDebug
+		return LevelDebug
 	case "info":
-		level = LevelInfo
+		return LevelInfo
 	case "error":
-		level = LevelError
+		return LevelError
 	case "fatal":
-		level = LevelFatal
+		return LevelFatal
 	}
 }
 
-func SetLogPrefix(logName string) {
-	prefix = logName
-}
-
 // Log makes use of Logger
 func Log(v ...interface{}) {
 	if len(prefix) > 0 {
@@ -191,8 +192,3 @@ func GetLevel() Level {
 func SetPrefix(p string) {
 	prefix = p
 }
-
-// Name sets service name
-func Name(name string) {
-	prefix = fmt.Sprintf("[%s]", name)
-}