pkg/logging/logger.go

// Copyright (c) Facebook, Inc. and its affiliates.
//
// This source code is licensed under the MIT license found in the
// LICENSE file in the root directory of this source tree.

package logging

import (
	"io/ioutil"

	log_prefixed "github.com/chappjc/logrus-prefix"
	"github.com/sirupsen/logrus"
)

var (
	log *logrus.Logger
)

// GetLogger returns a configured logger instance
func GetLogger(prefix string) *logrus.Entry {
	return log.WithField("prefix", prefix)
}

// AddField add a field to an existing logrus.Entry
func AddField(e *logrus.Entry, name string, value interface{}) *logrus.Entry {
	l := log.WithField(name, value)
	for k, v := range e.Data {
		l = l.WithField(k, v)
	}
	return l
}

// AddFields adds multiple fields to an existing logrus.Entry
func AddFields(e *logrus.Entry, fields map[string]interface{}) *logrus.Entry {
	l := e
	for k, v := range fields {
		l = AddField(l, k, v)
	}
	return l
}

// Disable sends all logging output to the bit bucket.
func Disable() {
	log.SetOutput(ioutil.Discard)
}

// Trace - Set Log Level to Trace
func Trace() {
	log.SetLevel(logrus.TraceLevel)
}

// Debug - Set Log Level to Debug
func Debug() {
	log.SetLevel(logrus.DebugLevel)
}

// Info - Set Log Level to Info
func Info() {
	log.SetLevel(logrus.InfoLevel)
}

// Warn - Set Log Level to Warn
func Warn() {
	log.SetLevel(logrus.WarnLevel)
}

// Error - Set Log Level to Error
func Error() {
	log.SetLevel(logrus.ErrorLevel)
}

// Fatal - Set Log Level to Fatal
func Fatal() {
	log.SetLevel(logrus.FatalLevel)
}

// Panic - Set Log Level to Panic
func Panic() {
	log.SetLevel(logrus.PanicLevel)
}

func init() {
	log = logrus.New()
	log.SetFormatter(&log_prefixed.TextFormatter{
		FullTimestamp: true,
	})
}