You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
570 B
32 lines
570 B
9 years ago
|
package logrus
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"io"
|
||
|
|
"runtime"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (logger *Logger) Writer() *io.PipeWriter {
|
||
|
|
reader, writer := io.Pipe()
|
||
|
|
|
||
|
|
go logger.writerScanner(reader)
|
||
|
|
runtime.SetFinalizer(writer, writerFinalizer)
|
||
|
|
|
||
|
|
return writer
|
||
|
|
}
|
||
|
|
|
||
|
|
func (logger *Logger) writerScanner(reader *io.PipeReader) {
|
||
|
|
scanner := bufio.NewScanner(reader)
|
||
|
|
for scanner.Scan() {
|
||
|
|
logger.Print(scanner.Text())
|
||
|
|
}
|
||
|
|
if err := scanner.Err(); err != nil {
|
||
|
|
logger.Errorf("Error while reading from Writer: %s", err)
|
||
|
|
}
|
||
|
|
reader.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
func writerFinalizer(writer *io.PipeWriter) {
|
||
|
|
writer.Close()
|
||
|
|
}
|