Function connectBatchWebSocket

static/js/app.js:2372–2488

(batchId)

// 连接批量任务 WebSocket
function connectBatchWebSocket(batchId) {
    activeBatchId = batchId;

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

    if (batchWsReconnectTimer) {
        clearTimeout(batchWsReconnectTimer);
        batchWsReconnectTimer = null;
    }
    batchWsManualClose = false;

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

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

        socket.onopen = () => {
            if (batchWebSocket !== socket) return;
            console.log('批量任务 WebSocket 连接成功');
            clearBatchWebSocketReconnect();
            // 停止轮询（如果有）
            stopBatchPolling();
            // 开始心跳
            startBatchWebSocketHeartbeat();
        };

        socket.onmessage = (event) => {
            if (batchWebSocket !== 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') {
                // 更新进度
                if (data.total !== undefined) {
                    updateBatchProgress({
                        total: data.total,
                        completed: data.completed || 0,
                        success: data.success || 0,
                        failed: data.failed || 0
                    });
                }

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

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

                    // 任务完成后再重置按钮