From 636b32d7f8ec2d4fff28f8c2921e7f8c55efa418 Mon Sep 17 00:00:00 2001
From: zijiren233 <pyh1670605849@gmail.com>
Date: Thu, 23 May 2024 19:03:18 +0800
Subject: [PATCH] Fix: gin log error levle

---
 server/middlewares/log.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/server/middlewares/log.go b/server/middlewares/log.go
index 38ad094..02bd5bc 100644
--- a/server/middlewares/log.go
+++ b/server/middlewares/log.go
@@ -2,6 +2,7 @@ package middlewares
 
 import (
 	"fmt"
+	"net/http"
 	"time"
 
 	"github.com/gin-gonic/gin"
@@ -42,7 +43,18 @@ func NewLog(l *logrus.Logger) gin.HandlerFunc {
 
 		param.Path = path
 
-		c.MustGet("log").(*logrus.Entry).Info(formatter(param))
+		logColor(c.MustGet("log").(*logrus.Entry), param)
+	}
+}
+
+func logColor(logger *logrus.Entry, p gin.LogFormatterParams) {
+	str := formatter(p)
+	code := p.StatusCode
+	switch {
+	case code >= http.StatusBadRequest && code < http.StatusInternalServerError:
+		logger.Error(str)
+	default:
+		logger.Info(str)
 	}
 }