1. 新增风险值的查询逻辑；

feiyu02

2023-11-14 8f069a80ed15dd431450f58304513aa3985e62ba

 src/main/java/com/flightfeather/monitor/analysis/dust/RiskAnalysisController.kt

@@ -1,7 +1,109 @@
package com.flightfeather.monitor.analysis.dust;
package com.flightfeather.monitor.analysis.dust

import com.flightfeather.monitor.analysis.dust.risk.*
import com.flightfeather.monitor.domain.ds1.repository.*
import com.flightfeather.monitor.enumration.dust.DeviceStatus
import com.flightfeather.monitor.utils.DateUtil
import org.springframework.stereotype.Component
import java.time.LocalDate
import java.time.LocalDateTime
import java.time.ZoneId

/**
 * 数据风险分析控制器
 */
public class RiskAnalysisController {
}
@Component
class RiskAnalysisController(
    private val dustExceptionSettingRep: DustExceptionSettingRep,
    private val riskValueRep: RiskValueRep,
    private val dustSiteStatusRep: DustSiteStatusRep,
    private val dustSiteDataRep: DustSiteDataRep,
    private val dustExceptionDataRep: DustExceptionDataRep,
    private val dustStatisticValueRep: DustStatisticsValueRep,
) {

    var running1 = false
    var running2 = false

    private var taskDaily: DustRiskAnalysis? = null
    private var taskMonthly: DustRiskMonthAnalysis? = null

    fun init() {
        dustExceptionSettingRep.findLatestSetting("金山区")?.let {
            taskDaily = DustRiskAnalysis(it)
            taskMonthly = DustRiskMonthAnalysis(it)
        }
    }

    fun autoRunDaily() {
        val data = riskValueRep.findLatestData("day")
        val yesterday = LocalDate.now().minusDays(1)
        if (data == null) {
            runDaily(yesterday)
        } else {
            val date = LocalDateTime.ofInstant(data.lst.toInstant(), ZoneId.systemDefault())
            val sT = if (date.hour == 0 && date.minute == 0 && date.second == 0) {
                date.toLocalDate()
            } else {
                date.plusDays(1).toLocalDate()
            }
            val du = DateUtil.findDurationDate(sT, yesterday)
            du.forEach {
                runDaily(it)
            }
        }
    }

    fun autoRunMonthly() {
        val data = riskValueRep.findLatestData("month")
        val lastMonth = LocalDate.now().minusMonths(1).withDayOfMonth(1)
        if (data == null) {
            runMonthly(lastMonth)
        } else {
            val date = LocalDateTime.ofInstant(data.lst.toInstant(), ZoneId.systemDefault())
            val sT = date.plusMonths(1).toLocalDate()
            val du = DateUtil.findDurationMonth(sT, lastMonth)
            du.forEach {
                runMonthly(it)
            }
        }
    }

    fun runDaily(date: LocalDate) {
        running1 = true
        taskDaily?.init()
        // 获取所有当前上线和停运的设备
        dustSiteStatusRep.select(listOf(DeviceStatus.ONLINE, DeviceStatus.STOP)).forEach { s ->
            s?.let {
                taskDaily?.roundInit(s.mnCode, date)
                val dustSiteDataList = dustSiteDataRep.select(s.mnCode, date)
                val dustExceptionDataList = dustExceptionDataRep.select(s.mnCode, date)
                val dustStatisticValueList = dustStatisticValueRep.select(s.mnCode, date)
                if (dustStatisticValueList.isEmpty()) return@forEach
                taskDaily?.roundCal(dustStatisticValueList[0]!!, dustSiteDataList, dustExceptionDataList)
            }
        }
        // 所有分析结果入库
        if (taskDaily?.resultList?.isNotEmpty() == true) {
            riskValueRep.insert(taskDaily?.resultList!!)
        }
        running1 = false
    }

    fun runMonthly(date: LocalDate) {
        running2 = true
        taskMonthly?.init()
        dustSiteStatusRep.select(listOf(DeviceStatus.ONLINE, DeviceStatus.STOP)).forEach { s ->
            s?.let {
                taskMonthly?.roundInit(s.mnCode, date)
                val riskValueList = riskValueRep.select(s.mnCode, date, "day")
                taskMonthly?.roundCal(riskValueList)
            }
        }
        // 所有分析结果入库
        if (taskMonthly?.resultList?.isNotEmpty() == true) {
            riskValueRep.insert(taskMonthly?.resultList!!)
        }
        running2 = false
    }
}