fix(client): 优化更新处理与WebSocket连接管理

- 在版本更新处理时，改为发送 SIGTERM 信号以实现优雅关闭，确保资源正确释放
- 修改WebSocket连接的读取限制为无限制，避免大消息被截断
- 移除读取超时设置，依靠心跳机制和TCP keepalive保持连接持久化
- 记录WebSocket连接关闭的详细信息，增强日志记录

Author: orange-juzipi

internal/client/update_version.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
+	"syscall"
 	"time"
 
 	"github.com/https-cert/deploy/internal/updater"
@@ -58,6 +59,21 @@ func (uh *UpdateHandler) HandleUpdate() {
 		return
 	}
 
-	time.Sleep(1 * time.Second)
-	os.Exit(0)
+	// 发送 SIGTERM 信号给当前进程，触发优雅关闭
+	// 这样可以让 HTTP 服务器和其他资源正确释放
+	process, err := os.FindProcess(os.Getpid())
+	if err != nil {
+		logger.Error("获取当前进程失败", "error", err)
+		// 降级方案：等待后强制退出
+		time.Sleep(3 * time.Second)
+		os.Exit(0)
+		return
+	}
+
+	if err := process.Signal(syscall.SIGTERM); err != nil {
+		logger.Error("发送退出信号失败", "error", err)
+		// 降级方案：等待后强制退出
+		time.Sleep(3 * time.Second)
+		os.Exit(0)
+	}
 }

internal/client/ws_connection.go

@@ -89,6 +89,9 @@ func (c *WSClient) connect() error {
 		return fmt.Errorf("WebSocket连接失败: %w", err)
 	}
 
+	// 设置 WebSocket 读取限制为无限制，避免大消息被截断
+	conn.SetReadLimit(-1)
+
 	c.connMu.Lock()
 	c.conn = conn
 	c.connMu.Unlock()

internal/client/ws_message_handler.go

@@ -49,19 +49,23 @@ func (c *WSClient) handleWSMessages() error {
 			return errors.New("连接已关闭")
 		}
 
-		// 设置读取超时（心跳间隔的 3 倍，确保有足够时间接收消息）
-		readCtx, readCancel := context.WithTimeout(c.ctx, heartbeatInterval*3)
-		_, data, err := conn.Read(readCtx)
-		readCancel()
+		// 不设置读取超时，让连接保持持久化
+		// 依靠心跳机制和 TCP keepalive 来检测连接状态
+		_, data, err := conn.Read(c.ctx)
 
 		if err != nil {
 			if errors.Is(err, context.Canceled) {
+				logger.Info("WebSocket 连接因 context 取消而关闭")
 				return nil
 			}
 			// 使用 CloseStatus 检查正常关闭
-			if websocket.CloseStatus(err) == websocket.StatusNormalClosure {
+			closeStatus := websocket.CloseStatus(err)
+			if closeStatus == websocket.StatusNormalClosure {
+				logger.Info("WebSocket 连接正常关闭")
 				return nil
 			}
+			// 记录详细的错误信息
+			logger.Warn("WebSocket 读取错误", "error", err, "closeStatus", closeStatus)
 			return err
 		}
 
@@ -157,7 +161,6 @@ func (c *WSClient) handleGetProvider(requestId string) {
 
 // handleUpdate 处理版本更新
 func (c *WSClient) handleUpdate() {
-	logger.Info("收到版本更新通知")
 	updateHandler := NewUpdateHandler(c.ctx)
 	updateHandler.HandleUpdate()
 }

internal/client/ws_sender.go

@@ -107,7 +107,7 @@ func (c *WSClient) sendHeartbeat(ctx context.Context) {
 			}
 
 			if err := c.sendNotifyRequest(req); err != nil {
-				logger.Warn("发送心跳失败，主动关闭连接以触发重连", "error", err)
+				logger.Warn("发送心跳失败，主动关闭连接以触发重连", "error", err, "interval", heartbeatInterval)
 				// 主动关闭连接，触发重连机制
 				c.connMu.Lock()
 				if c.conn != nil {

internal/scheduler/scheduler.go

@@ -6,6 +6,7 @@ import (
 	"time"
 
 	"github.com/https-cert/deploy/internal/client"
+	"github.com/https-cert/deploy/internal/config"
 	"github.com/https-cert/deploy/internal/server"
 	"github.com/https-cert/deploy/pkg/logger"
 )
@@ -60,6 +61,8 @@ func Start(ctx context.Context) {
 
 	// 启动 HTTP-01 验证服务器
 	go func() {
+		cfg := config.GetConfig()
+		logger.Info("HTTP-01 验证服务启动", "port", cfg.Server.Port)
 		if err := scheduler.httpServer.Start(); err != nil {
 			logger.Error("HTTP-01 验证服务启动失败", "error", err)
 		}

internal/server/http_server.go

@@ -65,9 +65,6 @@ func NewHTTPServer() *HTTPServer {
 
 // Start 启动 HTTP 服务器
 func (s *HTTPServer) Start() error {
-	cfg := config.GetConfig()
-	logger.Info("HTTP-01 验证服务启动", "port", cfg.Server.Port)
-
 	if err := s.server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
 		return fmt.Errorf("HTTP 服务器启动失败: %w", err)
 	}