42 lines
745 B
Go
42 lines
745 B
Go
|
package kafka
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"github.com/IBM/sarama"
|
||
|
|
|
||
|
|
"github.com/influxdata/telegraf"
|
||
|
|
"github.com/influxdata/telegraf/logger"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
log = logger.New("sarama", "", "")
|
||
|
|
once sync.Once
|
||
|
|
)
|
||
|
|
|
||
|
|
type debugLogger struct{}
|
||
|
|
|
||
|
|
func (*debugLogger) Print(v ...interface{}) {
|
||
|
|
log.Trace(v...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*debugLogger) Printf(format string, v ...interface{}) {
|
||
|
|
log.Tracef(format, v...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *debugLogger) Println(v ...interface{}) {
|
||
|
|
l.Print(v...)
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetLogger configures a debug logger for kafka (sarama)
|
||
|
|
func SetLogger(level telegraf.LogLevel) {
|
||
|
|
// Set-up the sarama logger only once
|
||
|
|
once.Do(func() {
|
||
|
|
sarama.Logger = &debugLogger{}
|
||
|
|
})
|
||
|
|
// Increase the log-level if needed.
|
||
|
|
if !log.Level().Includes(level) {
|
||
|
|
log.SetLevel(level)
|
||
|
|
}
|
||
|
|
}
|