refactor(scheduler): 优化调度器停止逻辑与WebSocket连接管理

- 在调度器停止时，增加对WebSocket客户端连接的关闭处理，并记录相关日志
- 修改停止调度器的日志信息，提供更清晰的状态反馈
- 在启动调度器时，使用goroutine以实现非阻塞的调度器启动
- 增加对调度器停止的超时处理，确保在超时情况下强制退出

Author: orange-juzipi

cmd/start.go

@@ -7,6 +7,7 @@ import (
 	"os/signal"
 	"path/filepath"
 	"syscall"
+	"time"
 
 	"github.com/https-cert/deploy/internal/config"
 	"github.com/https-cert/deploy/internal/scheduler"
@@ -41,15 +42,28 @@ func CreateStartCmd() *cobra.Command {
 			ctx, cancel := context.WithCancel(context.Background())
 			defer cancel()
 
-			scheduler.Start(ctx)
+			// 在 goroutine 中启动调度器
+			done := make(chan struct{})
+			go func() {
+				scheduler.Start(ctx)
+				close(done)
+			}()
 
 			sigChan := make(chan os.Signal, 1)
 			signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
 
 			<-sigChan
-			logger.Info("停止中...")
+			logger.Info("收到停止信号，正在关闭...")
 			cancel()
-			logger.Info("已停止")
+
+			// 等待调度器完全停止，最多等待 10 秒
+			select {
+			case <-done:
+				logger.Info("已停止")
+			case <-time.After(10 * time.Second):
+				logger.Warn("停止超时，强制退出")
+			}
+
 			return nil
 		},
 	}

internal/client/ws_connection.go

@@ -154,6 +154,8 @@ func (c *WSClient) StartWSNotify() {
 			busyOps := c.busyOperations.Load()
 			if busyOps > 0 {
 				logger.Warn("WebSocket连接意外断开(有业务正在执行)", "error", err, "busyOps", busyOps)
+			} else {
+				logger.Info("WebSocket连接断开", "error", err)
 			}
 
 			isConnected.Store(false)

internal/client/ws_sender.go

@@ -29,7 +29,8 @@ func (c *WSClient) sendNotifyRequest(req *deployPB.NotifyRequest) error {
 	}
 
 	// 发送 JSON 消息（WebSocket Text 消息）
-	ctx, cancel := context.WithTimeout(c.ctx, 10*time.Second)
+	// 使用较长的超时时间，避免网络慢时误判为失败
+	ctx, cancel := context.WithTimeout(c.ctx, 30*time.Second)
 	defer cancel()
 
 	return conn.Write(ctx, websocket.MessageText, data)
@@ -85,8 +86,8 @@ func (c *WSClient) sendHeartbeat(ctx context.Context) {
 			// 获取系统信息用于心跳
 			systemInfo, err := c.getSystemInfo()
 			if err != nil {
-				logger.Warn("获取系统信息失败", "error", err)
-				return
+				logger.Warn("获取系统信息失败，跳过本次心跳", "error", err)
+				continue // 跳过本次心跳，不要退出整个心跳循环
 			}
 
 			// 发送心跳消息（使用 RegisterResponse 格式）

internal/scheduler/scheduler.go

@@ -77,17 +77,32 @@ func Start(ctx context.Context) {
 
 // stop 停止调度器
 func (s *Scheduler) stop() {
+	logger.Info("正在停止调度器...")
+
 	if s.ticker != nil {
 		s.ticker.Stop()
 	}
 
+	// 关闭 WebSocket 客户端连接
+	if s.client != nil {
+		if err := s.client.Close(); err != nil {
+			logger.Error("关闭 WebSocket 客户端失败", "error", err)
+		} else {
+			logger.Info("WebSocket 客户端已关闭")
+		}
+	}
+
 	// 停止 HTTP 服务器
 	if s.httpServer != nil {
 		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
 		defer cancel()
 
 		if err := s.httpServer.Stop(ctx); err != nil {
 			logger.Error("停止 HTTP-01 验证服务失败", "error", err)
+		} else {
+			logger.Info("HTTP-01 验证服务已停止")
 		}
 	}
+
+	logger.Info("调度器已停止")
 }