golang-utils/utils/http/logger/logger.go

package logger

import (
	"log"
	"net/http"
	"strings"
	"time"

	"github.com/rollbar/rollbar-go"
	"github.com/vladimirok5959/golang-utils/utils/http/helpers"
)

var RollBarEnabled = false

type ResponseWriter struct {
	http.ResponseWriter
	Content []byte
	Size    int
	Status  int
}

func (w *ResponseWriter) Write(b []byte) (int, error) {
	if RollBarEnabled {
		if !(w.Status == http.StatusOK ||
			w.Status == http.StatusNotModified ||
			w.Status == http.StatusTemporaryRedirect ||
			w.Status == http.StatusNotFound ||
			w.Status == http.StatusMethodNotAllowed) {
			w.Content = append(w.Content, b...)
		}
	}
	size, err := w.ResponseWriter.Write(b)
	w.Size += size
	return size, err
}

func (w *ResponseWriter) WriteHeader(status int) {
	w.Status = status
	w.ResponseWriter.WriteHeader(status)
}

func LogInternalError(err error) {
	log.Printf("%s\n", err.Error())
	if RollBarEnabled {
		rollbar.Error(err)
	}
}

func LogRequests(handler http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		start := time.Now()
		nw := &ResponseWriter{
			ResponseWriter: w,
			Content:        []byte{},
			Size:           0,
			Status:         http.StatusOK,
		}
		handler.ServeHTTP(nw, r)
		ua := strings.TrimSpace(r.Header.Get("User-Agent"))
		if ua == "" || len(ua) > 256 {
			ua = "-"
		}
		log.Printf(
			"\"%s\" \"%s %s\" %d %d \"%.3f ms\" \"%s\"\n",
			strings.Join(helpers.ClientIPs(r), ", "),
			r.Method,
			r.URL,
			nw.Status,
			nw.Size,
			time.Since(start).Seconds(),
			ua,
		)
		if RollBarEnabled {
			if !(nw.Status == http.StatusOK ||
				nw.Status == http.StatusNotModified ||
				nw.Status == http.StatusTemporaryRedirect ||
				nw.Status == http.StatusNotFound ||
				nw.Status == http.StatusMethodNotAllowed) {
				rollbar.Error(r, string(nw.Content))
			}
		}
	})
}