kaspad/stability-tests/common/log-writer.go
Svarog a7299c1b87
Add stability tests (#1587)
* Add stability-tests

* Fix requires

* Fix golint errors

* Update README.md

* Remove payloadHash from everywhere

* don't run vet on kaspad in stability-tests/install_and_test
2021-03-09 15:01:08 +02:00

30 lines
726 B
Go

package common
import (
"strings"
"github.com/kaspanet/kaspad/infrastructure/logger"
)
// LogWriter writes to the given log with the given log level and prefix
type LogWriter struct {
log *logger.Logger
level logger.Level
prefix string
}
func (clw LogWriter) Write(p []byte) (n int, err error) {
logWithoutNewLine := strings.TrimSuffix(string(p), "\n")
clw.log.Writef(clw.level, "%s: %s", clw.prefix, logWithoutNewLine)
return len(p), nil
}
// NewLogWriter returns a new LogWriter that forwards to `log` all data written to it using at `level` level
func NewLogWriter(log *logger.Logger, level logger.Level, prefix string) LogWriter {
return LogWriter{
log: log,
level: level,
prefix: prefix,
}
}