70 lines
1.3 KiB
Go
70 lines
1.3 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log/slog"
|
||
|
|
"net/http"
|
||
|
|
"strings"
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type responseWriter struct {
|
||
|
|
http.ResponseWriter
|
||
|
|
statusCode int
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rw *responseWriter) WriteHeader(code int) {
|
||
|
|
rw.statusCode = code
|
||
|
|
rw.ResponseWriter.WriteHeader(code)
|
||
|
|
}
|
||
|
|
func (rw *responseWriter) Write(b []byte) (int, error) {
|
||
|
|
if rw.statusCode == 0 {
|
||
|
|
rw.statusCode = http.StatusOK
|
||
|
|
}
|
||
|
|
return rw.ResponseWriter.Write(b)
|
||
|
|
}
|
||
|
|
|
||
|
|
var rwPool = sync.Pool{
|
||
|
|
New: func() any {
|
||
|
|
return &responseWriter{}
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
// LogMiddleware
|
||
|
|
func LogMiddleware(next http.Handler) http.Handler {
|
||
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
start := time.Now()
|
||
|
|
|
||
|
|
rw := rwPool.Get().(*responseWriter)
|
||
|
|
defer func() {
|
||
|
|
rw.ResponseWriter = nil
|
||
|
|
rwPool.Put(rw)
|
||
|
|
}()
|
||
|
|
|
||
|
|
rw.ResponseWriter = w
|
||
|
|
// rw.statusCode = http.StatusOK
|
||
|
|
|
||
|
|
next.ServeHTTP(rw, r)
|
||
|
|
|
||
|
|
Logger.Info("request handled",
|
||
|
|
slog.String("ip", getIP(r)),
|
||
|
|
slog.String("method", r.Method),
|
||
|
|
slog.Int("status", rw.statusCode),
|
||
|
|
slog.String("path", r.URL.Path),
|
||
|
|
slog.Duration("duration", time.Since(start)),
|
||
|
|
)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
func getIP(r *http.Request) string {
|
||
|
|
if ip := r.Header.Get("X-Real-IP"); ip != "" {
|
||
|
|
return ip
|
||
|
|
}
|
||
|
|
if forwarded := r.Header.Get("X-Forwarded-For"); forwarded != "" {
|
||
|
|
if i := strings.Index(forwarded, ","); i != -1 {
|
||
|
|
return forwarded[:i]
|
||
|
|
}
|
||
|
|
return forwarded
|
||
|
|
}
|
||
|
|
return r.RemoteAddr
|
||
|
|
}
|