BJ_ZTEITS-Developers / parking-cloud-yudao

Commit a3f58be5714b0c5c24e998ed4f322db427f45629

Authored by YunaiV
1 parent 97260b8e

feat：【IoT 物联网】新增告警恢复场景规则执行类 IotAlertRecoverSceneRuleAction

Inline Side-by-side
Showing 1 changed file with 16 additions and 36 deletions
yudao-module-iot/yudao-module-iot-biz/src/main/java/cn/iocoder/yudao/module/iot/service/rule/scene/action/IotAlertRecoverSceneRuleAction.java
  View file @a3f58be
1 package cn.iocoder.yudao.module.iot.service.rule.scene.action; 1 package cn.iocoder.yudao.module.iot.service.rule.scene.action;
2 2
3 import cn.hutool.core.collection.CollUtil; 3 import cn.hutool.core.collection.CollUtil;
4 -import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum; 4 +import cn.hutool.core.util.StrUtil;
5 import cn.iocoder.yudao.module.iot.core.mq.message.IotDeviceMessage; 5 import cn.iocoder.yudao.module.iot.core.mq.message.IotDeviceMessage;
6 -import cn.iocoder.yudao.module.iot.dal.dataobject.alert.IotAlertConfigDO; 6 +import cn.iocoder.yudao.module.iot.dal.dataobject.alert.IotAlertRecordDO;
7 import cn.iocoder.yudao.module.iot.dal.dataobject.rule.IotRuleSceneDO; 7 import cn.iocoder.yudao.module.iot.dal.dataobject.rule.IotRuleSceneDO;
8 import cn.iocoder.yudao.module.iot.enums.rule.IotRuleSceneActionTypeEnum; 8 import cn.iocoder.yudao.module.iot.enums.rule.IotRuleSceneActionTypeEnum;
9 -import cn.iocoder.yudao.module.iot.service.alert.IotAlertConfigService;  
10 import cn.iocoder.yudao.module.iot.service.alert.IotAlertRecordService; 9 import cn.iocoder.yudao.module.iot.service.alert.IotAlertRecordService;
11 -import cn.iocoder.yudao.module.system.api.mail.MailSendApi;  
12 -import cn.iocoder.yudao.module.system.api.notify.NotifyMessageSendApi;  
13 -import cn.iocoder.yudao.module.system.api.sms.SmsSendApi;  
14 import jakarta.annotation.Resource; 10 import jakarta.annotation.Resource;
15 import org.springframework.stereotype.Component; 11 import org.springframework.stereotype.Component;
16 12
17 -import javax.annotation.Nullable;  
18 import java.util.List; 13 import java.util.List;
19 14
  15 +import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertList;
  16 +
20 // TODO @puhui999、@芋艿：未测试；需要场景联动开发完 17 // TODO @puhui999、@芋艿：未测试；需要场景联动开发完
21 /** 18 /**
22 - * IoT 告警触发的 {@link IotSceneRuleAction} 实现类 19 + * IoT 告警恢复的 {@link IotSceneRuleAction} 实现类
23 * 20 *
24 * @author 芋道源码 21 * @author 芋道源码
25 */ 22 */
26 @Component 23 @Component
27 -public class IotAlertTriggerSceneRuleAction implements IotSceneRuleAction { 24 +public class IotAlertRecoverSceneRuleAction implements IotSceneRuleAction {
28 25
29 - @Resource  
30 - private IotAlertConfigService alertConfigService;  
31 - @Resource  
32 - private IotAlertRecordService alertRecordService; 26 + private static final String PROCESS_REMARK = "告警自动回复，基于【{}】场景联动规则";
33 27
34 @Resource 28 @Resource
35 - private SmsSendApi smsSendApi;  
36 - @Resource  
37 - private MailSendApi mailSendApi;  
38 - @Resource  
39 - private NotifyMessageSendApi notifyMessageSendApi; 29 + private IotAlertRecordService alertRecordService;
40 30
41 @Override 31 @Override
42 - public void execute(@Nullable IotDeviceMessage message, 32 + public void execute(IotDeviceMessage message,
43 IotRuleSceneDO rule, IotRuleSceneDO.ActionConfig actionConfig) throws Exception { 33 IotRuleSceneDO rule, IotRuleSceneDO.ActionConfig actionConfig) throws Exception {
44 - List<IotAlertConfigDO> alertConfigs = alertConfigService.getAlertConfigListBySceneRuleIdAndStatus(  
45 - rule.getId(), CommonStatusEnum.ENABLE.getStatus());  
46 - if (CollUtil.isEmpty(alertConfigs)) { 34 + Long deviceId = message != null ? message.getDeviceId() : null;
  35 + List<IotAlertRecordDO> alertRecords = alertRecordService.getAlertRecordListBySceneRuleId(
  36 + rule.getId(), deviceId, false);
  37 + if (CollUtil.isEmpty(alertRecords)) {
47 return; 38 return;
48 } 39 }
49 - alertConfigs.forEach(alertConfig -> {  
50 - // 记录告警记录  
51 - alertRecordService.createAlertRecord(alertConfig, message);  
52 - // 发送告警消息  
53 - sendAlertMessage(alertConfig, message);  
54 - });  
55 - }  
56 -  
57 - private void sendAlertMessage(IotAlertConfigDO config, IotDeviceMessage deviceMessage) {  
58 - // TODO @芋艿：等场景联动开发完，再实现  
59 - // TODO @芋艿：短信  
60 - // TODO @芋艿：邮箱  
61 - // TODO @芋艿：站内信 40 + alertRecordService.processAlertRecordList(convertList(alertRecords, IotAlertRecordDO::getId),
  41 + StrUtil.format(PROCESS_REMARK, rule.getName()));
62 } 42 }
63 43
64 @Override 44 @Override
65 public IotRuleSceneActionTypeEnum getType() { 45 public IotRuleSceneActionTypeEnum getType() {
66 - return IotRuleSceneActionTypeEnum.ALERT_TRIGGER; 46 + return IotRuleSceneActionTypeEnum.ALERT_RECOVER;
67 } 47 }
68 48
69 } 49 }