package com.ruoyi.system.web;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
import com.github.pagehelper.PageHelper;
|
import com.github.pagehelper.PageInfo;
|
import com.ruoyi.common.core.domain.AjaxResult;
|
import com.ruoyi.common.utils.DateUtils;
|
import com.ruoyi.common.utils.StringUtils;
|
import com.ruoyi.system.domain.company.CompanyInfo;
|
import com.ruoyi.system.domain.company.CompanyWarning;
|
import com.ruoyi.system.domain.vo.CompanyWarningVO;
|
import com.ruoyi.system.service.company.ICompanyInfoService;
|
import com.ruoyi.system.service.company.ICompanyWarningService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.transaction.annotation.Transactional;
|
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.text.ParseException;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
@RestController
|
public class CompanyWarningController {
|
|
|
@Value("${expire-warning-day}")
|
private Integer expireWarningData;
|
|
@Autowired
|
ICompanyWarningService companyWarningService;
|
|
@Autowired
|
ICompanyInfoService companyInfoService;
|
|
@RequestMapping("/checkCompanyIsSoonExpire")
|
@Transactional(rollbackFor = Exception.class)
|
public void checkCompanyIsSoonExpire() {
|
Date now;
|
try {
|
now = DateUtils.parseDate(DateUtils.parseDateToStr(DateUtils.YYYY_MM_DD, new Date()), DateUtils.YYYY_MM_DD);
|
} catch (ParseException e) {
|
throw new RuntimeException(e);
|
}
|
Date expireDate = DateUtils.addDays(now, expireWarningData);
|
//查询还有30天即将过期的企业
|
QueryWrapper expireQueryWrapper = new QueryWrapper();
|
expireQueryWrapper.eq(CompanyInfo.IS_DELETE, 0);
|
expireQueryWrapper.lt(CompanyInfo.VALIDITY_DATE_END, expireDate);
|
//过期的企业列表
|
List<CompanyInfo> soonExpireCompanyInfoList = companyInfoService.list(expireQueryWrapper);
|
|
|
QueryWrapper companyWarningQueryWrapper = new QueryWrapper();
|
companyWarningQueryWrapper.eq(CompanyInfo.IS_DELETE, 0);
|
List<CompanyWarning> companyWarningList = companyWarningService.list(companyWarningQueryWrapper);
|
|
Map<String, CompanyWarning> companyWarningMap = companyWarningList
|
.stream().collect(Collectors.toMap(CompanyWarning::getCompanyId, CompanyWarning -> CompanyWarning));
|
|
List<CompanyWarning> updateWarning = new ArrayList<>();
|
List<CompanyWarning> addWarningList = new ArrayList<>();
|
|
soonExpireCompanyInfoList.forEach(companyInfo -> {
|
//如果已经产生报警信息
|
CompanyWarning companyWarning = companyWarningMap.get(companyInfo.getId().toString());
|
if (companyWarning == null) {
|
CompanyWarning newCompanyWarn = new CompanyWarning();
|
String msg = "";
|
if (companyInfo.getValidityDateEnd().before(now)) {
|
msg = "【" + companyInfo.getCompanyName() + "】证书已到期！";
|
newCompanyWarn.setIsExpire(1);
|
} else {
|
msg = "【" + companyInfo.getCompanyName() + "】证书有效期不足" + expireWarningData + "天！";
|
newCompanyWarn.setIsExpire(0);
|
}
|
newCompanyWarn.setMsg(msg);
|
newCompanyWarn.setCompanyId(String.valueOf(companyInfo.getId()));
|
newCompanyWarn.setCreateTime(new Date());
|
addWarningList.add(newCompanyWarn);
|
} else {
|
Integer isExpire = companyWarning.getIsExpire();
|
String msg = "";
|
String oldMsg = companyWarning.getMsg();
|
if (companyInfo.getValidityDateEnd().before(now)) {
|
msg = "【" + companyInfo.getCompanyName() + "】证书已到期！";
|
companyWarning.setIsExpire(1);
|
} else {
|
msg = "【" + companyInfo.getCompanyName() + "】证书有效期不足" + expireWarningData + "天！";
|
companyWarning.setIsExpire(0);
|
}
|
companyWarning.setMsg(msg);
|
companyWarning.setUpdateTime(new Date());
|
//如果预警装填变更（即将到期-》到期），或者预警信息如预警天数发生变动，更新预警类型
|
if (isExpire != companyWarning.getIsExpire() || !StringUtils.equals(msg, oldMsg)) {
|
companyWarning.setIsRead(0);
|
updateWarning.add(companyWarning);
|
}
|
|
}
|
});
|
|
List<String> companyIds = soonExpireCompanyInfoList.stream().map(CompanyInfo::getIdStr).collect(Collectors.toList());
|
|
List<CompanyWarning> deleteWarning = companyWarningList.stream()
|
.filter(n -> !companyIds.contains(n.getCompanyId())).collect(Collectors.toList());
|
|
//报警信息修改，报警信息修改或者为过去-》过期 过期-》未过期
|
if (CollectionUtils.isNotEmpty(updateWarning)) {
|
companyWarningService.updateBatchById(updateWarning);
|
}
|
//添加新的警报信息
|
if (CollectionUtils.isNotEmpty(addWarningList)) {
|
companyWarningService.saveBatch(addWarningList);
|
}
|
// 证书续期成功删除对应报警信息
|
if (CollectionUtils.isNotEmpty(deleteWarning)) {
|
deleteWarning.forEach(n -> {
|
n.setIsDelete(1);
|
n.setUpdateTime(new Date());
|
});
|
companyWarningService.updateBatchById(deleteWarning);
|
}
|
}
|
|
@PostMapping("/companyWarn/findCompanyWarningPage")
|
public AjaxResult findCompanyWarningPage(@RequestBody CompanyWarningVO companyWarningVO) {
|
if (companyWarningVO.getPageSize() == null || companyWarningVO.getPageNum() == null) {
|
return AjaxResult.error("请求参数【pageSize，pageNum】不能为空！");
|
}
|
PageHelper.startPage(companyWarningVO.getPageNum(), companyWarningVO.getPageSize());
|
QueryWrapper queryWrapper = new QueryWrapper();
|
queryWrapper.eq(CompanyWarning.IS_DELETE, 0);
|
queryWrapper.orderByDesc(CompanyWarning.CREATE_TIME);
|
queryWrapper.orderByAsc(CompanyWarningVO.IS_READ);
|
PageInfo pageInfo = new PageInfo(companyWarningService.list(queryWrapper));
|
return AjaxResult.success(pageInfo);
|
}
|
|
@PostMapping("/companyWarn/warningRead")
|
public AjaxResult warningRead(@RequestBody CompanyWarning companyWarning) {
|
companyWarning.setIsRead(1);
|
companyWarningService.updateById(companyWarning);
|
return AjaxResult.success("已读成功！");
|
}
|
|
|
}
|
