cronnoss / tk-api

import (

	"net/http"

	"time"

)

type statusWriter struct {

	http.ResponseWriter

	status int

}

func (w *statusWriter) WriteHeader(status int) {

	w.status = status

	w.ResponseWriter.WriteHeader(status)

}

type MiddlewareLogger struct{}

func NewMiddlewareLogger() *MiddlewareLogger {

	return &MiddlewareLogger{}

}

func (a *MiddlewareLogger) loggingMiddleware(next http.Handler) http.Handler {

	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {

		sw := &statusWriter{ResponseWriter: w}

		l := r.Context().Value(KeyLoggerID).(Logger)

		start := time.Now()

		next.ServeHTTP(sw, r)

		l.Debugf("%s [%s] %s %s %s %d %s %s %s\n",

			r.RemoteAddr,

			start.Format("02/Jan/2006:15:04:05 -0700"),

			r.Method,

			r.RequestURI,

			sw.status,

			http.StatusText(sw.status),

			time.Since(start).String(),

			r.Header.Get("User-Agent"),

		)

	})

}

}