logger.go

package main

import (
	"github.com/fatih/color"
	"io"
	"log"
	"os"
	"sync"
)

var logger = NewLogger(os.Stdout /*ioutil.Discard*/, os.Stdout, os.Stdout, os.Stderr)

func NewLogger(traceHandle, infoHandle, warningHandle, errorHandle io.Writer) *Logger {
	return &Logger{
		colorTraceSprintfFunc:   color.New().SprintfFunc(),
		colorInfoSprintfFunc:    color.New(color.FgHiGreen).SprintfFunc(),
		colorWarningSprintfFunc: color.New(color.FgHiYellow).SprintfFunc(),
		colorErrSprintfFunc:     color.New(color.FgHiRed).SprintfFunc(),

		trace:   log.New(traceHandle, "[T] ", log.Ldate|log.Ltime /*|log.Lshortfile*/),
		info:    log.New(infoHandle, "[I] ", log.Ldate|log.Ltime /*|log.Lshortfile*/),
		warning: log.New(warningHandle, "[W] ", log.Ldate|log.Ltime /*|log.Lshortfile*/),
		err:     log.New(errorHandle, "[E] ", log.Ldate|log.Ltime /*|log.Lshortfile*/),
	}
}

type Logger struct {
	sync.RWMutex

	colorTraceSprintfFunc   func(format string, args ...interface{}) string
	colorInfoSprintfFunc    func(format string, args ...interface{}) string
	colorWarningSprintfFunc func(format string, args ...interface{}) string
	colorErrSprintfFunc     func(format string, args ...interface{}) string

	trace   *log.Logger
	info    *log.Logger
	warning *log.Logger
	err     *log.Logger
}

func (l *Logger) Tracelnf(format string, args ...interface{}) {
	l.Lock()
	defer l.Unlock()
	l.trace.Println(l.colorTraceSprintfFunc(format, args...))
}
func (l *Logger) Infolnf(format string, args ...interface{}) {
	l.Lock()
	defer l.Unlock()
	l.info.Println(l.colorInfoSprintfFunc(format, args...))
}
func (l *Logger) Warninglnf(format string, args ...interface{}) {
	l.Lock()
	defer l.Unlock()
	l.warning.Println(l.colorWarningSprintfFunc(format, args...))
}
func (l *Logger) Errorlnf(format string, args ...interface{}) {
	l.Lock()
	defer l.Unlock()
	l.err.Println(l.colorErrSprintfFunc(format, args...))
}
func (l *Logger) Fatallnf(format string, args ...interface{}) {
	l.Lock()
	defer l.Unlock()
	l.err.Fatalln(l.colorErrSprintfFunc(format, args...))
}