Add an Atoi funtion for loglevel string conversion

Author: crooks

log.go

@@ -1,5 +1,10 @@
 package log
 
+import (
+	"fmt"
+	"strings"
+)
+
 // Fields represents a map of key-value pairs where the value can be any Go
 // type. The value must be able to be converted to a string.
 type Fields map[string]interface{}
@@ -84,6 +89,32 @@ type Logger interface {
 	Fatalw(msg string, fields Fields)
 }
 
+// Atoi returns the loglevel integer associated with a common loglevel
+// string representation.
+func Atoi(loglevelStr string) (level int, err error) {
+	switch strings.ToLower(loglevelStr) {
+	case "trace":
+		level = TraceLevel
+	case "debug":
+		level = DebugLevel
+	case "info":
+		level = InfoLevel
+	case "warn":
+		level = WarnLevel
+	case "warning":
+		level = WarnLevel
+	case "error":
+		level = ErrorLevel
+	case "panic":
+		level = PanicLevel
+	case "fatal":
+		level = FatalLevel
+	default:
+		err = fmt.Errorf("unknown loglevel: %s", loglevelStr)
+	}
+	return
+}
+
 // Current contains the logger used for the package level logging functions
 var Current Logger
 

log_test.go

@@ -154,6 +154,29 @@ func TestInterface(t *testing.T) {
 	buf.Reset()
 }
 
+func TestAtoi(t *testing.T) {
+	names := []string{"Trace", "Debug", "Info", "Warn", "Warning", "Error", "Panic", "Fatal"}
+	levels := []int{TraceLevel, DebugLevel, InfoLevel, WarnLevel, WarnLevel, ErrorLevel, PanicLevel, FatalLevel}
+	if len(names) != len(levels) {
+		t.Fatal("inconsistent number of elements in test")
+	}
+	for n := range names {
+		name := names[n]
+		level := levels[n]
+		testLevel, err := Atoi(name)
+		if err != nil {
+			t.Errorf("name to level translation returned: %v", err)
+		}
+		if testLevel != level {
+			t.Errorf("incorrect loglevel (%d) set by name \"%s\"", level, name)
+		}
+	}
+	_, err := Atoi("fakeName")
+	if err == nil {
+		t.Error("fake loglevel name failed to return an error")
+	}
+}
+
 type testLogger struct {
 	logger *bytes.Buffer
 }