package cn.iocoder.yudao.module.mes.service.energy;
|
|
import cn.iocoder.yudao.framework.common.exception.ErrorCode;
|
import cn.iocoder.yudao.framework.common.pojo.*;
|
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
|
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
|
import cn.iocoder.yudao.module.mes.controller.admin.energy.vo.EnergyAlertRuleSaveReqVO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.energy.*;
|
import cn.iocoder.yudao.module.mes.dal.mysql.energy.*;
|
import cn.iocoder.yudao.module.system.api.notify.NotifyMessageSendApi;
|
import cn.iocoder.yudao.module.system.api.notify.dto.NotifySendSingleToUserReqDTO;
|
import cn.iocoder.yudao.module.system.service.user.AdminUserService;
|
import jakarta.annotation.Resource;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.*;
|
import java.math.BigDecimal;
|
import java.time.LocalDate;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
|
@Service @Slf4j
|
public class EnergyAlertService {
|
@Resource private EnergyAlertRuleMapper energyAlertRuleMapper;
|
@Resource private EnergyAlertLogMapper energyAlertLogMapper;
|
@Resource private EnergyRecordMapper energyRecordMapper;
|
@Resource private AdminUserService adminUserService;
|
@Resource private NotifyMessageSendApi notifyMessageSendApi;
|
|
public PageResult<EnergyAlertRuleDO> page(PageParam page) {
|
return energyAlertRuleMapper.selectPage(page, new LambdaQueryWrapperX<EnergyAlertRuleDO>().orderByDesc(EnergyAlertRuleDO::getId));
|
}
|
public PageResult<EnergyAlertLogDO> logs(PageParam page, LocalDate date, String status) {
|
return energyAlertLogMapper.selectPage(page, new LambdaQueryWrapperX<EnergyAlertLogDO>()
|
.eqIfPresent(EnergyAlertLogDO::getStatus, status)
|
.and(date != null, w -> w.and(day -> day.eq(EnergyAlertLogDO::getPeriodType, "DAY").eq(EnergyAlertLogDO::getRecordDate, date))
|
.or(month -> month.eq(EnergyAlertLogDO::getPeriodType, "MONTH").eq(EnergyAlertLogDO::getRecordDate, date.withDayOfMonth(1))))
|
.orderByDesc(EnergyAlertLogDO::getId));
|
}
|
public Long save(EnergyAlertRuleSaveReqVO req) {
|
if (req.getId() != null) get(req.getId());
|
adminUserService.validateUserList(req.getRecipientIds());
|
EnergyAlertRuleDO rule = BeanUtils.toBean(req, EnergyAlertRuleDO.class);
|
rule.setName(req.getName().trim());
|
rule.setUsageArea(req.getUsageArea() == null ? "" : req.getUsageArea().trim());
|
rule.setProductCategory(req.getProductCategory() == null ? "" : req.getProductCategory().trim());
|
rule.setRecipientIds(req.getRecipientIds().stream().distinct().map(String::valueOf).collect(Collectors.joining(",")));
|
if (rule.getId() == null) energyAlertRuleMapper.insert(rule);
|
else if (energyAlertRuleMapper.updateById(rule) == 0) throw missing();
|
return rule.getId();
|
}
|
public void delete(Long id) {
|
get(id);
|
if (energyAlertRuleMapper.deleteById(id) == 0) throw missing();
|
}
|
private EnergyAlertRuleDO get(Long id) {
|
var rule = energyAlertRuleMapper.selectById(id);
|
if (rule == null) throw missing();
|
return rule;
|
}
|
private RuntimeException missing() { return exception(new ErrorCode(1_040_951_000, "能耗预警规则不存在")); }
|
|
/** Check the selected day and its calendar month; successful sends commit with their history. */
|
@Transactional(propagation = Propagation.REQUIRES_NEW, isolation = Isolation.READ_COMMITTED, rollbackFor = Exception.class)
|
public Map<String, Integer> check(LocalDate date) {
|
int sent = 0, failed = 0;
|
var rules = energyAlertRuleMapper.selectList(new LambdaQueryWrapperX<EnergyAlertRuleDO>()
|
.eq(EnergyAlertRuleDO::getEnabled, true).orderByAsc(EnergyAlertRuleDO::getId));
|
for (var candidate : rules) {
|
var rule = energyAlertRuleMapper.lock(candidate.getId());
|
if (rule == null || !Boolean.TRUE.equals(rule.getEnabled())) continue;
|
boolean monthly = rule.getMetric().startsWith("MONTH_");
|
LocalDate periodDate = monthly ? date.withDayOfMonth(1) : date;
|
String periodType = monthly ? "MONTH" : "DAY";
|
boolean amount = rule.getMetric().endsWith("AMOUNT");
|
BigDecimal actual = energyRecordMapper.selectDailyValue(periodDate, rule);
|
if (!exceeds(actual, rule.getThreshold())) continue;
|
for (String id : rule.getRecipientIds().split(",")) {
|
long recipient = Long.parseLong(id);
|
var log = energyAlertLogMapper.selectOne(new LambdaQueryWrapperX<EnergyAlertLogDO>()
|
.eq(EnergyAlertLogDO::getRuleId, rule.getId()).eq(EnergyAlertLogDO::getRecordDate, periodDate)
|
.eq(EnergyAlertLogDO::getPeriodType, periodType)
|
.eq(EnergyAlertLogDO::getRecipientId, recipient));
|
if (log != null && "SENT".equals(log.getStatus())) continue;
|
if (log == null) log = new EnergyAlertLogDO();
|
log.setRuleId(rule.getId()); log.setRuleName(rule.getName()); log.setRecordDate(periodDate); log.setPeriodType(periodType);
|
log.setEnergyType(rule.getEnergyType()); log.setMetric(rule.getMetric());
|
log.setScope((rule.getUsageArea().isEmpty() ? "全部区域" : rule.getUsageArea()) + " / "
|
+ (rule.getProductCategory().isEmpty() ? "全部产品" : rule.getProductCategory()));
|
log.setActualValue(actual); log.setThreshold(rule.getThreshold()); log.setRecipientId(recipient);
|
log.setRecipientName(String.valueOf(recipient));
|
try {
|
adminUserService.validateUserList(List.of(recipient));
|
log.setRecipientName(adminUserService.getUser(recipient).getNickname());
|
NotifySendSingleToUserReqDTO message = new NotifySendSingleToUserReqDTO();
|
message.setUserId(recipient); message.setTemplateCode(monthly ? "energy_monthly_warning" : "energy_daily_warning");
|
message.setTemplateParams(Map.of("ruleName", rule.getName(), "date", monthly ? java.time.YearMonth.from(date).toString() : date.toString(),
|
"energyName", energyName(rule.getEnergyType()), "scope", log.getScope(),
|
"metricName", amount ? "费用" : "用量",
|
"actual", actual.toPlainString(), "threshold", rule.getThreshold().toPlainString(),
|
"unit", amount ? "元" : unit(rule.getEnergyType())));
|
Long messageId = notifyMessageSendApi.sendSingleMessageToAdmin(message);
|
if (messageId == null) throw new IllegalStateException("能耗预警通知模板已停用");
|
log.setMessageId(messageId); log.setStatus("SENT"); log.setErrorMessage(""); sent++;
|
} catch (Exception e) {
|
logFailure(rule.getId(), date, recipient, e);
|
log.setStatus("FAILED");
|
log.setErrorMessage("发送失败，请检查接收人是否启用以及通知模板 " + (monthly ? "energy_monthly_warning" : "energy_daily_warning") + " 是否可用，然后重新检查该日期。");
|
failed++;
|
}
|
if (log.getId() == null) energyAlertLogMapper.insert(log); else energyAlertLogMapper.updateById(log);
|
}
|
}
|
return Map.of("sent", sent, "failed", failed);
|
}
|
private void logFailure(Long ruleId, LocalDate date, long recipient, Exception cause) {
|
log.warn("能耗预警站内通知发送失败 rule={} date={} recipient={}", ruleId, date, recipient, cause);
|
}
|
static boolean exceeds(BigDecimal actual, BigDecimal threshold) { return actual != null && actual.compareTo(threshold) > 0; }
|
public static String energyName(String type) { return switch (type) { case "WATER" -> "水"; case "ELECTRICITY" -> "电"; default -> "蒸汽"; }; }
|
public static String unit(String type) { return switch (type) { case "WATER" -> "m³"; case "ELECTRICITY" -> "kWh"; default -> "t"; }; }
|
}
|
