podman

1
package logrus
2

3
import (
4
	"bufio"
5
	"io"
6
	"runtime"
7
)
8

9
// Writer at INFO level. See WriterLevel for details.
10
func (logger *Logger) Writer() *io.PipeWriter {
11
	return logger.WriterLevel(InfoLevel)
12
}
13

14
// WriterLevel returns an io.Writer that can be used to write arbitrary text to
15
// the logger at the given log level. Each line written to the writer will be
16
// printed in the usual way using formatters and hooks. The writer is part of an
17
// io.Pipe and it is the callers responsibility to close the writer when done.
18
// This can be used to override the standard library logger easily.
19
func (logger *Logger) WriterLevel(level Level) *io.PipeWriter {
20
	return NewEntry(logger).WriterLevel(level)
21
}
22

23
func (entry *Entry) Writer() *io.PipeWriter {
24
	return entry.WriterLevel(InfoLevel)
25
}
26

27
func (entry *Entry) WriterLevel(level Level) *io.PipeWriter {
28
	reader, writer := io.Pipe()
29

30
	var printFunc func(args ...interface{})
31

32
	switch level {
33
	case TraceLevel:
34
		printFunc = entry.Trace
35
	case DebugLevel:
36
		printFunc = entry.Debug
37
	case InfoLevel:
38
		printFunc = entry.Info
39
	case WarnLevel:
40
		printFunc = entry.Warn
41
	case ErrorLevel:
42
		printFunc = entry.Error
43
	case FatalLevel:
44
		printFunc = entry.Fatal
45
	case PanicLevel:
46
		printFunc = entry.Panic
47
	default:
48
		printFunc = entry.Print
49
	}
50

51
	go entry.writerScanner(reader, printFunc)
52
	runtime.SetFinalizer(writer, writerFinalizer)
53

54
	return writer
55
}
56

57
func (entry *Entry) writerScanner(reader *io.PipeReader, printFunc func(args ...interface{})) {
58
	scanner := bufio.NewScanner(reader)
59
	for scanner.Scan() {
60
		printFunc(scanner.Text())
61
	}
62
	if err := scanner.Err(); err != nil {
63
		entry.Errorf("Error while reading from Writer: %s", err)
64
	}
65
	reader.Close()
66
}
67

68
func writerFinalizer(writer *io.PipeWriter) {
69
	writer.Close()
70
}
71