Implement gin request logger with param filtering

pkg/api/logger.go

@@ -0,0 +1,54 @@
+package api
+
+import (
+	"net/http"
+	"strings"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"github.com/sirupsen/logrus"
+)
+
+const loggerMessage = "http_request"
+
+func RequestLogger(logger *logrus.Logger) gin.HandlerFunc {
+	return func(c *gin.Context) {
+		start := time.Now()
+		path := c.Request.URL.Path
+
+		// Process request
+		c.Next()
+
+		// Skip logging static assets
+		if strings.Contains(path, "/static/") {
+			return
+		}
+
+		status := c.Writer.Status()
+		end := time.Now()
+		latency := end.Sub(start)
+
+		fields := logrus.Fields{
+			"status":      status,
+			"method":      c.Request.Method,
+			"path":        path,
+			"remote_addr": c.ClientIP(),
+			"duration":    latency,
+		}
+
+		if err := c.Errors.Last(); err != nil {
+			fields["error"] = err.Error()
+		}
+
+		entry := logrus.WithFields(fields)
+
+		switch {
+		case status >= http.StatusBadRequest && status < http.StatusInternalServerError:
+			entry.Warn(loggerMessage)
+		case status >= http.StatusInternalServerError:
+			entry.Error(loggerMessage)
+		default:
+			entry.Info(loggerMessage)
+		}
+	}
+}

pkg/cli/cli.go

@@ -11,6 +11,7 @@ import (
 
 	"github.com/gin-gonic/gin"
 	"github.com/jessevdk/go-flags"
+	"github.com/sirupsen/logrus"
 
 	"github.com/sosedoff/pgweb/pkg/api"
 	"github.com/sosedoff/pgweb/pkg/bookmarks"
@@ -183,7 +184,10 @@ func printVersion() {
 }
 
 func startServer() {
-	router := gin.Default()
+	router := gin.New()
+
+	router.Use(api.RequestLogger(logrus.New()))
+	router.Use(gin.Recovery())
 
 	// Enable HTTP basic authentication only if both user and password are set
 	if options.AuthUser != "" && options.AuthPass != "" {