优化流式传输&fix bugs

internal/router/router.go

@@ -36,6 +36,7 @@ func NewRouter(
 	settingHandler *handlers.SettingHandler,
 	dashboardHandler *handlers.DashboardHandler,
 	taskHandler *handlers.TaskHandler,
+	wsHandler *handlers.WebSocketHandler,
 	// Web Page Handlers
 	webAuthHandler *webhandlers.WebAuthHandler,
 	pageHandler *webhandlers.PageHandler,
@@ -75,6 +76,7 @@ func NewRouter(
 	registerWebRoutes(router, webAdminAuth, webAuthHandler, pageHandler)
 	registerAdminRoutes(router, apiAdminAuth, keyGroupHandler, tokensHandler, apiKeyHandler,
 		logHandler, settingHandler, dashboardHandler, taskHandler, upstreamModule, proxyModule)
+	registerWebSocketRoutes(router, wsHandler)
 	registerProxyRoutes(router, proxyHandler, securityService, logger)
 
 	return router
@@ -408,3 +410,7 @@ func registerPublicAPIRoutes(
 		// publicAPI.POST("/forgot-password", ipBanMiddleware, apiAuthHandler.HandleForgotPassword)
 	}
 }
+
+func registerWebSocketRoutes(router *gin.Engine, wsHandler *handlers.WebSocketHandler) {
+	router.GET("/ws/system-logs", wsHandler.HandleSystemLogs)
+}