138 lignes
3.2 KiB
Go
138 lignes
3.2 KiB
Go
package log
|
|
|
|
import (
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/NyaaPantsu/nyaa/config"
|
|
"github.com/Sirupsen/logrus"
|
|
lumberjack "gopkg.in/natefinch/lumberjack.v2"
|
|
)
|
|
|
|
// LumberJackLogger : Initialize logger
|
|
func LumberJackLogger(filePath string, maxSize int, maxBackups int, maxAge int) *lumberjack.Logger {
|
|
return &lumberjack.Logger{
|
|
Filename: filePath,
|
|
MaxSize: maxSize, // megabytes
|
|
MaxBackups: maxBackups,
|
|
MaxAge: maxAge, //days
|
|
}
|
|
}
|
|
|
|
// InitLogToStdoutDebug : set logrus to debug param
|
|
func InitLogToStdoutDebug() {
|
|
logrus.SetFormatter(&logrus.TextFormatter{ForceColors: true})
|
|
logrus.SetOutput(os.Stdout)
|
|
logrus.SetLevel(logrus.DebugLevel)
|
|
}
|
|
|
|
// InitLogToStdout : set logrus to stdout
|
|
func InitLogToStdout() {
|
|
logrus.SetFormatter(&logrus.TextFormatter{})
|
|
logrus.SetOutput(os.Stdout)
|
|
logrus.SetLevel(logrus.WarnLevel)
|
|
}
|
|
|
|
// InitLogToFile : set logrus to output in file
|
|
func InitLogToFile() {
|
|
logrus.SetFormatter(&logrus.JSONFormatter{})
|
|
|
|
out := LumberJackLogger(config.Conf.Log.ErrorLogFilePath+config.Conf.Log.ErrorLogFileExtension, config.Conf.Log.ErrorLogMaxSize, config.Conf.Log.ErrorLogMaxBackups, config.Conf.Log.ErrorLogMaxAge)
|
|
|
|
logrus.SetOutput(out)
|
|
logrus.SetLevel(logrus.WarnLevel)
|
|
}
|
|
|
|
// Init logrus
|
|
func Init(environment string) {
|
|
switch environment {
|
|
case "DEVELOPMENT":
|
|
InitLogToStdoutDebug()
|
|
case "TEST":
|
|
InitLogToFile()
|
|
case "PRODUCTION":
|
|
InitLogToFile()
|
|
}
|
|
logrus.Debugf("Environment : %s", environment)
|
|
}
|
|
|
|
// Debug logs a message with debug log level.
|
|
func Debug(msg string) {
|
|
logrus.Debug(msg)
|
|
}
|
|
|
|
// Debugf logs a formatted message with debug log level.
|
|
func Debugf(msg string, args ...interface{}) {
|
|
logrus.Debugf(msg, args...)
|
|
}
|
|
|
|
// Info logs a message with info log level.
|
|
func Info(msg string) {
|
|
logrus.Info(msg)
|
|
}
|
|
|
|
// Infof logs a formatted message with info log level.
|
|
func Infof(msg string, args ...interface{}) {
|
|
logrus.Infof(msg, args...)
|
|
}
|
|
|
|
// Warn logs a message with warn log level.
|
|
func Warn(msg string) {
|
|
logrus.Warn(msg)
|
|
}
|
|
|
|
// Warnf logs a formatted message with warn log level.
|
|
func Warnf(msg string, args ...interface{}) {
|
|
logrus.Warnf(msg, args...)
|
|
}
|
|
|
|
// Error logs a message with error log level.
|
|
func Error(msg string) {
|
|
logrus.Error(msg)
|
|
}
|
|
|
|
// Errorf logs a formatted message with error log level.
|
|
func Errorf(msg string, args ...interface{}) {
|
|
logrus.Errorf(msg, args...)
|
|
}
|
|
|
|
// Fatal logs a message with fatal log level.
|
|
func Fatal(msg string) {
|
|
logrus.Fatal(msg)
|
|
}
|
|
|
|
// Fatalf logs a formatted message with fatal log level.
|
|
func Fatalf(msg string, args ...interface{}) {
|
|
logrus.Fatalf(msg, args...)
|
|
}
|
|
|
|
// Panic logs a message with panic log level.
|
|
func Panic(msg string) {
|
|
logrus.Panic(msg)
|
|
}
|
|
|
|
// Panicf logs a formatted message with panic log level.
|
|
func Panicf(msg string, args ...interface{}) {
|
|
logrus.Panicf(msg, args...)
|
|
}
|
|
|
|
// DebugResponse : log response body data for debugging
|
|
func DebugResponse(response *http.Response) string {
|
|
bodyBuffer := make([]byte, 5000)
|
|
var str string
|
|
count, err := response.Body.Read(bodyBuffer)
|
|
if err != nil {
|
|
Debug(err.Error())
|
|
return ""
|
|
}
|
|
for ; count > 0; count, err = response.Body.Read(bodyBuffer) {
|
|
if err != nil {
|
|
Debug(err.Error())
|
|
continue
|
|
}
|
|
str += string(bodyBuffer[:count])
|
|
}
|
|
Debugf("response data : %v", str)
|
|
return str
|
|
}
|