package log import ( "bufio" "io" "runtime" "github.com/sirupsen/logrus" ) // Debug logs a message at level Debug on the standard logger. // Deprecated func Debug(args ...interface{}) { mainLogger.Debug(args...) } // Debugf logs a message at level Debug on the standard logger. // Deprecated func Debugf(format string, args ...interface{}) { mainLogger.Debugf(format, args...) } // Info logs a message at level Info on the standard logger. // Deprecated func Info(args ...interface{}) { mainLogger.Info(args...) } // Infof logs a message at level Info on the standard logger. // Deprecated func Infof(format string, args ...interface{}) { mainLogger.Infof(format, args...) } // Warn logs a message at level Warn on the standard logger. // Deprecated func Warn(args ...interface{}) { mainLogger.Warn(args...) } // Warnf logs a message at level Warn on the standard logger. // Deprecated func Warnf(format string, args ...interface{}) { mainLogger.Warnf(format, args...) } // Error logs a message at level Error on the standard logger. // Deprecated func Error(args ...interface{}) { mainLogger.Error(args...) } // Errorf logs a message at level Error on the standard logger. // Deprecated func Errorf(format string, args ...interface{}) { mainLogger.Errorf(format, args...) } // Panic logs a message at level Panic on the standard logger. // Deprecated func Panic(args ...interface{}) { mainLogger.Panic(args...) } // Fatal logs a message at level Fatal on the standard logger. // Deprecated func Fatal(args ...interface{}) { mainLogger.Fatal(args...) } // Fatalf logs a message at level Fatal on the standard logger. // Deprecated func Fatalf(format string, args ...interface{}) { mainLogger.Fatalf(format, args...) } // AddHook adds a hook to the standard logger hooks. func AddHook(hook logrus.Hook) { logrus.AddHook(hook) } // CustomWriterLevel logs writer for a specific level. (with a custom scanner buffer size.) // adapted from github.com/Sirupsen/logrus/writer.go func CustomWriterLevel(level logrus.Level, maxScanTokenSize int) *io.PipeWriter { reader, writer := io.Pipe() var printFunc func(args ...interface{}) switch level { case logrus.DebugLevel: printFunc = Debug case logrus.InfoLevel: printFunc = Info case logrus.WarnLevel: printFunc = Warn case logrus.ErrorLevel: printFunc = Error case logrus.FatalLevel: printFunc = Fatal case logrus.PanicLevel: printFunc = Panic default: printFunc = mainLogger.Print } go writerScanner(reader, maxScanTokenSize, printFunc) runtime.SetFinalizer(writer, writerFinalizer) return writer } // extract from github.com/Sirupsen/logrus/writer.go // Hack the buffer size func writerScanner(reader io.ReadCloser, scanTokenSize int, printFunc func(args ...interface{})) { scanner := bufio.NewScanner(reader) if scanTokenSize > bufio.MaxScanTokenSize { buf := make([]byte, bufio.MaxScanTokenSize) scanner.Buffer(buf, scanTokenSize) } for scanner.Scan() { printFunc(scanner.Text()) } if err := scanner.Err(); err != nil { Errorf("Error while reading from Writer: %s", err) } reader.Close() } func writerFinalizer(writer *io.PipeWriter) { writer.Close() }