From 15bb12e2e4c719128d7ad20c1dff06c4f80b73f4 Mon Sep 17 00:00:00 2001
From: Dennis Hermsmeier <d.hermsmeier@mittwald.de>
Date: Tue, 18 Aug 2020 09:15:07 +0200
Subject: [PATCH] optionally skip paths to log

---
 logger.go | 22 +++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/logger.go b/logger.go
index 57c817d..f673998 100644
--- a/logger.go
+++ b/logger.go
@@ -3,6 +3,7 @@ package ginlogrus
 import (
 	"fmt"
 	"math"
+	"net/http"
 	"os"
 	"time"
 
@@ -19,11 +20,22 @@ import (
 var timeFormat = "02/Jan/2006:15:04:05 -0700"
 
 // Logger is the logrus logger handler
-func Logger(logger logrus.FieldLogger) gin.HandlerFunc {
+func Logger(logger logrus.FieldLogger, notLogged ...string) gin.HandlerFunc {
 	hostname, err := os.Hostname()
 	if err != nil {
 		hostname = "unknow"
 	}
+
+	var skip map[string]struct{}
+
+	if length := len(notLogged); length > 0 {
+		skip = make(map[string]struct{}, length)
+
+		for _, p := range notLogged {
+			skip[p] = struct{}{}
+		}
+	}
+
 	return func(c *gin.Context) {
 		// other handler can change c.Path so:
 		path := c.Request.URL.Path
@@ -40,6 +52,10 @@ func Logger(logger logrus.FieldLogger) gin.HandlerFunc {
 			dataLength = 0
 		}
 
+		if _, ok := skip[path]; ok {
+			return
+		}
+
 		entry := logger.WithFields(logrus.Fields{
 			"hostname":   hostname,
 			"statusCode": statusCode,
@@ -56,9 +72,9 @@ func Logger(logger logrus.FieldLogger) gin.HandlerFunc {
 			entry.Error(c.Errors.ByType(gin.ErrorTypePrivate).String())
 		} else {
 			msg := fmt.Sprintf("%s - %s [%s] \"%s %s\" %d %d \"%s\" \"%s\" (%dms)", clientIP, hostname, time.Now().Format(timeFormat), c.Request.Method, path, statusCode, dataLength, referer, clientUserAgent, latency)
-			if statusCode > 499 {
+			if statusCode >= http.StatusInternalServerError {
 				entry.Error(msg)
-			} else if statusCode > 399 {
+			} else if statusCode >= http.StatusBadRequest {
 				entry.Warn(msg)
 			} else {
 				entry.Info(msg)