new logger

Author: xjasonlyu

common/log/log.go

@@ -0,0 +1,28 @@
+package log
+
+const (
+	DEBUG Level = iota
+	INFO
+	WARNING
+	ERROR
+	SILENT
+)
+
+type Level int
+
+func (l Level) String() string {
+	switch l {
+	case INFO:
+		return "info"
+	case WARNING:
+		return "warning"
+	case ERROR:
+		return "error"
+	case DEBUG:
+		return "debug"
+	case SILENT:
+		return "silent"
+	default:
+		return "unknown"
+	}
+}

common/log/logger.go

@@ -0,0 +1,78 @@
+package log
+
+import (
+	"fmt"
+	"os"
+
+	log "github.com/sirupsen/logrus"
+)
+
+var (
+	level = INFO
+)
+
+func init() {
+	log.SetOutput(os.Stdout)
+	log.SetLevel(log.DebugLevel)
+}
+
+type Event struct {
+	LogLevel Level
+	Payload  string
+}
+
+func (e *Event) Type() string {
+	return e.LogLevel.String()
+}
+
+func Infof(format string, v ...interface{}) {
+	event := newLog(INFO, format, v...)
+	printf(event)
+}
+
+func Warnf(format string, v ...interface{}) {
+	event := newLog(WARNING, format, v...)
+	printf(event)
+}
+
+func Errorf(format string, v ...interface{}) {
+	event := newLog(ERROR, format, v...)
+	printf(event)
+}
+
+func Debugf(format string, v ...interface{}) {
+	event := newLog(DEBUG, format, v...)
+	printf(event)
+}
+
+func Fatalf(format string, v ...interface{}) {
+	log.Fatalf(format, v...)
+}
+
+func SetLevel(newLevel Level) {
+	level = newLevel
+}
+
+func printf(data *Event) {
+	if data.LogLevel < level {
+		return
+	}
+
+	switch data.LogLevel {
+	case INFO:
+		log.Infoln(data.Payload)
+	case WARNING:
+		log.Warnln(data.Payload)
+	case ERROR:
+		log.Errorln(data.Payload)
+	case DEBUG:
+		log.Debugln(data.Payload)
+	}
+}
+
+func newLog(logLevel Level, format string, v ...interface{}) *Event {
+	return &Event{
+		LogLevel: logLevel,
+		Payload:  fmt.Sprintf(format, v...),
+	}
+}