feiyu02
2025-08-14 dac47617b37ccfb834cd73ce0ee725e1101de214
src/main/kotlin/com/flightfeather/uav/biz/sourcetrace/model/PollutedSummary.kt
@@ -1,12 +1,27 @@
package com.flightfeather.uav.biz.sourcetrace.model
import com.flightfeather.uav.biz.sourcetrace.config.RTExcWindLevelConfig
import com.flightfeather.uav.common.net.AMapService
import com.flightfeather.uav.common.utils.DateUtil
import com.flightfeather.uav.common.utils.MapUtil
import com.flightfeather.uav.domain.entity.BaseRealTimeData
import com.flightfeather.uav.domain.entity.SceneInfo
import java.math.BigDecimal
import java.time.LocalDateTime
import java.util.Date
import java.util.Timer
import java.util.TimerTask
// 异常数据生成回调类
typealias NewPolluteSummaryCallback = (ex: AnalysisResult) -> Unit
/**
 * 污染情况汇总
 * 针对单次走航,定时统计已有污染线索[PollutedClue],按照策略给出走航建议
 * @date 2025/5/27
 * @author feiyu02
 */
class PollutedSummary {
class PollutedSummary(private val config: RTExcWindLevelConfig, private val callback: NewPolluteSummaryCallback) {
    /**
@@ -14,6 +29,185 @@
     * 每一刻钟对历史线索进行统计,提出会商建议(离污染源较远、污染源数量、出现次数)、走航路线调整建议(离污染源较近、走航轨迹未接近溯源场景)
     */
    // 污染线索
    var clueList = mutableListOf<PollutedClue>()
    /**
     * 实时统计
     */
    inner class AnalysisStatistic {
        // 按照被扫描次数降序排列的污染源列表
        var sortedSceneList: List<Pair<SceneInfo?, Int>>? = null
    }
    // 最新实时走航监测数据
    val realTimeDataList = mutableListOf<BaseRealTimeData>()
    // 未分析的污染线索
    val clueList = mutableListOf<PollutedClue>()
    // 已分析的污染线索
    private val historyClueList = mutableListOf<PollutedClue>()
    // 定时污染分析任务控制
    private var analysisTimer: Timer? = null
    // 定时污染分析任务
    private var lastAnalysisOnTimeTask: TimerTask? = null
    // 定时污染分析任务运行状态
    private var analysisTaskIsRunning = false
    // 上一次定时污染分析任务结束时间
    private lateinit var lastAnalysisTime: LocalDateTime
    init {
        clear()
    }
    // 新增一条污染线索
    fun addClue(pollutedClue: PollutedClue) {
        // 当溯源未找到风险源时,此次溯源信息不作为线索统计项
        if (pollutedClue.pollutedSource?.sceneList?.isNotEmpty() == true)
            clueList.add(pollutedClue)
//        realTimeSummary()
        analysisOnClueCount()
    }
    // 新增一条污染线索
    fun addClueList(pollutedClues: List<PollutedClue>) {
        pollutedClues.forEach { addClue(it) }
    }
    // 刷新当前最新的走航监测数据
    fun refreshLatestMonitorData(data: BaseRealTimeData) {
//        realTimeDataList.clear()
        realTimeDataList.add(data)
    }
    fun clear() {
        realTimeDataList.clear()
        clueList.clear()
        historyClueList.clear()
        analysisTimer?.cancel()
        analysisTimer = null
        analysisTaskIsRunning = false
        lastAnalysisTime = LocalDateTime.now()
        resetAnalysisOnTime()
    }
    /**
     * 重置定时分析线索任务
     */
    private fun resetAnalysisOnTime() {
        // 取消原有的分析任务计时
        analysisTimer?.cancel()
        lastAnalysisOnTimeTask?.cancel()
        // 以当前时间为起点,重新开始新的一轮等待计时
        analysisTimer = Timer()
        val period = config.analysisPeriod * 60 * 1000L
        lastAnalysisOnTimeTask = newAnalysisTask()
        analysisTimer?.schedule(lastAnalysisOnTimeTask, period, period)
    }
    /**
     * 在定时污染线索分析任务等待周期时间内,若污染线索量超过设定值,直接触发分析线索任务
     * 并重置定时分析任务
     */
    private fun analysisOnClueCount() {
        if (clueList.size >= config.analysisCount && !analysisTaskIsRunning) {
            newAnalysisTask().run()
            resetAnalysisOnTime()
        }
    }
    /**
     * 实时线索统计
     */
    private fun realTimeSummary() {
        val statistic = AnalysisStatistic()
        // 共有多少相关污染源,哪些污染源被扫描次数较多
        val sceneMap = mutableMapOf<String?, Pair<SceneInfo?, Int>>()
        clueList.forEach { c ->
            c.pollutedSource?.sceneList?.forEach { s ->
                if (!sceneMap.containsKey(s?.guid)) {
                    sceneMap[s?.guid] = s to 1
                } else {
                    sceneMap[s?.guid] = s to (sceneMap[s?.guid]?.second!! + 1)
                }
            }
        }
        val res = sceneMap.entries.sortedBy { it.value.second }
        statistic.sortedSceneList = res.map { it.value }
    }
    /**
     * 线索分析
     */
    private fun analysis() {
        if (clueList.isEmpty()) return
        val result = AnalysisResult().apply { deviceCode = clueList.first().deviceCode }
        // 共有多少相关污染源,哪些污染源被扫描次数较多
        val sceneMap = mutableMapOf<String?, Pair<SceneInfo?, Int>>()
        clueList.forEach { c ->
            c.pollutedSource?.sceneList?.forEach { s ->
                if (!sceneMap.containsKey(s?.guid)) {
                    sceneMap[s?.guid] = s to 1
                } else {
                    sceneMap[s?.guid] = s to (sceneMap[s?.guid]?.second!! + 1)
                }
            }
        }
        val res = sceneMap.entries.sortedByDescending { it.value.second }
        result.sortedSceneList = res.map { it.value }
        // 当前的走航数据的定位和污染源距离是否是逐渐接近,若走航远离了主要污染源,提示用户调整走航路线
        if (!result.sortedSceneList.isNullOrEmpty()) {
            val sT =
                DateUtil.instance.dateToString(clueList.first().pollutedData?.startTime, DateUtil.DateStyle.HH_MM_SS)
            val eT = DateUtil.instance.dateToString(clueList.last().pollutedData?.endTime, DateUtil.DateStyle.HH_MM_SS)
            val closetScene = result.sortedSceneList?.first()
            // 走航路线调整建议
            result.advice =
                "根据${sT}至${eT}的${clueList.size}个溯源切片,风险源【" +
                        "${closetScene?.first?.name}】被多次溯源,具有较高污染风险,现提供最新直达走航路线。"
            val lastP = realTimeDataList.last()
            // 建议对应的数据采样时间
            result.time = lastP.dataTime
            if (lastP.longitude != null && lastP.latitude != null &&
                lastP.longitude!! > BigDecimal.ZERO && lastP.latitude!! > BigDecimal.ZERO
                && closetScene?.first?.longitude != null && closetScene.first?.latitude != null &&
                closetScene.first?.longitude!! > BigDecimal.ZERO && closetScene.first?.latitude!! > BigDecimal.ZERO
            ) {
                val origin = MapUtil.wgs84ToGcj02(lastP.longitude!!.toDouble() to lastP.latitude!!.toDouble())
                val destination = closetScene.first!!.longitude.toDouble() to closetScene.first!!.latitude.toDouble()
                if (config.isSearchAddress) {
                    // 建议的走航路线
                    result.direction = AMapService.directionDriving(origin, destination)
                }
            }
            // 线索分析完成后,移动至历史线索列表
            historyClueList.addAll(clueList)
            clueList.clear()
            realTimeDataList.clear()
            callback(result)
        }
//        TODO()
    }
    // 定时污染分析任务
    private fun newAnalysisTask(): TimerTask {
        return object : TimerTask() {
            override fun run() {
                // 记录任务运行状态
                analysisTaskIsRunning = true
                analysis()
                // 记录上一次的任务结束时间
                lastAnalysisTime = LocalDateTime.now()
                analysisTaskIsRunning = false
            }
        }
    }
}