mirror of
https://github.com/hamburghammer/gsave.git
synced 2024-12-22 18:37:39 +01:00
41 lines
1.1 KiB
Go
41 lines
1.1 KiB
Go
|
package middleware
|
||
|
|
||
|
import (
|
||
|
"net/http"
|
||
|
"time"
|
||
|
|
||
|
"github.com/sirupsen/logrus"
|
||
|
)
|
||
|
|
||
|
type statusCodeLogger struct {
|
||
|
http.ResponseWriter
|
||
|
statusCode int
|
||
|
}
|
||
|
|
||
|
func newStatusCodeLogger(rw http.ResponseWriter) *statusCodeLogger {
|
||
|
return &statusCodeLogger{ResponseWriter: rw, statusCode: http.StatusOK}
|
||
|
}
|
||
|
|
||
|
func (sl *statusCodeLogger) WriteHeader(code int) {
|
||
|
sl.statusCode = code
|
||
|
sl.ResponseWriter.WriteHeader(code)
|
||
|
}
|
||
|
|
||
|
// RequestTimeLoggingHandler logs the time a request needs to be processed.
|
||
|
// This handler should be add at the beginning of a handler chain.
|
||
|
// Every request will be logged with the Trace logging level.
|
||
|
func RequestTimeLoggingHandler(next http.Handler) http.Handler {
|
||
|
return http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
|
||
|
sl := newStatusCodeLogger(rw)
|
||
|
requestBeginn := time.Now()
|
||
|
next.ServeHTTP(sl, r)
|
||
|
requestDuration := time.Since(requestBeginn)
|
||
|
logPackage.WithFields(logrus.Fields{
|
||
|
"RequestTime": requestDuration.Milliseconds(),
|
||
|
"RequestPath": r.URL.String(),
|
||
|
"RequestMethod": r.Method,
|
||
|
"StatusCode": sl.statusCode,
|
||
|
}).Tracef("[%s] %q %v\n", r.Method, r.URL.String(), requestDuration)
|
||
|
})
|
||
|
}
|