304 lines
12 KiB
Java
304 lines
12 KiB
Java
package com.whdc.controller;
|
|
|
|
import com.alibaba.fastjson.JSON;
|
|
import com.alibaba.fastjson.JSONObject;
|
|
import com.whdc.exception.MyException;
|
|
import com.whdc.model.dto.ApiDto;
|
|
import com.whdc.model.dto.GroupWarningDto;
|
|
import com.whdc.model.vo.WarningData;
|
|
import com.whdc.model.vo.WarningHistoryListVo;
|
|
import com.whdc.model.vo.WarningListVo;
|
|
import com.whdc.service.IAddressBookService;
|
|
import com.whdc.utils.HttpUtil;
|
|
import com.whdc.utils.ResultJson;
|
|
import io.swagger.annotations.Api;
|
|
import io.swagger.annotations.ApiOperation;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.apache.commons.compress.utils.Lists;
|
|
import org.apache.commons.lang3.StringUtils;
|
|
import org.apache.http.HttpStatus;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.web.bind.annotation.PostMapping;
|
|
import org.springframework.web.bind.annotation.RequestBody;
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.Comparator;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.stream.Collectors;
|
|
|
|
/**
|
|
* Description:
|
|
* Created by XuSan on 2024/5/23.
|
|
*
|
|
* @author XuSan
|
|
* @version 1.0
|
|
*/
|
|
@Slf4j
|
|
@Api(tags = "预警 - Controller")
|
|
@RestController
|
|
@RequestMapping("/warning")
|
|
public class WarningController {
|
|
|
|
|
|
@Autowired
|
|
private IAddressBookService addressBookService;
|
|
|
|
/**
|
|
* 气象预警
|
|
*
|
|
* @param dto
|
|
* @return
|
|
*/
|
|
@ApiOperation(value = "气象预警")
|
|
@PostMapping("/getGroupWarning")
|
|
public ResultJson<List<WarningListVo>> getGroupWarning(
|
|
@RequestBody GroupWarningDto dto) {
|
|
|
|
ApiDto apiDto = new ApiDto();
|
|
apiDto.setStartTime(dto.getStartTime());
|
|
apiDto.setEndTime(dto.getEndTime());
|
|
apiDto.setPageNumber(dto.getPageNumber());
|
|
apiDto.setPageSize(dto.getPageSize());
|
|
ArrayList<ApiDto.Filter> filter = Lists.newArrayList();
|
|
|
|
String publishUnit = dto.getPublishUnit();
|
|
if (StringUtils.isNotBlank(publishUnit)) {
|
|
String[] split = publishUnit.split(",");
|
|
if (split.length > 1) {
|
|
String str = "(";
|
|
for (String s : split) {
|
|
str += s + ",";
|
|
}
|
|
str = str.substring(0, str.length() - 1) + ")";
|
|
filter.add(new ApiDto.Filter("publishUnit", "in " + str, ""));
|
|
|
|
} else {
|
|
filter.add(new ApiDto.Filter("publishUnit", "like", String.format("%s%s%s", "%", publishUnit, "%")));
|
|
}
|
|
}
|
|
String warnSignalLevel = dto.getWarnSignalLevel();
|
|
if (StringUtils.isNotBlank(warnSignalLevel)) {
|
|
String[] split = warnSignalLevel.split(",");
|
|
if (split.length > 1) {
|
|
String str = "(";
|
|
for (String s : split) {
|
|
str += s + ",";
|
|
}
|
|
str = str.substring(0, str.length() - 1) + ")";
|
|
filter.add(new ApiDto.Filter("warnSignalLevel", "in " + str, ""));
|
|
|
|
} else {
|
|
filter.add(new ApiDto.Filter("warnSignalLevel", "like", String.format("%s%s%s", "%", warnSignalLevel, "%")));
|
|
}
|
|
}
|
|
String warnSignalType = dto.getWarnSignalType();
|
|
if (StringUtils.isNotBlank(warnSignalType)) {
|
|
String[] split = warnSignalType.split(",");
|
|
if (split.length > 1) {
|
|
String str = "(";
|
|
for (String s : split) {
|
|
str += s + ",";
|
|
}
|
|
str = str.substring(0, str.length() - 1) + ")";
|
|
filter.add(new ApiDto.Filter("warnSignalType", "in " + str, ""));
|
|
|
|
} else {
|
|
filter.add(new ApiDto.Filter("warnSignalType", "like", String.format("%s%s%s", "%", warnSignalType, "%")));
|
|
}
|
|
}
|
|
|
|
apiDto.setFilter(filter);
|
|
// String str = HttpUtil.sendPost("http://223.75.53.141:8000/shzh/met/zyqxfw/api/warning/getGroupWarning", JSON.toJSONString(apiDto));
|
|
String str = HttpUtil.sendPost("http://127.0.0.1:20000/shzh/met/zyqxfw/api/warning/getGroupWarning", JSON.toJSONString(apiDto));
|
|
JSONObject json = JSON.parseObject(str);
|
|
if (json != null && json.getInteger("code") == HttpStatus.SC_OK) {
|
|
List<WarningData> data = json.getJSONArray("data").toJavaList(WarningData.class);
|
|
return ResultJson.ok(getList(data));
|
|
}
|
|
|
|
return ResultJson.ok(json);
|
|
}
|
|
|
|
/**
|
|
* 历史气象预警统计
|
|
*
|
|
* @param dto
|
|
* @return
|
|
*/
|
|
@ApiOperation(value = "历史气象预警统计")
|
|
@PostMapping("/getHistoryGroupWarning")
|
|
public ResultJson<List<WarningListVo>> getHistoryGroupWarning(@RequestBody GroupWarningDto dto) {
|
|
|
|
ApiDto apiDto = new ApiDto();
|
|
apiDto.setStartTime(dto.getStartTime());
|
|
apiDto.setEndTime(dto.getEndTime());
|
|
apiDto.setPageNumber(dto.getPageNumber());
|
|
apiDto.setPageSize(dto.getPageSize());
|
|
ArrayList<ApiDto.Filter> filter = Lists.newArrayList();
|
|
|
|
String publishUnit = dto.getPublishUnit();
|
|
if (StringUtils.isNotBlank(publishUnit)) {
|
|
String[] split = publishUnit.split(",");
|
|
if (split.length > 1) {
|
|
String str = "(";
|
|
for (String s : split) {
|
|
str += s + ",";
|
|
}
|
|
str = str.substring(0, str.length() - 1) + ")";
|
|
filter.add(new ApiDto.Filter("publishUnit", "in " + str, ""));
|
|
|
|
} else {
|
|
filter.add(new ApiDto.Filter("publishUnit", "like", String.format("%s%s%s", "%", publishUnit, "%")));
|
|
}
|
|
}
|
|
String warnSignalLevel = dto.getWarnSignalLevel();
|
|
if (StringUtils.isNotBlank(warnSignalLevel)) {
|
|
String[] split = warnSignalLevel.split(",");
|
|
if (split.length > 1) {
|
|
String str = "(";
|
|
for (String s : split) {
|
|
str += s + ",";
|
|
}
|
|
str = str.substring(0, str.length() - 1) + ")";
|
|
filter.add(new ApiDto.Filter("warnSignalLevel", "in " + str, ""));
|
|
|
|
} else {
|
|
filter.add(new ApiDto.Filter("warnSignalLevel", "like", String.format("%s%s%s", "%", warnSignalLevel, "%")));
|
|
}
|
|
}
|
|
String warnSignalType = dto.getWarnSignalType();
|
|
if (StringUtils.isNotBlank(warnSignalType)) {
|
|
String[] split = warnSignalType.split(",");
|
|
if (split.length > 1) {
|
|
String str = "(";
|
|
for (String s : split) {
|
|
str += s + ",";
|
|
}
|
|
str = str.substring(0, str.length() - 1) + ")";
|
|
filter.add(new ApiDto.Filter("warnSignalType", "in " + str, ""));
|
|
|
|
} else {
|
|
filter.add(new ApiDto.Filter("warnSignalType", "like", String.format("%s%s%s", "%", warnSignalType, "%")));
|
|
}
|
|
}
|
|
|
|
|
|
apiDto.setFilter(filter);
|
|
// String str = HttpUtil.sendPost("http://223.75.53.141:8000/shzh/met/zyqxfw/api/warning/getGroupWarning", JSON.toJSONString(apiDto));
|
|
String str = HttpUtil.sendPost("http://127.0.0.1:20000/shzh/met/zyqxfw/api/warning/getGroupWarning", JSON.toJSONString(apiDto));
|
|
JSONObject json = JSON.parseObject(str);
|
|
if (json != null && json.getInteger("code") == HttpStatus.SC_OK) {
|
|
List<WarningData> data = json.getJSONArray("data").toJavaList(WarningData.class);
|
|
return ResultJson.ok(getCount(data));
|
|
}
|
|
|
|
return ResultJson.ok(json);
|
|
}
|
|
|
|
/**
|
|
* 正在生效预警
|
|
*
|
|
* @param data
|
|
* @return
|
|
*/
|
|
private List<WarningListVo> getList(List<WarningData> data) {
|
|
|
|
List<WarningListVo> respList = Lists.newArrayList();
|
|
for (WarningData datum : data) {
|
|
for (WarningData.TypeList typeList : datum.getTypeList()) {
|
|
// 预警类型
|
|
String type = typeList.getType();
|
|
for (WarningData.Warning warning : typeList.getWarnList()) {
|
|
WarningListVo vo = new WarningListVo();
|
|
String publishUnit = warning.getPublishUnit();
|
|
vo.setTitle(publishUnit + "发布" + type + "预警");
|
|
vo.setPublishTime(warning.getPublishTime());
|
|
vo.setContent(warning.getContent());
|
|
vo.setWarnSignalType(warning.getWarnSignalType());
|
|
vo.setWarnSignalLevel(warning.getWarnSignalLevel());
|
|
vo.setAddressBooks(addressBookService.getListByAdnm(publishUnit.replace("气象台", "")));
|
|
respList.add(vo);
|
|
}
|
|
}
|
|
}
|
|
|
|
return respList;
|
|
}
|
|
|
|
/**
|
|
* 正在生效预警
|
|
*
|
|
* @param data
|
|
* @return
|
|
*/
|
|
private List<WarningHistoryListVo> getCount(List<WarningData> data) {
|
|
|
|
List<WarningData.Warning> warnList = Lists.newArrayList();
|
|
for (WarningData datum : data) {
|
|
WarningHistoryListVo vo = new WarningHistoryListVo();
|
|
vo.setWarnSignalType("全部");
|
|
|
|
for (WarningData.TypeList typeList : datum.getTypeList()) {
|
|
warnList.addAll(typeList.getWarnList());
|
|
}
|
|
}
|
|
|
|
Map<String, List<WarningData.Warning>> warnMap = warnList.stream()
|
|
.collect(Collectors
|
|
.groupingBy(WarningData.Warning::getPublishUnit, Collectors.toList()));
|
|
|
|
List<WarningHistoryListVo> respList = Lists.newArrayList();
|
|
|
|
warnMap.forEach((k,v) ->{
|
|
WarningHistoryListVo vo = new WarningHistoryListVo();
|
|
vo.setEffectArea(k.split("气象")[0]);
|
|
|
|
// 类型
|
|
Map<String, List<WarningData.Warning>> types = v.stream()
|
|
.collect(Collectors
|
|
.groupingBy(WarningData.Warning::getWarnSignalType, Collectors.toList()));
|
|
types.forEach((kt,vt) ->{
|
|
WarningHistoryListVo voType = new WarningHistoryListVo();
|
|
voType.setWarnSignalType(kt);
|
|
|
|
// 等级分组
|
|
Map<String, Long> levels = vt.stream()
|
|
.collect(Collectors.groupingBy(WarningData.Warning::getWarnSignalLevel, Collectors.counting()));
|
|
|
|
levels.forEach((k1, v1) -> {
|
|
switch (k1) {
|
|
case "红色":
|
|
voType.setRedCount(v1.intValue());
|
|
vo.setRedCount(v1.intValue() + vo.getRedCount());
|
|
break;
|
|
case "橙色":
|
|
voType.setOraCount(v1.intValue());
|
|
vo.setOraCount(v1.intValue() + vo.getOraCount());
|
|
break;
|
|
case "黄色":
|
|
voType.setYelCount(v1.intValue());
|
|
vo.setYelCount(v1.intValue() + vo.getYelCount());
|
|
break;
|
|
case "蓝色":
|
|
voType.setBluCount(v1.intValue());
|
|
vo.setBluCount(v1.intValue() + vo.getBluCount());
|
|
break;
|
|
default:
|
|
throw new MyException("当前预警等级不存在: " + k1);
|
|
}
|
|
});
|
|
|
|
|
|
});
|
|
vo.setSumCount(v.size());
|
|
respList.add(vo);
|
|
});
|
|
|
|
return respList.stream().sorted(Comparator.comparing(WarningHistoryListVo::getSumCount).reversed()).collect(Collectors.toList());
|
|
}
|
|
|
|
}
|