package com.ruoyi.safe.specialequipment.task;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.ruoyi.approve.pojo.NotificationManagement;
|
import com.ruoyi.approve.service.NotificationManagementService;
|
import com.ruoyi.safe.specialequipment.mapper.SpecialEquipmentLedgerMapper;
|
import com.ruoyi.safe.specialequipment.pojo.SpecialEquipmentLedger;
|
import lombok.RequiredArgsConstructor;
|
import org.springframework.scheduling.annotation.Scheduled;
|
import org.springframework.stereotype.Component;
|
import java.time.LocalDate;
|
import java.time.temporal.ChronoUnit;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Optional;
|
@Component
|
@RequiredArgsConstructor
|
public class SpecialEquipmentReminderTask {
|
private final SpecialEquipmentLedgerMapper ledgerMapper;
|
private final NotificationManagementService notificationManagementService;
|
private static final int REMIND_DAYS = 30;
|
|
@Scheduled(cron = "0 0 8 * * ?")
|
public void remindExpire() {
|
LocalDate now = LocalDate.now();
|
LocalDate threshold = now.plusDays(REMIND_DAYS);
|
List<SpecialEquipmentLedger> list = ledgerMapper.selectList(Wrappers.<SpecialEquipmentLedger>lambdaQuery()
|
.and(wrapper -> wrapper
|
.isNotNull(SpecialEquipmentLedger::getNextInspectionDate)
|
.le(SpecialEquipmentLedger::getNextInspectionDate, threshold)
|
.or()
|
.isNotNull(SpecialEquipmentLedger::getNextMaintenanceDate)
|
.le(SpecialEquipmentLedger::getNextMaintenanceDate, threshold)));
|
for (SpecialEquipmentLedger ledger : list) {
|
syncReminder(ledger, now, threshold, true);
|
syncReminder(ledger, now, threshold, false);
|
}
|
}
|
|
private void syncReminder(SpecialEquipmentLedger ledger, LocalDate now, LocalDate threshold, boolean inspection) {
|
LocalDate dueDate = inspection ? ledger.getNextInspectionDate() : ledger.getNextMaintenanceDate();
|
if (dueDate == null || dueDate.isAfter(threshold)) {
|
return;
|
}
|
long daysLeft = ChronoUnit.DAYS.between(now, dueDate);
|
String reminderType = inspection ? "检验到期提醒" : "维保到期提醒";
|
String title = ledger.getEquipmentName() + "-" + reminderType;
|
String content = String.format("设备【%s】(%s) %s，%s，请及时安排处理。",
|
ledger.getEquipmentName(),
|
Optional.ofNullable(ledger.getEquipmentNo()).orElse(""),
|
inspection ? "检验即将到期" : "维保即将到期",
|
daysLeft < 0 ? ("已逾期 " + Math.abs(daysLeft) + " 天") : ("距离到期还有 " + daysLeft + " 天"));
|
NotificationManagement notification = notificationManagementService.lambdaQuery()
|
.eq(NotificationManagement::getTitle, title)
|
.eq(NotificationManagement::getType, reminderType)
|
.oneOpt()
|
.orElseGet(NotificationManagement::new);
|
notification.setTitle(title);
|
notification.setType(reminderType);
|
notification.setPriority(daysLeft <= 7 ? "高" : "中");
|
notification.setStatus("1");
|
notification.setExpireDate(dueDate);
|
notification.setContent(content);
|
notification.setDepartments(new ArrayList<>());
|
notification.setSyncMethods(new ArrayList<>());
|
if (notification.getId() == null) {
|
notificationManagementService.save(notification);
|
} else {
|
notificationManagementService.updateById(notification);
|
}
|
}
|
}
|
