mirror of
https://github.com/empayre/fleet.git
synced 2024-11-07 01:15:22 +00:00
d827a9d6b7
Modify the loggerDebug function to check whether the error is nil before choosing the log level. Fixes #473
32 lines
874 B
Go
32 lines
874 B
Go
package service
|
|
|
|
import (
|
|
"github.com/fleetdm/fleet/server/kolide"
|
|
kitlog "github.com/go-kit/kit/log"
|
|
"github.com/go-kit/kit/log/level"
|
|
)
|
|
|
|
// logging middleware logs the service actions
|
|
type loggingMiddleware struct {
|
|
kolide.Service
|
|
logger kitlog.Logger
|
|
}
|
|
|
|
// NewLoggingService takes an existing service and adds a logging wrapper
|
|
func NewLoggingService(svc kolide.Service, logger kitlog.Logger) kolide.Service {
|
|
return loggingMiddleware{Service: svc, logger: logger}
|
|
}
|
|
|
|
// loggerDebug returns the the info level if there error is non-nil, otherwise defaulting to the debug level.
|
|
func (mw loggingMiddleware) loggerDebug(err error) kitlog.Logger {
|
|
if err != nil {
|
|
return level.Info(mw.logger)
|
|
}
|
|
return level.Debug(mw.logger)
|
|
}
|
|
|
|
// loggerInfo returns the info level
|
|
func (mw loggingMiddleware) loggerInfo(err error) kitlog.Logger {
|
|
return level.Info(mw.logger)
|
|
}
|