luzifer/go_helpers
1
0
Fork 0
mirror of https://github.com/Luzifer/go_helpers.git synced 2024-12-25 13:31:21 +00:00
Code Issues Releases Wiki Activity

Allow to pass in a logger for HTTP logs

Browse source 
Signed-off-by: Knut Ahlers <knut@ahlers.me>
This commit is contained in:
Knut Ahlers 2021-02-06 22:39:56 +01:00
parent e7a1201f43
commit 171f74b174
Signed by: luzifer
GPG key ID: 0066F03ED215AD7D
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
http/logHandler.go
View file

@ -3,18 +3,26 @@ package http
import ( import (
"log" "log"
"net/http" "net/http"
"os"
"strings" "strings"
"time" "time"
"github.com/Luzifer/go_helpers/v2/accessLogger" "github.com/Luzifer/go_helpers/v2/accessLogger"
"github.com/sirupsen/logrus"
) )
type HTTPLogHandler struct { type HTTPLogHandler struct {
Handler http.Handler Handler http.Handler
TrustedIPHeaders []string TrustedIPHeaders []string
logger logrus.StdLogger
} }
func NewHTTPLogHandler(h http.Handler) http.Handler { func NewHTTPLogHandler(h http.Handler) http.Handler {
return NewHTTPLogHandlerWithLogger(h, log.New(os.Stderr, "", log.LstdFlags))
}
func NewHTTPLogHandlerWithLogger(h http.Handler, l logrus.StdLogger) http.Handler {
return HTTPLogHandler{ return HTTPLogHandler{
Handler: h, Handler: h,
TrustedIPHeaders: []string{"X-Forwarded-For", "RemoteAddr", "X-Real-IP"}, TrustedIPHeaders: []string{"X-Forwarded-For", "RemoteAddr", "X-Real-IP"},
@ -32,7 +40,7 @@ func (l HTTPLogHandler) ServeHTTP(res http.ResponseWriter, r *http.Request) {
path = path + "?" + q path = path + "?" + q
} }
log.Printf("%s - \"%s %s\" %d %d \"%s\" \"%s\" %s", l.logger.Printf("%s - \"%s %s\" %d %d \"%s\" \"%s\" %s",
l.findIP(r), l.findIP(r),
r.Method, r.Method,
path, path,