66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
|
package zerolog
|
||
|
|
|
||
|
|
import "io"
|
||
|
|
|
||
|
|
// LevelWriter defines as interface a writer may implement in order
|
||
|
|
// to receive level information with payload.
|
||
|
|
type LevelWriter interface {
|
||
|
|
io.Writer
|
||
|
|
WriteLevel(level Level, p []byte) (n int, err error)
|
||
|
|
}
|
||
|
|
|
||
|
|
type levelWriterAdapter struct {
|
||
|
|
io.Writer
|
||
|
|
}
|
||
|
|
|
||
|
|
func (lw levelWriterAdapter) WriteLevel(level Level, p []byte) (n int, err error) {
|
||
|
|
return lw.Write(p)
|
||
|
|
}
|
||
|
|
|
||
|
|
type multiLevelWriter struct {
|
||
|
|
writers []LevelWriter
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t multiLevelWriter) Write(p []byte) (n int, err error) {
|
||
|
|
for _, w := range t.writers {
|
||
|
|
n, err = w.Write(p)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if n != len(p) {
|
||
|
|
err = io.ErrShortWrite
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return len(p), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t multiLevelWriter) WriteLevel(l Level, p []byte) (n int, err error) {
|
||
|
|
for _, w := range t.writers {
|
||
|
|
n, err = w.WriteLevel(l, p)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if n != len(p) {
|
||
|
|
err = io.ErrShortWrite
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return len(p), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// MultiLevelWriter creates a writer that duplicates its writes to all the
|
||
|
|
// provided writers, similar to the Unix tee(1) command. If some writers
|
||
|
|
// implement LevelWriter, their WriteLevel method will be used instead of Write.
|
||
|
|
func MultiLevelWriter(writers ...io.Writer) LevelWriter {
|
||
|
|
lwriters := make([]LevelWriter, 0, len(writers))
|
||
|
|
for _, w := range writers {
|
||
|
|
if lw, ok := w.(LevelWriter); ok {
|
||
|
|
lwriters = append(lwriters, lw)
|
||
|
|
} else {
|
||
|
|
lwriters = append(lwriters, levelWriterAdapter{w})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return multiLevelWriter{lwriters}
|
||
|
|
}
|