196 lines
7.4 KiB
Java
196 lines
7.4 KiB
Java
package com.gunshi.project.xyt.service;
|
||
|
||
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
||
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
||
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
|
||
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
||
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
||
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
||
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
||
import com.gunshi.project.xyt.entity.so.WarnPageSo;
|
||
import com.gunshi.project.xyt.entity.so.WarnSo;
|
||
import com.gunshi.project.xyt.entity.vo.OsmoticWarnVo;
|
||
import com.gunshi.project.xyt.mapper.OsmoticWarnRMapper;
|
||
import com.gunshi.project.xyt.mapper.OsmoticWarnRuleMapper;
|
||
import com.gunshi.project.xyt.model.OsmoticShiftR;
|
||
import com.gunshi.project.xyt.model.OsmoticWarnR;
|
||
import com.gunshi.project.xyt.model.OsmoticWarnRule;
|
||
import jakarta.annotation.Resource;
|
||
import lombok.extern.slf4j.Slf4j;
|
||
import org.springframework.stereotype.Service;
|
||
import org.springframework.transaction.annotation.Transactional;
|
||
|
||
import java.math.BigDecimal;
|
||
import java.util.Date;
|
||
import java.util.List;
|
||
import java.util.Map;
|
||
import java.util.Optional;
|
||
import java.util.stream.Collectors;
|
||
|
||
/**
|
||
* 描述: 隐患预警记录表
|
||
* author: xusan
|
||
* date: 2024-07-08 17:30:37
|
||
*/
|
||
@Service
|
||
@Slf4j
|
||
@Transactional(rollbackFor = Exception.class)
|
||
public class OsmoticWarnRService extends ServiceImpl<OsmoticWarnRMapper, OsmoticWarnR>
|
||
{
|
||
@Resource
|
||
private OsmoticWarnRuleMapper warnRuleMapper;
|
||
|
||
@Resource
|
||
private MessageCenterService messageCenterService;
|
||
|
||
public Page<OsmoticWarnVo> queryPage(WarnPageSo warnPageSo) {
|
||
return this.baseMapper.queryPage(warnPageSo.getPageSo().toPage(),warnPageSo);
|
||
}
|
||
|
||
public Map<Integer, Long> stat(WarnSo warnSo) {
|
||
LambdaQueryWrapper<OsmoticWarnR> wrapper = Wrappers.lambdaQuery();
|
||
if(warnSo.getType() != null){
|
||
wrapper.eq(OsmoticWarnR::getType,warnSo.getType());
|
||
}
|
||
if(warnSo.getLevel() != null){
|
||
wrapper.eq(OsmoticWarnR::getLevel,warnSo.getLevel());
|
||
}
|
||
if(StringUtils.isNotEmpty(warnSo.getStationCode())){
|
||
wrapper.like(OsmoticWarnR::getStationCode,warnSo.getStationCode());
|
||
}
|
||
if(warnSo.getDateTimeRangeSo() != null && warnSo.getDateTimeRangeSo().getStart() != null){
|
||
wrapper.ge(OsmoticWarnR::getTm,warnSo.getDateTimeRangeSo().getStart());
|
||
}
|
||
if(warnSo.getDateTimeRangeSo() != null && warnSo.getDateTimeRangeSo().getEnd() != null){
|
||
wrapper.le(OsmoticWarnR::getTm,warnSo.getDateTimeRangeSo().getEnd());
|
||
}
|
||
List<OsmoticWarnR> list = this.list(wrapper);
|
||
return list.stream().collect(Collectors.groupingBy(OsmoticWarnR::getLevel, Collectors.counting()));
|
||
}
|
||
|
||
/**
|
||
* 生成位移预警
|
||
*/
|
||
public void saveShiftWarn(OsmoticShiftR shiftR){
|
||
String stationCode = shiftR.getStationCode();
|
||
Date tm = shiftR.getTm();
|
||
BigDecimal x = shiftR.getX();
|
||
savePressWarn(stationCode,x,tm,"x");
|
||
BigDecimal y = shiftR.getY();
|
||
savePressWarn(stationCode,y,tm,"y");
|
||
BigDecimal h = shiftR.getH();
|
||
savePressWarn(stationCode,h,tm,"h");
|
||
}
|
||
|
||
/**
|
||
* 生成渗压/渗流预警
|
||
* @param
|
||
*/
|
||
public void savePressWarn(String stationCode,BigDecimal value,Date tm,String direction) {
|
||
//查询预警规则,每个测点至多只有2条预警规则
|
||
LambdaQueryWrapper<OsmoticWarnRule> queryWrapper = Wrappers.lambdaQuery();
|
||
queryWrapper.eq(OsmoticWarnRule::getStationCode,stationCode)
|
||
.eq(OsmoticWarnRule::getStatus,1);
|
||
if(StringUtils.isNotEmpty(direction)){
|
||
queryWrapper.eq(OsmoticWarnRule::getDirection,direction);
|
||
}
|
||
List<OsmoticWarnRule> ruleList = warnRuleMapper.selectList(queryWrapper);
|
||
if(CollectionUtils.isNotEmpty(ruleList)){
|
||
Boolean isRedWarn = false;
|
||
//先判断是否符合红色预警
|
||
Optional<OsmoticWarnRule> red = ruleList.stream().filter(o -> o.getLevel() == 2).findFirst();
|
||
if(red.isPresent()){
|
||
OsmoticWarnRule redRule = red.get();
|
||
isRedWarn = saveWarnData(redRule,value,tm);
|
||
if (isRedWarn){
|
||
return;
|
||
}
|
||
//没有生成红色预警，再判断是否符合黄色预警
|
||
Optional<OsmoticWarnRule> yellow = ruleList.stream().filter(o -> o.getLevel() == 1).findFirst();
|
||
if(yellow.isPresent()){
|
||
OsmoticWarnRule yellowRule = yellow.get();
|
||
saveWarnData(yellowRule,value,tm);
|
||
}
|
||
}else {
|
||
OsmoticWarnRule yellowRule = ruleList.get(0);
|
||
saveWarnData(yellowRule,value,tm);
|
||
}
|
||
}
|
||
}
|
||
|
||
private Boolean saveWarnData(OsmoticWarnRule rule,BigDecimal value,Date tm){
|
||
String conditionOne = rule.getConditionOne();
|
||
BigDecimal valueOne = rule.getValueOne();
|
||
Integer condition = rule.getCondition();
|
||
Boolean flagOne = getFlag(conditionOne,valueOne,value);
|
||
/**
|
||
* 没有第二条件时，条件1满足就生成预警
|
||
* 有第二条件时，当为或时，条件1满足就生成预警，条件1不满足就判断条件2是否满足；当为且时，需条件1和条件2同时满足才生成预警
|
||
*/
|
||
if(condition == null){
|
||
if(flagOne){
|
||
//生成预警
|
||
return geneWarn(rule,value,tm);
|
||
}
|
||
}else{
|
||
String conditionTwo = rule.getConditionTwo();
|
||
BigDecimal valueTwo = rule.getValueTwo();
|
||
Boolean flagTwo = getFlag(conditionTwo,valueTwo,value);
|
||
if(condition == 1){
|
||
if(flagOne && flagTwo){
|
||
//生成预警
|
||
return geneWarn(rule,value,tm);
|
||
}
|
||
}else{
|
||
if(flagOne || flagTwo){
|
||
//生成预警
|
||
return geneWarn(rule,value,tm);
|
||
}
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
private Boolean geneWarn(OsmoticWarnRule rule,BigDecimal value,Date tm) {
|
||
OsmoticWarnR warnR = new OsmoticWarnR();
|
||
warnR.setId(IdWorker.getId());
|
||
warnR.setStationCode(rule.getStationCode());
|
||
warnR.setRuleId(rule.getId());
|
||
warnR.setValue(value);
|
||
warnR.setTm(tm);
|
||
warnR.setType(rule.getType());
|
||
warnR.setLevel(rule.getLevel());
|
||
this.baseMapper.insert(warnR);
|
||
//生成消息
|
||
messageCenterService.osmoticWarnMes(warnR,rule.getRuleDesc());
|
||
return true;
|
||
}
|
||
|
||
private Boolean getFlag(String condition,BigDecimal ruleValue,BigDecimal value){
|
||
Boolean flag = false;
|
||
switch (condition) {
|
||
case ">" :
|
||
flag = value.compareTo(ruleValue) > 0;
|
||
break;
|
||
case ">=" :
|
||
flag = value.compareTo(ruleValue) >= 0;
|
||
break;
|
||
case "<" :
|
||
flag = value.compareTo(ruleValue) < 0;
|
||
break;
|
||
case "<=" :
|
||
flag = value.compareTo(ruleValue) <= 0;
|
||
break;
|
||
case "=" :
|
||
flag = value.compareTo(ruleValue) == 0;
|
||
break;
|
||
case "!=" :
|
||
flag = value.compareTo(ruleValue) != 0;
|
||
break;
|
||
}
|
||
return flag;
|
||
}
|
||
}
|
||
|
||
|