go-zero/core/logx/lesswriter.go
2021-02-20 22:45:58 +08:00

23 lines
396 B
Go

package logx
import "io"
type lessWriter struct {
*limitedExecutor
writer io.Writer
}
func newLessWriter(writer io.Writer, milliseconds int) *lessWriter {
return &lessWriter{
limitedExecutor: newLimitedExecutor(milliseconds),
writer: writer,
}
}
func (w *lessWriter) Write(p []byte) (n int, err error) {
w.logOrDiscard(func() {
w.writer.Write(p)
})
return len(p), nil
}