debug.go

package goku_log

import (
	"fmt"
	"github.com/sirupsen/logrus"
	"os"
)

func StartDebug() {

	logger.AddHook(new(debugHook))
}

type debugHook struct {
}

func (h *debugHook) Levels() []logrus.Level {
	return []Level{
		TraceLevel,
		DebugLevel,
		InfoLevel,
		WarnLevel,
		ErrorLevel,
		FatalLevel,
		PanicLevel,
	}
}

func (h *debugHook) Fire(entry *logrus.Entry) error {
	s, e := logger.Formatter.Format(entry)
	if e != nil {
		fmt.Println(entry)
		return nil
	}
	os.Stdout.Write(s)
	return nil
}