forked from gsn/predictor
30 lines
695 B
Go
30 lines
695 B
Go
package middleware
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/ogen-go/ogen/middleware"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// Logging returns an ogen middleware that logs request duration.
|
|
func Logging(log *zap.Logger) middleware.Middleware {
|
|
return func(req middleware.Request, next func(req middleware.Request) (middleware.Response, error)) (middleware.Response, error) {
|
|
lg := log.With(zap.String("operation", req.OperationID))
|
|
|
|
start := time.Now()
|
|
resp, err := next(req)
|
|
dur := time.Since(start)
|
|
|
|
if err != nil {
|
|
lg.Error("request failed",
|
|
zap.Duration("duration", dur),
|
|
zap.Error(err))
|
|
} else {
|
|
lg.Info("request completed",
|
|
zap.Duration("duration", dur))
|
|
}
|
|
|
|
return resp, err
|
|
}
|
|
}
|