Function connectWebSocket

static/js/app.js:1304–1419

(taskUuid)

// 连接 WebSocket
function connectWebSocket(taskUuid) {
    activeTaskUuid = taskUuid;

    if (webSocket && [WebSocket.OPEN, WebSocket.CONNECTING].includes(webSocket.readyState)) {
        return;
    }

    if (wsReconnectTimer) {
        clearTimeout(wsReconnectTimer);
        wsReconnectTimer = null;
    }
    wsManualClose = false;

    const protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
    const wsUrl = `${protocol}//${window.location.host}/api/ws/task/${taskUuid}`;

    try {
        const socket = new WebSocket(wsUrl);
        webSocket = socket;

        socket.onopen = () => {
            if (webSocket !== socket) return;
            console.log('WebSocket 连接成功');
            useWebSocket = true;
            clearWebSocketReconnect();
            // 停止轮询（如果有）
            stopLogPolling();
            // 开始心跳
            startWebSocketHeartbeat();
        };

        socket.onmessage = (event) => {
            if (webSocket !== socket) return;
            const data = JSON.parse(event.data);

            if (data.type === 'log') {
                const logType = getLogType(data.message);
                addLog(logType, data.message);
            } else if (data.type === 'status') {
                updateTaskStatus(data.status);
                if (data.email) {
                    elements.taskEmail.textContent = data.email;
                }
                if (data.email_service) {
                    elements.taskService.textContent = getServiceTypeText(data.email_service);
                }

                // 检查是否完成
                if (['completed', 'failed', 'cancelled', 'cancelling'].includes(data.status)) {
                    // 保存最终状态，用于 onclose 判断
                    taskFinalStatus = data.status;
                    taskCompleted = true;

                    // 断开 WebSocket（异步操作）
                    disconnectWebSocket();

                    // 任务完成后再重置按钮
                    resetButtons();