feat: polish & windviz & deploy
This commit is contained in:
parent
81b8e763bd
commit
465ad00f7b
78 changed files with 20622 additions and 2154 deletions
|
|
@ -1,51 +1,34 @@
|
|||
// Package middleware contains HTTP and ogen middleware used by the API layer.
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"github.com/ogen-go/ogen/middleware"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
// OgenLogging is an ogen middleware that logs request duration and outcome.
|
||||
// statusCoder is implemented by ogen's *...StatusCode error wrappers.
|
||||
type statusCoder interface{ GetStatusCode() int }
|
||||
|
||||
// OgenLogging is an ogen middleware that logs each operation's duration and
|
||||
// outcome. Handler errors carrying a 4xx/5xx-class status are logged at the
|
||||
// appropriate level: client errors (and expected 503s during startup) at
|
||||
// warn without a stacktrace, server errors at error.
|
||||
func OgenLogging(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("op", 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))
|
||||
lg := log.With(zap.String("operation", req.OperationID), zap.Duration("duration", time.Since(start)))
|
||||
|
||||
if err == nil {
|
||||
lg.Info("request completed")
|
||||
return resp, err
|
||||
}
|
||||
if sc, ok := err.(statusCoder); ok && sc.GetStatusCode() < 500 {
|
||||
lg.Warn("request rejected", zap.Int("status", sc.GetStatusCode()), zap.NamedError("reason", err))
|
||||
} else {
|
||||
lg.Info("request completed", zap.Duration("duration", dur))
|
||||
lg.Error("request failed", zap.Error(err))
|
||||
}
|
||||
return resp, err
|
||||
}
|
||||
}
|
||||
|
||||
// statusRecorder captures the response status for HTTPLogging.
|
||||
type statusRecorder struct {
|
||||
http.ResponseWriter
|
||||
status int
|
||||
}
|
||||
|
||||
func (r *statusRecorder) WriteHeader(code int) {
|
||||
r.status = code
|
||||
r.ResponseWriter.WriteHeader(code)
|
||||
}
|
||||
|
||||
// HTTPLogging wraps the given http.Handler with a per-request log line.
|
||||
func HTTPLogging(log *zap.Logger, next http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
start := time.Now()
|
||||
rec := &statusRecorder{ResponseWriter: w, status: 200}
|
||||
next.ServeHTTP(rec, r)
|
||||
log.Info("http",
|
||||
zap.String("method", r.Method),
|
||||
zap.String("path", r.URL.Path),
|
||||
zap.Int("status", rec.status),
|
||||
zap.Duration("duration", time.Since(start)))
|
||||
})
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue