37 lines
661 B
Go
37 lines
661 B
Go
|
package logging
|
||
|
|
|
||
|
|
import (
|
||
|
|
"go.uber.org/zap/zapcore"
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
type (
|
||
|
|
// hookable logger container
|
||
|
|
loggerWithHookImpl struct {
|
||
|
|
loggerImpl
|
||
|
|
hookerLock sync.RWMutex
|
||
|
|
hooker LoggerHook
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func (l *loggerWithHookImpl) hook(entry zapcore.Entry) (err error) {
|
||
|
|
l.hookerLock.RLock()
|
||
|
|
defer l.hookerLock.RUnlock()
|
||
|
|
if l.hooker != nil {
|
||
|
|
err = l.hooker(entry.Level, entry.LoggerName, entry.Message, entry.Time)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *loggerWithHookImpl) SetHook(hook LoggerHook) (err error) {
|
||
|
|
l.hookerLock.Lock()
|
||
|
|
defer l.hookerLock.Unlock()
|
||
|
|
l.hooker = hook
|
||
|
|
if hook == nil {
|
||
|
|
l.Info("log hook cleared")
|
||
|
|
} else {
|
||
|
|
l.Info("log hook set")
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|