gunshi-project-ss/src/main/java/com/gunshi/project/hsz/service/StQxWarnRService.java

package com.gunshi.project.hsz.service;

import cn.hutool.json.JSONArray;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.gunshi.db.dto.DateTimeRangeSo;
import com.gunshi.project.hsz.common.model.JcskByR;
import com.gunshi.project.hsz.common.model.StRiverR;
import com.gunshi.project.hsz.common.model.StRiverRReal;
import com.gunshi.project.hsz.entity.vo.*;
import com.gunshi.project.hsz.mapper.AttResBaseMapper;
import com.gunshi.project.hsz.mapper.StQxWarnRMapper;
import com.gunshi.project.hsz.model.*;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.compress.utils.Lists;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.util.EntityUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 描述: 气象预警表
 * author: xusan
 * date: 2024-07-08 17:30:38
 */
@Service
@Slf4j
@Transactional(rollbackFor = Exception.class)
public class StQxWarnRService extends ServiceImpl<StQxWarnRMapper, StQxWarnR> {

    @Value("${shqxjsWarnPath}")
    private String shqxjsWarnPath;

    @Resource
    private StImgWarnRService imgWarnRService;

    @Resource
    private AttResBaseMapper attResBaseMapper;

    @Resource
    private StRsvrRRealService stRsvrRRealService;

    @Resource
    private AlarmSetService alarmSetService;

    @Resource
    private StRiverRRealService stRiverRRealService;


    @Resource
    private StRiverRService stRiverRService;

    @Autowired
    private ISCAIEventService iscaiEventService;

    @Resource
    private JcskByRService jcskByRService;

    public HomeWarnVo homeWarn(DateTimeRangeSo dateTimeRangeSo) {
        HomeWarnVo vo = new HomeWarnVo();

        List<AttResBaseVo> attResBaseVos = attResBaseMapper.queryList();
        AttResBaseVo attResBaseVo = attResBaseVos.get(0);
        Date tm = attResBaseVo.getTm();
        //水位的最新采集时间不在24小时内
        if (tm.before(dateTimeRangeSo.getStart())) {
            vo.setRzWarn("-");
        } else {
            BigDecimal rz = attResBaseVo.getRz();
            if (attResBaseVo.getCalState() == 1) {
                BigDecimal aboveCal = rz.subtract(attResBaseVo.getCalFloodLev());
                vo.setRzWarn("超校核(" + aboveCal.setScale(2, RoundingMode.HALF_UP) + "m)");
                attResBaseVo.setDesState(0);
                attResBaseVo.setFlState(0);
            } else if (attResBaseVo.getDesState() == 1) {
                BigDecimal aboveDes = rz.subtract(attResBaseVo.getDesFloodLev());
                vo.setRzWarn("超设计(" + aboveDes.setScale(2, RoundingMode.HALF_UP) + "m)");
                attResBaseVo.setFlState(0);
            } else if (attResBaseVo.getFlState() == 1) {
                vo.setRzWarn("超汛限(" + attResBaseVo.getAFsltdz().setScale(2, RoundingMode.HALF_UP) + "m)");
            } else if (rz.compareTo(attResBaseVo.getDeadLev()) < 0) {
                BigDecimal aboveDead = rz.subtract(attResBaseVo.getDeadLev());
                vo.setRzWarn("低水位(" + aboveDead.setScale(1, RoundingMode.HALF_UP) + "m)");
            } else {
                vo.setRzWarn("正常(" + attResBaseVo.getAFsltdz().setScale(2, RoundingMode.HALF_UP) + "m)");
            }
        }

//        vo.setQxWarn(queryQxWarn());

        List<OsmoticWarnVo> warnVos = baseMapper.queryOsmoticWarn(dateTimeRangeSo);
        vo.setPressWarn(warnVos.stream().filter(o -> o.getType() == 1).collect(Collectors.toList()));
        vo.setFlowWarn(warnVos.stream().filter(o -> o.getType() == 2).collect(Collectors.toList()));
        vo.setShiftWarn(warnVos.stream().filter(o -> o.getType() == 3).collect(Collectors.toList()));
        Long aiWarnCount = iscaiEventService.lambdaQuery()
                .ge(ISCAIEvent::getStartTime, dateTimeRangeSo.getStart())
                .le(ISCAIEvent::getStartTime,dateTimeRangeSo.getEnd())
                .count();
        vo.setAiWarnCount(aiWarnCount);
        //TODO 超河道上游
        String upperStcd = "999999999";//固定的站点编码
        setUpAndDownOverRvUp(upperStcd,vo,dateTimeRangeSo);

        //TODO 超河道下游
        String lowerStcd = "61610700";//固定的站点编码
        setUpAndDownOverRvDown(lowerStcd,vo,dateTimeRangeSo);
        return vo;
    }

    private void setUpAndDownOverRvUp(String stcd, HomeWarnVo vo, DateTimeRangeSo dateTimeRangeSo){
        //StRsvrRReal stRsvrRReal = stRsvrRRealService.getBaseMapper().queryRzByStcd(stcd);
        //StRiverRReal stRiverRReal = stRiverRRealService.getBaseMapper().queryQByStcd(stcd);
        StRiverR stRiverR = stRiverRService.getBaseMapper().queryByOneByTimeRange(stcd,dateTimeRangeSo);
        if(stRiverR != null){
            AlarmSet alarmSet = alarmSetService.getBaseMapper().queryByStcd(stcd);
            //警告水位
            BigDecimal warnWaterLevel = alarmSet.getWarnWaterLevel();
            //保证水位
            BigDecimal promiseWaterLevel = alarmSet.getPromiseWaterLevel();
            if(alarmSet == null){
                vo.setOverRvUp("告警设置未配置，请检查");
            }else{
                BigDecimal rz = stRiverR.getZ();
                if(rz == null){
                    vo.setOverRvUp("无水位数据");
                    return;
                }
                if(rz.compareTo(warnWaterLevel) >=0){
                    BigDecimal gap = rz.subtract(warnWaterLevel);
                    vo.setOverRvUp("超警戒(" +gap.setScale(2, RoundingMode.HALF_UP)+")m");
                }else{
                    BigDecimal gap = rz.subtract(warnWaterLevel);
                    vo.setOverRvUp("正常(" +gap.setScale(2, RoundingMode.HALF_UP)+")m");
                }
            }
        }else{
            vo.setOverRvUp("-");
        }
    }

    private void setUpAndDownOverRvDown(String stcd, HomeWarnVo vo, DateTimeRangeSo dateTimeRangeSo){
        StRiverR stRiverR = stRiverRService.getBaseMapper().queryByOneByTimeRange(stcd,dateTimeRangeSo);
        if(stRiverR != null){
            AlarmSet alarmSet = alarmSetService.getBaseMapper().queryByStcd(stcd);
            //警告水位
            BigDecimal warnWaterLevel = alarmSet.getWarnWaterLevel();
            //保证水位
            BigDecimal promiseWaterLevel = alarmSet.getPromiseWaterLevel();
            if(alarmSet == null){
                vo.setOverRvDown("告警设置未配置，请检查");

            }else{
                BigDecimal rz = stRiverR.getZ();
                if(rz == null){
                    vo.setOverRvUp("无水位数据");
                    return;
                }
                if(rz.compareTo(warnWaterLevel) >=0){
                    BigDecimal gap = rz.subtract(warnWaterLevel);
                    vo.setOverRvDown("超警戒(" +gap.setScale(2, RoundingMode.HALF_UP)+")m");
                }else{
                    BigDecimal gap = rz.subtract(warnWaterLevel);
                    vo.setOverRvDown("正常(" +gap.setScale(2, RoundingMode.HALF_UP)+")m");
                }
            }
        }else{
            vo.setOverRvDown("-");
        }
    }

    private List<QXWarningVo> queryQxWarn() {
        List<QXWarningVo> list = new ArrayList<>();
        ApiVo apiVo = new ApiVo();
        apiVo.setFilter(Lists.newArrayList());
        ObjectMapper om = new ObjectMapper();
        HttpClient httpClient = HttpClientBuilder.create().build();
        HttpPost httpPost = new HttpPost(shqxjsWarnPath);
        log.info("请求气象预警接口：{}", shqxjsWarnPath);
        StringEntity stringEntity;
        long start = System.currentTimeMillis();
        try {
            String params = om.writeValueAsString(apiVo);
            log.info("请求气象预警接口参数：{}", params);
            stringEntity = new StringEntity(params, ContentType.APPLICATION_JSON);
            httpPost.setEntity(stringEntity);
            HttpResponse response = httpClient.execute(httpPost);
            HttpEntity entity = response.getEntity();
            JSONObject dataObject = JSONUtil.parseObj(EntityUtils.toString(entity));
            JSONArray data = dataObject.getJSONArray("data");
            if (data == null) return list;
            List<WarningData> dataList = data.toList(WarningData.class);
            list = getList(dataList);
        } catch (Exception e) {
            e.printStackTrace();
        }
        long end = System.currentTimeMillis();
        log.info("查询气象预警耗时：{} 秒", (end - start) / 1000);
        return list;
    }

    private List<QXWarningVo> getList(List<WarningData> data) {
        List<QXWarningVo> respList = Lists.newArrayList();
        for (WarningData datum : data) {//最外层的列表
            String ctnm = datum.getEffectArea();//市级范围
            for (WarningData.TypeList typeList : datum.getTypeList()) {//里面的
                // 预警类型
                String type = typeList.getType();
                List<WarningData.Warning> warnList = typeList.getWarnList();

                for (WarningData.Warning warning : warnList) {
                    String cnnm = warning.getEffectArea();
                    if("红安".equals(cnnm)){
                        QXWarningVo vo = new QXWarningVo();
                        String publishUnit = warning.getPublishUnit();
                        vo.setCtnm(ctnm);//市级名称
                        vo.setCnnm(cnnm);//县级名称
                        vo.setPublishUnit(publishUnit);//发布单位
                        vo.setPublishTime(warning.getPublishTime());//预警发布时间
                        vo.setWarnSignalType(warning.getWarnSignalType());//预警类型
                        vo.setWarnSignalLevel(warning.getWarnSignalLevel());//预警级别
                        vo.setContent(warning.getContent());//预警内容
                        vo.setWarnid(warning.getId());
                        vo.setCreateTime(warning.getCreateTime());
                        vo.setStartTime(warning.getStartTime());
                        vo.setEndTime(warning.getEndTime());
                        respList.add(vo);
                    }

                }
            }
        }
        respList = respList.stream().filter(o -> "暴雨".equals(o.getWarnSignalType()) || "雷雨大风".equals(o.getWarnSignalType())).collect(Collectors.toList());
        respList = respList.stream().filter(o -> "红色".equals(o.getWarnSignalLevel()) || "橙色".equals(o.getWarnSignalLevel())).collect(Collectors.toList());
        return respList.stream().sorted(Comparator.comparing(QXWarningVo::getPublishTime).reversed())
                .collect(Collectors.toList());
    }
}