logger.go

package artnet

import (
	"os"

	"github.com/sirupsen/logrus"
)

// Fields are a representation of formatted log fields
type Fields map[string]interface{}

// Logger is the interface for a logger
type Logger interface {
	logrus.StdLogger
	With(fields Fields) *logger
}

type logger struct {
	*logrus.Entry
}

// NewLogger returns a Logger based on logrus
func NewDefaultLogger() Logger {
	log := &struct {
		*logrus.Logger
	}{
		Logger: logrus.New(),
	}

	customFormatter := new(logrus.TextFormatter)
	customFormatter.TimestampFormat = "2006-01-02 15:04:05.0000"
	customFormatter.DisableColors = true
	customFormatter.FullTimestamp = true

	log.Formatter = customFormatter

	// Output to stdout instead of the default stderr
	// Can be any io.Writer, see below for File example
	log.Out = os.Stdout

	// Only log the debug severity or above.
	log.Level = logrus.DebugLevel

	// Disable concurrency mutex as we use Stdout
	log.SetNoLock()
	return &logger{Entry: log.WithFields(nil)}
}

// NewLogger creates a new logger from given logrus logger
func NewLogger(log *logrus.Entry) Logger {
	return &logger{log}
}

// With will add the fields to the formatted log entry
func (l *logger) With(fields Fields) *logger {
	return &logger{Entry: l.WithFields(logrus.Fields(fields))}
}