From ed84d537693adb736bfed5c8947e15d8f6f40ec4 Mon Sep 17 00:00:00 2001 From: tbphp Date: Tue, 10 Jun 2025 14:12:23 +0800 Subject: [PATCH] fix: error log --- cmd/gpt-load/main.go | 1 + internal/proxy/server.go | 8 ++++---- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/cmd/gpt-load/main.go b/cmd/gpt-load/main.go index b7330e6..cfbc76b 100644 --- a/cmd/gpt-load/main.go +++ b/cmd/gpt-load/main.go @@ -116,6 +116,7 @@ func setupRoutes(handlers *handler.Handler, proxyServer *proxy.ProxyServer, conf // Add middleware router.Use(middleware.Recovery()) + router.Use(middleware.ErrorHandler()) router.Use(middleware.Logger(configManager.GetLogConfig())) router.Use(middleware.CORS(configManager.GetCORSConfig())) router.Use(middleware.RateLimiter(configManager.GetPerformanceConfig())) diff --git a/internal/proxy/server.go b/internal/proxy/server.go index baccb54..45283f8 100644 --- a/internal/proxy/server.go +++ b/internal/proxy/server.go @@ -261,9 +261,9 @@ func (ps *ProxyServer) executeRequestWithRetry(c *gin.Context, startTime time.Ti // Log failure if retryCount > 0 { - logrus.Debugf("Retry request failed (attempt %d): %v (response time: %v)", retryCount+1, err, responseTime) + logrus.Warnf("Retry request failed (attempt %d): %v (response time: %v)", retryCount+1, err, responseTime) } else { - logrus.Debugf("Initial request failed: %v (response time: %v)", err, responseTime) + logrus.Warnf("Initial request failed: %v (response time: %v)", err, responseTime) } // Record failure asynchronously @@ -312,9 +312,9 @@ func (ps *ProxyServer) executeRequestWithRetry(c *gin.Context, startTime time.Ti } if err := json.Unmarshal([]byte(errorMessage), &jsonError); err == nil && jsonError.Error.Message != "" { - logrus.Errorf("Error message: %s", jsonError.Error.Message) + logrus.Warnf("Http Error: %s", jsonError.Error.Message) } else { - logrus.Errorf("Error message: %s", errorMessage) + logrus.Warnf("Http Error: %s", errorMessage) } // Record failure asynchronously