优化

liusuyi

2026-05-14 9958bc1501d0daee954d9bba383475e55e24ebab

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
package com.ard.qs.utils;
 
import com.ard.qs.api.domain.QsDevice;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.bytedeco.javacv.FFmpegFrameGrabber;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import org.springframework.stereotype.Component;
 
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.TimeUnit;
 
@Slf4j
@Component
public class StreamDetector {
 
    /**
     * 批量检测流状态
     */
    public List<StreamResult> batchDetect(List<QsDevice> qsDeviceList, ThreadPoolTaskExecutor taskExecutor) {
        List<StreamResult> results = new ArrayList<>();
        List<CompletableFuture<StreamResult>> futures = new ArrayList<>();
 
        for (QsDevice device : qsDeviceList) {
            // 使用自定义线程池异步执行
            CompletableFuture<StreamResult> future = CompletableFuture.supplyAsync(() ->
                    detectSingle(device.getId(), device.getLiveAddress()), taskExecutor).orTimeout(15, TimeUnit.SECONDS); // 兜底超时
 
            futures.add(future);
        }
 
        // 等待所有任务完成
        for (CompletableFuture<StreamResult> future : futures) {
            try {
                results.add(future.join());
            } catch (Exception e) {
                log.error("[流检测失败] 批量检测时发生异常: {}", e.getMessage());
            }
        }
        return results;
    }
 
    /**
     * 单个流检测逻辑
     */
    public StreamResult detectSingle(Long id, String url) {
        if (url == null || url.isEmpty()) {
            return new StreamResult(id, "OFFLINE");
        }
 
        FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(url);
 
        try {
            // --- 核心配置 ---
            grabber.setOption("stimeout", "3000000"); // 3秒超时
            grabber.setOption("reconnect", "0");      // 禁用重连
            grabber.setOption("protocol_whitelist", "file,http,https,tcp,tls,ws,wss,crypto,udp,rtp,rtcp,rtmp,rtmpt,subfile,pipe,data");
            grabber.setOption("user_agent", "Mozilla/5.0 (compatible; StreamChecker)");
 
            grabber.start();
 
            if (grabber.getFormat() != null) {
                return new StreamResult(id, "ON");
            }
 
            return new StreamResult(id, "OFFLINE");
 
        } catch (Exception e) {
            log.debug("[流检测] 设备ID: {}, URL: {}, 检测结果: OFFLINE, 原因: {}", id, url, e.getMessage());
            return new StreamResult(id, "OFFLINE");
        } finally {
            try {
                if (grabber != null) {
                    grabber.stop();
                    grabber.release();
                }
            } catch (Exception e) {
                log.debug("[流检测] 释放资源失败: {}", e.getMessage());
            }
        }
    }
 
    // 结果封装类
    @Data
    public static class StreamResult {
        public Long id;
        public String status;
 
        public StreamResult(Long id, String status) {
            this.id = id;
            this.status = status;
        }
    }
}