mirror of
https://github.com/gomods/athens
synced 2026-02-03 12:10:32 +00:00
34f55365a2
* always use request logging middleware * reads better * rm http-url as it is redundant * gopls being redundant * add color name
50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
package middleware
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/fatih/color"
|
|
"github.com/gomods/athens/pkg/log"
|
|
logrus "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type responseWriter struct {
|
|
http.ResponseWriter
|
|
statusCode int
|
|
}
|
|
|
|
func (rw *responseWriter) WriteHeader(statusCode int) {
|
|
rw.statusCode = statusCode
|
|
rw.ResponseWriter.WriteHeader(statusCode)
|
|
}
|
|
|
|
// RequestLogger logs request params to standard output
|
|
// it should only be used during dev.
|
|
func RequestLogger(h http.Handler) http.Handler {
|
|
f := func(w http.ResponseWriter, r *http.Request) {
|
|
rw := &responseWriter{w, 0}
|
|
h.ServeHTTP(rw, r)
|
|
log.EntryFromContext(r.Context()).WithFields(logrus.Fields{
|
|
"http-status": fmtResponseCode(rw.statusCode),
|
|
}).Infof("incoming request")
|
|
}
|
|
return http.HandlerFunc(f)
|
|
}
|
|
|
|
func fmtResponseCode(statusCode int) string {
|
|
if statusCode == 0 {
|
|
statusCode = 200
|
|
}
|
|
status := fmt.Sprint(statusCode)
|
|
switch {
|
|
case statusCode < http.StatusBadRequest:
|
|
status = color.GreenString("%v", status)
|
|
case statusCode >= http.StatusBadRequest && statusCode < http.StatusInternalServerError:
|
|
status = color.HiYellowString("%v", status)
|
|
default:
|
|
status = color.HiRedString("%v", status)
|
|
}
|
|
return status
|
|
}
|