5376b9e271
* New config files
As decided, config files are parsed at runtime.
I decided to go for YAML config files because there can be comments in
it.
There are 2 files:
* config/default_config.yml <= which shouldn't be edited unless we add a
config parameter
* config/config.yml <= which is the user-defined config. This file
shouldn't be commited
Changed every call to config.XXX to config.Conf.XXX (look to the new
stucture of config in config/types.go)
Of course, putting config parameters in config.yml overrides config in
config_default.yml. You don't have to put everything in it, just add
what you want to override.
* Fixing test
Replacing conf.New by config.Conf
* Fixing call to config.Conf to config.Config{} in test files
* Might have fixed testing with this
Printf instead of Fatalf
* Renaming config.yml in example file
* Forbid commiting config.yml
* Should be now fixed
* Do not need this file anymore
137 lignes
3,2 Kio
Go
137 lignes
3,2 Kio
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
|
|
}
|