middleware: make Logger argument optional

pull/447/head
Demian 3 years ago
parent 6ecbc6adae
commit 35526d3d53

@ -7,14 +7,18 @@ import (
tele "gopkg.in/tucnak/telebot.v3"
)
func Logger(logger *log.Logger) tele.MiddlewareFunc {
if logger == nil {
logger = log.Default()
func Logger(logger ...*log.Logger) tele.MiddlewareFunc {
var l *log.Logger
if len(logger) > 0 {
l = logger[0]
} else {
l = log.Default()
}
return func(next tele.HandlerFunc) tele.HandlerFunc {
return func(c tele.Context) error {
data, _ := json.MarshalIndent(c.Update(), "", " ")
logger.Println(string(data))
l.Println(string(data))
return next(c)
}
}

Loading…
Cancel
Save